milvus/internal/proxy/dummyreq_test.go
Cai Yudong 2926a78968
Rename proxynode to proxy (#5961)
* rename package name from proxynode to proxy

Signed-off-by: yudong.cai <yudong.cai@zilliz.com>

* move distributed/proxynode to distributed/proxy

Signed-off-by: yudong.cai <yudong.cai@zilliz.com>

* move internal/proxynode to internal/proxy

Signed-off-by: yudong.cai <yudong.cai@zilliz.com>

* rename proxynode to proxy

Signed-off-by: yudong.cai <yudong.cai@zilliz.com>
2021-06-22 14:40:07 +08:00

60 lines
1.9 KiB
Go

// Copyright (C) 2019-2020 Zilliz. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance
// with the License. You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software distributed under the License
// is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
// or implied. See the License for the specific language governing permissions and limitations under the License.
package proxy
import (
"testing"
"github.com/stretchr/testify/assert"
)
func TestParseDummyRequestType(t *testing.T) {
invalidStr := `{"request_type":"retrieve"`
_, err := parseDummyRequestType(invalidStr)
assert.NotNil(t, err)
retrievetypeStr := `{"request_type":"retrieve"}`
drt, err := parseDummyRequestType(retrievetypeStr)
assert.Nil(t, err)
assert.Equal(t, drt.RequestType, "retrieve")
}
func TestParseDummyRetrieveRequest(t *testing.T) {
invalidStr := `{"request_type":"retrieve"`
_, err := parseDummyRetrieveRequest(invalidStr)
assert.NotNil(t, err)
onlytypeStr := `{"request_type":"retrieve"}`
drr, err := parseDummyRetrieveRequest(onlytypeStr)
assert.Nil(t, err)
assert.Equal(t, drr.RequestType, "retrieve")
assert.Equal(t, len(drr.DbName), 0)
fulltypeStr := `{
"request_type":"retrieve",
"dbname":"",
"collection_name":"test",
"partition_names": [],
"ids": [100, 101],
"output_fields": ["_id", "age"]
}`
drr2, err := parseDummyRetrieveRequest(fulltypeStr)
assert.Nil(t, err)
assert.Equal(t, drr2.RequestType, "retrieve")
assert.Equal(t, len(drr2.DbName), 0)
assert.Equal(t, drr2.CollectionName, "test")
assert.Equal(t, len(drr2.PartitionNames), 0)
assert.Equal(t, len(drr2.Ids), 2)
assert.Equal(t, drr2.Ids, []int64{100, 101})
assert.Equal(t, drr2.OutputFields, []string{"_id", "age"})
}