Pytest макет моего собственного модуля, импортированного в проверенную функцию - PullRequest
Новая
       8

Pytest макет моего собственного модуля, импортированного в проверенную функцию

0 голосов
/ 18 октября 2019

Я хочу проверить функцию, которая импортирует другой модуль.

Итак, я хочу смоделировать модуль.

Итак, моя тестовая функция: 

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 голосов
/ 19 октября 2019

Я не уверен, что у вас на самом деле нет других проблем, связанных с импортом, но вы пытаетесь издеваться над неправильным объектом . Вам нужно смоделировать get_last_transaction_id импортированный в модуль, где validate_transaction. Вы не упомянули его имя, поэтому давайте предположим, что оно называется xyz: 

from full_path.to.xyz import validate_transaction

def test_validate_transaction():
    dataset_id = "idx"
    transaction_id = "idy"
    transaction = {
            "transaction_rid" : transaction_id,
            "is_successful" : True
        }

    with mock.patch('xyz.get_last_transaction_id', return_value=transaction):
        assert validate_transaction(dataset_rid,transaction_rid) == False
...