88
config/consul/consul_test.go
Normal file
88
config/consul/consul_test.go
Normal file
@@ -0,0 +1,88 @@
|
||||
package consul_test
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"os"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
jsoncodec "github.com/unistack-org/micro-codec-json/v3"
|
||||
consul "github.com/unistack-org/micro-config-consul/v3"
|
||||
"github.com/unistack-org/micro/v3/config"
|
||||
)
|
||||
|
||||
type Cfg struct {
|
||||
StringValue string `default:"string_value"`
|
||||
IgnoreValue string `json:"-"`
|
||||
StructValue struct {
|
||||
StringValue string `default:"string_value"`
|
||||
}
|
||||
IntValue int `default:"99"`
|
||||
}
|
||||
|
||||
func TestWatch(t *testing.T) {
|
||||
if tr := os.Getenv("INTEGRATION_TESTS"); len(tr) > 0 {
|
||||
t.Skip()
|
||||
}
|
||||
|
||||
addrs := ""
|
||||
if addr := os.Getenv("CONSUL_ADDRS"); len(addr) == 0 {
|
||||
addrs = "127.0.0.1:8500"
|
||||
} else {
|
||||
addrs = addr
|
||||
}
|
||||
|
||||
ctx := context.Background()
|
||||
|
||||
conf := &Cfg{IntValue: 10}
|
||||
|
||||
cfg := consul.NewConfig(config.Struct(conf), consul.Address(addrs), consul.Path("test/consul"), config.Codec(jsoncodec.NewCodec()))
|
||||
if err := cfg.Init(); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if err := cfg.Load(ctx); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
w, err := cfg.Watch(ctx, config.WatchInterval(700*time.Millisecond, 2000*time.Millisecond))
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
defer func() {
|
||||
_ = w.Stop()
|
||||
}()
|
||||
|
||||
done := make(chan struct{})
|
||||
|
||||
go func() {
|
||||
for {
|
||||
mp, err := w.Next()
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if len(mp) != 1 {
|
||||
close(done)
|
||||
t.Fatal(fmt.Errorf("consul watcher err: %v", mp))
|
||||
return
|
||||
}
|
||||
v, ok := mp["IntValue"]
|
||||
if !ok {
|
||||
close(done)
|
||||
t.Fatal(fmt.Errorf("consul watcher err: %v", v))
|
||||
return
|
||||
}
|
||||
if nv, ok := v.(int); !ok || nv != 5 {
|
||||
close(done)
|
||||
t.Fatal(fmt.Errorf("consul watcher err: %v", v))
|
||||
return
|
||||
}
|
||||
close(done)
|
||||
return
|
||||
}
|
||||
}()
|
||||
|
||||
<-done
|
||||
|
||||
}
|
Reference in New Issue
Block a user