mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-02-05 19:03:30 -05:00
54 lines
1.5 KiB
Python
54 lines
1.5 KiB
Python
import importlib
|
|
import os
|
|
from unittest import mock
|
|
|
|
import pytest
|
|
|
|
from app import config
|
|
|
|
|
|
def cf_conf():
|
|
os.environ['REDIS_URL'] = 'rediss://xxx:6379'
|
|
|
|
|
|
@pytest.fixture
|
|
def reload_config(os_environ):
|
|
"""
|
|
Reset config, by simply re-running config.py from a fresh environment
|
|
"""
|
|
old_env = os.environ.copy()
|
|
os.environ.clear()
|
|
|
|
yield
|
|
|
|
os.environ.clear()
|
|
for k, v in old_env.items():
|
|
os.environ[k] = v
|
|
importlib.reload(config)
|
|
|
|
|
|
def test_load_cloudfoundry_config_if_available(reload_config):
|
|
os.environ['REDIS_URL'] = 'some uri'
|
|
os.environ['VCAP_SERVICES'] = 'some json blob'
|
|
|
|
with mock.patch('app.cloudfoundry_config.extract_cloudfoundry_config', side_effect=cf_conf):
|
|
# reload config so that its module level code (ie: all of it) is re-instantiated
|
|
importlib.reload(config)
|
|
|
|
assert os.environ['REDIS_URL'] == 'rediss://xxx:6379'
|
|
assert config.Config.REDIS_URL == 'rediss://xxx:6379'
|
|
|
|
|
|
def test_load_config_if_cloudfoundry_not_available(reload_config):
|
|
os.environ['REDIS_URL'] = 'redis://xxx:6379'
|
|
os.environ.pop('VCAP_SERVICES', None)
|
|
|
|
with mock.patch('app.cloudfoundry_config.extract_cloudfoundry_config') as cf_config:
|
|
# reload config so that its module level code (ie: all of it) is re-instantiated
|
|
importlib.reload(config)
|
|
|
|
assert not cf_config.called
|
|
|
|
assert os.environ['REDIS_URL'] == 'redis://xxx:6379'
|
|
assert config.Config.REDIS_URL == 'redis://xxx:6379'
|