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)