def test_empty_graph(self): h = HubDispatch() with self.assertRaises(Exception) as exc: h.remove_hub('foo') self.assertEqual(exc.exception.message, "Hub 'foo' does not exist") with self.assertRaises(Exception) as exc: h.link('foo', 'bar') self.assertEqual(exc.exception.message, "Hub 'foo' does not exist") with self.assertRaises(Exception) as exc: h.unlink('foo', 'bar') self.assertEqual(exc.exception.message, "Hub 'foo' does not exist")
def test_add_hub_linked_to_node(self): h = HubDispatch() h.add_hub('foo') self.assertEqual(h._changes.assignments, [('foo', 'foo')]) h.link('foo', 'bar') self.assertEqual( h._topology.nodes, {'bar': 'foo', 'foo': 'foo'}, ) self.assertEqual( h._topology.hubs, {'foo': 2}, ) self.assertEqual(h._changes.assignments, [ ('foo', 'foo'), ('foo', 'bar') ]) h._changes._clear()
def test_add_hub_linked_to_node(self): h = HubDispatch() h.add_hub('foo') self.assertEqual(h._changes.assignments, [('foo', 'foo')]) h.link('foo', 'bar') self.assertEqual( h._topology.nodes, { 'bar': 'foo', 'foo': 'foo' }, ) self.assertEqual( h._topology.hubs, {'foo': 2}, ) self.assertEqual(h._changes.assignments, [('foo', 'foo'), ('foo', 'bar')]) h._changes._clear()
def test_empty_graph(self): h = HubDispatch() with self.assertRaises(Exception) as exc: h.remove_hub('foo') self.assertEqual( exc.exception.message, "Hub 'foo' does not exist" ) with self.assertRaises(Exception) as exc: h.link('foo', 'bar') self.assertEqual( exc.exception.message, "Hub 'foo' does not exist" ) with self.assertRaises(Exception) as exc: h.unlink('foo', 'bar') self.assertEqual( exc.exception.message, "Hub 'foo' does not exist" )
def test_add_too_much_node(self): h = HubDispatch(max_nodes_per_hub=2).add_hub('h').link('h', 'n1') h._changes._clear() with self.assertRaises(NotImplementedError): h.link('h', 'n2')