micro-config-flag/flag_test.go

package flag

import (
	"context"
	"os"
	"testing"
	"time"

	"go.unistack.org/micro/v3/config"
)

func TestLoad(t *testing.T) {
	time.Local = time.UTC
	os.Args = append(os.Args, "-broker", "5566:33")
	os.Args = append(os.Args, "-verbose")
	os.Args = append(os.Args, "-wait", "5s")
	os.Args = append(os.Args, "-addr", "33,44")
	os.Args = append(os.Args, "-time", time.RFC822)
	os.Args = append(os.Args, "-metadata", "key=20")
	os.Args = append(os.Args, "-components", "all=info,api=debug")
	type NestedConfig struct {
		Value string `flag:"name=nested_value"`
	}
	type Config struct {
		Broker         string            `flag:"name=broker,desc='description with, comma',default='127.0.0.1:9092'"`
		Verbose        bool              `flag:"name=verbose,desc='verbose output',default='false'"`
		Addr           []string          `flag:"name=addr,desc='addrs',default='127.0.0.1:9092'"`
		Wait           time.Duration     `flag:"name=wait,desc='wait time',default='2s'"`
		Time           time.Time         `flag:"name=time,desc='some time',default='02 Jan 06 15:04 MST'"`
		Metadata       map[string]int    `flag:"name=metadata,desc='some meta',default=''"`
		WithoutDefault string            `flag:"name=without_default,desc='with'"`
		WithoutDesc    string            `flag:"name=without_desc,default='without_default'"`
		WithoutAll     string            `flag:"name=without_all"`
		Components     map[string]string `flag:"name=components,desc='components logging'"`
		Nested         *NestedConfig
	}

	ctx := context.Background()
	cfg := &Config{Nested: &NestedConfig{}}

	c := NewConfig(config.Struct(cfg), TimeFormat(time.RFC822))
	if err := c.Init(); err != nil {
		t.Fatalf("init failed: %v", err)
	}

	if err := c.Load(ctx); err != nil {
		t.Fatalf("load failed: %v", err)
	}

	if cfg.Broker != "5566:33" {
		t.Fatalf("failed to parse flags broker value invalid: %#+v", cfg)
	}
	if tf := cfg.Time.Format(time.RFC822); tf != "02 Jan 06 15:04 MST" {
		t.Fatalf("parse time error: %s != %s", tf, "02 Jan 06 15:04 MST")
	}

	if len(cfg.Components) != 2 {
		t.Fatalf("cant parse map components %#+v", cfg)
	}
}