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