mirror of
https://github.com/etcd-io/etcd.git
synced 2024-09-27 06:25:44 +00:00
78 lines
2.2 KiB
Go
78 lines
2.2 KiB
Go
// Copyright 2016 The etcd Authors
|
|
//
|
|
// 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 clientv3
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
|
|
pb "go.etcd.io/etcd/api/v3/etcdserverpb"
|
|
)
|
|
|
|
// TestOpWithSort tests if WithSort(ASCEND, KEY) and WithLimit are specified,
|
|
// RangeRequest ignores the SortOption to avoid unnecessarily fetching
|
|
// the entire key-space.
|
|
func TestOpWithSort(t *testing.T) {
|
|
opReq := OpGet("foo", WithSort(SortByKey, SortAscend), WithLimit(10)).toRequestOp().Request
|
|
q, ok := opReq.(*pb.RequestOp_RequestRange)
|
|
if !ok {
|
|
t.Fatalf("expected range request, got %v", reflect.TypeOf(opReq))
|
|
}
|
|
req := q.RequestRange
|
|
wreq := &pb.RangeRequest{Key: []byte("foo"), SortOrder: pb.RangeRequest_NONE, Limit: 10}
|
|
if !reflect.DeepEqual(req, wreq) {
|
|
t.Fatalf("expected %+v, got %+v", wreq, req)
|
|
}
|
|
}
|
|
|
|
func TestIsSortOptionValid(t *testing.T) {
|
|
rangeReqs := []struct {
|
|
sortOrder pb.RangeRequest_SortOrder
|
|
sortTarget pb.RangeRequest_SortTarget
|
|
expectedValid bool
|
|
}{
|
|
{
|
|
sortOrder: pb.RangeRequest_ASCEND,
|
|
sortTarget: pb.RangeRequest_CREATE,
|
|
expectedValid: true,
|
|
},
|
|
{
|
|
sortOrder: pb.RangeRequest_ASCEND,
|
|
sortTarget: 100,
|
|
expectedValid: false,
|
|
},
|
|
{
|
|
sortOrder: 200,
|
|
sortTarget: pb.RangeRequest_MOD,
|
|
expectedValid: false,
|
|
},
|
|
}
|
|
|
|
for _, req := range rangeReqs {
|
|
getOp := Op{
|
|
sort: &SortOption{
|
|
Order: SortOrder(req.sortOrder),
|
|
Target: SortTarget(req.sortTarget),
|
|
},
|
|
}
|
|
|
|
actualRet := getOp.IsSortOptionValid()
|
|
if actualRet != req.expectedValid {
|
|
t.Errorf("expected sortOrder (%d) and sortTarget (%d) to be %t, but got %t",
|
|
req.sortOrder, req.sortTarget, req.expectedValid, actualRet)
|
|
}
|
|
}
|
|
}
|