def getShows():
"""Returns a list of show objects.
@rtype: list<Show>
@return: a List of show objects"""
showSeq = Cuebot.getStub('show').GetShows(show_pb2.ShowGetShowsRequest(),
timeout=Cuebot.Timeout).shows
return [Show(s) for s in showSeq.shows]
def testGetShows(self, getStubMock):
stubMock = mock.Mock()
stubMock.GetShows.return_value = show_pb2.ShowGetShowsResponse(
shows=show_pb2.ShowSeq(shows=[show_pb2.Show(name=TEST_SHOW_NAME)]))
getStubMock.return_value = stubMock
showList = opencue.api.getShows()
stubMock.GetShows.assert_called_with(show_pb2.ShowGetShowsRequest(), timeout=mock.ANY)
self.assertEqual([TEST_SHOW_NAME], [show.name() for show in showList])
