-#!/usr/bin/env python3
+#!/usr/bin/env python
import unittest
-from vpp_serializer import VPPType, VPPEnumType, VPPUnionType
+from vpp_papi.vpp_serializer import VPPType, VPPEnumType
+from vpp_papi.vpp_serializer import VPPUnionType, VPPMessage
from socket import inet_pton, AF_INET, AF_INET6
import logging
-
-logging.basicConfig(format='%(levelname)s:%(message)s', level=logging.DEBUG)
-es_logger = logging.getLogger('vpp_serializer')
-es_logger.setLevel(logging.DEBUG)
-
-class VPPMessage(VPPType):
- pass
+import sys
class TestAddType(unittest.TestCase):
[['u8', 'is_bool'],
['u32', 'is_int']])
- b = un.pack({'is_int': 0x1234})
+ b = un.pack({'is_int': 0x12345678})
self.assertEqual(len(b), 4)
nt = un.unpack(b)
- self.assertEqual(nt.is_bool, 52)
- self.assertEqual(nt.is_int, 0x1234)
+ self.assertEqual(nt.is_bool, 0x12)
+ self.assertEqual(nt.is_int, 0x12345678)
def test_address(self):
af = VPPEnumType('vl_api_address_family_t', [["ADDRESS_IP4", 0],
self.assertEqual(len(b), 20)
nt = address.unpack(b)
- print('NT union', nt)
self.assertEqual(nt.af, af.ADDRESS_IP4)
self.assertEqual(nt.un.ip4.address,
inet_pton(AF_INET, '2.2.2.2'))
self.assertEqual(nt.un.ip6.address,
- inet_pton(AF_INET6, '::0202:0202'))
+ inet_pton(AF_INET6, '0202:0202::'))
def test_arrays(self):
# Test cases
valistip4_legacy = VPPType('list_ip4_t',
[['u8', 'foo'],
['ip4_address', 'addresses', 0]])
-
+
addresses = []
for i in range(4):
addresses.append({'address': inet_pton(AF_INET, '2.2.2.2')})
self.assertEqual(len(b), 17)
nt = valistip4.unpack(b)
- print('NT', nt)
+ self.assertEqual(nt.count, 4)
+ self.assertEqual(nt.addresses[0].address,
+ inet_pton(AF_INET, '2.2.2.2'))
b = valistip4_legacy.pack({'foo': 1, 'addresses': addresses})
self.assertEqual(len(b), 17)
nt = valistip4_legacy.unpack(b)
- print('NT', nt)
+ self.assertEqual(len(nt.addresses), 4)
+ self.assertEqual(nt.addresses[0].address,
+ inet_pton(AF_INET, '2.2.2.2'))
-
def test_message(self):
foo = VPPMessage('foo', [['u16', '_vl_msg_id'],
['u8', 'client_index'],
'something': 200})
self.assertEqual(len(b), 4)
nt = foo.unpack(b)
- print('NT', nt)
+ self.assertEqual(nt.something, 200)
+
+ def test_abf(self):
+
+ fib_mpls_label = VPPType('vl_api_fib_mpls_label_t',
+ [['u8', 'is_uniform'],
+ ['u32', 'label'],
+ ['u8', 'ttl'],
+ ['u8', 'exp']])
+
+ label_stack = {'is_uniform': 0,
+ 'label': 0,
+ 'ttl': 0,
+ 'exp': 0}
+
+ b = fib_mpls_label.pack(label_stack)
+ self.assertEqual(len(b), 7)
+
+ fib_path = VPPType('vl_api_fib_path_t',
+ [['u32', 'sw_if_index'],
+ ['u32', 'table_id'],
+ ['u8', 'weight'],
+ ['u8', 'preference'],
+ ['u8', 'is_local'],
+ ['u8', 'is_drop'],
+ ['u8', 'is_udp_encap'],
+ ['u8', 'is_unreach'],
+ ['u8', 'is_prohibit'],
+ ['u8', 'is_resolve_host'],
+ ['u8', 'is_resolve_attached'],
+ ['u8', 'is_dvr'],
+ ['u8', 'is_source_lookup'],
+ ['u8', 'afi'],
+ ['u8', 'next_hop', 16],
+ ['u32', 'next_hop_id'],
+ ['u32', 'rpf_id'],
+ ['u32', 'via_label'],
+ ['u8', 'n_labels'],
+ ['vl_api_fib_mpls_label_t', 'label_stack', 16]])
+ label_stack_list = []
+ for i in range(16):
+ label_stack_list.append(label_stack)
+
+ paths = {'is_udp_encap': 0,
+ 'next_hop': b'\x10\x02\x02\xac',
+ 'table_id': 0,
+ 'afi': 0,
+ 'weight': 1,
+ 'next_hop_id': 4294967295,
+ 'label_stack': label_stack_list,
+ 'n_labels': 0,
+ 'sw_if_index': 4294967295,
+ 'preference': 0}
+
+ b = fib_path.pack(paths)
+ self.assertEqual(len(b), (7*16) + 49)
+
+ abf_policy = VPPType('vl_api_abf_policy_t',
+ [['u32', 'policy_id'],
+ ['u32', 'acl_index'],
+ ['u8', 'n_paths'],
+ ['vl_api_fib_path_t', 'paths', 0, 'n_paths']])
+
+ policy = {
+ 'n_paths': 1,
+ 'paths': [paths],
+ 'acl_index': 0,
+ 'policy_id': 10}
+
+ b = abf_policy.pack(policy)
+ self.assertEqual(len(b), (7*16) + 49 + 9)
+
+ abf_policy_add_del = VPPMessage('abf_policy_add_del',
+ [['u16', '_vl_msg_id'],
+ ['u32', 'client_index'],
+ ['u32', 'context'],
+ ['u8', 'is_add'],
+ ['vl_api_abf_policy_t', 'policy']])
+
+ b = abf_policy_add_del.pack({'is_add': 1,
+ 'context': 66,
+ '_vl_msg_id': 1066,
+ 'policy': policy})
+
+ nt = abf_policy_add_del.unpack(b)
+ self.assertEqual(nt.policy.paths[0].next_hop,
+ b'\x10\x02\x02\xac\x00\x00\x00\x00'
+ b'\x00\x00\x00\x00\x00\x00\x00\x00')
+
+ def test_bier(self):
+
+ bier_table_id = VPPType('vl_api_bier_table_id_t',
+ [['u8', 'bt_set'],
+ ['u8', 'bt_sub_domain'],
+ ['u8', 'bt_hdr_len_id']])
+
+ bier_imp_add = VPPMessage('bier_imp_add',
+ [['u32', 'client_index'],
+ ['u32', 'context'],
+ ['vl_api_bier_table_id_t', 'bi_tbl_id'],
+ ['u16', 'bi_src'],
+ ['u8', 'bi_n_bytes'],
+ ['u8', 'bi_bytes', 0, 'bi_n_bytes']])
+
+ table_id = {'bt_set': 0,
+ 'bt_sub_domain': 0,
+ 'bt_hdr_len_id': 0}
+
+ bibytes = b'foobar'
+
+ b = bier_imp_add.pack({'bi_tbl_id': table_id,
+ 'bi_n_bytes': len(bibytes),
+ 'bi_bytes': bibytes})
+
+ self.assertEqual(len(b), 20)
+
if __name__ == '__main__':
unittest.main()