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'