diff --git a/contrib/config/consul/config_test.go b/contrib/config/consul/config_test.go index 8e952736f..a82a20369 100644 --- a/contrib/config/consul/config_test.go +++ b/contrib/config/consul/config_test.go @@ -3,7 +3,9 @@ package consul import ( "reflect" "testing" + "time" + "github.com/go-kratos/kratos/v2/config" "github.com/hashicorp/consul/api" ) @@ -99,3 +101,158 @@ func TestExtToFormat(t *testing.T) { t.Errorf("kvs[0].Format is %s", kvs[0].Format) } } + +func Test_source_Watch(t *testing.T) { + client, err := api.NewClient(&api.Config{ + Address: "127.0.0.1:8500", + }) + if err != nil { + t.Fatal(err) + } + + source, err := New(client, WithPath(testPath)) + if err != nil { + t.Fatal(err) + } + + type fields struct { + source config.Source + } + + type args struct { + key string + value string + } + + tests := []struct { + name string + fields fields + args args + want string + wantErr bool + deferFunc func(t *testing.T) + }{ + { + name: "normal", + fields: fields{source: source}, + args: args{ + key: testKey, + value: "test value", + }, + want: "test value", + wantErr: false, + deferFunc: func(t *testing.T) { + _, err := client.KV().Delete(testKey, nil) + if err != nil { + t.Error(err) + } + }, + }, + } + for _, tt := range tests { + t.Run(tt.name, func(t *testing.T) { + if tt.deferFunc != nil { + defer tt.deferFunc(t) + } + + got, err := tt.fields.source.Watch() + if (err != nil) != tt.wantErr { + t.Errorf("Watch() error = %v, wantErr %v", err, tt.wantErr) + return + } + + time.Sleep(100 * time.Millisecond) + _, err = client.KV().Put(&api.KVPair{Key: tt.args.key, Value: []byte(tt.args.value)}, nil) + if err != nil { + t.Error(err) + } + + next, err := got.Next() + if (err != nil) != tt.wantErr { + t.Errorf("Watch() error = %v, wantErr %v", err, tt.wantErr) + return + } + + if len(next) != 1 { + t.Error("watch is error") + } + + if !reflect.DeepEqual(string(next[0].Value), tt.want) { + t.Errorf("Watch got = %v, want %v", string(next[0].Value), tt.want) + } + }) + } +} + +func Test_source_Load(t *testing.T) { + client, err := api.NewClient(&api.Config{ + Address: "127.0.0.1:8500", + }) + if err != nil { + t.Fatal(err) + } + + source, err := New(client, WithPath(testPath)) + if err != nil { + t.Fatal(err) + } + + type args struct { + key string + value string + } + type fields struct { + source config.Source + } + tests := []struct { + name string + args args + fields fields + want []*config.KeyValue + wantErr bool + deferFunc func(t *testing.T) + }{ + { + name: "normal", + args: args{ + key: testKey, + value: "test value", + }, + fields: fields{ + source: source, + }, + want: []*config.KeyValue{ + { + Key: "key", + Value: []byte("test value"), + }, + }, + deferFunc: func(t *testing.T) { + _, err1 := client.KV().Delete(testKey, nil) + if err1 != nil { + t.Error(err) + } + }, + }, + } + for _, tt := range tests { + t.Run(tt.name, func(t *testing.T) { + if tt.deferFunc != nil { + defer tt.deferFunc(t) + } + _, err = client.KV().Put(&api.KVPair{Key: tt.args.key, Value: []byte(tt.args.value)}, nil) + if err != nil { + t.Error(err) + } + got, err := tt.fields.source.Load() + if (err != nil) != tt.wantErr { + t.Errorf("Load() error = %v, wantErr %v", err, tt.wantErr) + return + } + + if !reflect.DeepEqual(got[0], tt.want[0]) { + t.Errorf("Load() got = %v, want %v", got, tt.want) + } + }) + } +}