You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
110 lines
2.2 KiB
110 lines
2.2 KiB
package etcd
|
|
|
|
import (
|
|
"context"
|
|
"reflect"
|
|
"testing"
|
|
"time"
|
|
|
|
clientv3 "go.etcd.io/etcd/client/v3"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
const testKey = "/kratos/test/config"
|
|
|
|
func TestConfig(t *testing.T) {
|
|
client, err := clientv3.New(clientv3.Config{
|
|
Endpoints: []string{"127.0.0.1:2379"},
|
|
DialTimeout: time.Second, DialOptions: []grpc.DialOption{grpc.WithBlock()},
|
|
})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer func() {
|
|
_ = client.Close()
|
|
}()
|
|
if _, err = client.Put(context.Background(), testKey, "test config"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
source, err := New(client, WithPath(testKey))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
kvs, err := source.Load()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if len(kvs) != 1 || kvs[0].Key != testKey || string(kvs[0].Value) != "test config" {
|
|
t.Fatal("config error")
|
|
}
|
|
|
|
w, err := source.Watch()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer func() {
|
|
_ = w.Stop()
|
|
}()
|
|
|
|
if _, err = client.Put(context.Background(), testKey, "new config"); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if kvs, err = w.Next(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if len(kvs) != 1 || kvs[0].Key != testKey || string(kvs[0].Value) != "new config" {
|
|
t.Fatal("config error")
|
|
}
|
|
|
|
if _, err := client.Delete(context.Background(), testKey); err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestExtToFormat(t *testing.T) {
|
|
client, err := clientv3.New(clientv3.Config{
|
|
Endpoints: []string{"127.0.0.1:2379"},
|
|
DialTimeout: time.Second, DialOptions: []grpc.DialOption{grpc.WithBlock()},
|
|
})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer func() {
|
|
_ = client.Close()
|
|
}()
|
|
|
|
tp := "/kratos/test/ext"
|
|
tn := "a.bird.json"
|
|
tk := tp + "/" + tn
|
|
tc := `{"a":1}`
|
|
if _, err = client.Put(context.Background(), tk, tc); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
source, err := New(client, WithPath(tp), WithPrefix(true))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
kvs, err := source.Load()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !reflect.DeepEqual(len(kvs), 1) {
|
|
t.Errorf("len(kvs) = %d", len(kvs))
|
|
}
|
|
if !reflect.DeepEqual(tk, kvs[0].Key) {
|
|
t.Errorf("kvs[0].Key is %s", kvs[0].Key)
|
|
}
|
|
if !reflect.DeepEqual(tc, string(kvs[0].Value)) {
|
|
t.Errorf("kvs[0].Value is %s", kvs[0].Value)
|
|
}
|
|
if !reflect.DeepEqual("json", kvs[0].Format) {
|
|
t.Errorf("kvs[0].Format is %s", kvs[0].Format)
|
|
}
|
|
}
|
|
|