def test_paasta_docker_stop(mock_get_status_for_instance, mock_pick_slave_from_status, mock_get_container_name, mock_subprocess): mock_status = mock.Mock(marathon=mock.Mock(app_id='appID1')) mock_get_status_for_instance.return_value = mock_status mock_args = mock.Mock(cluster='cluster1', service='mock_service', instance='mock_instance', host='host1', mesos_id=None) mock_pick_slave_from_status.return_value = 'host1' docker_stop.paasta_docker_stop(mock_args) mock_get_status_for_instance.assert_called_with(cluster='cluster1', service='mock_service', instance='mock_instance') mock_pick_slave_from_status.assert_called_with(status=mock_status, host='host1') mock_get_container_name.assert_called_with('appID1', slave_hostname='host1', task_id=None) expected = [ "ssh", "-o", "LogLevel=QUIET", "-tA", mock_pick_slave_from_status.return_value, "sudo docker stop {0}".format(mock_get_container_name.return_value) ] mock_subprocess.call.assert_called_with(expected)
def test_paasta_docker_stop( mock_get_task_from_instance, mock_get_container_name, mock_subprocess, ): mock_task = mock.Mock(slave={'hostname': 'host1'}) mock_get_task_from_instance.return_value = mock_task mock_args = mock.Mock( cluster='cluster1', service='mock_service', instance='mock_instance', host='host1', mesos_id=None, ) docker_stop.paasta_docker_stop(mock_args) mock_get_task_from_instance.assert_called_with( cluster='cluster1', service='mock_service', instance='mock_instance', slave_hostname='host1', task_id=None, ) mock_get_container_name.assert_called_with(mock_task) expected = [ "ssh", "-o", "LogLevel=QUIET", "-tA", 'host1', f"sudo docker stop {mock_get_container_name.return_value}", ] mock_subprocess.call.assert_called_with(expected)
def test_paasta_docker_stop(mock_get_task_from_instance, mock_get_container_name, mock_subprocess): mock_task = mock.Mock(slave={'hostname': 'host1'}) mock_get_task_from_instance.return_value = mock_task mock_args = mock.Mock(cluster='cluster1', service='mock_service', instance='mock_instance', host='host1', mesos_id=None) docker_stop.paasta_docker_stop(mock_args) mock_get_task_from_instance.assert_called_with(cluster='cluster1', service='mock_service', instance='mock_instance', slave_hostname='host1', task_id=None) mock_get_container_name.assert_called_with(mock_task) expected = ["ssh", "-o", "LogLevel=QUIET", "-tA", 'host1', "sudo docker stop {0}".format(mock_get_container_name.return_value)] mock_subprocess.call.assert_called_with(expected)