用Python进行gRPC接口测试(一)

gRPC 是一个高性能、通用的开源RPC框架，其由 Google 主要面向移动应用开发并基于HTTP/2 协议标准而设计，基于 ProtoBuf(Protocol Buffers) 序列化协议开发，且支持众多开发语言。

自gRPC推出以来，已经广泛应用于各种服务之中。在测试中，我们也越来越多地遇到gRPC接口相关的测试内容。测试一个gRPC接口，我们往往需要一个测试用的客户端，本文就为大家介绍如何用python来实现一个简易的gRPC客户端程序。

一、前期准备

gRPC 默认使用 protocol buffers，这是 Google 开源的一套成熟的结构数据序列化机制。在开始编写客户端之前，我们首先要安装一些必要的模块和工具。

1. pip install grpcio

2. pip install grpcio-tools

安装好了必要的模块和工具（编译器）之后，我们就可以根据proto协议文件生成所需的模块和方法。比如我们要测试的接口协议文档为helloword.proto，文档内容为：

1. // helloworld.proto

2. syntax = "proto3";

5. service Greeter {

6.     rpc SayHello(HelloRequest) returns (HelloReply) {}

7. }

8. message HelloRequest {

9.     string name = 1;

10. }

13. message HelloReply {

14.     string message = 1;

15. }

就可以执行以下命令来快速获取：

python -m grpc_tools.protoc --python_out=. --grpc_python_out=. -I. helloworld.proto

生成的模块、方法会保存在两个文件之中，分别为helloworld_pb2.py、helloworld_pb2_grpc.py。这两个文件是实现客户端时不可缺少的。

二、客户端实现

在前期准备完成之后，实现一个客户端并非难事，我们只需将自动生成的文件和模块引用进来，就可以调用一些内置方法来完成与gRPC接口的交互。

比如上面生成的文件是helloworld_pb2.py、helloworld_pb2_grpc.py，我们就在客户端引入这两个模块：

1. import helloworld_pb2

2. import helloworld_pb2_grpc

同时再引入必要的grpc模块：

import grpc

接下来就是调用模块中的方法来进行grpc接口的连接、测试了。如果服务端起在本地的50051端口，完整实现的代码如下：

1. import helloworld_pb2

2. import helloworld_pb2_grpc

3. import grpc

5. def run():

7.     # 连接 rpc 服务器

8.     channel = grpc.insecure_channel('localhost:50051')

9.     # 调用 rpc 服务

10.     stub = helloworld_pb2_grpc.GreeterStub(channel)

11.     response = stub.SayHello(helloworld_pb2.HelloRequest(name='test'))

12.     print("Greeter client received: " + response.message)

15. if __name__ == '__main__':

16.     run()

其中，关键的步骤为：

1. 1、连接rpc服务器；

3. 2、对service获取一个stub用于调用接口；

5. 3、发送数据、接收数据。

将代码保存为client.py，运行脚本就可以成功获取服务端返回的结果了，假设服务端返回的文本为"Hello"：

1. # python client.py

2. # Greeter client received: Hello

到这里，一次简单的gRPC接口连接、交互就算完成了，并实现了一个简易的gRPC客户端。

三、实际应用

在知音平台项目中，官网控制台需要获取某个用户对于某种服务的用量。官网控制台与后端服务之间使用gRPC接口，通信方式为客服端一次请求, 服务器一次应答。

按照上面介绍的方法，我们可以制作一个简单的测试客户端，模拟官网控制台对后端服务进行请求，实现对用户用量的查询。

接口协议如下（仅为演示需要，只展示部分内容）

1. service test {

2.   rpc GetUsage(GetUsageRequest) returns (GetUsageResponse);

3. }

6. message GetUsageRequest {

7.   string appid = 1;

8.   string account_id = 4;

9.   string audience = 2;

10.   string metric = 3;

11. }

14. message GetUsageResponse {

15.   Usage usage = 1;

16. }

实现客户端关键方法如下：

1. def subcmd_GetUsage(args):

2.     # 连接 rpc 服务器

3.     channel = grpc.insecure_channel(ADDRESS)

4.     # 调用 rpc 服务

5.     stub = test_pb2_grpc.testStub(channel)

6.     response = stub.GetUsage(auth_pb2.GetUsageRequest(appid=args.appid,account_id=args.account_id,audience=args.audience,metric=args.metric))

7.     print("GetUsage received: \n" + str(response.usage))

运行客户端（已利用argparse模块做成命令行形式），可以成功返回结果：

进一步，如果需要对接口进行并发下的稳定性测试，可以将客户端编译成可执行程序或利用shell脚本，再结合jmeter等自动化测试工具进行测试，以编译可执行程序的方法为例。

首先利用pyinstaller工具将脚本编译为可执行程序：

pyinstaller -F auth_client.py

可以在dist目录下得到当前系统下的可执行程序。

接下来用jmeter编写自动化测试脚本，在线程组下添加OS Process Sampler，传入所需参数：

运行脚本，即可自动化进行测试并得到结果，从中可以得到性能、稳定性等指标。

此外还可以结合jmeter的参数化功能，对不同用户对于不同服务的用量进行查询，从而更加真实地模拟线上请求的情景进行测试：

小结

本文介绍了用python实现gRPC接口测试的方法，可以快速地制作一个简易客户端应用于gRPC接口相关的测试，有一定实用价值。对于gRPC接口，本期主要讲了gRPC四种通信方式中最简单的一种，即客服端一次请求, 服务器一次应答（简单RPC）。关于其他三种通信方式的实现，感兴趣的同学可以参考gRPC相关文档，在之后的文章中，小编会再进行介绍。好了，本期就到这里，我们下期再见~

总结：

感谢每一个认真阅读我文章的人！！！

作为一位过来人也是希望大家少走一些弯路，如果你不想再体验一次学习时找不到资料，没人解答问题，坚持几天便放弃的感受的话，在这里我给大家分享一些自动化测试的学习资源，希望能给你前进的路上带来帮助。

1. 文档获取方式：

3. 加入我的软件测试交流群：680748947免费获取~（同行大佬一起学术交流，每晚都有大佬直播分享技术知识点）

这份文档，对于想从事【软件测试】的朋友来说应该是最全面最完整的备战仓库，这个仓库也陪伴我走过了最艰难的路程，希望也能帮助到你！

以上均可以分享，只需要你搜索vx公众号：程序员雨果，即可免费领取