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])
def testGetActiveShows(self, getStubMock): show1Name = 'first-show' show2Name = 'second-show' stubMock = mock.Mock() stubMock.GetActiveShows.return_value = show_pb2.ShowGetActiveShowsResponse( shows=show_pb2.ShowSeq( shows=[show_pb2.Show(name=show1Name), show_pb2.Show(name=show2Name)])) getStubMock.return_value = stubMock showList = opencue.api.getActiveShows() stubMock.GetActiveShows.assert_called_with( show_pb2.ShowGetActiveShowsRequest(), timeout=mock.ANY) self.assertEqual([show1Name, show2Name], [show.name() for show in showList])