def test_consumer_returns_last_commit(db):
conn, _ = db
consumer = Consumer(conn, "foo", "offsets")
new_offset = consumer.new_offset()
consumer.commit(new_offset)
offset = consumer.get_offset()
assert offset == new_offset
def test_consumer_passes_offsets_to_work_fn(db):
conn, _ = db
consumer = Consumer(conn, "foo", "offsets")
new_offset = consumer.new_offset()
consumer.commit(new_offset)
def work(old, new):
assert old == new_offset
consumer.do_work(work)
def test_consumer_throws_consumer_work_error_on_exception(db):
conn, _ = db
consumer = Consumer(conn, "foo", "offsets")
new_offset = consumer.new_offset()
consumer.commit(new_offset)
def work(old, new):
assert old != new_offset
with pytest.raises(ConsumerWorkError):
consumer.do_work(work)