【发布时间】:2017-11-26 20:03:26
【问题描述】:
我们使用 grpc 作为我们所有内部系统的 RPC 协议。大多数系统是用Java编写的。
在 Java 中,我们可以使用 InprocessServerBuilder 进行单元测试。但是,我在 Python 中没有找到类似的类。
任何人都可以提供如何在 python 中进行 GRPC 单元测试的示例代码吗?
【问题讨论】:
标签: python python-3.x grpc
我们使用 grpc 作为我们所有内部系统的 RPC 协议。大多数系统是用Java编写的。
在 Java 中,我们可以使用 InprocessServerBuilder 进行单元测试。但是,我在 Python 中没有找到类似的类。
任何人都可以提供如何在 python 中进行 GRPC 单元测试的示例代码吗?
【问题讨论】:
标签: python python-3.x grpc
你今天问这个问题真是太巧了; our unit test framework just entered code review。所以目前测试的方式是use the full production stack to connect your client-side and server-side code(或者违反API并模拟很多内部的东西),但希望在几天到几周内你可以使用更好的解决方案。
【讨论】:
一些示例代码开始使用:
原型
syntax = "proto3";
service MyLibrary {
rpc Search (Request) returns (Response);
}
message Request {
string id = 1;
}
message Response {
string status = 1;
}
python 单元测试
#!/usr/bin/env python
# coding=utf-8
import unittest
from grpc import StatusCode
from grpc_testing import server_from_dictionary, strict_real_time
import mylibrary_pb2
class TestCase(unittest.TestCase):
def __init__(self, methodName) -> None:
super().__init__(methodName)
myServicer = MyLibraryServicer()
servicers = {
mylibrary_pb2.DESCRIPTOR.services_by_name['MyLibrary']: myServicer
}
self.test_server = server_from_dictionary(
servicers, strict_real_time())
def test_search(self):
request = mylibrary_pb2.Request(
id=2,
)
method = self.test_server.invoke_unary_unary(
method_descriptor=(mylibrary_pb2.DESCRIPTOR
.services_by_name['Library']
.methods_by_name['Search']),
invocation_metadata={},
request=request, timeout=1)
response, metadata, code, details = method.termination()
self.assertTrue(bool(response.status))
self.assertEqual(code, StatusCode.OK)
if __name__ == '__main__':
unittest.main()
【讨论】:
我发现 pytest-grpc 很容易上手并在几分钟内就可以运行。
【讨论】: