def createFilter(self, name):
"""Create a filter on the show
@type: string
@param: Name of the filter to create
@rtype: show_pb2.ShowCreateFilterResponse
@return: response is empty
"""
response = self.stub.CreateFilter(show_pb2.ShowCreateFilterRequest(
show=self.data, name=name), timeout=Cuebot.Timeout)
return opencue.wrappers.filter.Filter(response.filter)
def createFilter(self, name):
"""Creates a filter on the show.
:type name: str
:param name: name of the filter to create
:rtype: opencue.wrappers.filter.Filter
:return: the new filter object
"""
response = self.stub.CreateFilter(show_pb2.ShowCreateFilterRequest(
show=self.data, name=name), timeout=Cuebot.Timeout)
return opencue.wrappers.filter.Filter(response.filter)
def testCreateFilter(self, getStubMock):
stubMock = mock.Mock()
stubMock.CreateFilter.return_value = show_pb2.ShowCreateFilterResponse(
filter=filter_pb2.Filter(name=TEST_FILTER_NAME))
getStubMock.return_value = stubMock
show = opencue.wrappers.show.Show(show_pb2.Show(name=TEST_SHOW_NAME))
filter = show.createFilter(TEST_FILTER_NAME)
stubMock.CreateFilter.assert_called_with(
show_pb2.ShowCreateFilterRequest(show=show.data, name=TEST_FILTER_NAME),
timeout=mock.ANY)
self.assertEqual(filter.name(), TEST_FILTER_NAME)
