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