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)