def test_select_add(self):
q = Queues()
q.select(['foo', 'bar'])
q.select_add('baz')
self.assertItemsEqual(keys(q._consume_from), ['foo', 'bar', 'baz'])
def test_add_default_exchange(self):
ex = Exchange('fff', 'fanout')
q = Queues(default_exchange=ex)
q.add(Queue('foo'))
assert q['foo'].exchange.name == 'fff'
def test_with_max_priority(self, queues_kwargs, qname, q, expected):
queues = Queues(**queues_kwargs)
queues.add(q)
assert queues[qname].queue_arguments == expected
def test_with_ha_policy(self, ha_policy, qname, q, qargs, expected):
queues = Queues(ha_policy=ha_policy, create_missing=False)
queues.add(q, queue_arguments=qargs)
assert queues[qname].queue_arguments == expected
def test_deselect(self):
q = Queues()
q.select(['foo', 'bar'])
q.deselect('bar')
assert sorted(keys(q._consume_from)) == ['foo']
def test_add(self):
q = Queues()
q.add('foo', exchange='ex', routing_key='rk')
self.assertIn('foo', q)
self.assertIsInstance(q['foo'], Queue)
self.assertEqual(q['foo'].routing_key, 'rk')
def test_select_add(self):
q = Queues()
q.select(["foo", "bar"])
q.select_add("baz")
self.assertItemsEqual(keys(q._consume_from), ["foo", "bar", "baz"])
def test_alias(self):
q = Queues()
q.add(Queue('foo', alias='barfoo'))
self.assertIs(q['barfoo'], q['foo'])
def test_deselect(self):
q = Queues()
q.select(["foo", "bar"])
q.deselect("bar")
assert sorted(keys(q._consume_from)) == ["foo"]
def test_with_ha_policy_compat(self):
q = Queues(ha_policy='all')
q.add('bar')
self.assertEqual(q['bar'].queue_arguments, {'x-ha-policy': 'all'})
def test_add_default_exchange(self):
ex = Exchange('fff', 'fanout')
q = Queues(default_exchange=ex)
q.add(Queue('foo'))
self.assertEqual(q['foo'].exchange, ex)
def test_deselect(self):
q = Queues()
q.select(['foo', 'bar'])
q.deselect('bar')
self.assertItemsEqual(keys(q._consume_from), ['foo'])
def test_select_add(self):
q = Queues()
q.select(['foo', 'bar'])
q.select_add('baz')
self.assertItemsEqual(keys(q._consume_from), ['foo', 'bar', 'baz'])
def test_with_ha_policy(self):
qn = Queues(ha_policy=None, create_missing=False)
qn.add('xyz')
self.assertIsNone(qn['xyz'].queue_arguments)
qn.add('xyx', queue_arguments={'x-foo': 'bar'})
self.assertEqual(qn['xyx'].queue_arguments, {'x-foo': 'bar'})
q = Queues(ha_policy='all', create_missing=False)
q.add(Queue('foo'))
self.assertEqual(q['foo'].queue_arguments, {'x-ha-policy': 'all'})
qq = Queue('xyx2', queue_arguments={'x-foo': 'bari'})
q.add(qq)
self.assertEqual(q['xyx2'].queue_arguments, {
'x-ha-policy': 'all',
'x-foo': 'bari',
})
q2 = Queues(ha_policy=['A', 'B', 'C'], create_missing=False)
q2.add(Queue('foo'))
self.assertEqual(q2['foo'].queue_arguments, {
'x-ha-policy': 'nodes',
'x-ha-policy-params': ['A', 'B', 'C'],
})
def test_with_ha_policy_compat(self):
q = Queues(ha_policy='all')
q.add('bar')
self.assertEqual(q['bar'].queue_arguments, {'x-ha-policy': 'all'})
def test_with_ha_policy_compat(self):
q = Queues(ha_policy="all")
q.add("bar")
assert q["bar"].queue_arguments == {"x-ha-policy": "all"}
def test_alias(self):
q = Queues()
q.add(Queue('foo', alias='barfoo'))
self.assertIs(q['barfoo'], q['foo'])
def test_add_default_exchange(self):
ex = Exchange("fff", "fanout")
q = Queues(default_exchange=ex)
q.add(Queue("foo"))
assert q["foo"].exchange.name == ""
def test_add(self):
q = Queues()
q.add('foo', exchange='ex', routing_key='rk')
self.assertIn('foo', q)
self.assertIsInstance(q['foo'], Queue)
self.assertEqual(q['foo'].routing_key, 'rk')
def test_alias(self):
q = Queues()
q.add(Queue("foo", alias="barfoo"))
assert q["barfoo"] is q["foo"]
def test_select_remove(self):
q = Queues()
q.select_subset(['foo', 'bar'])
q.select_remove('bar')
self.assertItemsEqual(keys(q._consume_from), ['foo'])
def test_add(self):
q = Queues()
q.add("foo", exchange="ex", routing_key="rk")
assert "foo" in q
assert isinstance(q["foo"], Queue)
assert q["foo"].routing_key == "rk"
def test_select_add(self):
q = Queues()
q.select(['foo', 'bar'])
q.select_add('baz')
assert sorted(keys(q._consume_from)) == ['bar', 'baz', 'foo']
def test_select_remove(self):
q = Queues()
q.select_subset(["foo", "bar"])
q.select_remove("bar")
self.assertItemsEqual(keys(q._consume_from), ["foo"])
def test_with_ha_policy_compat(self):
q = Queues(ha_policy='all')
q.add('bar')
assert q['bar'].queue_arguments == {'x-ha-policy': 'all'}
def test_deselect(self):
q = Queues()
q.select(["foo", "bar"])
q.deselect("bar")
self.assertItemsEqual(keys(q._consume_from), ["foo"])
def test_alias(self):
q = Queues()
q.add(Queue('foo', alias='barfoo'))
assert q['barfoo'] is q['foo']
def test_with_ha_policy_compat(self):
q = Queues(ha_policy="all")
q.add("bar")
self.assertEqual(q["bar"].queue_arguments, {"x-ha-policy": "all"})
def test_add(self):
q = Queues()
q.add('foo', exchange='ex', routing_key='rk')
assert 'foo' in q
assert isinstance(q['foo'], Queue)
assert q['foo'].routing_key == 'rk'
def test_add_default_exchange(self):
ex = Exchange("fff", "fanout")
q = Queues(default_exchange=ex)
q.add(Queue("foo"))
self.assertEqual(q["foo"].exchange, ex)
def test_deselect(self):
q = Queues()
q.select(['foo', 'bar'])
q.deselect('bar')
self.assertItemsEqual(keys(q._consume_from), ['foo'])
def test_alias(self):
q = Queues()
q.add(Queue("foo", alias="barfoo"))
self.assertIs(q["barfoo"], q["foo"])
def test_add_default_exchange(self):
ex = Exchange('fff', 'fanout')
q = Queues(default_exchange=ex)
q.add(Queue('foo'))
self.assertEqual(q['foo'].exchange.name, '')
def test_add(self):
q = Queues()
q.add("foo", exchange="ex", routing_key="rk")
self.assertIn("foo", q)
self.assertIsInstance(q["foo"], Queue)
self.assertEqual(q["foo"].routing_key, "rk")
def test_with_max_priority(self):
qs1 = Queues(max_priority=10)
qs1.add('foo')
self.assertEqual(qs1['foo'].queue_arguments, {'x-max-priority': 10})
q1 = Queue('xyx', queue_arguments={'x-max-priority': 3})
qs1.add(q1)
self.assertEqual(qs1['xyx'].queue_arguments, {
'x-max-priority': 3,
})
q1 = Queue('moo', queue_arguments=None)
qs1.add(q1)
self.assertEqual(qs1['moo'].queue_arguments, {
'x-max-priority': 10,
})
qs2 = Queues(ha_policy='all', max_priority=5)
qs2.add('bar')
self.assertEqual(qs2['bar'].queue_arguments, {
'x-ha-policy': 'all',
'x-max-priority': 5
})
q2 = Queue('xyx2', queue_arguments={'x-max-priority': 2})
qs2.add(q2)
self.assertEqual(qs2['xyx2'].queue_arguments, {
'x-ha-policy': 'all',
'x-max-priority': 2,
})
qs3 = Queues(max_priority=None)
qs3.add('foo2')
self.assertEqual(qs3['foo2'].queue_arguments, None)
q3 = Queue('xyx3', queue_arguments={'x-max-priority': 7})
qs3.add(q3)
self.assertEqual(qs3['xyx3'].queue_arguments, {
'x-max-priority': 7,
})
def test_with_ha_policy(self):
qn = Queues(ha_policy=None, create_missing=False)
qn.add("xyz")
self.assertIsNone(qn["xyz"].queue_arguments)
qn.add("xyx", queue_arguments={"x-foo": "bar"})
self.assertEqual(qn["xyx"].queue_arguments, {"x-foo": "bar"})
q = Queues(ha_policy="all", create_missing=False)
q.add(Queue("foo"))
self.assertEqual(q["foo"].queue_arguments, {"x-ha-policy": "all"})
qq = Queue("xyx2", queue_arguments={"x-foo": "bari"})
q.add(qq)
self.assertEqual(q["xyx2"].queue_arguments, {"x-ha-policy": "all", "x-foo": "bari"})
q2 = Queues(ha_policy=["A", "B", "C"], create_missing=False)
q2.add(Queue("foo"))
self.assertEqual(q2["foo"].queue_arguments, {"x-ha-policy": "nodes", "x-ha-policy-params": ["A", "B", "C"]})
def test_with_ha_policy(self):
qn = Queues(ha_policy=None, create_missing=False)
qn.add('xyz')
self.assertIsNone(qn['xyz'].queue_arguments)
qn.add('xyx', queue_arguments={'x-foo': 'bar'})
self.assertEqual(qn['xyx'].queue_arguments, {'x-foo': 'bar'})
q = Queues(ha_policy='all', create_missing=False)
q.add(Queue('foo'))
self.assertEqual(q['foo'].queue_arguments, {'x-ha-policy': 'all'})
qq = Queue('xyx2', queue_arguments={'x-foo': 'bari'})
q.add(qq)
self.assertEqual(q['xyx2'].queue_arguments, {
'x-ha-policy': 'all',
'x-foo': 'bari',
})
q2 = Queues(ha_policy=['A', 'B', 'C'], create_missing=False)
q2.add(Queue('foo'))
self.assertEqual(q2['foo'].queue_arguments, {
'x-ha-policy': 'nodes',
'x-ha-policy-params': ['A', 'B', 'C'],
})
def test_with_defaults(self):
self.assertEqual(Queues(None), {})
