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)