Mirrored_Repos/InvenTree
1
0
Fork 0
mirror of https://github.com/inventree/InvenTree synced 2024-08-30 18:33:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use unierest mock for env setting

Browse Source
This commit is contained in:
Matthias Mair 2022-05-17 19:23:50 +02:00 committed by GitHub
parent b630fb2856
commit a40f189c7a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
InvenTree/InvenTree/tests.py
View File

@ -1,6 +1,6 @@
import json
from test import support
from unittest import mock
from django.test import TestCase, override_settings
import django.core.exceptions as django_exceptions
@ -450,12 +450,11 @@ class TestSettings(TestCase):
def setUp(self) -> None:
self.user_mdl = get_user_model()
self.env = support.EnvironmentVarGuard()
def run_reload(self):
def run_reload(self, envs):
from plugin import registry
with self.env:
with mock.patch.dict(os.environ, envs):
settings.USER_ADDED = False
registry.reload_plugins()
@ -471,15 +470,17 @@ class TestSettings(TestCase):
self.assertEqual(user_count(), 0)
# not enough set
self.env.set('INVENTREE_ADMIN_USER', 'admin') # set username
self.run_reload()
envs = {}
envs['INVENTREE_ADMIN_USER'] = 'admin'
self.run_reload(envs)
self.assertEqual(user_count(), 0)
# enough set
self.env.set('INVENTREE_ADMIN_USER', 'admin') # set username
self.env.set('INVENTREE_ADMIN_EMAIL', 'info@example.com') # set email
self.env.set('INVENTREE_ADMIN_PASSWORD', 'password123') # set password
self.run_reload()
envs = {'INVENTREE_ADMIN_USER': 'admin', # set username
'INVENTREE_ADMIN_EMAIL': 'info@example.com', # set email
'INVENTREE_ADMIN_PASSWORD': 'password123' # set password
}
self.run_reload(envs)
self.assertEqual(user_count(), 1)
# make sure to clean up