def testGetDepend(self, getStubMock): arbitraryId = '00000000-0000-0000-0000-012345678980' stubMock = mock.Mock() stubMock.GetDepend.return_value = depend_pb2.DependGetDependResponse( depend=depend_pb2.Depend(id=arbitraryId)) getStubMock.return_value = stubMock depend = opencue.api.getDepend(arbitraryId) stubMock.GetDepend.assert_called_with( depend_pb2.DependGetDependRequest(id=arbitraryId), timeout=mock.ANY) self.assertEqual(arbitraryId, depend.id())
def testGetWhatThisDependsOn(self, getStubMock): dependId = 'ddd-dddd-ddd' stubMock = mock.Mock() stubMock.GetWhatThisDependsOn.return_value = job_pb2.JobGetWhatThisDependsOnResponse( depends=depend_pb2.DependSeq(depends=[depend_pb2.Depend(id=dependId)])) getStubMock.return_value = stubMock job = opencue.wrappers.job.Job( job_pb2.Job(name=TEST_JOB_NAME)) depends = job.getWhatThisDependsOn() stubMock.GetWhatThisDependsOn.assert_called_with( job_pb2.JobGetWhatThisDependsOnRequest(job=job.data), timeout=mock.ANY) self.assertEqual(len(depends), 1) self.assertEqual(depends[0].id(), dependId)
def testGetWhatThisDependsOn(self, getStubMock): dependId = 'ddd-dddd-ddd' stubMock = mock.Mock() stubMock.GetWhatThisDependsOn.return_value = job_pb2.FrameGetWhatThisDependsOnResponse( depends=depend_pb2.DependSeq(depends=[depend_pb2.Depend(id=dependId)])) getStubMock.return_value = stubMock frame = opencue.wrappers.frame.Frame( job_pb2.Frame(name=TEST_FRAME_NAME, state=job_pb2.RUNNING)) depends = frame.getWhatThisDependsOn() stubMock.GetWhatThisDependsOn.assert_called_with( job_pb2.FrameGetWhatThisDependsOnRequest(frame=frame.data), timeout=mock.ANY) self.assertEqual(len(depends), 1) self.assertEqual(depends[0].id(), dependId)
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)
def testGetWhatDependsOnThis(self, getStubMock): dependId = 'dddd-ddd-dddd' stubMock = mock.Mock() stubMock.GetWhatDependsOnThis.return_value = job_pb2.LayerGetWhatDependsOnThisResponse( depends=depend_pb2.DependSeq(depends=[depend_pb2.Depend(id=dependId)])) getStubMock.return_value = stubMock layer = opencue.wrappers.layer.Layer( job_pb2.Layer(name=TEST_LAYER_NAME)) depends = layer.getWhatDependsOnThis() stubMock.GetWhatDependsOnThis.assert_called_with( job_pb2.LayerGetWhatDependsOnThisRequest(layer=layer.data), timeout=mock.ANY) self.assertEqual(len(depends), 1) self.assertEqual(depends[0].id(), dependId)
def testCreateDependencyOnFrame(self, getStubMock): dependId = 'ddd-dddd-ddd' dependFrame = 'testFrame' stubMock = mock.Mock() stubMock.CreateDependencyOnFrame.return_value = job_pb2.JobCreateDependencyOnFrameResponse( depend=depend_pb2.Depend(id=dependId)) getStubMock.return_value = stubMock onFrame = job_pb2.Frame(name=dependFrame) job = opencue.wrappers.job.Job( job_pb2.Job(name=TEST_JOB_NAME)) depend = job.createDependencyOnFrame(onFrame) stubMock.CreateDependencyOnFrame.assert_called_with( job_pb2.JobCreateDependencyOnFrameRequest(job=job.data, frame=onFrame), timeout=mock.ANY) self.assertEqual(depend.id(), dependId)
def testCreateDependencyOnLayer(self, getStubMock): dependId = 'ddd-dddd-ddd' dependLayer = 'testLayer' stubMock = mock.Mock() stubMock.CreateDependencyOnLayer.return_value = job_pb2.JobCreateDependencyOnLayerResponse( depend=depend_pb2.Depend(id=dependId)) getStubMock.return_value = stubMock onLayer = job_pb2.Layer(name=dependLayer) job = opencue.wrappers.job.Job( job_pb2.Job(name=TEST_JOB_NAME)) depend = job.createDependencyOnLayer(onLayer) stubMock.CreateDependencyOnLayer.assert_called_with( job_pb2.JobCreateDependencyOnLayerRequest(job=job.data, layer=onLayer), timeout=mock.ANY) self.assertEqual(depend.id(), dependId)
def testCreateDependencyOnFrame(self, getStubMock): dependId = 'dddd-ddd-dddd' frameId = 'ffff-fff-ffff' stubMock = mock.Mock() stubMock.CreateDependOnFrame.return_value = job_pb2.LayerCreateDependOnFrameResponse( depend=depend_pb2.Depend(id=dependId)) getStubMock.return_value = stubMock layer = opencue.wrappers.layer.Layer( job_pb2.Layer(name=TEST_LAYER_NAME)) frame = job_pb2.Frame(id=frameId) depend = layer.createDependencyOnFrame(frame) stubMock.CreateDependOnFrame.assert_called_with( job_pb2.LayerCreateDependOnFrameRequest(layer=layer.data, frame=frame), timeout=mock.ANY) self.assertEqual(depend.id(), dependId)
def testCreateDependencyOnLayer(self, getStubMock): dependId = 'dddd-ddd-dddd' layerId = 'llll-lll-llll' stubMock = mock.Mock() stubMock.CreateDependOnLayer.return_value = job_pb2.LayerCreateDependOnLayerResponse( depend=depend_pb2.Depend(id=dependId)) getStubMock.return_value = stubMock layer = opencue.wrappers.layer.Layer( job_pb2.Layer(name=TEST_LAYER_NAME)) dependLayer = job_pb2.Layer(id=layerId) depend = layer.createDependencyOnLayer(dependLayer) stubMock.CreateDependOnLayer.assert_called_with( job_pb2.LayerCreateDependOnLayerRequest(layer=layer.data, depend_on_layer=dependLayer), timeout=mock.ANY) self.assertEqual(depend.id(), dependId)
def testCreateDependencyOnJob(self, getStubMock): dependId = 'dddd-ddd-dddd' jobId = 'jjjj-jjj-jjjj' stubMock = mock.Mock() stubMock.CreateDependOnJob.return_value = job_pb2.LayerCreateDependOnJobResponse( depend=depend_pb2.Depend(id=dependId)) getStubMock.return_value = stubMock layer = opencue.wrappers.layer.Layer( job_pb2.Layer(name=TEST_LAYER_NAME)) job = job_pb2.Job(id=jobId) depend = layer.createDependencyOnJob(job) stubMock.CreateDependOnJob.assert_called_with( job_pb2.LayerCreateDependOnJobRequest(layer=layer.data, job=job), timeout=mock.ANY) self.assertEqual(depend.id(), dependId)
def testCreateDependencyOnJob(self, getStubMock): dependId = 'ddd-dddd-ddd' stubMock = mock.Mock() stubMock.CreateDependencyOnJob.return_value = \ job_pb2.FrameCreateDependencyOnJobResponse(depend=depend_pb2.Depend(id=dependId)) getStubMock.return_value = stubMock dependJobName = 'jobDependTest' frame = opencue.wrappers.frame.Frame( job_pb2.Frame(name=TEST_FRAME_NAME, state=job_pb2.RUNNING)) dependOnJob = opencue.wrappers.job.Job(job_pb2.Job(name=dependJobName)) depend = frame.createDependencyOnJob(dependOnJob) stubMock.CreateDependencyOnJob.assert_called_with( job_pb2.FrameCreateDependencyOnJobRequest(frame=frame.data, job=dependOnJob.data), timeout=mock.ANY) self.assertEqual(depend.id(), dependId)
def testCreateDependencyOnLayer(self, getStubMock): dependId = 'ddd-dddd-ddd' stubMock = mock.Mock() stubMock.CreateDependencyOnLayer.return_value = \ job_pb2.FrameCreateDependencyOnLayerResponse(depend=depend_pb2.Depend(id=dependId)) getStubMock.return_value = stubMock dependLayerName = 'layerDependTest' frame = opencue.wrappers.frame.Frame( job_pb2.Frame(name=TEST_FRAME_NAME, state=job_pb2.RUNNING)) dependOnLayer = job_pb2.Layer(name=dependLayerName) depend = frame.createDependencyOnLayer(dependOnLayer) stubMock.CreateDependencyOnLayer.assert_called_with( job_pb2.FrameCreateDependencyOnLayerRequest(frame=frame.data, layer=dependOnLayer), timeout=mock.ANY) self.assertEqual(depend.id(), dependId)
def testCreateDependencyOnFrame(self, getStubMock): dependId = 'ddd-dddd-ddd' stubMock = mock.Mock() stubMock.CreateDependencyOnFrame.return_value = \ job_pb2.FrameCreateDependencyOnFrameResponse(depend=depend_pb2.Depend(id=dependId)) getStubMock.return_value = stubMock dependFrameName = 'frameDependTest' frame = opencue.wrappers.frame.Frame( job_pb2.Frame(name=TEST_FRAME_NAME)) dependOnFrame = job_pb2.Frame(name=dependFrameName) depend = frame.createDependencyOnFrame(dependOnFrame) stubMock.CreateDependencyOnFrame.assert_called_with( job_pb2.FrameCreateDependencyOnFrameRequest( frame=frame.data, depend_on_frame=dependOnFrame), timeout=mock.ANY) self.assertEqual(depend.id(), dependId)