def test_get_job_status_returns_completed_if_deployment_job_is_queued(
self):
deployment = Mock()
deployment.get_job_status.return_value = 'Queued'
deployment_wrapper = DeploymentWrapper(deployment)
self.assertEqual(deployment_wrapper.get_job_status(), 'Queued')
def test_get_job_logs_returns_deployment_object_get_job_logs_return_value(
self):
deployment = Mock()
deployment.get_job_logs.return_value = 'some fancy logs'
deployment_wrapper = DeploymentWrapper(deployment)
self.assertEqual(deployment_wrapper.get_job_logs(), 'some fancy logs')
def test_is_job_complete_returns_false_if_deployment_job_is_not_complete(
self):
deployment = Mock()
deployment.is_job_complete.return_value = False
deployment_wrapper = DeploymentWrapper(deployment)
self.assertFalse(deployment_wrapper.is_job_complete())
def test_get_job_logs_returns_error_message_if_deployment_object_does_not_have_get_job_logs_method(
self):
deployment = object()
deployment_wrapper = DeploymentWrapper(deployment)
self.assertEqual(
deployment_wrapper.get_job_logs(),
'Current deployment method does not support get_job_logs()')
def test_stream_job_logs_returns_deployment_object_stream_job_logs_different_return_value(
self):
deployment = Mock()
deployment.stream_job_logs.return_value = 'some fancier logs'
deployment_wrapper = DeploymentWrapper(deployment)
self.assertEqual(deployment_wrapper.stream_job_logs(),
'some fancier logs')
def test_wait_for_deployment_to_complete_only_sleeps_for_specified_time_if_job_not_complete(
self, mock_time_sleep):
deployment = Mock()
states = [False, True]
deployment.job_name.return_value = 'whatever'
deployment.is_job_complete = lambda: states.pop(0)
deployment_wrapper = DeploymentWrapper(deployment)
deployment_wrapper.wait_for_deployment_to_complete(2)
mock_time_sleep.assert_called_once_with(2)
def test_stream_job_logs_raises_exception_if_deployment_object_does_not_have_stream_job_logs_method(
self):
deployment = object()
deployment_wrapper = DeploymentWrapper(deployment)
with self.assertRaises(NotImplementedError) as error_context:
deployment_wrapper.stream_job_logs()
self.assertIn(
'Current deployment method does not support stream_job_logs()',
error_context.exception.args)
def deploy_job(pipeline_context_wrapper, job_name, job_params):
from foundations_contrib.global_state import deployment_manager, log_manager
from foundations_contrib.deployment_wrapper import DeploymentWrapper
logger = log_manager.get_logger(__name__)
logger.info("Job submission started. Ctrl-C to cancel.")
job_deployment = deployment_manager.simple_deploy(pipeline_context_wrapper, job_name, job_params)
return DeploymentWrapper(job_deployment)
def test_job_name_returns_deployment_job_name(self):
deployment = Mock()
deployment.job_name.return_value = 'job1'
deployment_wrapper = DeploymentWrapper(deployment)
self.assertEqual(deployment_wrapper.job_name(), 'job1')
def test_is_job_complete_returns_true_if_deployment_job_is_complete(self):
deployment = Mock()
deployment.is_job_complete.return_value = True
deployment_wrapper = DeploymentWrapper(deployment)
self.assertTrue(deployment_wrapper.is_job_complete())