编辑“︁Python 单元测试”︁（章节）

== 高级单元测试技巧 ==

=== 测试夹具（Fixtures） ===
测试夹具用于准备测试环境和清理资源。在unittest中，可以使用<code>setUp()</code>和<code>tearDown()</code>方法：

<syntaxhighlight lang="python">
class TestDatabase(unittest.TestCase):
    def setUp(self):
        self.conn = create_test_connection()
    
    def tearDown(self):
        self.conn.close()
    
    def test_query(self):
        result = self.conn.execute("SELECT 1")
        self.assertEqual(result, [1])
</syntaxhighlight>

在pytest中，可以使用装饰器定义fixture：

<syntaxhighlight lang="python">
import pytest

@pytest.fixture
def database_connection():
    conn = create_test_connection()
    yield conn
    conn.close()

def test_query(database_connection):
    result = database_connection.execute("SELECT 1")
    assert result == [1]
</syntaxhighlight>

=== 模拟对象（Mocking） ===
使用<code>unittest.mock</code>可以创建模拟对象，避免测试依赖外部系统：

<syntaxhighlight lang="python">
from unittest.mock import Mock

def test_api_call():
    mock_response = Mock()
    mock_response.status_code = 200
    mock_response.json.return_value = {"key": "value"}
    
    # 测试代码使用mock_response而不是真实API调用
    result = process_response(mock_response)
    assert result == "value"
</syntaxhighlight>