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)