openbilibili/library/cache/memcache/client_test.go

package memcache

import (
	"context"
	"fmt"
	"reflect"
	"testing"
	"time"
)

var testClient *Memcache

func Test_client_Set(t *testing.T) {
	type args struct {
		c    context.Context
		item *Item
	}
	tests := []struct {
		name    string
		args    args
		wantErr bool
	}{
		{name: "set value", args: args{c: context.Background(), item: &Item{Key: "Test_client_Set", Value: []byte("abc")}}, wantErr: false},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if err := testClient.Set(tt.args.c, tt.args.item); (err != nil) != tt.wantErr {
				t.Errorf("client.Set() error = %v, wantErr %v", err, tt.wantErr)
			}
		})
	}
}

func Test_client_Add(t *testing.T) {
	type args struct {
		c    context.Context
		item *Item
	}
	key := fmt.Sprintf("Test_client_Add_%d", time.Now().Unix())
	tests := []struct {
		name    string
		args    args
		wantErr bool
	}{
		{name: "add not exist value", args: args{c: context.Background(), item: &Item{Key: key, Value: []byte("abc")}}, wantErr: false},
		{name: "add exist value", args: args{c: context.Background(), item: &Item{Key: key, Value: []byte("abc")}}, wantErr: true},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if err := testClient.Add(tt.args.c, tt.args.item); (err != nil) != tt.wantErr {
				t.Errorf("client.Add() error = %v, wantErr %v", err, tt.wantErr)
			}
		})
	}
}

func Test_client_Replace(t *testing.T) {
	key := fmt.Sprintf("Test_client_Replace_%d", time.Now().Unix())
	ekey := "Test_client_Replace_exist"
	testClient.Set(context.Background(), &Item{Key: ekey, Value: []byte("ok")})
	type args struct {
		c    context.Context
		item *Item
	}
	tests := []struct {
		name    string
		args    args
		wantErr bool
	}{
		{name: "not exist value", args: args{c: context.Background(), item: &Item{Key: key, Value: []byte("abc")}}, wantErr: true},
		{name: "exist value", args: args{c: context.Background(), item: &Item{Key: ekey, Value: []byte("abc")}}, wantErr: false},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if err := testClient.Replace(tt.args.c, tt.args.item); (err != nil) != tt.wantErr {
				t.Errorf("client.Replace() error = %v, wantErr %v", err, tt.wantErr)
			}
		})
	}
}

func Test_client_CompareAndSwap(t *testing.T) {
	key := fmt.Sprintf("Test_client_CompareAndSwap_%d", time.Now().Unix())
	ekey := "Test_client_CompareAndSwap_k"
	testClient.Set(context.Background(), &Item{Key: ekey, Value: []byte("old")})
	cas := testClient.Get(context.Background(), ekey).Item().cas
	type args struct {
		c    context.Context
		item *Item
	}
	tests := []struct {
		name    string
		args    args
		wantErr bool
	}{
		{name: "not exist value", args: args{c: context.Background(), item: &Item{Key: key, Value: []byte("abc")}}, wantErr: true},
		{name: "exist value", args: args{c: context.Background(), item: &Item{Key: ekey, cas: cas, Value: []byte("abc")}}, wantErr: false},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if err := testClient.CompareAndSwap(tt.args.c, tt.args.item); (err != nil) != tt.wantErr {
				t.Errorf("client.CompareAndSwap() error = %v, wantErr %v", err, tt.wantErr)
			}
		})
	}
}

func Test_client_Get(t *testing.T) {
	key := fmt.Sprintf("Test_client_Get_%d", time.Now().Unix())
	ekey := "Test_client_Get_k"
	testClient.Set(context.Background(), &Item{Key: ekey, Value: []byte("old")})
	type args struct {
		c   context.Context
		key string
	}
	tests := []struct {
		name    string
		args    args
		want    string
		wantErr bool
	}{
		{name: "not exist value", args: args{c: context.Background(), key: key}, wantErr: true},
		{name: "exist value", args: args{c: context.Background(), key: ekey}, wantErr: false, want: "old"},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			var res string
			if err := testClient.Get(tt.args.c, tt.args.key).Scan(&res); (err != nil) != tt.wantErr || res != tt.want {
				t.Errorf("client.Get() = %v, want %v, got err: %v, want err: %v", err, tt.want, err, tt.wantErr)
			}
		})
	}
}

func Test_client_Touch(t *testing.T) {
	key := fmt.Sprintf("Test_client_Touch_%d", time.Now().Unix())
	ekey := "Test_client_Touch_k"
	testClient.Set(context.Background(), &Item{Key: ekey, Value: []byte("old")})
	type args struct {
		c       context.Context
		key     string
		timeout int32
	}
	tests := []struct {
		name    string
		args    args
		wantErr bool
	}{
		{name: "not exist value", args: args{c: context.Background(), key: key, timeout: 100000}, wantErr: true},
		{name: "exist value", args: args{c: context.Background(), key: ekey, timeout: 100000}, wantErr: false},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if err := testClient.Touch(tt.args.c, tt.args.key, tt.args.timeout); (err != nil) != tt.wantErr {
				t.Errorf("client.Touch() error = %v, wantErr %v", err, tt.wantErr)
			}
		})
	}
}

func Test_client_Delete(t *testing.T) {
	key := fmt.Sprintf("Test_client_Delete_%d", time.Now().Unix())
	ekey := "Test_client_Delete_k"
	testClient.Set(context.Background(), &Item{Key: ekey, Value: []byte("old")})
	type args struct {
		c   context.Context
		key string
	}
	tests := []struct {
		name    string
		args    args
		wantErr bool
	}{
		{name: "not exist value", args: args{c: context.Background(), key: key}, wantErr: true},
		{name: "exist value", args: args{c: context.Background(), key: ekey}, wantErr: false},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if err := testClient.Delete(tt.args.c, tt.args.key); (err != nil) != tt.wantErr {
				t.Errorf("client.Delete() error = %v, wantErr %v", err, tt.wantErr)
			}
		})
	}
}

func Test_client_Increment(t *testing.T) {
	key := fmt.Sprintf("Test_client_Increment_%d", time.Now().Unix())
	ekey := "Test_client_Increment_k"
	testClient.Set(context.Background(), &Item{Key: ekey, Value: []byte("1")})
	type args struct {
		c     context.Context
		key   string
		delta uint64
	}
	tests := []struct {
		name         string
		args         args
		wantNewValue uint64
		wantErr      bool
	}{
		{name: "not exist value", args: args{c: context.Background(), key: key, delta: 10}, wantErr: true},
		{name: "exist value", args: args{c: context.Background(), key: ekey, delta: 10}, wantErr: false, wantNewValue: 11},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			gotNewValue, err := testClient.Increment(tt.args.c, tt.args.key, tt.args.delta)
			if (err != nil) != tt.wantErr {
				t.Errorf("client.Increment() error = %v, wantErr %v", err, tt.wantErr)
				return
			}
			if gotNewValue != tt.wantNewValue {
				t.Errorf("client.Increment() = %v, want %v", gotNewValue, tt.wantNewValue)
			}
		})
	}
}

func Test_client_Decrement(t *testing.T) {
	key := fmt.Sprintf("Test_client_Decrement_%d", time.Now().Unix())
	ekey := "Test_client_Decrement_k"
	testClient.Set(context.Background(), &Item{Key: ekey, Value: []byte("100")})
	type args struct {
		c     context.Context
		key   string
		delta uint64
	}
	tests := []struct {
		name         string
		args         args
		wantNewValue uint64
		wantErr      bool
	}{
		{name: "not exist value", args: args{c: context.Background(), key: key, delta: 10}, wantErr: true},
		{name: "exist value", args: args{c: context.Background(), key: ekey, delta: 10}, wantErr: false, wantNewValue: 90},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			gotNewValue, err := testClient.Decrement(tt.args.c, tt.args.key, tt.args.delta)
			if (err != nil) != tt.wantErr {
				t.Errorf("client.Decrement() error = %v, wantErr %v", err, tt.wantErr)
				return
			}
			if gotNewValue != tt.wantNewValue {
				t.Errorf("client.Decrement() = %v, want %v", gotNewValue, tt.wantNewValue)
			}
		})
	}
}

func Test_client_GetMulti(t *testing.T) {
	key := fmt.Sprintf("Test_client_GetMulti_%d", time.Now().Unix())
	ekey1 := "Test_client_GetMulti_k1"
	ekey2 := "Test_client_GetMulti_k2"
	testClient.Set(context.Background(), &Item{Key: ekey1, Value: []byte("1")})
	testClient.Set(context.Background(), &Item{Key: ekey2, Value: []byte("2")})
	keys := []string{key, ekey1, ekey2}
	rows, err := testClient.GetMulti(context.Background(), keys)
	if err != nil {
		t.Errorf("client.GetMulti() error = %v, wantErr %v", err, nil)
	}
	tests := []struct {
		key          string
		wantNewValue string
		wantErr      bool
		nilItem      bool
	}{
		{key: key, wantErr: true, nilItem: true},
		{key: ekey1, wantErr: false, wantNewValue: "1", nilItem: false},
		{key: ekey2, wantErr: false, wantNewValue: "2", nilItem: false},
	}
	if reflect.DeepEqual(keys, rows.Keys()) {
		t.Errorf("got %v, expect: %v", rows.Keys(), keys)
	}
	for _, tt := range tests {
		t.Run(tt.key, func(t *testing.T) {
			var gotNewValue string
			err = rows.Scan(tt.key, &gotNewValue)
			if (err != nil) != tt.wantErr {
				t.Errorf("rows.Scan() error = %v, wantErr %v", err, tt.wantErr)
				return
			}
			if gotNewValue != tt.wantNewValue {
				t.Errorf("rows.Value() = %v, want %v", gotNewValue, tt.wantNewValue)
			}
			if (rows.Item(tt.key) == nil) != tt.nilItem {
				t.Errorf("rows.Item() = %v, want %v", rows.Item(tt.key) == nil, tt.nilItem)
			}
		})
	}
	err = rows.Close()
	if err != nil {
		t.Errorf("client.Replies.Close() error = %v, wantErr %v", err, nil)
	}
}

func Test_client_Conn(t *testing.T) {
	conn := testClient.Conn(context.Background())
	defer conn.Close()
	if conn == nil {
		t.Errorf("expect get conn, get nil")
	}
}