def test_request_tracker_start_adds_request():
req_tracker = RequestTimeTracker(INSTANCE_COUNT)
digest = "digest"
now = 1.0
req_tracker.start(digest, now)
assert digest in req_tracker
assert req_tracker.started(digest) == now
assert digest in [
digest for digest, _ in req_tracker.unhandled_unordered()
]
assert digest not in req_tracker.handled_unordered()
def test_request_tracker_order_by_backup_returns_time_to_order():
req_tracker = RequestTimeTracker(INSTANCE_COUNT)
digest = "digest"
now = 1.0
req_tracker.start(digest, now)
tto = req_tracker.order(1, digest, now + 5)
assert digest in [
digest for digest, _ in req_tracker.unhandled_unordered()
]
assert digest not in req_tracker.handled_unordered()
assert int(tto) == 5
def test_request_tracker_handle_makes_request_handled_unordered():
req_tracker = RequestTimeTracker(INSTANCE_COUNT)
digest = "digest"
now = 1.0
req_tracker.start(digest, now)
req_tracker.handle(digest)
assert digest in req_tracker
assert digest not in [
digest for digest, _ in req_tracker.unhandled_unordered()
]
assert digest in req_tracker.handled_unordered()
def test_request_tracker_reset_clears_all_requests():
req_tracker = RequestTimeTracker(INSTANCE_COUNT)
digest = "digest"
now = 1.0
req_tracker.start(digest, now)
req_tracker.handle(digest)
req_tracker.reset()
assert digest not in req_tracker
assert digest not in req_tracker.unordered()
assert digest not in [
digest for digest, _ in req_tracker.unhandled_unordered()
]
assert digest not in req_tracker.handled_unordered()
