def test_serve_ticket():
lock_store = InMemoryLockStore()
conversation_id = "my id 1"
lock = lock_store.create_lock(conversation_id)
lock_store.save_lock(lock)
# issue ticket with long lifetime
ticket_0 = lock_store.issue_ticket(conversation_id, 10)
assert ticket_0 == 0
lock = lock_store.get_lock(conversation_id)
assert lock.last_issued == ticket_0
assert lock.now_serving == ticket_0
assert lock.is_someone_waiting()
# issue another ticket
ticket_1 = lock_store.issue_ticket(conversation_id, 10)
# finish serving ticket_0
lock_store.finish_serving(conversation_id, ticket_0)
lock = lock_store.get_lock(conversation_id)
assert lock.last_issued == ticket_1
assert lock.now_serving == ticket_1
assert lock.is_someone_waiting()
# serve second ticket and no one should be waiting
lock_store.finish_serving(conversation_id, ticket_1)
lock = lock_store.get_lock(conversation_id)
assert not lock.is_someone_waiting()
def test_create_lock_store():
lock_store = InMemoryLockStore()
conversation_id = "my id 0"
# create and lock
lock = lock_store.create_lock(conversation_id)
lock_store.save_lock(lock)
lock = lock_store.get_lock(conversation_id)
assert lock
assert lock.conversation_id == conversation_id
def test_lock_expiration():
lock_store = InMemoryLockStore()
conversation_id = "my id 2"
lock = lock_store.create_lock(conversation_id)
lock_store.save_lock(lock)
# issue ticket with long lifetime
ticket = lock.issue_ticket(10)
assert ticket == 0
assert not lock._ticket_for_ticket_number(ticket).has_expired()
# issue ticket with short lifetime
ticket = lock.issue_ticket(0.00001)
time.sleep(0.00002)
assert ticket == 1
assert lock._ticket_for_ticket_number(ticket) is None
# newly assigned ticket should get number 1 again
assert lock.issue_ticket(10) == 1
