Docker容器内用Python取消环境变量测试配置模块失败，求解决方案

我写了一个从环境变量读取值的Python配置模块，这些环境变量由docker-compose文件设置。为了测试这个配置模块，我想在测试代码里先清空相关环境变量，再导入配置模块，验证它是否会使用默认值。但我试了os.unsetenv('VAR')、os.system('unset VAR')和del os.environ['VAR']三种方法都没成功，怀疑是Docker重写了环境变量，想问问当前做法对不对，以及可行的解决办法。

测试代码 test_config.py #

import os
from unittest import TestCase

def test_without_env():
    os.unsetenv('GITLAB_URL')
    os.unsetenv('GITLAB_PROJECT_ID')
    os.unsetenv('GITLAB_API_TOKEN')
    
    os.unsetenv('JIRA_API_TOKEN')
    os.unsetenv('JIRA_BOARD_ID')
    os.unsetenv('JIRA_URL')

    os.unsetenv('ROCKET_CHAT_API_TOKEN')
    os.unsetenv('ROCKET_CHAT_USER_ID')
    os.unsetenv('ROCKET_CHAT_URL')

    import src.config as conf

    assert conf.GITLAB_URL == 'URL'
    assert conf.GITLAB_PROJECT_ID == '1234'
    assert conf.GITLAB_API_TOKEN == 'TOKEN'

    assert conf.JIRA_API_TOKEN == 'TOKEN'
    assert conf.JIRA_BOARD_ID == '1234'
    assert conf.JIRA_URL == 'URL'

    assert conf.ROCKET_CHAT_API_TOKEN == 'TOKEN'
    assert conf.ROCKET_CHAT_URL == 'URL'
    assert conf.ROCKET_CHAT_USER_ID == 'USER_ID'

配置模块 config.py #

import os

# GitLab
if 'GITLAB_URL' in os.environ: GITLAB_URL = os.environ['GITLAB_URL']
else: GITLAB_URL = 'URL'

if 'GITLAB_PROJECT_ID' in os.environ: GITLAB_PROJECT_ID = os.environ['GITLAB_PROJECT_ID']
else: GITLAB_PROJECT_ID = 1234

if 'GITLAB_API_TOKEN' in os.environ: GITLAB_API_TOKEN = os.environ['GITLAB_API_TOKEN']
else: GITLAB_API_TOKEN = 'TOKEN'

# Jira
if 'JIRA_URL' in os.environ: JIRA_URL = os.environ['JIRA_URL']
else: JIRA_URL = 'URL'

if 'JIRA_BOARD_ID' in os.environ: JIRA_BOARD_ID = os.environ['JIRA_BOARD_ID']
else: JIRA_BOARD_ID = 1234

if 'JIRA_API_TOKEN' in os.environ: JIRA_API_TOKEN = os.environ['JIRA_API_TOKEN']
else: JIRA_API_TOKEN = 'TOKEN'

# RocketChat
if 'ROCKET_CHAT_API_TOKEN' in os.environ: ROCKET_CHAT_API_TOKEN = os.environ['ROCKET_CHAT_API_TOKEN']
else: ROCKET_CHAT_API_TOKEN = 'TOKEN'

if 'ROCKET_CHAT_URL' in os.environ: ROCKET_CHAT_URL = os.environ['ROCKET_CHAT_URL']
else: ROCKET_CHAT_URL = 'URL'

if 'ROCKET_CHAT_USER_ID' in os.environ: ROCKET_CHAT_USER_ID = os.environ['ROCKET_CHAT_USER_ID']
else: ROCKET_CHAT_USER_ID = 'USER_ID'

核心问题 #

1. 模块导入缓存：Python模块一旦导入就会被缓存，若测试过程中已导入过src.config，后续import会直接用缓存版本，不会重新执行配置逻辑，导致环境变量修改无效。
2. 环境变量修改方式错误：
   • os.system('unset VAR')是在子shell执行，不会影响当前Python进程的环境变量。
   • os.unsetenv不会同步更新os.environ字典，部分系统下无效；del os.environ['VAR']是正确方式，但需处理变量不存在的情况，避免抛出KeyError。

可行方案 #

方案1：清理模块缓存+安全删除环境变量 #

import os
import sys
from unittest import TestCase

def test_without_env():
    # 安全删除环境变量，避免KeyError
    env_vars = [
        'GITLAB_URL', 'GITLAB_PROJECT_ID', 'GITLAB_API_TOKEN',
        'JIRA_API_TOKEN', 'JIRA_BOARD_ID', 'JIRA_URL',
        'ROCKET_CHAT_API_TOKEN', 'ROCKET_CHAT_USER_ID', 'ROCKET_CHAT_URL'
    ]
    for var in env_vars:
        if var in os.environ:
            del os.environ[var]
    
    # 清理模块缓存，确保重新导入时执行配置逻辑
    if 'src.config' in sys.modules:
        del sys.modules['src.config']
    
    # 重新导入配置模块
    import src.config as conf

    assert conf.GITLAB_URL == 'URL'
    assert conf.GITLAB_PROJECT_ID == 1234
    assert conf.GITLAB_API_TOKEN == 'TOKEN'

    assert conf.JIRA_API_TOKEN == 'TOKEN'
    assert conf.JIRA_BOARD_ID == 1234
    assert conf.JIRA_URL == 'URL'

    assert conf.ROCKET_CHAT_API_TOKEN == 'TOKEN'
    assert conf.ROCKET_CHAT_URL == 'URL'
    assert conf.ROCKET_CHAT_USER_ID == 'USER_ID'

方案2：重构配置模块，支持重新加载 #

修改config.py，把配置逻辑封装成函数，无需重新导入模块即可刷新配置：

import os

def load_config():
    config = {}
    # GitLab
    config['GITLAB_URL'] = os.environ.get('GITLAB_URL', 'URL')
    config['GITLAB_PROJECT_ID'] = os.environ.get('GITLAB_PROJECT_ID', 1234)
    config['GITLAB_API_TOKEN'] = os.environ.get('GITLAB_API_TOKEN', 'TOKEN')

    # Jira
    config['JIRA_URL'] = os.environ.get('JIRA_URL', 'URL')
    config['JIRA_BOARD_ID'] = os.environ.get('JIRA_BOARD_ID', 1234)
    config['JIRA_API_TOKEN'] = os.environ.get('JIRA_API_TOKEN', 'TOKEN')

    # RocketChat
    config['ROCKET_CHAT_API_TOKEN'] = os.environ.get('ROCKET_CHAT_API_TOKEN', 'TOKEN')
    config['ROCKET_CHAT_URL'] = os.environ.get('ROCKET_CHAT_URL', 'URL')
    config['ROCKET_CHAT_USER_ID'] = os.environ.get('ROCKET_CHAT_USER_ID', 'USER_ID')
    
    return config

# 初始化配置
config = load_config()

# 导出变量，保持原有使用方式
GITLAB_URL = config['GITLAB_URL']
GITLAB_PROJECT_ID = config['GITLAB_PROJECT_ID']
GITLAB_API_TOKEN = config['GITLAB_API_TOKEN']

JIRA_URL = config['JIRA_URL']
JIRA_BOARD_ID = config['JIRA_BOARD_ID']
JIRA_API_TOKEN = config['JIRA_API_TOKEN']

ROCKET_CHAT_API_TOKEN = config['ROCKET_CHAT_API_TOKEN']
ROCKET_CHAT_URL = config['ROCKET_CHAT_URL']
ROCKET_CHAT_USER_ID = config['ROCKET_CHAT_USER_ID']

测试代码修改为：

import os
import src.config as conf

def test_without_env():
    env_vars = [
        'GITLAB_URL', 'GITLAB_PROJECT_ID', 'GITLAB_API_TOKEN',
        'JIRA_API_TOKEN', 'JIRA_BOARD_ID', 'JIRA_URL',
        'ROCKET_CHAT_API_TOKEN', 'ROCKET_CHAT_USER_ID', 'ROCKET_CHAT_URL'
    ]
    for var in env_vars:
        if var in os.environ:
            del os.environ[var]
    
    # 重新加载配置并更新模块变量
    new_config = conf.load_config()
    conf.GITLAB_URL = new_config['GITLAB_URL']
    conf.GITLAB_PROJECT_ID = new_config['GITLAB_PROJECT_ID']
    conf.GITLAB_API_TOKEN = new_config['GITLAB_API_TOKEN']
    
    conf.JIRA_URL = new_config['JIRA_URL']
    conf.JIRA_BOARD_ID = new_config['JIRA_BOARD_ID']
    conf.JIRA_API_TOKEN = new_config['JIRA_API_TOKEN']
    
    conf.ROCKET_CHAT_API_TOKEN = new_config['ROCKET_CHAT_API_TOKEN']
    conf.ROCKET_CHAT_URL = new_config['ROCKET_CHAT_URL']
    conf.ROCKET_CHAT_USER_ID = new_config['ROCKET_CHAT_USER_ID']

    assert conf.GITLAB_URL == 'URL'
    # 其他断言...

方案3：使用unittest.mock临时修改环境变量 #

用mock.patch.dict临时替换os.environ，更简洁可靠：

import os
from unittest import TestCase, mock
import sys

def test_without_env():
    # 临时清空目标环境变量
    with mock.patch.dict(os.environ, {}, clear=True):
        # 清理模块缓存
        if 'src.config' in sys.modules:
            del sys.modules['src.config']
        import src.config as conf

        assert conf.GITLAB_URL == 'URL'
        assert conf.GITLAB_PROJECT_ID == 1234
        assert conf.GITLAB_API_TOKEN == 'TOKEN'

        assert conf.JIRA_API_TOKEN == 'TOKEN'
        assert conf.JIRA_BOARD_ID == 1234
        assert conf.JIRA_URL == 'URL'

        assert conf.ROCKET_CHAT_API_TOKEN == 'TOKEN'
        assert conf.ROCKET_CHAT_URL == 'URL'
        assert conf.ROCKET_CHAT_USER_ID == 'USER_ID'

额外提示 #

• Docker不会在Python进程运行时重写环境变量：Docker设置的环境变量是在进程启动时加载的，测试时修改的是当前进程的环境变量，与Docker无关。
• 推荐用os.environ.get(var, default)替代if-else逻辑，代码更简洁，功能完全一致。

内容的提问来源于stack exchange，提问作者DasMoorhuhn