def setUp(self): self.qm = QueueManager() self.q = Queue() self.qq = Queue() self.qm.register(self.q, {"foo": "bar", "baz": "quux"}) self.qm.register(self.qq, {"foo": "quux"})
class QueueManager_Test(TestCase): def setUp(self): self.qm = QueueManager() self.q = Queue() self.qq = Queue() self.qm.register(self.q, {"foo": "bar", "baz": "quux"}) self.qm.register(self.qq, {"foo": "quux"}) def tearDown(self): self.qm.deregister(self.q) self.qm.deregister(self.qq) def test_fanout(self): message = object() self.qm.fanout(message) assert self.q.get(False) is message assert self.qq.get(False) is message def test_rules(self): message = object() self.qm.fanout(message, {"foo": "bar"}) assert self.q.get(False) is message with self.assertRaises(Empty): self.qq.get(False) def test_already_registered(self): with self.assertRaises(AlreadyRegistered): self.qm.register(self.q) def test_not_found(self): with self.assertRaises(NotFound): self.qm.deregister(Queue())