def create_default_network_data_prefix_sub_tlvs_factories():
return {
network_data.TlvType.HAS_ROUTE: network_data.HasRouteFactory(
routes_factory=network_data.RoutesFactory(
route_factory=network_data.RouteFactory())
),
network_data.TlvType.BORDER_ROUTER: network_data.BorderRouterFactory(),
network_data.TlvType.LOWPAN_ID: network_data.LowpanIdFactory()
}
def create_default_network_data_prefix_sub_tlvs_factories():
return {
0:
network_data.HasRouteFactory(routes_factory=network_data.RoutesFactory(
route_factory=network_data.RouteFactory())),
2:
network_data.BorderRouterFactory(),
3:
network_data.LowpanIdFactory()
}
def test_should_create_Route_list_from_bytearray_when_parse_method_is_called(self):
# GIVEN
routes = any_routes()
factory = network_data.RoutesFactory(network_data.RouteFactory())
data = convert_routes_to_bytearray(routes)
# WHEN
actual_routes = factory.parse(io.BytesIO(data), None)
# THEN
self.assertTrue(isinstance(actual_routes, list))
self.assertEqual(routes, actual_routes)
def test_should_create_Route_from_bytearray_when_parse_method_is_called(self):
# GIVEN
border_router_16 = any_border_router_16()
prf = any_prf()
factory = network_data.RouteFactory()
data = convert_route_to_bytearray(network_data.Route(border_router_16, prf))
# WHEN
actual_route = factory.parse(io.BytesIO(data), None)
# THEN
self.assertTrue(isinstance(actual_route, network_data.Route))
self.assertEqual(border_router_16, actual_route.border_router_16)
self.assertEqual(prf, actual_route.prf)
def test_should_create_HasRoute_from_bytearray_when_parse_method_is_called(self):
# GIVEN
routes = any_routes()
stable = any_stable()
factory = network_data.HasRouteFactory(network_data.RoutesFactory(network_data.RouteFactory()))
data = convert_routes_to_bytearray(routes)
message_info = common.MessageInfo()
message_info.stable = stable
# WHEN
actual_has_route = factory.parse(io.BytesIO(data), message_info)
# THEN
self.assertTrue(isinstance(actual_has_route, network_data.HasRoute))
self.assertEqual(routes, actual_has_route.routes)
self.assertEqual(stable, actual_has_route.stable)
