python—模拟测试类方法的更好方法

bvpmtnay  于 2021-09-08  发布在  Java
关注(0)|答案(0)|浏览(230)

我有一个类,它在init中使用了一些http和弹性,我想找到一种更好的方法来运行模拟测试,而不必修补每个测试用例。

class MyClass:
    def __init(self):
        self.es=ElasticSearch()
        self.http = urllib3.PoolManager()

    def method_1():
        xxxxx

    def method_2():
        xxxx

    def method_100():
        xxxx

测试用例

@mock.patch(ElasticSearch)
@mock.patch(PoolManager)
test_method_1():
    class_instance = MyClass()
    class_instance.method_1()
    assert xxxxx

@mock.patch(ElasticSearch)
@mock.patch(PoolManager)
test_method_2():
    class_instance = MyClass()
    class_instance.method_2()
    assert xxx

是否有一种简单的方法可以避免每次测试都重复指定mock.patch?

暂无答案!

目前还没有任何答案,快来回答吧!

相关问题