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)
