Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
BaiXuePrincess
milvus
提交
8ad5d6c2
milvus
项目概览
BaiXuePrincess
/
milvus
与 Fork 源项目一致
从无法访问的项目Fork
通知
7
Star
4
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
milvus
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
8ad5d6c2
编写于
10月 14, 2019
作者:
P
peng.xu
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
add test_grpc
上级
c4f7b7c4
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
77 addition
and
0 deletion
+77
-0
mishards/grpc_utils/test_grpc.py
mishards/grpc_utils/test_grpc.py
+77
-0
未找到文件。
mishards/grpc_utils/test_grpc.py
0 → 100644
浏览文件 @
8ad5d6c2
import
logging
import
opentracing
from
mishards.grpc_utils
import
GrpcSpanDecorator
,
is_grpc_method
from
milvus.grpc_gen
import
status_pb2
,
milvus_pb2
logger
=
logging
.
getLogger
(
__name__
)
class
TestTracer
(
opentracing
.
Tracer
):
pass
class
TestSpan
(
opentracing
.
Span
):
def
__init__
(
self
,
context
,
tracer
,
**
kwargs
):
super
(
TestSpan
,
self
).
__init__
(
tracer
,
context
)
self
.
reset
()
def
set_tag
(
self
,
key
,
value
):
self
.
tags
.
append
({
key
:
value
})
def
log_kv
(
self
,
key_values
,
timestamp
=
None
):
self
.
logs
.
append
(
key_values
)
def
reset
(
self
):
self
.
tags
=
[]
self
.
logs
=
[]
class
TestRpcInfo
:
def
__init__
(
self
,
request
,
response
):
self
.
request
=
request
self
.
response
=
response
class
TestGrpcUtils
:
def
test_span_deco
(
self
):
request
=
'request'
OK
=
status_pb2
.
Status
(
error_code
=
status_pb2
.
SUCCESS
,
reason
=
'Success'
)
response
=
OK
rpc_info
=
TestRpcInfo
(
request
=
request
,
response
=
response
)
span
=
TestSpan
(
context
=
None
,
tracer
=
TestTracer
())
span_deco
=
GrpcSpanDecorator
()
span_deco
(
span
,
rpc_info
)
assert
len
(
span
.
logs
)
==
0
assert
len
(
span
.
tags
)
==
0
response
=
milvus_pb2
.
BoolReply
(
status
=
OK
,
bool_reply
=
False
)
rpc_info
=
TestRpcInfo
(
request
=
request
,
response
=
response
)
span
=
TestSpan
(
context
=
None
,
tracer
=
TestTracer
())
span_deco
=
GrpcSpanDecorator
()
span_deco
(
span
,
rpc_info
)
assert
len
(
span
.
logs
)
==
0
assert
len
(
span
.
tags
)
==
0
response
=
1
rpc_info
=
TestRpcInfo
(
request
=
request
,
response
=
response
)
span
=
TestSpan
(
context
=
None
,
tracer
=
TestTracer
())
span_deco
=
GrpcSpanDecorator
()
span_deco
(
span
,
rpc_info
)
logger
.
error
(
span
.
logs
)
assert
len
(
span
.
logs
)
==
1
assert
len
(
span
.
tags
)
==
1
response
=
0
rpc_info
=
TestRpcInfo
(
request
=
request
,
response
=
response
)
span
=
TestSpan
(
context
=
None
,
tracer
=
TestTracer
())
span_deco
=
GrpcSpanDecorator
()
span_deco
(
span
,
rpc_info
)
logger
.
error
(
span
.
logs
)
assert
len
(
span
.
logs
)
==
0
assert
len
(
span
.
tags
)
==
0
def
test_is_grpc_method
(
self
):
target
=
1
assert
not
is_grpc_method
(
target
)
target
=
None
assert
not
is_grpc_method
(
target
)
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录