def test_consumer_creates_table_if_not_exists(db):
conn, _ = db
consumer = Consumer(conn, "foo", "offsets")
offset = consumer.get_offset()
assert offset == datetime(1970, 1, 1)
offset = consumer.get_offset()
assert offset == datetime(1970, 1, 1)
def test_consumer_no_commits_returns_unix_start(db):
conn, cur = db
consumer = Consumer(conn, "foo", "offsets")
cur.execute(consumer.create_query)
conn.commit()
offset = consumer.get_offset()
assert offset == datetime(1970, 1, 1)
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
