Return 或 python 模拟中的副作用?
Return or break as a side effect in python mock?
我有一个多处理应用程序,在 "while True" 个循环中有工作人员 运行。为了进行测试,我希望能够以这样的方式模拟 sys.exit():
with mock.patch('sys.exit') as sys_mock:
sys_mock.side_effect = break
或
with mock.patch('sys.exit') as sys_mock:
sys_mock.side_effect = return
所以我可以跳出循环并完成我的测试。这些都不起作用,但是有没有不同的方法来完成我想要完成的事情?
您可以将 Exception
用作 side_effect
来模拟 sys.exit
行为而无需退出测试。
side_effect
文档说:
This can either be a function to be called when the mock is called, an iterable or an exception (class or instance) to be raised.
所以你不能使用像 break
或 return
这样的 语句 但你想要做的是退出你的 运行 循环这可以通过引发异常来获得......我希望你不要在线程的主循环中使用疯狂的 try
-except
。
我写了一个 简单的 例子来测试它,我使用装饰器 patch
语法并放入内联 side_effect=Exception
使测试更具可读性:
import sys
import threading
import unittest
from unittest.mock import patch
class T(threading.Thread):
def __init__(self, *args, **kwargs):
super(T, self).__init__(*args, **kwargs)
self._interrupt = threading.Event()
self.started = threading.Event() #Used to be sure that we test run() behavior
self.started.clear()
self.terminated = False
def interrupt(self):
self._interrupt.set()
def run(self, *args, **kwargs):
self._interrupt.clear()
self.started.set()
while not self._interrupt.is_set():
self._interrupt.wait(timeout=1)
self.terminated = True
sys.exit()
class TestInterrupt(unittest.TestCase):
@patch("sys.exit", side_effect=Exception("Ignore it... just close thread"))
def test_interrupt(self, mock_sys_exit):
t = T()
t.start()
if not t.started.is_set():
t.started.wait(timeout=0.2)
self.assertTrue(t.started.is_set(), "t not started!")
#Ok t is in run() main cycle: we can test interrupt
t.interrupt()
t.join(0.1)
self.assertTrue(t.terminated)
self.assertFalse(t.isAlive())
我有一个多处理应用程序,在 "while True" 个循环中有工作人员 运行。为了进行测试,我希望能够以这样的方式模拟 sys.exit():
with mock.patch('sys.exit') as sys_mock:
sys_mock.side_effect = break
或
with mock.patch('sys.exit') as sys_mock:
sys_mock.side_effect = return
所以我可以跳出循环并完成我的测试。这些都不起作用,但是有没有不同的方法来完成我想要完成的事情?
您可以将 Exception
用作 side_effect
来模拟 sys.exit
行为而无需退出测试。
side_effect
文档说:
This can either be a function to be called when the mock is called, an iterable or an exception (class or instance) to be raised.
所以你不能使用像 break
或 return
这样的 语句 但你想要做的是退出你的 运行 循环这可以通过引发异常来获得......我希望你不要在线程的主循环中使用疯狂的 try
-except
。
我写了一个 简单的 例子来测试它,我使用装饰器 patch
语法并放入内联 side_effect=Exception
使测试更具可读性:
import sys
import threading
import unittest
from unittest.mock import patch
class T(threading.Thread):
def __init__(self, *args, **kwargs):
super(T, self).__init__(*args, **kwargs)
self._interrupt = threading.Event()
self.started = threading.Event() #Used to be sure that we test run() behavior
self.started.clear()
self.terminated = False
def interrupt(self):
self._interrupt.set()
def run(self, *args, **kwargs):
self._interrupt.clear()
self.started.set()
while not self._interrupt.is_set():
self._interrupt.wait(timeout=1)
self.terminated = True
sys.exit()
class TestInterrupt(unittest.TestCase):
@patch("sys.exit", side_effect=Exception("Ignore it... just close thread"))
def test_interrupt(self, mock_sys_exit):
t = T()
t.start()
if not t.started.is_set():
t.started.wait(timeout=0.2)
self.assertTrue(t.started.is_set(), "t not started!")
#Ok t is in run() main cycle: we can test interrupt
t.interrupt()
t.join(0.1)
self.assertTrue(t.terminated)
self.assertFalse(t.isAlive())