config_test.go

package config

import (
	"testing"
)

type Config struct {
	Version string `config:"version"`
	// gzingress
	Server struct {
		Ports   []int64 `config:"ports"`
		Cleanup string  `config:"cleanup"`
	} `config:"server"`
	Logger struct {
		Level string `config:"level"`
		Trace bool   `config:"trace"`
	} `config:"logger"`
	Rules []struct {
		Host    string `config:"host"`
		Backend struct {
			ServiceName string `config:"service_name"`
			ServicePort int64  `config:"service_port"`
		} `config:"backend"`
	} `config:"rules"`
	Struc struct {
		Field1 string `config:"field1"`
		Field2 string `config:"field2"`
	} `config:"struct"`

	// gzfly
	Relay  string `config:"relay"`
	Auth   string `config:"auth"`
	Crypto string `config:"crypto"`
	//
	Actions map[string]Action `config:"actions"`
}

type Action struct {
	Target string `config:"target"`
	Bind   string `config:"bind"`
	Socks5 string `config:"socks5"`
}

func TestConfig(t *testing.T) {
	var cfg Config
	if err := Load(&cfg); err != nil {
		t.Fatal(err)
	}

	// j, _ := json.MarshalIndent(cfg, "", "  ")
	// fmt.Println(string(j))
	if cfg.Version != "1.0.0" {
		t.Fatal("version is not 1.0.0")
	}

	if len(cfg.Server.Ports) != 2 {
		t.Fatal("ports is not 2")
	}

	if cfg.Server.Ports[0] != 8080 {
		t.Fatal("ports[0] is not 8080")
	}

	if cfg.Server.Ports[1] != 9090 {
		t.Fatal("ports[1] is not 9090")
	}

	if cfg.Server.Cleanup != "1h" {
		t.Fatal("cleanup is not 1h")
	}

	if cfg.Logger.Level != "warn" {
		t.Fatal("level is not warn")
	}

	if cfg.Logger.Trace != false {
		t.Fatal("trace is not false")
	}

	if len(cfg.Rules) != 2 {
		t.Fatal("rules is not 2")
	}

	if cfg.Rules[0].Host != "backend-a.example.com" {
		t.Fatal("rules[0].host is not backend-a.example.com")
	}

	if cfg.Rules[0].Backend.ServiceName != "backend-a" {
		t.Fatal("rules[0].backend.service_name is not backend-a")
	}

	if cfg.Rules[0].Backend.ServicePort != 80 {
		t.Fatal("rules[0].backend.service_port is not 80")
	}

	if cfg.Rules[1].Host != "backend-b.example.com" {
		t.Fatal("rules[1].host is not backend-b.example.com")
	}

	if cfg.Rules[1].Backend.ServiceName != "backend-b" {
		t.Fatal("rules[1].backend.service_name is not backend-b")
	}

	if cfg.Rules[1].Backend.ServicePort != 8080 {
		t.Fatal("rules[1].backend.service_port is not 8080")
	}

	// gzfly
	if cfg.Actions["action1"].Target != "client_name:pk" {
		t.Fatalf("actions.action1.target is not %s, but got %s", "client_name:pk", cfg.Actions["action1"].Target)
	}

	if cfg.Actions["action1"].Bind != "tcp:0.0.0.0:17890:192.168.1.2:17890" {
		t.Fatalf("actions.action1.bind is not %s, but got %s", "tcp:0.0.0.0:17890:192.168.1.2:17890", cfg.Actions["action1"].Bind)
	}

	if cfg.Actions["action1"].Socks5 != "" {
		t.Fatal("actions.action1.socks5 is not empty string")
	}
}

func TestConfigFromURL(t *testing.T) {
	type Config2 struct {
		Title   string `config:"title"`
		TagLine string `config:"tagline"`
		Author  struct {
			Name     string `config:"name"`
			Nickname string `config:"nickname"`
			Email    string `config:"email"`
			GitHub   string `config:"github"`
		} `config:"author"`
	}
	var cfg Config2
	err := Load(&cfg, &LoadOptions{
		FilePath: "https://raw.githubusercontent.com/whatwewant/whatwewant.github.io/refs/heads/master/_config.yml",
	})
	if err != nil {
		t.Fatal(err)
	}

	if cfg.Title != "Cole" {
		t.Fatal("title is not Cole")
	}

	if cfg.TagLine != "Free as the wind." {
		t.Fatal("tagline is not Free as the wind.")
	}
}