Vasiliy Tolstov
fa8fb3aed7
* util/router: sync from github * config: add watcher interface Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
105 lines
2.1 KiB
Go
105 lines
2.1 KiB
Go
package config_test
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"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) {
|
|
ctx := context.Background()
|
|
|
|
conf := &Cfg{IntValue: 10}
|
|
|
|
cfg := config.NewConfig(config.Struct(conf))
|
|
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(500*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 && err != config.ErrWatcherStopped {
|
|
t.Fatal(err)
|
|
} else if err == config.ErrWatcherStopped {
|
|
return
|
|
}
|
|
if len(mp) != 1 {
|
|
t.Fatal(fmt.Errorf("default watcher err: %v", mp))
|
|
}
|
|
|
|
v, ok := mp["IntValue"]
|
|
if !ok {
|
|
t.Fatal(fmt.Errorf("default watcher err: %v", v))
|
|
}
|
|
if nv, ok := v.(int); !ok || nv != 99 {
|
|
t.Fatal(fmt.Errorf("default watcher err: %v", v))
|
|
}
|
|
close(done)
|
|
return
|
|
}
|
|
}()
|
|
|
|
<-done
|
|
}
|
|
|
|
func TestDefault(t *testing.T) {
|
|
ctx := context.Background()
|
|
conf := &Cfg{IntValue: 10}
|
|
blfn := func(ctx context.Context, cfg config.Config) error {
|
|
nconf, ok := cfg.Options().Struct.(*Cfg)
|
|
if !ok {
|
|
return fmt.Errorf("failed to get Struct from options: %v", cfg.Options())
|
|
}
|
|
nconf.StringValue = "before_load"
|
|
return nil
|
|
}
|
|
alfn := func(ctx context.Context, cfg config.Config) error {
|
|
nconf, ok := cfg.Options().Struct.(*Cfg)
|
|
if !ok {
|
|
return fmt.Errorf("failed to get Struct from options: %v", cfg.Options())
|
|
}
|
|
nconf.StringValue = "after_load"
|
|
return nil
|
|
}
|
|
|
|
cfg := config.NewConfig(config.Struct(conf), config.BeforeLoad(blfn), config.AfterLoad(alfn))
|
|
if err := cfg.Init(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := cfg.Load(ctx); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if conf.StringValue != "after_load" {
|
|
t.Fatal("AfterLoad option not working")
|
|
}
|
|
_ = conf
|
|
//t.Logf("%#+v\n", conf)
|
|
}
|