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 testGetLayers(self, getStubMock):
layerNames = ['testLayerA', 'testLayerB']
stubMock = mock.Mock()
stubMock.GetLayers.return_value = job_pb2.JobGetLayersResponse(
layers=job_pb2.LayerSeq(layers=[job_pb2.Layer(name=layerNames[0]),
job_pb2.Layer(name=layerNames[1])]))
getStubMock.return_value = stubMock
job = opencue.wrappers.job.Job(
job_pb2.Job(name=TEST_JOB_NAME))
layers = job.getLayers()
stubMock.GetLayers.assert_called_with(
job_pb2.JobGetLayersRequest(job=job.data), timeout=mock.ANY)
self.assertTrue(len(layers), 2)
self.assertTrue(layers[0].name(), layerNames[0])
self.assertTrue(layers[1].name(), layerNames[1])
def testEat(self, getStubMock):
stubMock = mock.Mock()
stubMock.EatFrames.return_value = job_pb2.LayerEatFramesResponse()
getStubMock.return_value = stubMock
layer = opencue.wrappers.layer.Layer(
job_pb2.Layer(name=TEST_LAYER_NAME))
layer.eat()
stubMock.EatFrames.assert_called_with(
job_pb2.LayerEatFramesRequest(layer=layer.data), timeout=mock.ANY)
def testGetLayer(self, getStubMock):
stubMock = mock.Mock()
stubMock.GetLayer.return_value = host_pb2.ProcGetLayerResponse(
layer=job_pb2.Layer(name=TEST_LAYER_NAME))
getStubMock.return_value = stubMock
proc = opencue.wrappers.proc.Proc(host_pb2.Proc(name=TEST_PROC_NAME))
layer = proc.getLayer()
stubMock.GetLayer.assert_called_with(
host_pb2.ProcGetLayerRequest(proc=proc.data), timeout=mock.ANY)
self.assertEqual(layer.name(), TEST_LAYER_NAME)
def testFindLayer(self, getStubMock):
stubMock = mock.Mock()
stubMock.FindLayer.return_value = job_pb2.LayerFindLayerResponse(
layer=job_pb2.Layer(name=TEST_LAYER_NAME))
getStubMock.return_value = stubMock
layer = opencue.api.findLayer(TEST_JOB_NAME, TEST_LAYER_NAME)
stubMock.FindLayer.assert_called_with(
job_pb2.LayerFindLayerRequest(job=TEST_JOB_NAME, layer=TEST_LAYER_NAME),
timeout=mock.ANY)
self.assertEqual(TEST_LAYER_NAME, layer.name())
def testGetLayer(self, getStubMock):
arbitraryId = '00000000-0000-0000-0000-012345678980'
stubMock = mock.Mock()
stubMock.GetLayer.return_value = job_pb2.LayerGetLayerResponse(
layer=job_pb2.Layer(id=arbitraryId))
getStubMock.return_value = stubMock
layer = opencue.api.getLayer(arbitraryId)
stubMock.GetLayer.assert_called_with(
job_pb2.LayerGetLayerRequest(id=arbitraryId), timeout=mock.ANY)
self.assertEqual(arbitraryId, layer.id())
def testSetTags(self, getStubMock):
tags = ['cloud', 'local']
stubMock = mock.Mock()
stubMock.SetTags.return_value = job_pb2.LayerSetTagsResponse()
getStubMock.return_value = stubMock
layer = opencue.wrappers.layer.Layer(
job_pb2.Layer(name=TEST_LAYER_NAME))
layer.setTags(tags)
stubMock.SetTags.assert_called_with(
job_pb2.LayerSetTagsRequest(layer=layer.data, tags=tags), timeout=mock.ANY)
def testEnableMemoryOptimizerFalse(self, getStubMock):
stubMock = mock.Mock()
stubMock.EnableMemoryOptimizer.return_value = job_pb2.LayerEnableMemoryOptimizerResponse()
getStubMock.return_value = stubMock
layer = opencue.wrappers.layer.Layer(
job_pb2.Layer(name=TEST_LAYER_NAME))
layer.enableMemoryOptimizer(False)
stubMock.EnableMemoryOptimizer.assert_called_with(
job_pb2.LayerEnableMemoryOptimizerRequest(layer=layer.data, value=False),
timeout=mock.ANY)
def testSetThreadable(self, getStubMock):
stubMock = mock.Mock()
stubMock.SetThreadable.return_value = job_pb2.LayerSetThreadableResponse()
getStubMock.return_value = stubMock
value = True
layer = opencue.wrappers.layer.Layer(
job_pb2.Layer(name=TEST_LAYER_NAME))
layer.setThreadable(value)
stubMock.SetThreadable.assert_called_with(
job_pb2.LayerSetThreadableRequest(layer=layer.data, threadable=value),
timeout=mock.ANY)
def testSetMinMemory(self, getStubMock):
stubMock = mock.Mock()
stubMock.SetMinMemory.return_value = job_pb2.LayerSetMinMemoryResponse()
getStubMock.return_value = stubMock
memory = 2048
layer = opencue.wrappers.layer.Layer(
job_pb2.Layer(name=TEST_LAYER_NAME))
layer.setMinMemory(memory)
stubMock.SetMinMemory.assert_called_with(
job_pb2.LayerSetMinMemoryRequest(layer=layer.data, memory=memory),
timeout=mock.ANY)
def testDropLimit(self, getStubMock):
test_limit_id = 'lll-llll-lll'
stubMock = mock.Mock()
stubMock.DropLimit.return_value = job_pb2.LayerDropLimitResponse()
getStubMock.return_value = stubMock
layer = opencue.wrappers.layer.Layer(
job_pb2.Layer(name=TEST_LAYER_NAME))
layer.dropLimit(test_limit_id)
stubMock.DropLimit.assert_called_with(job_pb2.LayerDropLimitRequest(
layer=layer.data, limit_id=test_limit_id),
timeout=mock.ANY)
def testRegisterOutputPath(self, getStubMock):
stubMock = mock.Mock()
stubMock.RegisterOutputPath.return_value = job_pb2.LayerRegisterOutputPathResponse()
getStubMock.return_value = stubMock
outputPath = '/test/output/path'
layer = opencue.wrappers.layer.Layer(
job_pb2.Layer(name=TEST_LAYER_NAME))
layer.registerOutputPath(outputPath)
stubMock.RegisterOutputPath.assert_called_with(
job_pb2.LayerRegisterOutputPathRequest(layer=layer.data, spec=outputPath),
timeout=mock.ANY)
def testSetMinGpu(self, getStubMock):
stubMock = mock.Mock()
stubMock.SetMinGpu.return_value = job_pb2.LayerSetMinGpuResponse()
getStubMock.return_value = stubMock
testCores = 100
layer = opencue.wrappers.layer.Layer(
job_pb2.Layer(name=TEST_LAYER_NAME))
layer.setMinGpu(testCores)
stubMock.SetMinGpu.assert_called_with(job_pb2.LayerSetMinGpuRequest(
layer=layer.data, gpu=testCores),
timeout=mock.ANY)
def testGetOutputPaths(self, getStubMock):
stubMock = mock.Mock()
stubMock.GetOutputPaths.return_value = job_pb2.LayerGetOutputPathsResponse(
output_paths=[TEST_OUTPUT_PATH])
getStubMock.return_value = stubMock
layer = opencue.wrappers.layer.Layer(
job_pb2.Layer(name=TEST_LAYER_NAME))
outputPaths = layer.getOutputPaths()
stubMock.GetOutputPaths.assert_called_with(
job_pb2.LayerGetOutputPathsRequest(layer=layer.data), timeout=mock.ANY)
self.assertEqual(len(outputPaths), 1)
self.assertEqual(outputPaths[0], TEST_OUTPUT_PATH)
def testSetMaxCores(self, getStubMock):
stubMock = mock.Mock()
stubMock.SetMaxCores.return_value = job_pb2.LayerSetMaxCoresResponse()
getStubMock.return_value = stubMock
testCores = 100
testCoresActual = testCores/100.0
layer = opencue.wrappers.layer.Layer(
job_pb2.Layer(name=TEST_LAYER_NAME))
layer.setMaxCores(testCores)
stubMock.SetMaxCores.assert_called_with(
job_pb2.LayerSetMaxCoresRequest(layer=layer.data, cores=testCoresActual),
timeout=mock.ANY)
def testStaggerFrames(self, getStubMock):
stubMock = mock.Mock()
stubMock.StaggerFrames.return_value = job_pb2.LayerStaggerFramesResponse()
getStubMock.return_value = stubMock
range = '1-10'
stagger = 4
layer = opencue.wrappers.layer.Layer(
job_pb2.Layer(name=TEST_LAYER_NAME))
layer.staggerFrames(range, stagger)
stubMock.StaggerFrames.assert_called_with(
job_pb2.LayerStaggerFramesRequest(layer=layer.data, range=range, stagger=stagger),
timeout=mock.ANY)
def testGetWhatThisDependsOn(self, getStubMock):
dependId = 'dddd-ddd-dddd'
stubMock = mock.Mock()
stubMock.GetWhatThisDependsOn.return_value = job_pb2.LayerGetWhatThisDependsOnResponse(
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.getWhatThisDependsOn()
stubMock.GetWhatThisDependsOn.assert_called_with(
job_pb2.LayerGetWhatThisDependsOnRequest(layer=layer.data),
timeout=mock.ANY)
self.assertEqual(len(depends), 1)
self.assertEqual(depends[0].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 testGetFrames(self, getStubMock):
stubMock = mock.Mock()
stubMock.GetFrames.return_value = job_pb2.LayerGetFramesResponse(
frames=job_pb2.FrameSeq(
frames=[job_pb2.Frame(layer_name=TEST_LAYER_NAME)]))
getStubMock.return_value = stubMock
layer = opencue.wrappers.layer.Layer(
job_pb2.Layer(name=TEST_LAYER_NAME))
frames = layer.getFrames()
stubMock.GetFrames.assert_called_with(job_pb2.LayerGetFramesRequest(
layer=layer.data,
s=opencue.search.FrameSearch.criteriaFromOptions()),
timeout=mock.ANY)
self.assertEqual(len(frames), 1)
self.assertEqual(frames[0].data.layer_name, TEST_LAYER_NAME)
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 testReorderFrames(self, getStubMock):
stubMock = mock.Mock()
stubMock.ReorderFrames.return_value = job_pb2.LayerReorderFramesResponse(
)
getStubMock.return_value = stubMock
frameRange = '1-10'
order = job_pb2.REVERSE
layer = opencue.wrappers.layer.Layer(
job_pb2.Layer(name=TEST_LAYER_NAME))
layer.reorderFrames(frameRange, order)
stubMock.ReorderFrames.assert_called_with(
job_pb2.LayerReorderFramesRequest(layer=layer.data,
range=frameRange,
order=order),
timeout=mock.ANY)
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 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)
