如何用Pytest模拟gRPC服务端响应测试Python gRPC客户端？

方法一：使用官方grpcio-testing库（推荐） #

这是gRPC官方提供的测试工具，能完整模拟gRPC服务端的各种行为（包括一元调用、流式调用），确保每次测试都返回固定响应。

步骤1：安装依赖 #

pip install grpcio-testing pytest

步骤2：编写测试代码 #

假设你已经通过proto生成了hello_pb2和hello_pb2_grpc模块，服务定义为HelloService，包含SayHello一元调用方法：

import pytest
import grpc_testing
import hello_pb2
import hello_pb2_grpc
from your_client_module import HelloClient  # 导入你的实际客户端类

@pytest.fixture
def mock_grpc_server():
    # 创建模拟服务端，指定支持的gRPC方法类型
    server = grpc_testing.server_from_dictionary(
        {
            hello_pb2.DESCRIPTOR.services_by_name['HelloService']: {
                'SayHello': lambda request, context: hello_pb2.HelloResponse(message=f"Hello, {request.name}!")
            }
        },
        grpc_testing.strict_real_time()
    )
    yield server
    server.stop(None)

@pytest.fixture
def test_client(mock_grpc_server):
    # 获取模拟服务端的存根，创建客户端实例
    stub = hello_pb2_grpc.HelloServiceStub(mock_grpc_server.channel())
    return HelloClient(stub)  # 假设你的客户端类接收存根作为参数

def test_say_hello(test_client):
    # 调用客户端方法，断言返回固定响应
    response = test_client.say_hello("Alice")
    assert response.message == "Hello, Alice!"

方法二：使用unittest.mock直接Mock存根方法 #

如果你的场景简单，不需要处理复杂的gRPC特性，可以直接Mock客户端依赖的存根方法，强制返回固定响应：

import pytest
from unittest.mock import Mock
import hello_pb2
from your_client_module import HelloClient

def test_say_hello_mocked():
    # 创建Mock存根
    mock_stub = Mock()
    # 指定存根方法返回固定响应
    mock_stub.SayHello.return_value = hello_pb2.HelloResponse(message="Hello, Mocked Alice!")
    
    # 初始化客户端并调用方法
    client = HelloClient(mock_stub)
    response = client.say_hello("Alice")
    
    # 断言结果
    assert response.message == "Hello, Mocked Alice!"
    # 验证存根方法被正确调用
    mock_stub.SayHello.assert_called_once_with(hello_pb2.HelloRequest(name="Alice"))

关键注意事项 #

• 每次测试前确保模拟对象被重置：使用Pytest的fixture（默认是function级别的）可以保证每个测试用例都拿到全新的模拟服务端/Mock对象，避免测试间的污染。
• 处理流式调用：如果你的客户端涉及服务器流、客户端流或双向流，grpcio-testing支持通过生成器返回固定的流式响应，比如：

  'ListMessages': lambda request, context: (hello_pb2.Message(content=f"Msg {i}") for i in range(3))
  

内容的提问来源于stack exchange，提问作者datahack