wehub-resource-sync
30 lines
896 B
Python
30 lines
896 B
Python
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']
|