Question

我正在使用Python 3.5.2。我正在尝试在不同的测试用例下模拟相同的方法，但是似乎sys.argv并没有被模拟。

我也尝试使用@patch装饰器，但无济于事。

main.py：

from os import path
from sys import argv

globals = {}

def get_target_dir():
    if len(argv) < 2 or not path.exists(argv[1]):
        print("You must specify valid full path of instances directory as command line argument.")
        exit(1)
    globals['TARGET_DIR'] = argv[1] + ('/' if argv[1][-1] != '/' else '')

tests.py：

import unittest
from unittest.mock import patch
from main import get_target_dir, globals

class TestMain(unittest.TestCase):
    def test_correct_target_dir(self):
        argv = [None, '/test']
        with patch('sys.argv', argv), patch('os.path.exists', lambda _: True):
            get_target_dir()
            assert globals['TARGET_DIR'] == argv[1] + '/'

    def test_invalid_target_dir(self):
        argv = [None, '/']
        with patch('sys.argv', argv), patch('os.path.exists', lambda _: False):
            try:
                get_target_dir()
            except SystemExit:
                assert True
            else:
                assert False

我运行测试时，由于上面的问题，它们无法正常工作。

Answer 1

我意识到我对gotchas一无所知，所以我错误地在sys.argv中包含了main.py方法-将导入from sys import argv替换为{{1 }}（并在代码中将import sys放在sys.之前）一切正常。

在测试中模拟特定方法不起作用

1 个答案: