def test_wait_for_delete_fast(self): fake_id = "my_deleted" fake_client = mock.Mock( spec="paasta_tools.setup_marathon_job.MarathonClient") fake_is_app_running_values = iter([False]) 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_delete(fake_id, fake_client) assert sleep_patch.call_count == 0 assert is_app_id_running_patch.call_count == 1
def test_wait_for_delete_slow(self): fake_id = 'my_deleted' fake_client = mock.Mock(spec='paasta_tools.setup_marathon_job.MarathonClient') fake_is_app_running_values = iter([True, True, False]) 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_delete(fake_id, fake_client) assert sleep_patch.call_count == 2 assert is_app_id_running_patch.call_count == 3
def test_wait_for_delete_fast(self): fake_id = 'my_deleted' fake_client = mock.Mock(spec='paasta_tools.setup_marathon_job.MarathonClient') fake_is_app_running_values = iter([False]) 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_delete(fake_id, fake_client) assert sleep_patch.call_count == 0 assert is_app_id_running_patch.call_count == 1