def setUp(self):
super().setUp()
self.owner = Cluster.get_or_create_owner(Cluster.load())
self.settings_service = DummySettingsService()
self.db_service = DummyDBService()
self.settings_service.option_manager = OptionManager()
self.db_service.option_manager = OptionManager()
self.settings_service.setup()
self.db_service.setup()
def setUp(self):
super().setUp()
self.owner = Owner.objects.get(name=Cluster.load().uuid)
self.settings_service = DummySettingsService()
self.db_service = DummyDBService()
self.settings_service.option_manager = OptionManager()
self.db_service.option_manager = OptionManager()
self.settings_service.setup()
self.db_service.setup()
class TestOptionManager(BaseTest):
def setUp(self):
self.manager = OptionManager()
super().setUp()
def test_subscribe(self):
self.assertEqual(len(self.manager.state), 0)
self.manager.subscribe(AdminViewEnabled)
assert len(self.manager.state) == 1
assert len(self.manager.items) == 1
assert len(self.manager.keys) == 1
assert len(self.manager.values) == 1
assert AdminViewEnabled.key in self.manager.state
assert self.manager.state[AdminViewEnabled.key] == AdminViewEnabled
# Adding the same event
self.manager.subscribe(AdminViewEnabled)
assert len(self.manager.state) == 1
assert len(self.manager.items) == 1
assert len(self.manager.items) == 1
assert len(self.manager.keys) == 1
assert len(self.manager.values) == 1
# Adding new event
self.manager.subscribe(PasswordLength)
assert len(self.manager.state) == 2
assert len(self.manager.items) == 2
assert len(self.manager.items) == 2
assert len(self.manager.keys) == 2
assert len(self.manager.values) == 2
# Adding new event with same event type
class DummyEvent(PasswordLength):
pass
with self.assertRaises(AssertionError):
self.manager.subscribe(DummyEvent)
def test_knows(self):
assert self.manager.knows(key=AdminViewEnabled.key) is False
self.manager.subscribe(AdminViewEnabled)
assert self.manager.knows(key=AdminViewEnabled.key) is True
# Adding same event
self.manager.subscribe(AdminViewEnabled)
assert self.manager.knows(key=AdminViewEnabled.key) is True
# New event
assert self.manager.knows(PasswordLength) is False
self.manager.subscribe(PasswordLength)
assert self.manager.knows(key=PasswordLength.key) is True
def test_get(self):
assert self.manager.get(key=AdminViewEnabled.key) is None
self.manager.subscribe(AdminViewEnabled)
assert self.manager.get(key=AdminViewEnabled.key) == AdminViewEnabled
def setUp(self):
self.manager = OptionManager()
super().setUp()
from options.option_manager import OptionManager option_manager = OptionManager()