def getLayers(self): """Returns the list of layers @rtype: list<Layer> @return: List of layers""" response = self.stub.GetLayers( job_pb2.JobGetLayersRequest(job=self.data), timeout=Cuebot.Timeout) layerSeq = response.layers return [layer.Layer(lyr) for lyr in layerSeq.layers]
def getLayers(self): """Returns the list of layers in the job. :rtype: list<opencue.wrappers.layer.Layer> :return: list of layers in the job """ response = self.stub.GetLayers( job_pb2.JobGetLayersRequest(job=self.data), timeout=Cuebot.Timeout) layerSeq = response.layers return [opencue.wrappers.layer.Layer(lyr) for lyr in layerSeq.layers]
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])