mirror of
https://github.com/etcd-io/etcd.git
synced 2024-09-27 06:25:44 +00:00
4df61af2df
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>
63 lines
1.9 KiB
Go
63 lines
1.9 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 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")
|
|
}
|
|
}
|