如何使用在另一个 python 文件中使用的 unittest 模拟变量的值

Question

I have the following file constants.py

# Here all the constants are written
variable_name = False
...

The above variable variable_name from constants.py is being used in another file b.py

#b.py

import constants
def func():
   if constants.variable_name:
       """ Do Something """
   else:
       """ Do Something """

Here, how can I mock the variable_name from constants.py to True to test the if condition of the function func using unittest?

Answer 1

If you just need to pre-set the value of a variable you can use the setUp method of the test case:

import constants
class test_class(unittest.TestCase):
    def setUp(self):
        constants.variable_name = True

Answer 2

You can use unittest.mock.patch to patch the object:

from unittest.mock import patch

with patch('constants.variable_name', True):
    func()

Answer 3

You can patch the variable as follows:

import constants
from mock import patch
@patch('constants.variable_name', True)