def test_paasta_status(system_paasta_config):
fake_dict = {
"git_sha": "fake_git_sha",
"instance": "fake_instance",
"service": "fake_service"
}
fake_dict2 = {
"error_message": None,
"desired_state": "start",
"app_id": "fake_app_id",
"app_count": 1,
"running_instance_count": 2,
"expected_instance_count": 2,
"deploy_status": "Running",
"bounce_method": "crossover",
}
fake_status_obj = Struct(**fake_dict)
fake_status_obj.marathon = Struct(**fake_dict2)
system_paasta_config = system_paasta_config
with mock.patch('bravado.http_future.HttpFuture.result',
autospec=True) as mock_result:
mock_result.return_value = fake_status_obj
paasta_status_on_api_endpoint(
'fake_cluster',
'fake_service',
'fake_instance',
system_paasta_config,
verbose=False,
)
def test_paasta_status_exception():
system_paasta_config = SystemPaastaConfig({
'api_endpoints': {
'fake_cluster': "http://fake_cluster:5054"
},
'cluster': 'fake_cluster'
}, 'fake_directory')
with mock.patch('paasta_tools.cli.cmds.status.get_paasta_api_client', autospec=True) as mock_get_paasta_api_client:
requests_response = mock.Mock(status_code=500, reason='Internal Server Error')
incoming_response = RequestsResponseAdapter(requests_response)
mock_swagger_client = mock.Mock()
mock_swagger_client.service.status_instance.side_effect = HTTPError(incoming_response)
mock_get_paasta_api_client.return_value = mock_swagger_client
paasta_status_on_api_endpoint('fake_cluster', 'fake_service', 'fake_instance',
system_paasta_config, verbose=False)
def test_paasta_status_exception(system_paasta_config):
system_paasta_config = system_paasta_config
with mock.patch('paasta_tools.cli.cmds.status.get_paasta_api_client',
autospec=True) as mock_get_paasta_api_client:
requests_response = mock.Mock(status_code=500,
reason='Internal Server Error')
incoming_response = RequestsResponseAdapter(requests_response)
mock_swagger_client = mock.Mock()
mock_swagger_client.service.status_instance.side_effect = HTTPError(
incoming_response)
mock_get_paasta_api_client.return_value = mock_swagger_client
paasta_status_on_api_endpoint(
'fake_cluster',
'fake_service',
'fake_instance',
system_paasta_config,
verbose=False,
)
def test_paasta_status():
fake_dict = {"git_sha": "fake_git_sha", "instance": "fake_instance", "service": "fake_service"}
fake_dict2 = {"error_message": None, "desired_state": "start",
"app_id": "fake_app_id", "app_count": 1,
"running_instance_count": 2, "expected_instance_count": 2,
"deploy_status": "Running", "bounce_method": "crossover"}
fake_status_obj = Struct(**fake_dict)
fake_status_obj.marathon = Struct(**fake_dict2)
system_paasta_config = SystemPaastaConfig({
'api_endpoints': {
'fake_cluster': "http://fake_cluster:5054"
},
'cluster': 'fake_cluster'
}, 'fake_directory')
with mock.patch('bravado.http_future.HttpFuture.result', autospec=True) as mock_result:
mock_result.return_value = fake_status_obj
paasta_status_on_api_endpoint('fake_cluster', 'fake_service', 'fake_instance',
system_paasta_config, verbose=False)
def paasta_status_via_api(context, service, instance):
output = []
system_paasta_config = load_system_paasta_config()
exit_code = paasta_status_on_api_endpoint(
cluster=system_paasta_config.get_cluster(),
service=service,
instance=instance,
output=output,
system_paasta_config=system_paasta_config,
verbose=0,
)
print(f"Got exitcode {exit_code} with output:\n{output}")
print() # sacrificial line for behave to eat instead of our output
assert exit_code == 0
assert len(output) > 0
