Added environ tests

This commit is contained in:
Luke Else 2024-01-29 22:07:36 +00:00
parent e4de5667b4
commit 482b6a633b
2 changed files with 30 additions and 1 deletions

View File

@ -1,4 +1,5 @@
import os
# Setup test environment variables
os.environ["ENVIRON"] = "test"
if os.environ.get("ENVIRON") is None:
os.environ["ENVIRON"] = "test"

28
tests/general/test_env.py Normal file
View File

@ -0,0 +1,28 @@
from os import environ
from warnings import warn
# Tests environment variables used within the projects domain are
# set in the correct environment
VARS = ['ENVIRON', 'APPSECRET', 'FILESTORE']
def test_env_vars():
""" Test that required environment variables are set
ahead of runtime
"""
for var in VARS:
env = environ.get(var)
# Check to see what variable we are comparing
if env is None:
warn(f"Variable {var} is not set!")
def test_environment_var_state():
""" Tests that the 'ENVIRON' Environment variable
is in a correct state
"""
var = environ.get('ENVIRON')
assert var is not None
assert (var == "test" or var == "prod")