Fix tests

1 files changed, 5 insertions, 3 deletions

diff --git a/app_test.py b/app_test.py
index 5dabfc68..0a9b2180 100644
--- a/app_test.py
+++ b/app_test.py
@@ -6,7 +6,9 @@ from flask_testing import TestCase
 
 from app import manager
 from gunicorn_config import when_ready
-from pysite import constants
+
+os.environ["BOT_API_KEY"] = "abcdefg"  # This is a constant, must be done first
+
 from pysite.constants import DISCORD_OAUTH_REDIRECT, DISCORD_OAUTH_AUTHORIZED
 
 when_ready()
@@ -152,7 +154,7 @@ class ApiEndpoints(SiteTest):
 
     def test_api_tags(self):
         """ Check tag API """
-        constants.BOT_API_KEY = 'abcdefg'
+        os.environ['BOT_API_KEY'] = 'abcdefg'
         headers = {'X-API-Key': 'abcdefg', 'Content-Type': 'application/json'}
 
         post_data = json.dumps({
@@ -221,7 +223,7 @@ class ApiEndpoints(SiteTest):
 
     def test_api_user(self):
         """ Check insert user """
-        constants.BOT_API_KEY = 'abcdefg'
+        os.environ['BOT_API_KEY'] = 'abcdefg'
         headers = {'X-API-Key': 'abcdefg', 'Content-Type': 'application/json'}
         single_data = json.dumps({'user_id': "1234", 'roles': ["5678"], "username": "test", "discriminator": "0000"})
         list_data = json.dumps([{'user_id': "1234", 'roles': ["5678"], "username": "test", "discriminator": "0000"}])