def test_wait_for_port_to_be_open_when_port_failed_to_open(self):
def mock_is_port_open(unused_port):
return False
def mock_sleep(unused_time):
mock_sleep.sleep_time += 1
def mock_exit(unused_exit_code):
return
mock_sleep.sleep_time = 0
is_port_open_swap = self.swap(common, 'is_port_open',
mock_is_port_open)
sleep_swap = self.swap_with_checks(time, 'sleep', mock_sleep)
exit_swap = self.swap_with_checks(sys, 'exit', mock_exit)
with is_port_open_swap, sleep_swap, exit_swap:
run_e2e_tests.wait_for_port_to_be_open(1)
self.assertEqual(mock_sleep.sleep_time,
run_e2e_tests.MAX_WAIT_TIME_FOR_PORT_TO_OPEN_SECS)
def test_wait_for_port_to_be_open_when_port_successfully_opened(self):
def mock_is_port_open(unused_port):
mock_is_port_open.wait_time += 1
if mock_is_port_open.wait_time > 10:
return True
return False
mock_is_port_open.wait_time = 0
def mock_sleep(unused_time):
mock_sleep.called_times += 1
return
mock_sleep.called_times = 0
is_port_open_swap = self.swap_with_checks(
common, 'is_port_open', mock_is_port_open)
sleep_swap = self.swap_with_checks(time, 'sleep', mock_sleep)
with is_port_open_swap, sleep_swap:
run_e2e_tests.wait_for_port_to_be_open(1)
self.assertEqual(mock_is_port_open.wait_time, 11)
self.assertEqual(mock_sleep.called_times, 10)
