我的工作流程中有几个任务。
class Task1():
def run(self):
try:
return 2,2
except Exception as e:
print e
raise Exception('Task1 Exception')
class Task2():
def run(self,x,y):
try:
return 2 * (x*y)
except Exception as e:
print e
raise Exception('Task2 Exception')
class Workflow():
def run(self,task1, task2):
x,y = task1.run()
print task2.run(x,y)
task1 = Task1()
task2 = Task2()
w = Workflow()
w.run(task1,task2)
这是我的UNITSEST -
import unittest
import mock
from test2 import Task1, Task2, Workflow
class TestWorkflow(unittest.TestCase):
def test_workflow(self):
self.task1 = mock.MagicMock(Task1()) # mocking task1
self.task2 = mock.MagicMock(Task2())
self.task1.side_effect = Exception() # setting task1 exception to an exception object
self.workflow = Workflow()
self.workflow.run(self.task1, self.task2)
self.task2.run.assert_not_called() # checking task2 is not called.
if __name__ == '__main__':
unittest.main()
例外: -
x,y = task1.run()
ValueError: need more than 0 values to unpack
鉴于语法,您使用的是Python 2,但是无论如何,我将在Python 3中参考文档。我相信解决方案适用。
第一个示例是:
from unittest.mock import MagicMock
thing = ProductionClass()
thing.method = MagicMock(return_value=3)
thing.method(3, 4, 5, key='value')
#prints 3
thing.method.assert_called_with(3, 4, 5, key='value')
带来这个问题,您将拥有以下代码而不是当前代码。请参阅我将task1.run和task2.run方法分配给magicMock实例。
import unittest
import mock
from t import Task1, Task2, Workflow
class TestWorkflow(unittest.TestCase):
def test_workflow(self):
self.task1 = Task1()
self.task1.run = mock.MagicMock() # mocking run from task1
self.task2 = Task2()
self.task2.run = mock.MagicMock()
self.task1.run.side_effect = Exception() # setting task1 exception to an exception object
self.workflow = Workflow()
self.workflow.run(self.task1, self.task2)
self.task2.run.assert_not_called() # checking task2 is not called.
if __name__ == '__main__':
unittest.main()