Python mock os.environ, используемый внутри класса

Question

Я пытаюсь высмеять os.environ внутри класса, но я просто не могу понять это правильно. Вот моя структура:

#file.py 
import os 

class MyClass(): 
    connection_url = os.environ['DB'] 

#some code 

А вот мой тест (последняя попытка, так или иначе):

#test.py 
from unittest import TestCase 
from unittest.mock import patch 
from file import MyClass 

class TestMyClass(TestCase): 
    @patch.dict('file.os.environ', {'DB' : 'Dummy' }) 
    def setUp(self): 
     self.class = MyClass() 

#some testing 

Это провал с треском, поднимая KeyError 'DB' ... Может кто-нибудь мне помочь? Я новичок в python unittesting. Я исследовал некоторые блоги и stackoverflow, пробовал некоторые решения, но не мог понять.

Заранее благодарен!

Answer 1

Проблема заключается в том, что connection_url задается при создании класса (при импорте). Если вы хотите, чтобы дразнить его, вы можете пропатчить атрибут самого класса, прежде чем использовать что атрибут:

class TestMyClass(TestCase): 
    @patch.object(file.MyClass, 'connection_url', 'Dummy') 
    def setUp(self): 
     self.instance = MyClass() 

Вы все еще будете иметь дело с потенциальным KeyError во время импорта - то есть вы» вам нужно убедиться, что ваш тестовый бегун имеет фиктивное значение в среде перед вами импортировать файл, иначе вам нужно будет изменить код в file.py, чтобы убедиться, что он не поднимает KeyError. Здесь есть несколько стратегий. Вы можете просто подавить KeyError позволяя connection_url = default_value, если он не был установлен в окружающей среде:

Или вы можете переместить connection_url забирающий в __init__:

class MyClass(): 
    def __init__(self): 
     connection_url = os.environ['DB'] 

Этот последний код имеет дополнительное преимущество, что ваш patch.dict должен начать работать.

Обратите внимание, что если вы выберете один из первых двух подходов, экземпляр будет иметь только исправленный connection_url во время метода setUp. Последующие тесты не будут исправлены. Вполне вероятно, что это будет шоу-стоппер, и вам необходимо создать новый экземпляр в каждом тесте, что у вас есть:

class TestMyClass(TestCase): 
    @patch.object(file.MyClass, 'connection_url', 'Dummy') 
    def test_first_thing(self): 
     self.instance = MyClass() 
     ... 

    @patch.object(file.MyClass, 'connection_url', 'Dummy') 
    def test_second_thing(self): 
     self.instance = MyClass() 
     ...