Тесты Django - объект патча во всех тестах

Question

Мне нужно создать какой-то MockMixin для моих тестов. Он должен включать в себя mocks для всего, что вызывает внешние источники. Например, каждый раз, когда я сохраняю модель в панели администратора, я вызываю несколько удаленных URL-адресов. Было бы хорошо, чтобы иметь, что издевались и использовать так:

class ExampleTestCase(MockedTestCase): 
    # tests 

Таким образом, каждый раз, когда я сохранить модель в администратора, например, в функциональных тестах, это насмешка применяется вместо вызова удаленных URL-адресов.

Возможно ли это? Я могу сделать это для одного конкретного теста, это не проблема. Но было бы более полезно иметь какой-то глобальный макет, потому что я использую его много.

Answer 1

Согласно mock documentation:

патч может быть использован в качестве класса TestCase декоратора. Он работает , украшая каждый метод тестирования в классе. Это уменьшает код шаблона , когда ваши методы тестирования имеют общий набор исправлений.

В основном это означает, что вы можете создать базовый тестовый класс с @patch декоратора нанесенным на него, что бы издеваться ваши внешние вызовы, в то время как каждый метод испытания внутри будет выполняться.

Кроме того, вы можете использовать start() and stop() методы Patcher в в setUp() и tearDown() методы соответственно:

class BaseTestCase(TestCase): 
    def setUp(self): 
     self.patcher = patch('mymodule.foo') 
     self.mock_foo = self.patcher.start() 

    def tearDown(self): 
     self.patcher.stop() 

Answer 2

Просто, чтобы добавить к ответу alecxe, в если вы используете teardown, то в соответствии с Документами

вам должен убедиться, что исправление «отменено», вызвав стоп

Итак, если в ваших тестах возникает исключение, исправление не будет отменено. Лучшим способом было бы позвонить addCleanup внутри вашего setUp. Тогда вы можете вообще опустить метод tearDown.

class BaseTestCase(TestCase): 
    def setUp(self): 
     self.patcher = patch('mymodule.foo') 
     self.mock_foo = self.patcher.start() 
     self.addCleanup(self.patcher.stop) # add this line