def createDependencyOnJob(self, job): """Create and return a job on job dependency @type job: Job @param job: the job you want this job to depend on @rtype: Depend @return: The new dependency""" response = self.stub.CreateDependencyOnJob( job_pb2.JobCreateDependencyOnJobRequest(job=self.data, on_job=job), timeout=Cuebot.Timeout) return depend.Depend(response.depend)
def createDependencyOnJob(self, job): """Create and return a job on job dependency. :type job: opencue.wrappers.job.Job :param job: the job you want this job to depend on :rtype: opencue.wrappers.depend.Depend :return: The new dependency""" response = self.stub.CreateDependencyOnJob( job_pb2.JobCreateDependencyOnJobRequest(job=self.data, on_job=job.data), timeout=Cuebot.Timeout) return opencue.wrappers.depend.Depend(response.depend)
def testCreateDependencyOnJob(self, getStubMock): dependId = 'ddd-dddd-ddd' stubMock = mock.Mock() stubMock.CreateDependencyOnJob.return_value = job_pb2.JobCreateDependencyOnJobResponse( depend=depend_pb2.Depend(id=dependId)) getStubMock.return_value = stubMock onJob = job_pb2.Job(name=TEST_JOB_NAME+"Depend") job = opencue.wrappers.job.Job( job_pb2.Job(name=TEST_JOB_NAME)) depend = job.createDependencyOnJob(onJob) stubMock.CreateDependencyOnJob.assert_called_with( job_pb2.JobCreateDependencyOnJobRequest(job=job.data, on_job=onJob), timeout=mock.ANY) self.assertEqual(depend.id(), dependId)