def test_wait_for_create_fast(self): fake_id = "my_created" fake_client = mock.Mock( spec="paasta_tools.setup_marathon_job.MarathonClient") fake_is_app_running_values = iter([True]) with mock.patch("paasta_tools.marathon_tools.is_app_id_running", autospec=True) as is_app_id_running_patch, mock.patch( "time.sleep", autospec=True) as sleep_patch: is_app_id_running_patch.side_effect = fake_is_app_running_values bounce_lib.wait_for_create(fake_id, fake_client) assert sleep_patch.call_count == 0 assert is_app_id_running_patch.call_count == 1
def test_wait_for_create_slow(self): fake_id = 'my_created' fake_client = mock.Mock(spec='paasta_tools.setup_marathon_job.MarathonClient') fake_is_app_running_values = iter([False, False, True]) with mock.patch( 'paasta_tools.marathon_tools.is_app_id_running', autospec=True, ) as is_app_id_running_patch, mock.patch( 'time.sleep', autospec=True, ) as sleep_patch: is_app_id_running_patch.side_effect = fake_is_app_running_values bounce_lib.wait_for_create(fake_id, fake_client) assert sleep_patch.call_count == 2 assert is_app_id_running_patch.call_count == 3
def test_wait_for_create_fast(self): fake_id = 'my_created' fake_client = mock.Mock(spec='paasta_tools.setup_marathon_job.MarathonClient') fake_is_app_running_values = iter([True]) with contextlib.nested( mock.patch('paasta_tools.marathon_tools.is_app_id_running', autospec=True), mock.patch('time.sleep', autospec=True), ) as ( is_app_id_running_patch, sleep_patch, ): is_app_id_running_patch.side_effect = fake_is_app_running_values bounce_lib.wait_for_create(fake_id, fake_client) assert sleep_patch.call_count == 0 assert is_app_id_running_patch.call_count == 1