Added tests for check_required_env_vars

kilobyteno · Oct 8, 2024 · 2c06b84 · 2c06b84
1 parent c190462
commit 2c06b84
Showing 1 changed file with 37 additions and 1 deletion.
diff --git a/tests/test_konfig.py b/tests/test_konfig.py
@@ -1,8 +1,9 @@
 import logging
+import os
 
 import pytest
 
-from tunsberg.konfig import uvicorn_log_config
+from tunsberg.konfig import check_required_env_vars, uvicorn_log_config
 
 
 class TestUvicornLogConfig:
@@ -55,3 +56,38 @@ def test_custom_log_level(self):
         assert config['loggers']['uvicorn']['level'] == logging.getLevelName(log_lvl)
         assert config['loggers']['uvicorn.error']['level'] == logging.getLevelName(log_lvl)
         assert config['loggers']['uvicorn.access']['level'] == logging.getLevelName(log_lvl)
+
+
+class TestCheckRequiredEnvVars:
+    # Set the environment variable for testing
+    os.environ['RANDOM_ENV_VAR'] = 'random_value'
+
+    def test_validate_if_true_in_local_development_env(self):
+        req_envs = {'RANDOM_ENV_VAR': {'runtime': True, 'build': True}}
+        check = check_required_env_vars(required_env_vars=req_envs, env='local')
+        assert check
+
+    def test_validate_if_true_in_staging_env(self):
+        req_envs = {'RANDOM_ENV_VAR': {'runtime': True, 'build': True}}
+        check = check_required_env_vars(required_env_vars=req_envs, env='staging', live_envs=['staging'])
+        assert check
+
+    def test_validate_if_true_in_prod_env(self):
+        req_envs = {'RANDOM_ENV_VAR': {'runtime': True, 'build': True}}
+        check = check_required_env_vars(required_env_vars=req_envs, env='prod', live_envs=['prod'])
+        assert check
+
+    def test_validate_that_it_fails_if_env_var_is_not_set(self):
+        req_envs = {'SOME_OTHER_RANDOM_ENV_VAR': {'runtime': True, 'build': True}}
+        with pytest.raises(ValueError):
+            check_required_env_vars(required_env_vars=req_envs, env='production', live_envs=['production'])
+
+    def test_validate_that_it_fails_for_code_build_if_env_var_is_not_set(self):
+        req_envs = {'SOME_OTHER_RANDOM_ENV_VAR': {'runtime': False, 'build': True}}
+        with pytest.raises(ValueError):
+            check_required_env_vars(required_env_vars=req_envs, env='production', live_envs=['production'], code_build=True)
+
+    def test_validate_that_it_fails_for_runtime_if_env_var_is_not_set(self):
+        req_envs = {'SOME_OTHER_RANDOM_ENV_VAR': {'runtime': True, 'build': False}}
+        with pytest.raises(ValueError):
+            check_required_env_vars(required_env_vars=req_envs, env='production', live_envs=['production'])