def setUp(self):
self.addon_config = AddonConfig(
short_name='test',
full_name='test',
owners=['node'],
added_to={'node': False},
categories=[],
settings_model=AddonNodeSettingsBase,
)
def init_mock_addon(short_name, user_settings, node_settings):
"""Add an addon to the settings, so that it is ready for app init
This is used to inject addons into the application context for tests."""
settings.ADDONS_REQUESTED.append(short_name)
addon_config = AddonConfig(
short_name=short_name,
full_name=short_name,
owners=['User', 'Node'],
categories=['Storage'],
user_settings_model=user_settings,
node_settings_model=node_settings,
models=[user_settings, node_settings],
)
settings.ADDONS_AVAILABLE_DICT[addon_config.short_name] = addon_config
settings.ADDONS_AVAILABLE.append(addon_config)
return addon_config
def init_mock_addon():
# TODO: Move to a base class, and have test cases inherit from that.
settings.ADDONS_REQUESTED.append(
MockUserSettings.oauth_provider.short_name)
addon_config = AddonConfig(
short_name=MockUserSettings.oauth_provider.short_name,
full_name=MockUserSettings.oauth_provider.name,
owners=['User'],
categories=['Storage'],
user_settings_model=MockUserSettings,
node_settings_model=MockNodeSettings,
models=[MockUserSettings, MockNodeSettings],
)
# add this so we can remove the mocked addon later.
settings.MOCKED_ADDON = addon_config
settings.ADDONS_AVAILABLE_DICT[addon_config.short_name] = addon_config
settings.ADDONS_AVAILABLE.append(addon_config)
def init_mock_addon(short_name, user_settings=None, node_settings=None):
"""Add an addon to the settings, so that it is ready for app init
This is used to inject addons into the application context for tests."""
#Importing within the function to prevent circular import problems.
import factories
user_settings = user_settings or factories.MockAddonUserSettings
node_settings = node_settings or factories.MockAddonNodeSettings
settings.ADDONS_REQUESTED.append(short_name)
addon_config = AddonConfig(
short_name=short_name,
full_name=short_name,
owners=['User', 'Node'],
categories=['Storage'],
user_settings_model=user_settings,
node_settings_model=node_settings,
models=[user_settings, node_settings],
)
settings.ADDONS_AVAILABLE_DICT[addon_config.short_name] = addon_config
settings.ADDONS_AVAILABLE.append(addon_config)
return addon_config
