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