def test_put_waits_to_consume():
("Queue#put should wait until someone consumes")
# Given a dummy manager
manager = Node()
consume = Queue("consume")
produce = Queue("produce")
step = DummyStep(consume, produce, manager)
step.start()
consume.put({"foo": "Bar"})
produce.get().should.equal({"cool": {"foo": "Bar"}})
consume.deactivate()
produce.deactivate()
def test_queue_adopt_producer_step():
("Queue#adopt_producer should take a step and know about its id and")
# Given a dummy manager
manager = Node()
# Given a Queue with maximum size of 10
queue = Queue("test-queue", maxsize=10)
q1 = Queue("name1")
q2 = Queue("name2")
# And a running step (so it gains a thread id)
step = DummyStep(q1, q2, manager)
step.start()
# When that queue adopts the step as a producer
consumers, producers = queue.adopt_producer(step)
# And the redis key should contain the step id
hostname = socket.gethostname()
pid = os.getpid()
tid = step.ident
value = "{hostname}|{pid}|{tid}|tests.functional.test_queue.DummyStep|lineup.framework.Node".format(**locals())
redis = StrictRedis()
members = redis.smembers("lineup:test-queue:producers")
members.should.contain(value)
queue.deactivate()
q1.deactivate()
q2.deactivate()
