Python 2.x:如何模拟子进程.如果 stdin=PIPE，则打开

Question

>我正在尝试模拟下面的函数

from subprocess import Popen, PIPE    
def run_query():
sql_cmd = "Some Query"
process = Popen(["sqlplus", "-S", "/", "as", "sysdba"], stdin=PIPE, stdout=PIPE, stderr=PIPE)
process.stdin.write(sql_cmd)
(process_stdout, process_stderr) = process.communicate()

下面是我写的测试函数：

@patch('subprocess.Popen')
def test_run_query(Popen):
Popen.return_value.communicate.return_value = (2, 1)

但是，我得到以下错误

Error occured while running sql command
Error output:
[Errno 2] No such file or directory
F

我尝试了其他堆栈溢出帖子，但这种示例不存在。请帮忙。

Answer 1

您在错误的命名空间中修补Popen。

您需要在查找名称的命名空间中修补名称Popen，而不是在定义名称的位置。假设mypackage/mymodule.py是定义run_query的模块：

from mypackage.mymodule import run_query
@patch('mypackage.mymodule.Popen')
def test_run_query(mock_popen):
proc = mock_popen.return_value
proc.communicate.return_value = "2", "1"
out, err = run_query()
assert out == "2"
assert err == "1"
proc.stdin.write.assert_called_once_with("Some Query")

有关详细信息，请参阅模拟文档中的修补位置。