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