test_feed.py

@pytest.mark.asyncio
@pytest.mark.django_db
async def test_request(zmq_client, feed_task, test_month):
    stocks, sessions, quotes = test_month

    subscription = zmq_client.subscribe('quotes', 'quote')

    reply, count = await zmq_client._async(feed_task.request,
                                           start=datetime(2017, 1, 1),
                                           end=datetime(2017, 1, 3))

    assert reply
    assert count == 2

    for date in (datetime(2017, 1, 1), datetime(2017, 1, 2)):
        *_, feed = await asyncio.wait_for(pop(subscription, schema=FeedSchema()), timeout=1.0)
        assert feed.session.date == date.date()
        assert set(feed.stock_data.keys()) == set(s.id for s in stocks.values())