Pytest — исправление класса не работает, вместо этого вызывает класс

Вам необходимо исправить символ «A» в тестируемом модуле, т.е. «B».

Когда вы делаете @mock.patch("full.path.A"), это должно быть:

@mock.patch("full.path.to.B.A")

Теперь символ A в модуле B исправлен вашим макетом.

Вы не пропустили имя самого файла?

@mock.patch( 'namespacename.mylibname.stats_collector.mylibname_stats_collector.StatsCollector')

Вы могли бы понять, как я пишу это.

Эмпирическое правило: не исправлять классы или функции там, где они определены, а исправлять их там, где они используются.

a.py

class A:
    def exponent(self, a, b)
        return a ** b
    

b.py    


from a import A
class B:
   def add_constat(a, b, c)
      return A().exponent(a, b) + c
      
  
  
  

Для тестирования метода add_constant у вас может возникнуть соблазн исправить A, как в

TestB:

@patch('a.A.exponent', mock_a)
    test_add_constant(self, mock_a) 

это неправильно, так как вы исправляете класс в файле, где дано его определение.

A используется в файле b в классе B. Следовательно, вместо этого вы должны исправить этот класс.

TestB:

    @patch('b.A')
    test_add_constant(self, mock_a):
    # mock_a is fake class of A, the return_value of mock_a gives us an instance (object) of the class(A)
    instance_a = mock_a.return_value # 
   
    # we now have instance of the class i.e A, hence it is possible to call the methods of class A
   
    instance_a.exponent.return_value = 16
   
    assert 26 = B().add_constant(2,4,10)

Я немного изменил ваш код, чтобы он работал в моей среде Python.

stats_collector.py

class StatsCollector:
    def __init__(self, user_id, app_id):
        self.user_id = user_id
        self.app_id = app_id

    def stat(self):
        return self.user_id + ':' + self.app_id


mydb.py

from stats_collector import StatsCollector
import logging

class Db:
    # constructor
    def __init__(self, db_name):
        self.db_name = db_name

    def begin_transaction(self, user_id, app_id):
        logging.info("Begin")
        stat = StatsCollector(user_id, app_id).stat()

        if stat:
            return user_id + ':' + app_id
        return "wrong User"

используя аналогию: для тестирования «begin_transaction» в классе Db файла mydb.py вам необходимо исправить класс StatsCollector, используемый в файле mydb.py

test_mydb.py

from unittest.mock import patch
from unittest import TestCase


class TestDb(TestCase):

    @patch('mydb.StatsCollector')
    def test_begin_transaction(self, db_class_mock):

        # db_class_mock is the mock of the class, it is not an instance of the DB class.
        # to create an instance of the class, you need to call return_value
       
        db_instance = db_class_mock.return_value
        db_instance.stat.return_value = 1
        # i prefere to do the above two lines in just one line as

        #db_class_mock.return_value.stat.return_value = 1
        

        db = Db('stat')
        expected = db.begin_transaction('Addis', 'Abeba')

        assert expected == 'Addis' + ':' + 'Abeba'

        # set the return value of stat method
        db_class_mock.return_value.stat.return_value = 0
        expected = db.begin_transaction('Addis', 'Abeba')

        assert expected == "wrong User"

Я надеюсь, что это поможет кому-то в сети