如何在pytest中全局模拟Enum枚举类（无需修改项目导入方式或直接操作sys.modules）

嗨，我完全理解你的困扰——既要保留项目里原有的导入习惯，又不想用太hacky的方式修改sys.modules，同时还要在测试时用枚举的子集。这里有两个实用的方案，你可以根据自己的情况选择：

方案一：动态修改原枚举的成员（无需重构代码） #

Enum类的核心成员存在__members__字典里，迭代行为由__iter__方法控制。我们可以利用pytest的autouse fixture，在测试开始前替换这些属性为测试用的子集，测试结束后再恢复原状。这样不管项目里怎么导入CoinName，只要是同一个类对象，修改都会全局生效。

首先定义测试用的精简枚举：

from enum import Enum

class MockCoinName(str, Enum):
    USDT = "USDT"
    BTC = "BTC"
    ETH = "ETH"  # 只保留测试需要的币种

然后在conftest.py里写自动生效的fixture：

import pytest
from trading_bot.definitions import CoinName

@pytest.fixture(autouse=True)
def mock_coin_name():
    # 先保存原枚举的核心属性，方便测试后恢复
    original_members = CoinName.__members__.copy()
    original_iter = CoinName.__iter__
    
    try:
        # 替换成员字典为测试用的子集
        CoinName.__members__.clear()
        CoinName.__members__.update(MockCoinName.__members__)
        
        # 重写迭代方法，确保list(CoinName)返回的是测试子集
        def mock_iter():
            return iter(MockCoinName)
        
        CoinName.__iter__ = mock_iter
        
        yield  # 执行测试用例
    finally:
        # 测试结束后恢复原枚举，避免影响其他测试
        CoinName.__members__.clear()
        CoinName.__members__.update(original_members)
        CoinName.__iter__ = original_iter

这个方案的好处是完全不需要修改项目里的任何导入或业务代码，测试时自动切换到枚举子集，测试后自动还原，非常省心。

方案二：重构代码为依赖注入（更优雅的长期方案） #

如果觉得动态修改枚举属性还是有点“取巧”，可以考虑用依赖注入的思路重构代码——把依赖CoinName的地方，改成接受一个可选的可迭代参数，默认值是原枚举，测试时传入子集即可。

比如原来的业务代码：

from trading_bot.definitions import CoinName

def fetch_all_historical_prices():
    for coin in CoinName:
        # 调用binance接口获取历史价格
        get_historical_price(coin)

改成：

from typing import Iterable
from trading_bot.definitions import CoinName

def fetch_all_historical_prices(coins: Iterable[CoinName] = CoinName):
    for coin in coins:
        get_historical_price(coin)

测试时直接传入测试用的枚举：

def test_fetch_prices(mock_binance_client):
    fetch_all_historical_prices(coins=MockCoinName)
    # 断言mock客户端被调用的次数/参数是否符合预期

这种方式的优势是代码更灵活、可测试性更强，不仅适合测试，未来如果需要针对特定币种批量操作，也可以直接传入对应子集，不需要修改枚举本身。如果项目里使用CoinName的地方不多，重构成本非常低。

总结 #

• 如果不想动现有代码，优先选方案一，通过动态修改枚举属性实现全局模拟，既不用改导入方式，也不用操作sys.modules；
• 如果想让代码结构更健壮、易扩展，优先选方案二，依赖注入的思路能从根源上降低代码对全局枚举的耦合。

备注：内容来源于stack exchange，提问作者nontrivial