一个grpc服务测试脚本工具（Python）

根据实际业务需要，写了一个方便测试grpc服务的脚本工具。
思路是如果进行服务的自动化测试，首先需要写出一个通用性的grpc_client调用方法。然后设计测试用例传参即可。

import grpc
import time

def common(service_dependence, request, _response, server='0.0.0.0:7000', **kwargs):
    # proto生成两个python文件：pd2_grpc结尾的service_stub的文件和pd2结尾的服务定义的文件
    pb2_grpc = service_dependence['pb2_grpc']
    # pb2_grpc_func 是生成stub对象的方法
    pb2_grpc_func = service_dependence['pb2_grpc_func']
    pb2 = service_dependence['pb2']
    # pb2_func 是关联服务中方法和请求参数的方法
    pb2_func = service_dependence['pb2_func']

    # 引用proto生成的两个文件
    import_pb2_grpc = 'import ' + pb2_grpc
    import_pb2 = 'import ' + pb2
    exec(import_pb2_grpc)
    exec(import_pb2)

    # 通用调用
    grpc_stub = pb2_grpc + '.' + pb2_grpc_func
    grpc_request = pb2 + '.' + pb2_func
    channel = grpc.insecure_channel(server)
    stub = eval(grpc_stub)(channel)
    start_time = time.time()
    response = eval("stub." + request)(eval(grpc_request)(**kwargs))
    end_time = time.time()
    # 服务耗时：以毫秒为单位
    elapsed = str(round((end_time - start_time)*1000, 3)) + 'ms'
    # 主要返回内容：如果方法的返回字段不能统一，也可以考虑传入一个返回字段的列表，把此处改写
    content = response.error_msg if response.code else eval("response." + _response)
    return response.code, content, elapsed

参考内容：https://github.com/grpc/grpc/tree/master/examples/python

版权声明：本文来源简书，感谢博主原创文章，遵循 CC 4.0 by-sa 版权协议，转载请附上原文出处链接和本声明。
原文链接：https://www.jianshu.com/p/471e44dd01b1
站方申明：本站部分内容来自社区用户分享，若涉及侵权，请联系站方删除。

发表于 2020-01-12 11:49:15
阅读 ( 2212 )
分类：

一个grpc服务测试脚本工具（Python）

你可能感兴趣的文章

精选的优质文章

0 条评论

官方社群

GO教程

推荐文章

猜你喜欢

随便看看