如何在Pytest中重写非本地fixture?
如何在Pytest中重写非本地fixture?
嘿,我来帮你搞定这个问题!你现在的情况是想修改一个外部库里的fixture,而且这个fixture所在的类还没在all.py里被引用对吧?别慌,Pytest有两个简单直接的方法能解决这个问题~
方法一:继承库中的测试类重写fixture
这是最直观的方式,因为那个fixture是定义在ReferenceFileTest这个类里的实例方法,你只需要在自己的测试文件里继承这个类,然后重写对应的fixture方法就行。Pytest会自动优先使用子类里的fixture,完全覆盖父类(也就是库中类)的原fixture。
举个实际的代码例子:
# 先从外部库导入那个包含fixture的类 from library._generic import ReferenceFileTest import pytest # 继承这个库中的测试类 class MyCustomTest(ReferenceFileTest): # 重写fixture,改成你想要的返回值或者逻辑 @pytest.fixture def reference_folder_name(self) -> str: return "my_custom_reference" # 这里替换成你需要的文件夹名 # 接下来写你的测试用例,就会自动使用重写后的fixture了 def test_my_custom_case(MyCustomTest, tmp_path): # 用重写后的fixture值创建文件夹 custom_ref_folder = tmp_path / MyCustomTest.reference_folder_name() custom_ref_folder.mkdir() # 剩下的测试逻辑就按你的需求来写啦 assert custom_ref_folder.exists()
方法二:用猴子补丁动态修改原类的fixture
如果你不想继承整个类,也可以用猴子补丁(monkeypatch)直接修改外部库类里的fixture方法,这样所有用到这个类的地方都会使用你修改后的逻辑。
代码示例如下:
import pytest from library._generic import ReferenceFileTest @pytest.fixture(autouse=True) def override_ref_folder_name(monkeypatch): # 定义你自己的fixture逻辑 def custom_ref_folder(self): return "dynamic_custom_ref" # 用猴子补丁替换原类中的fixture方法 monkeypatch.setattr(ReferenceFileTest, "reference_folder_name", custom_ref_folder) # 测试用例里直接使用原类,就会用到重写后的fixture了 def test_dynamic_override(ReferenceFileTest, tmp_path): ref_folder = tmp_path / ReferenceFileTest.reference_folder_name() ref_folder.mkdir() assert ref_folder.exists() assert ref_folder.name == "dynamic_custom_ref"
需要注意的是,第二种方法用autouse=True的话,会自动应用到所有测试用例上;如果只想在特定用例里用,去掉autouse=True,然后在测试用例上加上@pytest.mark.usefixtures("override_ref_folder_name")就行。
内容来源于stack exchange
相关产品推荐
相关产品推荐

