Mock object ที่ถูกสร้างใน method

โจทย์ของผมคือต้องใช้ Popen module subprocess แต่มีปัญหาว่าเราจะ mock ยังไงในเมื่อมันไป new object ใน method ไม่ได้มาจากการ inject ผ่าน constructor หรือ setter แต่อย่างใด ก็เลยเกิดไอเดียจากการนั่งอ่าน stackoverflow ทำให้เห็นภาพเป็นประมาณนี้

	import unittest
	import mock
	import subprocess
	class PythonSubprocessTest(unittest.TestCase):
	class MockPopen(object):
	def __init__(self):
	pass
	def communicate(self, input=None):
	pass
	@property
	def returncode(self):
	pass
	def test_process_with_mock_subprocess(self):
	mock_popen = PythonSubprocessTest.MockPopen()
	mock_popen.communicate = mock.Mock(return_value=('hello mock subprocess stdout', 'hello mock subprocess stderr'))
	mock_returncode = mock.PropertyMock(return_value=1)
	type(mock_popen).returncode = mock_returncode
	'''
	set attribute Popen return mock_popen when create Popen object
	'''
	setattr(subprocess, 'Popen', lambda *args, **kargs: mock_popen)
	expected_object = Process()
	self.assertEqual(expected_object.call_process(), ('hello mock subprocess stdout', 'hello mock subprocess stderr', 1))
	'''
	expected communicate must be call once time!!!
	'''
	mock_popen.communicate.assert_called_once_with()
	mock_returncode.assert_called_once_with()
	class Process(object):
	def __init__(self):
	pass
	def call_process(self):
	p = subprocess.Popen(['ls'], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
	stdout, stderr = p.communicate()
	exitcode = p.returncode
	return stdout, stderr, exitcode

view raw test_subprocess.py hosted with ❤ by GitHub

จากที่เห็นเราต้องสร้าง class MockPopen เพื่อเอามา mock ก่อนจากนั้นก็จัดการ mock method และ getter ต่อมาก็คือใช้ setattr เพื่อหลอกว่า Popen คือ MockPopen ที่เราสร้างขึ้น เมื่อ class Process เรียกใช้ก็ยังเห็นว่ามันเป็น Popen และที่เราเทสนั้นก็ยังกำหนด return_value และยังสามารถ verify ได้ว่า method ได้ถูก call หรือไม่