def findFilter(show_name, filter_name): """Returns the matching filter (for testing). @type show_name: str @param show_name: a show name @type filter_name: str @param filter_name: a filter name @rtype: Filter @return: the matching Filter object""" return Filter( Cuebot.getStub('filter').FindFilter(filter_pb2.FilterFindFilterRequest( show=show_name, name=filter_name), timeout=Cuebot.Timeout).filter)
def testFindFilter(self, getStubMock): filterName = 'arbitrary-name' stubMock = mock.Mock() stubMock.FindFilter.return_value = filter_pb2.FilterFindFilterResponse( filter=filter_pb2.Filter(name=filterName)) getStubMock.return_value = stubMock filterReturned = opencue.api.findFilter(TEST_SHOW_NAME, filterName) stubMock.FindFilter.assert_called_with( filter_pb2.FilterFindFilterRequest(show=TEST_SHOW_NAME, name=filterName), timeout=mock.ANY) self.assertEqual(filterName, filterReturned.name())