mirror of
https://github.com/etcd-io/etcd.git
synced 2024-09-27 06:25:44 +00:00
8637c54bcc
Make sure that WithPrefix correctly set the flag, and add test. Also, add test for WithFromKey. fixes #14056 Signed-off-by: Sahdev Zala <spzala@us.ibm.com>
102 lines
2.8 KiB
Go
102 lines
2.8 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)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestIsOptsWithPrefix(t *testing.T) {
|
|
optswithprefix := []OpOption{WithPrefix()}
|
|
if !IsOptsWithPrefix(optswithprefix) {
|
|
t.Errorf("IsOptsWithPrefix = false, expected true")
|
|
}
|
|
|
|
optswithfromkey := []OpOption{WithFromKey()}
|
|
if IsOptsWithPrefix(optswithfromkey) {
|
|
t.Errorf("IsOptsWithPrefix = true, expected false")
|
|
}
|
|
}
|
|
|
|
func TestIsOptsWithFromKey(t *testing.T) {
|
|
optswithfromkey := []OpOption{WithFromKey()}
|
|
if !IsOptsWithFromKey(optswithfromkey) {
|
|
t.Errorf("IsOptsWithFromKey = false, expected true")
|
|
}
|
|
|
|
optswithprefix := []OpOption{WithPrefix()}
|
|
if IsOptsWithFromKey(optswithprefix) {
|
|
t.Errorf("IsOptsWithFromKey = true, expected false")
|
|
}
|
|
}
|