blob: cd142fdc0b260f0ed3e4997a4935ce21ea6f3a1b [file] [log] [blame]
#
# Copyright 2016 the original author or authors.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
import json
import os
from unittest import TestCase
import time
from tests.itests.docutests.test_utils import run_command_to_completion_with_raw_stdout
from tests.utests.chameleon.protoc_plugins.test_utils import load_file
pyang_cmd = "pyang --plugindir /voltha/experiments/plugin -f protobuf " \
"-p /voltha/tests/utests/netconf/yang " \
"/voltha/tests/utests/netconf/yang/{}"
class YangToProtoBufTests(TestCase):
def _compare_file(self, response, expected_response):
# compare two files and strip empty lines, blanks, etc
def _filter(x):
x.strip()
return x is not None
response = filter(_filter,response.split())
expected_response = filter(_filter,expected_response.split())
self.assertEqual(set(response), set(expected_response))
def test_01_basic_def(self):
print "Test_01_basic_def_Start:------------------"
t0 = time.time()
# input file: /voltha/tests/utests/netconf/yang/basic.yang
expected_response = """
syntax = "proto3";
package basic;
message commonAttributes {
uint32 my-id = 1 ;
string my-name = 2 ;
bool my-status = 3 ;
}
"""
try:
cmd = pyang_cmd.format('basic.yang')
print 'running command: {}'.format(cmd)
response, err, rc = run_command_to_completion_with_raw_stdout(cmd)
self.assertEqual(rc, 0)
self._compare_file(response, expected_response)
finally:
print "Test_01_basic_def_End:------------------ took {} " \
"secs\n\n".format(time.time() - t0)
def test_02_container_def(self):
print "Test_02_container_def_Start:------------------"
t0 = time.time()
# input file: /voltha/tests/utests/netconf/yang/container.yang
expected_response = """
syntax = "proto3";
package container;
message int-container {
int32 eight = 1 ;
int32 nine = 2 ;
}
message int-container {
int32 eight = 1 ;
int32 nine = 2 ;
}
"""
try:
cmd = pyang_cmd.format('container.yang')
print 'running command: {}'.format(cmd)
response, err, rc = run_command_to_completion_with_raw_stdout(cmd)
self.assertEqual(rc, 0)
self._compare_file(response, expected_response)
finally:
print "Test_02_container_def_End:------------------ took {} " \
"secs\n\n".format(time.time() - t0)
def test_03_mix_simple_types(self):
print "Test_03_mix_simple_types_Start:------------------"
t0 = time.time()
# input file: /voltha/tests/utests/netconf/yang/mix_simple_types.yang
expected_response = """
syntax = "proto3";
package mix_simple_types;
message user {
string name = 1 ;
string full-name = 2 ;
string class = 3 ;
}
message user {
string name = 1 ;
string full-name = 2 ;
string class = 3 ;
}
message int-container {
int32 eight = 1 ;
int32 nine = 2 ;
int32 ten = 3 ;
}
message container1 {
bool a = 1 ;
Any b = 2 ;
string mleaf = 3 ;
repeated string mleaf_list = 4 ;
message inner-container {
string mleaf1 = 1 ;
string mleaf2 = 2 ;
}
}
message int-container {
int32 eight = 1 ;
int32 nine = 2 ;
int32 ten = 3 ;
}
message container1 {
bool a = 1 ;
Any b = 2 ;
string mleaf = 3 ;
repeated string mleaf_list = 4 ;
message inner-container {
string mleaf1 = 1 ;
string mleaf2 = 2 ;
}
}
"""
try:
cmd = pyang_cmd.format('mix_simple_types.yang')
print 'running command: {}'.format(cmd)
response, err, rc = run_command_to_completion_with_raw_stdout(cmd)
self.assertEqual(rc, 0)
self._compare_file(response, expected_response)
finally:
print "Test_03_mix_simple_types_End:------------------ took {} " \
"secs\n\n".format(time.time() - t0)
def test_04_cord_tenant(self):
print "Test_04_cord_tenant_Start:------------------"
t0 = time.time()
# input file: /voltha/tests/utests/netconf/yang/cord-tenant.yang
expected_response = """
syntax = "proto3";
package cord_tenant;
message subscriber {
string label = 1 ;
enum status
{
violation = 0 ;
enabled = 1 ;
delinquent = 2 ;
suspended = 3 ;
}
bool demo = 3 ;
}
"""
try:
cmd = pyang_cmd.format('cord-tenant.yang')
print 'running command: {}'.format(cmd)
response, err, rc = run_command_to_completion_with_raw_stdout(cmd)
self.assertEqual(rc, 0)
self._compare_file(response, expected_response)
finally:
print "Test_04_cord_tenant_End:------------------ took {} " \
"secs\n\n".format(time.time() - t0)