1234567891011121314151617181920212223242526272829 |
- import sys
- import os
- sys.path.append(os.path.join(os.path.dirname(__file__), os.pardir))
- from env_config import set_env
- def test_set_env():
- """
- Tests that the environment is set correctly and temporarily inside the context manager.
- """
- env = {
- 'HOSTNAME': 'localhost',
- 'OPENAI_API_KEY': '<key_1>',
- }
- os.environ.update(env)
- assert os.environ['OPENAI_API_KEY'] == '<key_1>'
- assert os.environ['HOSTNAME'] == 'localhost'
- assert 'UNSET' not in os.environ['HOSTNAME']
- with set_env(OPENAI_API_KEY = '<key_2>', UNSET = 'whatever'):
- assert os.environ['OPENAI_API_KEY'] == '<key_2>'
- assert os.environ['HOSTNAME'] == 'localhost'
- assert os.environ['UNSET'] == 'whatever'
- assert os.environ['OPENAI_API_KEY'] == '<key_1>'
- assert os.environ['HOSTNAME'] == 'localhost'
- assert 'UNSET' not in os.environ['HOSTNAME']
|