inventree/InvenTree
2
0
Fork 0
mirror of https://github.com/inventree/InvenTree.git synced 2025-04-28 11:36:44 +00:00
Code

generalise function to make new methods simpler

Browse Source
This commit is contained in:
Matthias 2022-05-18 02:04:15 +02:00
parent ca7fb691ac
commit a570dab5e5
No known key found for this signature in database
GPG Key ID: AB6D0E6C4CB65093
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
InvenTree/InvenTree/tests.py
View File

@ -457,10 +457,14 @@ class TestSettings(TestCase):
self.user = user.objects.create_superuser('testuser1', 'test1@testing.com', 'password1') self.user = user.objects.create_superuser('testuser1', 'test1@testing.com', 'password1')
self.client.login(username='testuser1', password='password1') self.client.login(username='testuser1', password='password1')
def in_env_context(self, envs={}):
"""Patch the env to include the given dict"""
return mock.patch.dict(os.environ, envs)
def run_reload(self, envs): def run_reload(self, envs):
from plugin import registry from plugin import registry
with mock.patch.dict(os.environ, envs): with self.in_env_context(envs):
settings.USER_ADDED = False settings.USER_ADDED = False
registry.reload_plugins() registry.reload_plugins()
@ -513,7 +517,7 @@ class TestSettings(TestCase):
self.assertIn('InvenTree/InvenTree/config.yaml', config.get_config_file()) self.assertIn('InvenTree/InvenTree/config.yaml', config.get_config_file())
# with env set # with env set
with mock.patch.dict(os.environ, {'INVENTREE_CONFIG_FILE': 'my_special_conf.yaml'}): with self.in_env_context({'INVENTREE_CONFIG_FILE': 'my_special_conf.yaml'}):
self.assertIn('InvenTree/InvenTree/my_special_conf.yaml', config.get_config_file()) self.assertIn('InvenTree/InvenTree/my_special_conf.yaml', config.get_config_file())
def test_helpers_plugin_file(self): def test_helpers_plugin_file(self):
@ -521,7 +525,7 @@ class TestSettings(TestCase):
self.assertIn('InvenTree/InvenTree/plugins.txt', config.get_plugin_file()) self.assertIn('InvenTree/InvenTree/plugins.txt', config.get_plugin_file())
# with env set # with env set
with mock.patch.dict(os.environ, {'INVENTREE_PLUGIN_FILE': 'my_special_plugins.txt'}): with self.in_env_context({'INVENTREE_PLUGIN_FILE': 'my_special_plugins.txt'}):
self.assertIn('my_special_plugins.txt', config.get_plugin_file()) self.assertIn('my_special_plugins.txt', config.get_plugin_file())
def test_helpers_setting(self): def test_helpers_setting(self):
@ -530,7 +534,7 @@ class TestSettings(TestCase):
self.assertEqual(config.get_setting(TEST_ENV_NAME, None, '123!'), '123!') self.assertEqual(config.get_setting(TEST_ENV_NAME, None, '123!'), '123!')
# with env set # with env set
with mock.patch.dict(os.environ, {'TEST_ENV_NAME': '321'}): with self.in_env_context({'TEST_ENV_NAME': '321'}):
self.assertEqual(config.get_setting(TEST_ENV_NAME, None), '321') self.assertEqual(config.get_setting(TEST_ENV_NAME, None), '321')