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')
