pytest模拟我自己的模块导入测试功能

时间:2019-10-18 16:18:24

标签: python mocking pytest

我想测试导入另一个模块的功能。

所以我想模拟模块。

所以我的测试功能是:

def test_validate_transaction():

    dataset_id = "idx"
    transaction_id = "idy"
    transaction = {
            "transaction_rid" : transaction_id,
            "is_successful" : True
        }
    with mock.patch('table_management.get_last_transaction_id', return_value=transaction) :
        assert validate_transaction(dataset_rid,transaction_rid) == False

我要测试的功能是

import json
import os
from table_management import get_last_transaction_id


def validate_transaction(dataset_id,transaction_id):
    try:

        transaction = get_last_transaction_id(dataset_rid)
        return True if transaction['transaction_id'] != transaction_rid or transaction['is_successful']==False else False
    except Exception as e:
        print("An exception occurred " + str(e))
        return {}

但是这样做,我会收到错误消息:

..\..\..\env\lib\site-packages\mock\mock.py:1378: in __enter__
     self.target = self.getter() ..\..\..\env\lib\site-packages\mock\mock.py:1548: in <lambda>
     getter = lambda: _importer(target)


 target = 'table_management'

     def _importer(target):
         components = target.split('.')
         import_path = components.pop(0)
      thing = __import__(import_path) 
E       ImportError: No module named table_management

 ..\..\..\env\lib\site-packages\mock\mock.py:1231: ImportError

您知道缺少什么吗?

1 个答案:

答案 0 :(得分:0)

我不确定您实际上没有其他与导入有关的问题,但您正在尝试模拟错误的对象。您需要模拟SetWindowPos(child, HWND_TOP, 0, 0, 0, 0, SWP_NOACTIVATE | SWP_NOMOVE | SWP_NOSIZE)所在模块中导入的get_last_transaction_id。您没有提到它的名称,所以我们假设它的名称为validate_transaction

xyz