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