def test_JobManager_get_job_dne(self, mock_tethys_job): mock_args = mock.MagicMock() mock_app_package = mock.MagicMock() mock_args.package = mock_app_package mock_tethys_job.DoesNotExist = TethysJob.DoesNotExist # Restore original exception mock_tethys_job.objects.get_subclass.side_effect = TethysJob.DoesNotExist mock_job_id = 'fooid' mock_user = '******' mgr = JobManager(mock_args) ret = mgr.get_job(job_id=mock_job_id, user=mock_user) self.assertEquals(ret, None) mock_tethys_job.objects.get_subclass.assert_called_once_with(id='fooid', label=mock_app_package, user='******')
def test_JobManager_get_job_dne(self, mock_tethys_job): mock_args = mock.MagicMock() mock_app_package = mock.MagicMock() mock_args.package = mock_app_package mock_tethys_job.DoesNotExist = TethysJob.DoesNotExist # Restore original exception mock_tethys_job.objects.get_subclass.side_effect = TethysJob.DoesNotExist mock_job_id = 'fooid' mock_user = '******' mgr = JobManager(mock_args) ret = mgr.get_job(job_id=mock_job_id, user=mock_user) self.assertEquals(ret, None) mock_tethys_job.objects.get_subclass.assert_called_once_with(id='fooid', label=mock_app_package, user='******')
def test_JobManager_get_job(self, mock_tethys_job): mock_args = mock.MagicMock() mock_app_package = mock.MagicMock() mock_args.package = mock_app_package mock_jobs = mock.MagicMock() mock_tethys_job.objects.get_subclass.return_value = mock_jobs mock_job_id = 'fooid' mock_user = '******' mgr = JobManager(mock_args) ret = mgr.get_job(job_id=mock_job_id, user=mock_user) self.assertEquals(ret, mock_jobs) mock_tethys_job.objects.get_subclass.assert_called_once_with(id='fooid', label=mock_app_package, user='******')
def test_JobManager_get_job(self, mock_tethys_job): mock_args = mock.MagicMock() mock_app_package = mock.MagicMock() mock_args.package = mock_app_package mock_jobs = mock.MagicMock() mock_tethys_job.objects.get_subclass.return_value = mock_jobs mock_job_id = 'fooid' mock_user = '******' mgr = JobManager(mock_args) ret = mgr.get_job(job_id=mock_job_id, user=mock_user) self.assertEquals(ret, mock_jobs) mock_tethys_job.objects.get_subclass.assert_called_once_with(id='fooid', label=mock_app_package, user='******')