29 lines
719 B
Python
29 lines
719 B
Python
|
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")
|