micro-store-redis/redis_test.go

package redis

import (
	"bytes"
	"context"
	"os"
	"sync/atomic"
	"testing"
	"time"

	goredis "github.com/redis/go-redis/v9"
	"go.unistack.org/micro/v3/store"
	"go.unistack.org/micro/v3/tracer"
)

func TestLazyConnect(t *testing.T) {
	ctx := context.Background()
	var err error

	r := NewStore()

	if err = r.Init(); err != nil {
		t.Fatal(err)
	}
	if err = r.Connect(ctx); err != nil {
		t.Logf("connect failed %v", err)
	}

	for {
		if err = r.Write(ctx, "mykey", "myval"); err != nil {
			t.Logf("failed to write %v", err)
		}
	}
}

func TestKeepTTL(t *testing.T) {
	ctx := context.Background()

	if tr := os.Getenv("INTEGRATION_TESTS"); len(tr) > 0 {
		t.Skip()
	}
	r := NewStore(store.Addrs(os.Getenv("STORE_NODES")))

	if err := r.Init(store.LazyConnect(true)); err != nil {
		t.Fatal(err)
	}
	if err := r.Connect(ctx); err != nil {
		t.Fatal(err)
	}

	key := "key"
	err := r.Write(ctx, key, "val1", store.WriteTTL(15*time.Second))
	if err != nil {
		t.Fatalf("Write error: %v", err)
	}
	time.Sleep(3 * time.Second)
	err = r.Write(ctx, key, "val2", store.WriteTTL(-1))
	if err != nil {
		t.Fatalf("Write error: %v", err)
	}
}

func Test_rkv_configure(t *testing.T) {
	type fields struct {
		options store.Options
		Client  goredis.UniversalClient
	}
	type wantValues struct {
		username string
		password string
		address  string
	}

	tests := []struct {
		name    string
		fields  fields
		wantErr bool
		want    wantValues
	}{
		{
			name: "No Url", fields: fields{options: store.Options{}, Client: nil},
			wantErr: false, want: wantValues{
				username: "",
				password: "",
				address:  "127.0.0.1:6379",
			},
		},
		{
			name: "legacy Url", fields: fields{options: store.Options{Tracer: tracer.DefaultTracer, Addrs: []string{"127.0.0.1:6379"}}, Client: nil},
			wantErr: false, want: wantValues{
				username: "",
				password: "",
				address:  "127.0.0.1:6379",
			},
		},
		{
			name: "New Url", fields: fields{options: store.Options{Tracer: tracer.DefaultTracer, Addrs: []string{"redis://127.0.0.1:6379"}}, Client: nil},
			wantErr: false, want: wantValues{
				username: "",
				password: "",
				address:  "127.0.0.1:6379",
			},
		},
		{
			name: "Url with Pwd", fields: fields{options: store.Options{Tracer: tracer.DefaultTracer, Addrs: []string{"redis://:password@redis:6379"}}, Client: nil},
			wantErr: false, want: wantValues{
				username: "",
				password: "password",
				address:  "redis:6379",
			},
		},
		{
			name: "Url with username and Pwd", fields: fields{options: store.Options{Tracer: tracer.DefaultTracer, Addrs: []string{"redis://username:password@redis:6379"}}, Client: nil},
			wantErr: false, want: wantValues{
				username: "username",
				password: "password",
				address:  "redis:6379",
			},
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			b := atomic.Bool{}
			rc := &Store{
				opts:      tt.fields.options,
				cli:       tt.fields.Client,
				connected: &b,
			}
			err := rc.configure()
			if (err != nil) != tt.wantErr {
				t.Errorf("configure() error = %v, wantErr %v", err, tt.wantErr)
				return
			}
		})
	}
}

func Test_Store(t *testing.T) {
	ctx := context.Background()

	if tr := os.Getenv("INTEGRATION_TESTS"); len(tr) > 0 {
		t.Skip()
	}
	r := NewStore(store.Addrs(os.Getenv("STORE_NODES")))

	if err := r.Init(); err != nil {
		t.Fatal(err)
	}
	if err := r.Connect(ctx); err != nil {
		t.Fatal(err)
	}

	key := "myTest"
	tval := []byte("myValue")
	var val []byte
	err := r.Write(ctx, key, tval, store.WriteTTL(2*time.Minute))
	if err != nil {
		t.Fatalf("Write error: %v", err)
	}
	err = r.Read(ctx, key, &val)
	if err != nil {
		t.Fatalf("Read error: %v\n", err)
	} else if !bytes.Equal(val, tval) {
		t.Fatalf("read err: data not eq")
	}
	keys, err := r.List(ctx)
	if err != nil {
		t.Fatalf("List error: %v\n", err)
	}
	_ = keys
	err = r.Delete(ctx, key)
	if err != nil {
		t.Fatalf("Delete error: %v\n", err)
	}
	// t.Logf("%v", keys)
}

func Test_MRead(t *testing.T) {
	ctx := context.Background()
	var err error
	if tr := os.Getenv("INTEGRATION_TESTS"); len(tr) > 0 {
		t.Skip()
	}
	r := NewStore(store.Addrs(os.Getenv("STORE_NODES")))

	if err = r.Init(); err != nil {
		t.Fatal(err)
	}
	if err = r.Connect(ctx); err != nil {
		t.Fatal(err)
	}

	key1 := "myTest1"
	key2 := "myTest2"
	tval1 := []byte("myValue1")
	tval2 := []byte("myValue2")
	var vals [][]byte
	err = r.Write(ctx, key1, tval1, store.WriteTTL(2*time.Minute))
	if err != nil {
		t.Fatalf("Write error: %v", err)
	}
	err = r.Write(ctx, key2, tval2, store.WriteTTL(2*time.Minute))
	if err != nil {
		t.Fatalf("Write error: %v", err)
	}
	err = r.MRead(ctx, []string{key1, key2}, &vals)
	if err != nil {
		t.Fatalf("Read error: %v\n", err)
	}
	// t.Logf("%s", vals)
	_ = vals
	keys, err := r.List(ctx)
	if err != nil {
		t.Fatalf("List error: %v\n", err)
	}
	_ = keys
	// t.Logf("%v", keys)
	err = r.Delete(ctx, key1)
	if err != nil {
		t.Fatalf("Delete error: %v\n", err)
	}
	err = r.Delete(ctx, key2)
	if err != nil {
		t.Fatalf("Delete error: %v\n", err)
	}
}

/*
func Test_MReadCodec(t *testing.T) {
	type mytype struct {
		Key string `json:"name"`
		Val string `json:"val"`
	}
	ctx := context.Background()
	var err error
	if tr := os.Getenv("INTEGRATION_TESTS"); len(tr) > 0 {
		t.Skip()
	}
	r := NewStore(store.Nodes(os.Getenv("STORE_NODES")), store.Codec(jsoncodec.NewCodec()))

	if err = r.Init(); err != nil {
		t.Fatal(err)
	}
	if err = r.Connect(ctx); err != nil {
		t.Fatal(err)
	}

	key1 := "myTest1"
	key2 := "myTest2"
	key3 := "myTest3"
	tval1 := &mytype{Key: "key1", Val: "val1"}
	tval2 := &mytype{Key: "key2", Val: "val2"}
	var vals []*mytype
	err = r.Write(ctx, key1, tval1, store.WriteTTL(2*time.Minute))
	if err != nil {
		t.Fatalf("Write error: %v", err)
	}
	err = r.Write(ctx, key2, tval2, store.WriteTTL(2*time.Minute))
	if err != nil {
		t.Fatalf("Write error: %v", err)
	}
	err = r.MRead(ctx, []string{key1, key3, key2}, &vals)
	if err != nil {
		t.Fatalf("Read error: %v\n", err)
	}
	if vals[0].Key != "key1" || vals[1] != nil || vals[2].Key != "key2" {
		t.Fatalf("read err: struct not filled")
	}
	keys, err := r.List(ctx)
	if err != nil {
		t.Fatalf("List error: %v\n", err)
	}
	_ = keys
	err = r.Delete(ctx, key1)
	if err != nil {
		t.Fatalf("Delete error: %v\n", err)
	}
	err = r.Delete(ctx, key2)
	if err != nil {
		t.Fatalf("Delete error: %v\n", err)
	}
}
*/