def getServiceOverrides(self):
"""Returns a list of service overrides on the show.
@rtype: list<ServiceOverride>
@return: a list of service override objects
"""
serviceOverrideSeq = self.stub.GetServiceOverrides(
show_pb2.ShowGetServiceOverridesRequest(show=self.data),
timeout=Cuebot.Timeout).service_overrides
return serviceOverrideSeq.service_overrides
def testGetServiceOverrides(self, getStubMock):
stubMock = mock.Mock()
stubMock.GetServiceOverrides.return_value = show_pb2.ShowGetServiceOverridesResponse(
service_overrides=service_pb2.ServiceOverrideSeq(
service_overrides=[
service_pb2.ServiceOverride(data=service_pb2.Service(name=TEST_SERVICE_NAME))
]))
getStubMock.return_value = stubMock
show = opencue.wrappers.show.Show(show_pb2.Show(name=TEST_SHOW_NAME))
overrides = show.getServiceOverrides()
stubMock.GetServiceOverrides.assert_called_with(
show_pb2.ShowGetServiceOverridesRequest(show=show.data),
timeout=mock.ANY)
self.assertEqual(overrides[0].data.name, TEST_SERVICE_NAME)
