模拟全局变量

Question

我一直在尝试为模块实现一些单元测试。一个名为 alphabet.py 的示例模块如下：

import database

def length_letters():
    return len(letters)

def contains_letter(letter):
    return True if letter in letters else False


letters = database.get('letters')   # returns a list of letters

我想用我选择的一些值来模拟来自数据库的响应，但下面的代码似乎不起作用。

import unittests  
import alphabet   
from unittest.mock import patch   


class TestAlphabet(unittest.TestCase): 
    @patch('alphabet.letters')
    def setUp(self, mock_letters):
        mock_letters.return_value = ['a', 'b', 'c']   

    def test_length_letters(self):
        self.assertEqual(3, alphabet.length_letters())

    def test_contains_letter(self):   
        self.assertTrue(alphabet.contains_letter('a'))

我见过很多例子，其中“补丁”应用于方法和类，但不应用于变量。我不想修补 database.get 方法，因为稍后我可能会再次使用它并使用不同的参数，所以我需要不同的响应。

我在这里做错了什么？

Answer 1

您不需要使用模拟。只需导入模块并更改 setUp() 中的全局值即可：

import alphabet

class TestAlphabet(unittest.TestCase): 
   def setUp(self):
        alphabet.letters = ['a', 'b', 'c']

Answer 2

试试这个：

import unittests  
import alphabet   
from unittest import mock 


class TestAlphabet(unittest.TestCase): 
    def setUp(self):
        self.mock_letters = mock.patch.object(
            alphabet, 'letters', return_value=['a', 'b', 'c']
        )

    def test_length_letters(self):
        with self.mock_letters:
            self.assertEqual(3, alphabet.length_letters())

    def test_contains_letter(self):
        with self.mock_letters:
            self.assertTrue(alphabet.contains_letter('a'))

您需要在各个测试实际运行时应用模拟，而不仅仅是在setUp() 中。我们可以在setUp() 中创建模拟，然后使用with ... 上下文管理器应用它。

Answer 3

我遇到了一个问题，我试图模拟出在任何函数或类之外使用的变量，这是有问题的，因为它们在你尝试模拟类的那一刻被使用，然后你才能模拟这些值。

我最终使用了一个环境变量。如果环境变量存在，则使用该值，否则使用应用程序默认值。这样我就可以在我的测试中设置环境变量的值了。

在我的测试中，我在导入类之前就有了这段代码

os.environ["PROFILER_LOG_PATH"] = "./"

在我的课堂上：

log_path = os.environ.get("PROFILER_LOG_PATH",config.LOG_PATH)

默认我的config.LOG_PATH是/var/log/<my app name>，但是现在运行测试时，日志路径设置为当前目录。这样您就不需要 root 访问权限来运行测试。

Answer 4

可以按如下方式修补变量：

from mock import patch
@patch('module.variable', new_value)    

例如：

import alphabet
from mock import patch
@patch('alphabet.letters', ['a', 'b', 'c'])
class TestAlphabet():

    def test_length_letters(self):
        assert 3 == alphabet.length_letters()

    def test_contains_letter(self):
       assert alphabet.contains_letter('a')