client/v3/op_test.go - main - githubmirror/etcd

client/v3/op_test.go
105 строк3 KB
caojiamingalan
expose op.isOptsWithFromKey and op.isOptsWithPrefix
11 июл 2023, 22:34
11 июл 2023, 22:3406579d9
// 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()}
	op := OpGet("key", optswithprefix...)
	if !IsOptsWithPrefix(optswithprefix) || !op.IsOptsWithPrefix() {
		t.Errorf("IsOptsWithPrefix = false, expected true")
	}

	optswithfromkey := []OpOption{WithFromKey()}
	op = OpGet("key", optswithfromkey...)
	if IsOptsWithPrefix(optswithfromkey) || op.IsOptsWithPrefix() {
		t.Errorf("IsOptsWithPrefix = true, expected false")
	}
}

func TestIsOptsWithFromKey(t *testing.T) {
	optswithfromkey := []OpOption{WithFromKey()}
	op := OpGet("key", optswithfromkey...)
	if !IsOptsWithFromKey(optswithfromkey) || !op.IsOptsWithFromKey() {
		t.Errorf("IsOptsWithFromKey = false, expected true")
	}

	optswithprefix := []OpOption{WithPrefix()}
	op = OpGet("key", optswithprefix...)
	if IsOptsWithFromKey(optswithprefix) || op.IsOptsWithFromKey() {
		t.Errorf("IsOptsWithFromKey = true, expected false")
	}
}