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 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)
