def setUp(self): self.settings = {'jobfactory.root_dir': '/somedir', 'restricted': False, 'ncpus': 1} self.config = testing.setUp(settings=self.settings) self.config.add_route('status.json', '/status/{jobid}.json') self.jobid = '3ad25048-26f6-11e1-851e-00012e260790' self.post = {'key': 'value'} self.jq = Mock(JobQuery) self.jobquery = Mock(JobQuery) jobmeta = JobMeta(uuid.UUID(self.jobid), owner='bob') self.job = Job(jobmeta, '/mydir', Mock(JobDb)) self.job.jobquery = Mock(return_value=self.jobquery) self.job.db.maxMSLevel.return_value = 1 self.job.db.moleculesTotalCount.return_value = 1 self.jobid2 = 'fe144547-cf16-44bb-8e97-e2ffe5675154' jobmeta2 = JobMeta(uuid.UUID(self.jobid2), owner='bob') self.job2 = Job(jobmeta2, '/mydir', Mock(JobDb)) self.job2.jobquery = Mock(return_value=self.jobquery) self.job2.db.maxMSLevel.return_value = 1 self.job2.db.moleculesTotalCount.return_value = 1 self.rpc = RpcViews(self.job, testing.DummyRequest(post=self.post)) self.rpc.new_job = Mock(return_value=self.job2) self.rpc.job_factory.submitQuery = Mock()
def test_construct(self): request = testing.DummyRequest() rpc = RpcViews(self.job, request) self.assertIsInstance(rpc.job_factory, JobFactory) self.assertEqual(request, rpc.request) self.assertEqual(self.job, rpc.job)
def test_new_job(self): request = testing.DummyRequest() request.user = User('bob', 'Bob Example', '*****@*****.**') parent_job = self.job rpc = RpcViews(parent_job, request) rpc.job_factory.cloneJob = Mock(return_value=self.job2) job = rpc.new_job() self.assertEqual(job, self.job2) rpc.job_factory.cloneJob.assert_called_with(parent_job, 'bob')