def test_close(self):
data = unittest.mock.Mock()
error1 = unittest.mock.Mock()
error2 = unittest.mock.Mock()
l1 = TagListener(data, error1)
l2 = TagListener(data, error2)
obj = object()
nh = TagDispatcher()
nh.add_listener("tag1", l1)
nh.add_listener("tag2", l2)
nh.close_all(obj)
data.assert_not_called()
error1.assert_called_once_with(obj)
error2.assert_called_once_with(obj)
with self.assertRaises(KeyError):
nh.remove_listener("tag1")
with self.assertRaises(KeyError):
nh.remove_listener("tag2")
with self.assertRaises(KeyError):
nh.unicast("tag1", None)
with self.assertRaises(KeyError):
nh.unicast("tag2", None)
def test_close(self):
data = unittest.mock.Mock()
error1 = unittest.mock.Mock()
error2 = unittest.mock.Mock()
l1 = TagListener(data, error1)
l2 = TagListener(data, error2)
obj = object()
nh = TagDispatcher()
nh.add_listener("tag1", l1)
nh.add_listener("tag2", l2)
nh.close_all(obj)
data.assert_not_called()
error1.assert_called_once_with(obj)
error2.assert_called_once_with(obj)
with self.assertRaises(KeyError):
nh.remove_listener("tag1")
with self.assertRaises(KeyError):
nh.remove_listener("tag2")
with self.assertRaises(KeyError):
nh.unicast("tag1", None)
with self.assertRaises(KeyError):
nh.unicast("tag2", None)
def test_remove_listener(self):
mock = unittest.mock.Mock()
nh = TagDispatcher()
nh.add_callback("tag", mock)
nh.remove_listener("tag")
with self.assertRaises(KeyError):
nh.unicast("tag", object())
mock.assert_not_called()
def test_remove_listener(self):
mock = unittest.mock.Mock()
nh = TagDispatcher()
nh.add_callback("tag", mock)
nh.remove_listener("tag")
with self.assertRaises(KeyError):
nh.unicast("tag", object())
mock.assert_not_called()
