def testBasicRouting(self): net = RoutingTable() def ar(ip, route): self.assertEquals(net.route(ip), route) ar('192.168.1.0', None) net.addSubnet('foo', '192.168.1.0', 24) ar('192.168.1.0', 'foo') ar('192.168.1.10', 'foo') ar('192.168.1.255', 'foo') ar('192.168.0.255', None) ar('192.168.2.0', None) net.addSubnet('foo', '192.168.2.0', 24) ar('192.168.0.255', None) ar('192.168.1.255', 'foo') ar('192.168.2.0', 'foo') net.removeSubnet('foo', '192.168.1.0', 24) net.removeSubnet('foo', '192.168.2.0', 24) ar('192.168.1.0', None) ar('192.168.1.10', None) ar('192.168.1.255', None) ar('192.168.0.255', None) ar('192.168.2.0', None)
def testAddRemove(self): net = RoutingTable() net.addSubnet('foo', '192.168.0.0', 24) net.addSubnet('foo', '192.168.1.0', 24) self.assertEquals(len(net), 2) net.removeSubnet('foo', '192.168.0.0', 24) net.removeSubnet('foo', '192.168.1.0', 24) self.assertEquals(len(net), 0)