def setEnabled(self, value):
"""Enables or disables the filter
@type value: bool
@param value: True to enable the filter and false to disable it"""
self.stub.SetEnabled(filter_pb2.FilterSetEnabledRequest(
filter=self.data, enabled=value),
timeout=Cuebot.Timeout)
def testSetEnabled(self, getStubMock):
stubMock = mock.Mock()
stubMock.SetEnabled.return_value = filter_pb2.FilterSetEnabledResponse()
getStubMock.return_value = stubMock
value = True
filterToEnable = opencue.wrappers.filter.Filter(filter_pb2.Filter(name=TEST_FILTER_NAME))
filterToEnable.setEnabled(value)
stubMock.SetEnabled.assert_called_with(
filter_pb2.FilterSetEnabledRequest(
filter=filterToEnable.data, enabled=value), timeout=mock.ANY)
