mfd-monitor-agent/internal/config/config_test.go

package config

import (
	"encoding/json"
	"os"
	"path/filepath"
	"testing"
	"time"
)

// TestDefaultConfigUsesAllInterfacesAndSaneIntervals 验证默认配置监听所有网卡且采集间隔合理。
func TestDefaultConfigUsesAllInterfacesAndSaneIntervals(t *testing.T) {
	cfg := Default()

	if cfg.HTTP.ListenAddress != "0.0.0.0:8080" {
		t.Fatalf("listen address = %q", cfg.HTTP.ListenAddress)
	}
	if cfg.HTTP.Token != "" {
		t.Fatalf("default config should not include token")
	}
	if cfg.Collect.Interval <= 0 || cfg.Collect.Timeout <= 0 {
		t.Fatalf("collect intervals must be positive: %+v", cfg.Collect)
	}
	if cfg.Report.Enabled {
		t.Fatal("reporting should be disabled by default")
	}
}

// TestResolvePathUsesDefaultFileName 验证空路径解析为 config.json。
func TestResolvePathUsesDefaultFileName(t *testing.T) {
	if got := ResolvePath(""); got != DefaultConfigFileName {
		t.Fatalf("path = %q", got)
	}
	if got := ResolvePath("/etc/agent.json"); got != "/etc/agent.json" {
		t.Fatalf("path = %q", got)
	}
}

// TestLoadOrCreateWritesConfigWithToken 验证缺失配置文件时会生成含 token 的 JSON。
func TestLoadOrCreateWritesConfigWithToken(t *testing.T) {
	dir := t.TempDir()
	path := filepath.Join(dir, "config.json")

	cfg, err := LoadOrCreate(path)
	if err != nil {
		t.Fatal(err)
	}
	if cfg.HTTP.Token == "" {
		t.Fatal("expected generated token")
	}
	if cfg.HTTP.ListenAddress != "0.0.0.0:8080" {
		t.Fatalf("listen address = %q", cfg.HTTP.ListenAddress)
	}

	info, err := os.Stat(path)
	if err != nil {
		t.Fatal(err)
	}
	if info.Mode().Perm() != 0o600 {
		t.Fatalf("file mode = %o", info.Mode().Perm())
	}

	data, err := os.ReadFile(path)
	if err != nil {
		t.Fatal(err)
	}
	var onDisk Config
	if err := json.Unmarshal(data, &onDisk); err != nil {
		t.Fatal(err)
	}
	if onDisk.HTTP.Token != cfg.HTTP.Token {
		t.Fatalf("token mismatch: disk=%q loaded=%q", onDisk.HTTP.Token, cfg.HTTP.Token)
	}
}

// TestLoadRejectsEmptyToken 验证未配置 token 时无法启动。
func TestLoadRejectsEmptyToken(t *testing.T) {
	file, err := os.CreateTemp(t.TempDir(), "config-*.json")
	if err != nil {
		t.Fatal(err)
	}
	_, _ = file.WriteString(`{"http":{"listen_address":"0.0.0.0:8080","token":""}}`)
	_ = file.Close()

	_, err = Load(file.Name())
	if err == nil {
		t.Fatal("expected error for empty token")
	}
}

// TestLoadAppliesEnvironmentOverrides 验证环境变量可覆盖默认配置。
func TestLoadAppliesEnvironmentOverrides(t *testing.T) {
	file, err := os.CreateTemp(t.TempDir(), "config-*.json")
	if err != nil {
		t.Fatal(err)
	}
	_, _ = file.WriteString(`{"http":{"listen_address":"0.0.0.0:8080","token":"base-token"}}`)
	_ = file.Close()

	t.Setenv("MFD_AGENT_LISTEN", "127.0.0.1:9090")
	t.Setenv("MFD_AGENT_TOKEN", "env-token")
	t.Setenv("MFD_AGENT_REPORT_ENABLED", "true")
	t.Setenv("MFD_AGENT_REPORT_ENDPOINT", "http://center/status")
	t.Setenv("MFD_AGENT_REPORT_INTERVAL", "2s")
	t.Setenv("MFD_AGENT_HOST_ID", "host-a")

	cfg, err := Load(file.Name())
	if err != nil {
		t.Fatal(err)
	}

	if cfg.HTTP.ListenAddress != "127.0.0.1:9090" {
		t.Fatalf("listen address = %q", cfg.HTTP.ListenAddress)
	}
	if cfg.HTTP.Token != "env-token" {
		t.Fatalf("token = %q", cfg.HTTP.Token)
	}
	if !cfg.Report.Enabled || cfg.Report.Endpoint != "http://center/status" {
		t.Fatalf("report config not applied: %+v", cfg.Report)
	}
	if cfg.Report.Interval.Duration() != 2*time.Second {
		t.Fatalf("report interval = %s", cfg.Report.Interval.Duration())
	}
	if cfg.HostID != "host-a" {
		t.Fatalf("host id = %q", cfg.HostID)
	}
}

// TestLoadJSONConfigFile 验证可从 JSON 配置文件解析 Config。
func TestLoadJSONConfigFile(t *testing.T) {
	file, err := os.CreateTemp(t.TempDir(), "config-*.json")
	if err != nil {
		t.Fatal(err)
	}
	_, _ = file.WriteString(`{"http":{"listen_address":"127.0.0.1:7070","token":"secret"},"collect":{"interval":"3s","timeout":"1s"}}`)
	_ = file.Close()

	cfg, err := Load(file.Name())
	if err != nil {
		t.Fatal(err)
	}

	if cfg.HTTP.ListenAddress != "127.0.0.1:7070" {
		t.Fatalf("listen address = %q", cfg.HTTP.ListenAddress)
	}
	if cfg.HTTP.Token != "secret" {
		t.Fatalf("token = %q", cfg.HTTP.Token)
	}
	if cfg.Collect.Interval.Duration() != 3*time.Second || cfg.Collect.Timeout.Duration() != time.Second {
		t.Fatalf("collect config = %+v", cfg.Collect)
	}
}