admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2026-01-07 19:31:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

enhance: add test for new parameters in access log (#42546)

Browse Source 
relate: https://github.com/milvus-io/milvus/issues/41801

Signed-off-by: aoiasd <zhicheng.yue@zilliz.com>
This commit is contained in:
aoiasd 2025-06-17 11:14:38 +08:00 committed by GitHub
parent b721eed3d6
commit 4e68c6d222
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 98 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
internal/proxy/accesslog/info/grpc_info_test.go
View File

@ -232,6 +232,55 @@ func (s *GrpcAccessInfoSuite) TestClusterPrefix() {
s.Equal(cluster, result[0])
}
func (s *GrpcAccessInfoSuite) TestNQ() {
nq := int64(10)
s.Equal(Unknown, Get(s.info, "$nq")[0])
s.info.req = &milvuspb.SearchRequest{
Nq: nq,
}
s.Equal(fmt.Sprintf("%d", nq), Get(s.info, "$nq")[0])
s.info.req = &milvuspb.HybridSearchRequest{
Requests: []*milvuspb.SearchRequest{{
Nq: nq,
}, {
Nq: nq,
}},
}
s.Equal("[\"10\", \"10\"]", Get(s.info, "$nq")[0])
}
func (s *GrpcAccessInfoSuite) TestSearchParams() {
params := []*commonpb.KeyValuePair{{Key: "test_key", Value: "test_value"}}
s.Equal(Unknown, Get(s.info, "$search_params")[0])
s.info.req = &milvuspb.SearchRequest{
SearchParams: params,
}
s.Equal(kvsToString(params), Get(s.info, "$search_params")[0])
s.info.req = &milvuspb.HybridSearchRequest{
Requests: []*milvuspb.SearchRequest{{SearchParams: params}, {SearchParams: params}},
}
s.Equal(listToString([]string{kvsToString(params), kvsToString(params)}), Get(s.info, "$search_params")[0])
}
func (s *GrpcAccessInfoSuite) TestQueryParams() {
params := []*commonpb.KeyValuePair{{Key: "test_key", Value: "test_value"}}
s.Equal(Unknown, Get(s.info, "$query_params")[0])
s.info.req = &milvuspb.QueryRequest{
QueryParams: params,
}
s.Equal(kvsToString(params), Get(s.info, "$query_params")[0])
}
func TestGrpcAccssInfo(t *testing.T) {
suite.Run(t, new(GrpcAccessInfoSuite))
}

49
internal/proxy/accesslog/info/restful_info_test.go
View File

@ -204,6 +204,55 @@ func (s *RestfulAccessInfoSuite) TestClusterPrefix() {
s.Equal(cluster, result[0])
}
func (s *RestfulAccessInfoSuite) TestNQ() {
nq := int64(10)
s.Equal(Unknown, Get(s.info, "$nq")[0])
s.info.req = &milvuspb.SearchRequest{
Nq: nq,
}
s.Equal(fmt.Sprintf("%d", nq), Get(s.info, "$nq")[0])
s.info.req = &milvuspb.HybridSearchRequest{
Requests: []*milvuspb.SearchRequest{{
Nq: nq,
}, {
Nq: nq,
}},
}
s.Equal("[\"10\", \"10\"]", Get(s.info, "$nq")[0])
}
func (s *RestfulAccessInfoSuite) TestSearchParams() {
params := []*commonpb.KeyValuePair{{Key: "test_key", Value: "test_value"}}
s.Equal(Unknown, Get(s.info, "$search_params")[0])
s.info.req = &milvuspb.SearchRequest{
SearchParams: params,
}
s.Equal(kvsToString(params), Get(s.info, "$search_params")[0])
s.info.req = &milvuspb.HybridSearchRequest{
Requests: []*milvuspb.SearchRequest{{SearchParams: params}, {SearchParams: params}},
}
s.Equal(listToString([]string{kvsToString(params), kvsToString(params)}), Get(s.info, "$search_params")[0])
}
func (s *RestfulAccessInfoSuite) TestQueryParams() {
params := []*commonpb.KeyValuePair{{Key: "test_key", Value: "test_value"}}
s.Equal(Unknown, Get(s.info, "$query_params")[0])
s.info.req = &milvuspb.QueryRequest{
QueryParams: params,
}
s.Equal(kvsToString(params), Get(s.info, "$query_params")[0])
}
func TestRestfulAccessInfo(t *testing.T) {
suite.Run(t, new(RestfulAccessInfoSuite))
}