def test_update_v6_message_adds_route(self): path_attributes = { "as_path" : "65032 65011 65002", "origin" : "EGP", "mp_reach_nlri" : { "next_hop" : [ IP6Address.from_string("2001:db8:1::242:ac11:2"), IP6Address.from_string("fe80::42:acff:fe11:2"), ], "nlri" : [ IP6Prefix.from_string("2001:db4::/127"), ] } } route_attributes = { "prefix" : IP6Prefix.from_string("2001:db4::/127"), "next_hop" : IP6Address.from_string("2001:db8:1::242:ac11:2"), "as_path" : "65032 65011 65002", "origin" : "EGP" } message = BgpUpdateMessage([], path_attributes, []) self.state_machine.event(EventMessageReceived(message), self.tick) self.assertEqual(self.state_machine.state, "established") self.assertEqual(self.state_machine.route_updates.qsize(), 1) self.assertEqual(self.state_machine.route_updates.get(), RouteAddition(**route_attributes))
def test_add_route_adds_route(self): self.beka.add_route("10.1.0.0/16", "192.168.1.3") self.assertEqual(self.beka.routes_to_advertise[0], RouteAddition( prefix=IPPrefix.from_string("10.1.0.0/16"), next_hop=IPAddress.from_string("192.168.1.3"), as_path="", origin="IGP" ) )
def test_keepalive_message_sends_all_routes_v6(self): self.tick += 3600 self.state_machine.routes_to_advertise = [ RouteAddition( IP6Prefix.from_string("2001:db4::/127"), IP6Address.from_string("2001:db1::1"), "", "IGP" ), RouteAddition( IP6Prefix.from_string("2001:db5::/127"), IP6Address.from_string("2001:db1::1"), "", "IGP" ), RouteAddition( IP6Prefix.from_string("2001:db6::/127"), IP6Address.from_string("2001:db1::2"), "", "IGP" ) ] message = BgpKeepaliveMessage() self.state_machine.event(EventMessageReceived(message), self.tick) self.assertEqual(self.state_machine.state, "established") self.assertTrue(self.state_machine.timers["hold"].running()) self.assertFalse(self.state_machine.timers["hold"].expired(self.tick)) self.assertEqual(self.state_machine.output_messages.qsize(), 2) first_update = self.state_machine.output_messages.get() second_update = self.state_machine.output_messages.get() self.assertTrue(isinstance(first_update, BgpUpdateMessage)) self.assertTrue(isinstance(second_update, BgpUpdateMessage)) self.assertEqual(first_update.path_attributes["mp_reach_nlri"]["next_hop"], [IP6Address.from_string("2001:db1::1")]) self.assertEqual(first_update.path_attributes["mp_reach_nlri"]["nlri"], [ IP6Prefix.from_string("2001:db4::/127"), IP6Prefix.from_string("2001:db5::/127") ]) self.assertEqual(second_update.path_attributes["mp_reach_nlri"]["next_hop"], [IP6Address.from_string("2001:db1::2")]) self.assertEqual(second_update.path_attributes["mp_reach_nlri"]["nlri"], [ IP6Prefix.from_string("2001:db6::/127") ])
def test_keepalive_message_sends_all_routes(self): self.tick += 3600 self.state_machine.routes_to_advertise = [ RouteAddition( IP4Prefix.from_string("10.0.0.0/8"), IP4Address.from_string("192.168.1.33"), "", "IGP" ), RouteAddition( IP4Prefix.from_string("192.168.64.0/23"), IP4Address.from_string("192.168.1.33"), "", "IGP" ), RouteAddition( IP4Prefix.from_string("192.168.128.0/23"), IP4Address.from_string("192.168.1.34"), "", "IGP" ) ] message = BgpKeepaliveMessage() self.state_machine.event(EventMessageReceived(message), self.tick) self.assertEqual(self.state_machine.state, "established") self.assertTrue(self.state_machine.timers["hold"].running()) self.assertFalse(self.state_machine.timers["hold"].expired(self.tick)) self.assertEqual(self.state_machine.output_messages.qsize(), 2) first_update = self.state_machine.output_messages.get() second_update = self.state_machine.output_messages.get() self.assertTrue(isinstance(first_update, BgpUpdateMessage)) self.assertTrue(isinstance(second_update, BgpUpdateMessage)) self.assertEqual(first_update.path_attributes["next_hop"], IP4Address.from_string("192.168.1.33")) self.assertEqual(first_update.nlri, [ IP4Prefix.from_string("10.0.0.0/8"), IP4Prefix.from_string("192.168.64.0/23") ]) self.assertEqual(second_update.path_attributes["next_hop"], IP4Address.from_string("192.168.1.34")) self.assertEqual(second_update.nlri, [ IP4Prefix.from_string("192.168.128.0/23") ])
def test_update_message_adds_route(self): path_attributes = { "next_hop" : IP4Address.from_string("5.4.3.2"), "as_path" : "65032 65011 65002", "origin" : "EGP" } route_attributes = { "prefix" : IP4Prefix.from_string("192.168.0.0/16"), "next_hop" : IP4Address.from_string("5.4.3.2"), "as_path" : "65032 65011 65002", "origin" : "EGP" } message = BgpUpdateMessage([], path_attributes, [IP4Prefix.from_string("192.168.0.0/16")]) self.state_machine.event(EventMessageReceived(message), self.tick) self.assertEqual(self.state_machine.state, "established") self.assertEqual(self.state_machine.route_updates.qsize(), 1) self.assertEqual(self.state_machine.route_updates.get(), RouteAddition(**route_attributes))