def test_transport_state_set_success():
state = TransportState()
state.status = state.ERROR
state.last_check = "foo"
state.retry_number = 5
state.set_success()
assert state.status == state.ONLINE
assert state.last_check is None
assert state.retry_number == -1
def test_transport_state_should_try_time_passed_error():
state = TransportState()
state.status = state.ERROR
state.last_check = timeit.default_timer() - 10
state.retry_number = 1
assert state.should_try() is True
def test_transport_state_set_fail():
state = TransportState()
state.set_fail()
assert state.status == state.ERROR
assert state.last_check is not None
assert state.retry_number == 0
def test_transport_state_should_try_online():
state = TransportState()
assert state.should_try() is True