class TestNetworkTestCase(unittest.TestCase): def setUp(self): options, args = sampleOptions("--configuration tests/mocks/veripy.cfg host") self.c = MockConfiguration(args, options, None) self.n = self.c.build_test_network() def tearDown(self): self.n.reset() def test_it_should_have_three_links(self): self.assertEqual(3, len(self.n.links())) self.assertEqual('A', self.n.link(1).name) self.assertEqual('B', self.n.link(2).name) self.assertEqual('C', self.n.link(3).name) def test_it_should_have_four_nodes(self): self.assertEqual(4, len(self.n.nodes())) self.assertEqual('TN1', self.n.node(1).name) self.assertEqual('TN2', self.n.node(2).name) self.assertEqual('TN3', self.n.node(3).name) self.assertEqual('TN4', self.n.node(4).name) def test_it_should_have_three_routers(self): self.assertEqual(3, len(self.n.routers())) self.assertEqual('TR1', self.n.router(1).name) self.assertEqual('TR2', self.n.router(2).name) self.assertEqual('TR3', self.n.router(3).name) def test_it_should_have_bound_internal_interfaces_to_links(self): self.assertEqual(5, len(self.n.link(1).bound_interfaces())) self.assertEqual(5, len(self.n.link(2).bound_interfaces())) self.assertEqual(2, len(self.n.link(3).bound_interfaces()))