def test_status_marathon_job_verbose_when_not_running():
client = mock.create_autospec(marathon.MarathonClient)
app = mock.create_autospec(marathon.models.app.MarathonApp)
client.get_app.return_value = app
service = 'my_service'
instance = 'my_instance'
with contextlib.nested(
mock.patch(
'paasta_tools.marathon_serviceinit.marathon_tools.get_matching_appids'
),
mock.patch(
'paasta_tools.marathon_serviceinit.get_verbose_status_of_marathon_app'
),
mock.patch('paasta_tools.marathon_tools.is_app_id_running',
return_value=False),
) as (
mock_get_matching_appids,
mock_get_verbose_app,
mock_is_app_id_running,
):
mock_get_matching_appids.return_value = ['/app1']
tasks, out = marathon_serviceinit.status_marathon_job_verbose(
service, instance, client)
mock_is_app_id_running.assert_called_once_with('/app1', client)
assert not mock_get_verbose_app.called
assert tasks == []
assert 'not running' in out
def test_status_marathon_job_verbose():
client = mock.create_autospec(marathon.MarathonClient)
app = mock.create_autospec(marathon.models.app.MarathonApp)
client.get_app.return_value = app
service = 'my_service'
instance = 'my_instance'
task = mock.Mock()
with contextlib.nested(
mock.patch(
'paasta_tools.marathon_serviceinit.marathon_tools.get_matching_appids',
autospec=True),
mock.patch(
'paasta_tools.marathon_serviceinit.get_verbose_status_of_marathon_app',
autospec=True),
mock.patch('paasta_tools.marathon_tools.is_app_id_running',
return_value=True,
autospec=True),
) as (
mock_get_matching_appids,
mock_get_verbose_app,
mock_is_app_id_running,
):
mock_get_matching_appids.return_value = ['/app1']
mock_get_verbose_app.return_value = ([task], 'fake_return')
tasks, out = marathon_serviceinit.status_marathon_job_verbose(
service, instance, client)
mock_is_app_id_running.assert_called_once_with('/app1', client)
mock_get_matching_appids.assert_called_once_with(
service, instance, client)
mock_get_verbose_app.assert_called_once_with(app)
assert tasks == [task]
assert 'fake_return' in out
def test_status_marathon_job_verbose_when_not_running():
client = mock.create_autospec(marathon.MarathonClient)
app = mock.create_autospec(marathon.models.app.MarathonApp)
client.get_app.return_value = app
service = 'my_service'
instance = 'my_instance'
with mock.patch(
'paasta_tools.marathon_serviceinit.marathon_tools.get_matching_appids',
autospec=True,
) as mock_get_matching_appids, mock.patch(
'paasta_tools.marathon_serviceinit.get_verbose_status_of_marathon_app',
autospec=True,
) as mock_get_verbose_app, mock.patch(
'paasta_tools.marathon_tools.is_app_id_running',
return_value=False,
autospec=True,
) as mock_is_app_id_running:
mock_get_matching_appids.return_value = ['/app1']
tasks, out = marathon_serviceinit.status_marathon_job_verbose(
service,
instance,
client,
'fake_cluster',
'/nail/blah',
)
mock_is_app_id_running.assert_called_once_with('/app1', client)
assert not mock_get_verbose_app.called
assert tasks == []
assert 'not running' in out
def test_status_marathon_job_verbose():
client = mock.create_autospec(marathon.MarathonClient)
app = mock.create_autospec(marathon.models.app.MarathonApp)
client.get_app.return_value = app
service = 'my_service'
instance = 'my_instance'
task = mock.Mock()
with mock.patch(
'paasta_tools.marathon_serviceinit.marathon_tools.get_matching_appids', autospec=True,
) as mock_get_matching_appids, mock.patch(
'paasta_tools.marathon_serviceinit.get_verbose_status_of_marathon_app', autospec=True,
) as mock_get_verbose_app, mock.patch(
'paasta_tools.marathon_tools.is_app_id_running', return_value=True, autospec=True,
) as mock_is_app_id_running:
mock_get_matching_appids.return_value = ['/app1']
mock_get_verbose_app.return_value = ([task], 'fake_return')
tasks, out = marathon_serviceinit.status_marathon_job_verbose(service, instance, client, 'fake_cluster',
'/nail/blah')
mock_is_app_id_running.assert_called_once_with('/app1', client)
mock_get_matching_appids.assert_called_once_with(service, instance, client)
mock_get_verbose_app.assert_called_once_with(marathon_client=client,
app=app,
service=service,
instance=instance,
cluster='fake_cluster',
soa_dir='/nail/blah')
assert tasks == [task]
assert 'fake_return' in out
def test_status_marathon_job_verbose():
client = mock.create_autospec(marathon.MarathonClient)
clients = mock.Mock(
get_all_clients_for_service=mock.Mock(return_value=[client]), )
app = mock.create_autospec(marathon.models.app.MarathonApp)
client.get_app.return_value = app
service = 'my_service'
instance = 'my_instance'
task = mock.Mock()
with mock.patch(
'paasta_tools.marathon_serviceinit.marathon_tools.get_marathon_apps_with_clients',
autospec=True,
) as mock_get_marathon_apps_with_clients, mock.patch(
'paasta_tools.marathon_serviceinit.marathon_tools.get_matching_apps_with_clients',
autospec=True,
) as mock_get_matching_apps_with_clients, mock.patch(
'paasta_tools.marathon_serviceinit.get_verbose_status_of_marathon_app',
autospec=True,
) as mock_get_verbose_app:
mock_get_matching_apps_with_clients.return_value = [(app, client)]
mock_get_verbose_app.return_value = ([task], 'fake_return')
tasks, out = marathon_serviceinit.status_marathon_job_verbose(
service=service,
instance=instance,
clients=clients,
cluster='fake_cluster',
soa_dir='/nail/blah',
job_config=mock.Mock(),
)
mock_get_marathon_apps_with_clients.assert_called_once_with(
[client],
embed_tasks=True,
)
mock_get_matching_apps_with_clients.assert_called_once_with(
service,
instance,
mock_get_marathon_apps_with_clients.return_value,
)
mock_get_verbose_app.assert_called_once_with(
marathon_client=client,
app=app,
service=service,
instance=instance,
cluster='fake_cluster',
soa_dir='/nail/blah',
)
assert tasks == [task]
assert 'fake_return' in out
def test_status_marathon_job_verbose():
client = mock.create_autospec(marathon.MarathonClient)
app = mock.create_autospec(marathon.models.app.MarathonApp)
client.get_app.return_value = app
service = "my_service"
instance = "my_instance"
task = mock.Mock()
with contextlib.nested(
mock.patch("paasta_tools.marathon_serviceinit.marathon_tools.get_matching_appids"),
mock.patch("paasta_tools.marathon_serviceinit.get_verbose_status_of_marathon_app"),
) as (mock_get_matching_appids, mock_get_verbose_app):
mock_get_matching_appids.return_value = ["/app1"]
mock_get_verbose_app.return_value = ([task], "fake_return")
tasks, out = marathon_serviceinit.status_marathon_job_verbose(service, instance, client)
mock_get_matching_appids.assert_called_once_with(service, instance, client)
mock_get_verbose_app.assert_called_once_with(app)
assert tasks == [task]
assert "fake_return" in out
def test_status_marathon_job_verbose_when_not_running():
client = mock.create_autospec(marathon.MarathonClient)
app = mock.create_autospec(marathon.models.app.MarathonApp)
client.get_app.return_value = app
service = 'my_service'
instance = 'my_instance'
with contextlib.nested(
mock.patch('paasta_tools.marathon_serviceinit.marathon_tools.get_matching_appids', autospec=True),
mock.patch('paasta_tools.marathon_serviceinit.get_verbose_status_of_marathon_app', autospec=True),
mock.patch('paasta_tools.marathon_tools.is_app_id_running', return_value=False, autospec=True),
) as (
mock_get_matching_appids,
mock_get_verbose_app,
mock_is_app_id_running,
):
mock_get_matching_appids.return_value = ['/app1']
tasks, out = marathon_serviceinit.status_marathon_job_verbose(service, instance, client)
mock_is_app_id_running.assert_called_once_with('/app1', client)
assert not mock_get_verbose_app.called
assert tasks == []
assert 'not running' in out
def test_status_marathon_job_verbose():
client = mock.create_autospec(marathon.MarathonClient)
app = mock.create_autospec(marathon.models.app.MarathonApp)
client.get_app.return_value = app
service = 'my_service'
instance = 'my_instance'
task = mock.Mock()
with contextlib.nested(
mock.patch('paasta_tools.marathon_serviceinit.marathon_tools.get_matching_appids'),
mock.patch('paasta_tools.marathon_serviceinit.get_verbose_status_of_marathon_app'),
mock.patch('paasta_tools.marathon_tools.is_app_id_running', return_value=True),
) as (
mock_get_matching_appids,
mock_get_verbose_app,
mock_is_app_id_running,
):
mock_get_matching_appids.return_value = ['/app1']
mock_get_verbose_app.return_value = ([task], 'fake_return')
tasks, out = marathon_serviceinit.status_marathon_job_verbose(service, instance, client)
mock_is_app_id_running.assert_called_once_with('/app1', client)
mock_get_matching_appids.assert_called_once_with(service, instance, client)
mock_get_verbose_app.assert_called_once_with(app)
assert tasks == [task]
assert 'fake_return' in out
