diff options
Diffstat (limited to 'src/vpp-api/python/tests/test_vpp_serializer.py')
| -rwxr-xr-x | src/vpp-api/python/tests/test_vpp_serializer.py | 226 |
1 files changed, 226 insertions, 0 deletions
diff --git a/src/vpp-api/python/tests/test_vpp_serializer.py b/src/vpp-api/python/tests/test_vpp_serializer.py new file mode 100755 index 00000000000..0c194b46d22 --- /dev/null +++ b/src/vpp-api/python/tests/test_vpp_serializer.py @@ -0,0 +1,226 @@ +#!/usr/bin/env python + +import unittest +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 +import sys + + +class TestAddType(unittest.TestCase): + + def test_union(self): + un = VPPUnionType('test_union', + [['u8', 'is_bool'], + ['u32', 'is_int']]) + + b = un.pack({'is_int': 0x1234}) + self.assertEqual(len(b), 4) + nt = un.unpack(b) + self.assertEqual(nt.is_bool, 52) + self.assertEqual(nt.is_int, 0x1234) + + def test_address(self): + af = VPPEnumType('vl_api_address_family_t', [["ADDRESS_IP4", 0], + ["ADDRESS_IP6", 1], + {"enumtype": "u32"}]) + ip4 = VPPType('vl_api_ip4_address_t', [['u8', 'address', 4]]) + ip6 = VPPType('vl_api_ip6_address_t', [['u8', 'address', 16]]) + VPPUnionType('vl_api_address_union_t', + [["vl_api_ip4_address_t", "ip4"], + ["vl_api_ip6_address_t", "ip6"]]) + + address = VPPType('address', [['vl_api_address_family_t', 'af'], + ['vl_api_address_union_t', 'un']]) + + b = ip4.pack({'address': inet_pton(AF_INET, '1.1.1.1')}) + self.assertEqual(len(b), 4) + nt = ip4.unpack(b) + self.assertEqual(nt.address, inet_pton(AF_INET, '1.1.1.1')) + + b = ip6.pack({'address': inet_pton(AF_INET6, '1::1')}) + self.assertEqual(len(b), 16) + + b = address.pack({'af': af.ADDRESS_IP4, + 'un': + {'ip4': + {'address': inet_pton(AF_INET, '2.2.2.2')}}}) + self.assertEqual(len(b), 20) + + nt = address.unpack(b) + 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')) + + def test_arrays(self): + # Test cases + # 1. Fixed list + # 2. Fixed list of variable length sub type + # 3. Variable length type + # + ip4 = VPPType('ip4_address', [['u8', 'address', 4]]) + listip4 = VPPType('list_ip4_t', [['ip4_address', 'addresses', 4]]) + valistip4 = VPPType('list_ip4_t', + [['u8', 'count'], + ['ip4_address', 'addresses', 0, 'count']]) + + 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')}) + b = listip4.pack({'addresses': addresses}) + self.assertEqual(len(b), 16) + nt = listip4.unpack(b) + + self.assertEqual(nt.addresses[0].address, + inet_pton(AF_INET, '2.2.2.2')) + + b = valistip4.pack({'count': len(addresses), 'addresses': addresses}) + self.assertEqual(len(b), 17) + + nt = valistip4.unpack(b) + 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) + 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'], + ['u8', 'something'], + {"crc": "0x559b9f3c"}]) + b = foo.pack({'_vl_msg_id': 1, 'client_index': 5, + 'something': 200}) + self.assertEqual(len(b), 4) + nt = foo.unpack(b) + 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() |