def testDiscoveryPacket(self):
r1 = RIPRouter()
r1.name = "r1"
r2 = RIPRouter()
r2.name = "r2"
r3 = RIPRouter()
r3.name = "r3"
r1.handle_rx(DiscoveryPacket(r2, 0), 1)
r1.handle_rx(DiscoveryPacket(r3, 0), 2)
self.failUnless(r1.distance_vector.dest_via_nbors == {r2:{None:0},r3:{None:0}})
self.failUnless(r1.neighbor_ports == {r2:1, r3:2})
def testDistanceVectorTable(self):
r1 = RIPRouter()
r1.name = "r1"
r2 = RIPRouter()
r2.name = "r2"
r3 = RIPRouter()
r3.name = "r3"
r4 = RIPRouter()
r4.name = "r4"
r5 = RIPRouter()
r5.name = "r5"
r3.handle_rx(DiscoveryPacket(r1,1), None)
r3.handle_rx(DiscoveryPacket(r4,4), None)
r3.handle_rx(DiscoveryPacket(r5,5), None)
packetOne = RoutingUpdate()
packetOne.src = r1
packetOne.paths = {r2:1,r3:1,r4:2,r5:2}
packetFour = RoutingUpdate()
packetFour.src = r4
packetFour.paths = {r1:2,r2:1,r3:1,r5:1}
packetFive = RoutingUpdate()
packetFive.src = r5
packetFive.paths = {r1:2,r2:2,r3:1,r4:1}
r3.handle_rx(packetOne, 1)
r3.handle_rx(packetFour, 4)
r3.handle_rx(packetFive, 5)
r3_table = {r1:{None:0,r4:2,r5:2},r2:{r1:1,r4:1,r5:2},r4:{r1:2,None:0,r5:1},r5:{r1:2,r4:1,None:0}}
self.failUnless(r3.distance_vector.dest_via_nbors == r3_table)
def testNaivePoisonReverse(self):
r1 = RIPRouter()
r1.name = "r1"
r2 = RIPRouter()
r2.name = "r2"
r3 = RIPRouter()
r3.name = "r3"
r1.handle_rx(DiscoveryPacket(r2,0),0)
packetOne = RoutingUpdate
packetOne.paths = {r1:1, r3:1}
packetOne.src = r2
r1.handle_rx(RoutingUpdate, 0)
self.failUnless(r1.distance_vector.get_routing_packet(r2).paths == {})
def testImplicitRemoval(self):
r1 = RIPRouter()
r1.name = "r1"
r2 = RIPRouter()
r2.name = "r2"
r3 = RIPRouter()
r3.name = "r3"
r4 = RIPRouter()
r4.name = "r4"
r5 = RIPRouter()
r5.name = "r5"
r6 = RIPRouter()
r6.name = "r6"
r1.neighbor_ports = {r2:2,r3:3}
r1.distance_vector.dest_via_nbors = {r2:{None:0},r3:{None:0},r4:{r2:1,r3:1},r5:{r2:3},r6:{r2:5}}
packet = RoutingUpdate()
packet.src = r2
packet.paths = {r1:1,r4:1}
r1.handle_rx(packet, 2)
self.failUnless(r1.distance_vector.dest_via_nbors == {r2:{None:0},r3:{None:0},r4:{r3:1,r2:1}})
def testDistanceVectorRouteDeletion(self):
r1 = RIPRouter()
r1.name = "r1"
r2 = RIPRouter()
r2.name = "r2"
r3 = RIPRouter()
r3.name = "r3"
r4 = RIPRouter()
r4.name = "r4"
r5 = RIPRouter()
r5.name = "r5"
r5.handle_rx(DiscoveryPacket(r3,0), 3)
r5.handle_rx(DiscoveryPacket(r4,0), 4)
packetThree = RoutingUpdate()
packetThree.src = r3
packetThree.paths = {r1:1,r2:2,r4:1,r5:1}
r5.handle_rx(packetThree, 3)
packetFour = RoutingUpdate()
packetFour.src = r4
packetFour.paths = {r1:2,r2:1,r3:1,r5:1}
r5.handle_rx(packetFour, 4)
r5_table = {r1:{r3:1,r4:2},r2:{r3:2,r4:1},r3:{None:0,r4:1},r4:{r3:1,None:0}}
self.failUnless(r5.distance_vector.dest_via_nbors == r5_table)
self.failUnless(r5.neighbor_ports == {r3:3,r4:4})
self.failUnless(r5.distance_vector.delete_link(r3))
r5_table = {r1:{r4:2},r2:{r4:1},r3:{r4:1},r4:{None:0}}
self.failUnless(r5.distance_vector.dest_via_nbors == r5_table)
