from __future__ import absolute_import
import unittest
+
try:
from mock import Mock, patch
except ImportError:
from unittest.mock import Mock, patch
-from . import KVStoreMockMixin
-from ..plugins.feature_toggles import FeatureToggles, Features
+from ..plugins.feature_toggles import Actions, Features, FeatureToggles
+from . import KVStoreMockMixin # pylint: disable=no-name-in-module
class SettingsTest(unittest.TestCase, KVStoreMockMixin):
import cherrypy
self.plugin.register_commands()['handle_command'](
- self.mgr, 'disable', ['cephfs'])
+ self.mgr, Actions.DISABLE, [Features.CEPHFS])
with patch.object(self.plugin, '_get_feature_from_request',
return_value=Features.CEPHFS):