wip6

2026-07-13 12:28:59 +08:00
parent 6c3de4b2ef
commit 1fca796f2c
56 changed files with 9709 additions and 406 deletions
@@ -0,0 +1,296 @@
+package cmd
+
+import (
+	"testing"
+
+	"axenov/iptv-checker/app"
+	"axenov/iptv-checker/app/config"
+
+	"github.com/spf13/cobra"
+)
+
+func newCmdWithFlags() *cobra.Command {
+	cmd := &cobra.Command{Use: "test"}
+	// persistent flags normally set on rootCmd
+	cmd.PersistentFlags().BoolVar(&app.Args.Debug, "debug", false, "")
+	cmd.PersistentFlags().StringVar(&app.Args.LogLevel, "log-level", "", "")
+	addCommonCheckFlags(cmd, 0)
+	return cmd
+}
+
+func resetArgs() {
+	app.Args = app.Arguments{}
+}
+
+func resetConfig() {
+	app.Config = &config.Config{
+		App: config.AppConfig{
+			Debug:     false,
+			LogLevel:  "info",
+			Playlists: "./playlists.ini",
+			Tags:      "./channels.json",
+		},
+		Check: config.CheckConfig{
+			Playlists: config.CheckPlaylistsConfig{
+				Timeout:     10000,
+				AllCooldown: config.IntRange{Min: 0, Max: 0},
+				OneCooldown: config.IntRange{Min: 0, Max: 0},
+				MaxRoutines: 5,
+				PerRoutine:  1,
+				UserAgent:   config.UserAgents{"default"},
+			},
+			Channels: config.CheckChannelsConfig{
+				Timeout:     10000,
+				ByteRange:   512,
+				Cooldown:    config.IntRange{Min: 0, Max: 0},
+				MaxRoutines: 50,
+				PerRoutine:  10,
+				UserAgent:   config.UserAgents{"default"},
+			},
+		},
+		Cache: config.CacheConfig{
+			Enabled: false,
+			Host:    "localhost",
+			Port:    6379,
+			Ttl:     1800,
+		},
+		Server: config.ServerConfig{Port: 8800},
+	}
+}
+
+func TestApplyAppOverrides_NothingChanged(t *testing.T) {
+	resetArgs()
+	resetConfig()
+	cmd := newCmdWithFlags()
+	cmd.ParseFlags([]string{})
+	applyAppOverrides(cmd)
+	if app.Config.App.Debug {
+		t.Error("debug should remain false")
+	}
+}
+
+func TestApplyAppOverrides_Debug(t *testing.T) {
+	resetArgs()
+	resetConfig()
+	cmd := newCmdWithFlags()
+	cmd.ParseFlags([]string{"--debug"})
+	applyAppOverrides(cmd)
+	if !app.Config.App.Debug {
+		t.Error("expected debug=true")
+	}
+}
+
+func TestApplyAppOverrides_LogLevel(t *testing.T) {
+	resetArgs()
+	resetConfig()
+	cmd := newCmdWithFlags()
+	cmd.ParseFlags([]string{"--log-level", "error"})
+	applyAppOverrides(cmd)
+	if app.Config.App.LogLevel != "error" {
+		t.Errorf("expected 'error', got '%s'", app.Config.App.LogLevel)
+	}
+}
+
+func TestApplyAppOverrides_Ini(t *testing.T) {
+	resetArgs()
+	resetConfig()
+	cmd := newCmdWithFlags()
+	cmd.ParseFlags([]string{"-i", "/custom/path.ini"})
+	applyAppOverrides(cmd)
+	if app.Config.App.Playlists != "/custom/path.ini" {
+		t.Errorf("expected '/custom/path.ini', got '%s'", app.Config.App.Playlists)
+	}
+}
+
+func TestApplyAppOverrides_Tags(t *testing.T) {
+	resetArgs()
+	resetConfig()
+	cmd := newCmdWithFlags()
+	cmd.ParseFlags([]string{"-t", "/custom/tags.json"})
+	applyAppOverrides(cmd)
+	if app.Config.App.Tags != "/custom/tags.json" {
+		t.Errorf("expected '/custom/tags.json', got '%s'", app.Config.App.Tags)
+	}
+}
+
+func TestApplyCacheOverrides_NothingChanged(t *testing.T) {
+	resetArgs()
+	resetConfig()
+	cmd := newCmdWithFlags()
+	cmd.ParseFlags([]string{})
+	applyCacheOverrides(cmd)
+	if app.Config.Cache.Enabled {
+		t.Error("cache should remain disabled")
+	}
+}
+
+func TestApplyCacheOverrides_Enable(t *testing.T) {
+	resetArgs()
+	resetConfig()
+	cmd := newCmdWithFlags()
+	cmd.ParseFlags([]string{"--cache-enabled"})
+	applyCacheOverrides(cmd)
+	if !app.Config.Cache.Enabled {
+		t.Error("expected cache enabled")
+	}
+}
+
+func TestApplyCacheOverrides_DisableWhenAlreadyEnabled(t *testing.T) {
+	resetArgs()
+	resetConfig()
+	app.Config.Cache.Enabled = true
+	cmd := newCmdWithFlags()
+	cmd.ParseFlags([]string{"--cache-enabled=false"})
+	applyCacheOverrides(cmd)
+	if app.Config.Cache.Enabled {
+		t.Error("expected cache disabled")
+	}
+}
+
+func TestApplyCacheOverrides_Host(t *testing.T) {
+	resetArgs()
+	resetConfig()
+	cmd := newCmdWithFlags()
+	cmd.ParseFlags([]string{"--cache-host", "db.local"})
+	applyCacheOverrides(cmd)
+	if app.Config.Cache.Host != "db.local" {
+		t.Errorf("expected 'db.local', got '%s'", app.Config.Cache.Host)
+	}
+}
+
+func TestApplyCacheOverrides_Port(t *testing.T) {
+	resetArgs()
+	resetConfig()
+	cmd := newCmdWithFlags()
+	cmd.ParseFlags([]string{"--cache-port", "6380"})
+	applyCacheOverrides(cmd)
+	if app.Config.Cache.Port != 6380 {
+		t.Errorf("expected 6380, got %d", app.Config.Cache.Port)
+	}
+}
+
+func TestApplyCacheOverrides_Ttl(t *testing.T) {
+	resetArgs()
+	resetConfig()
+	cmd := newCmdWithFlags()
+	cmd.ParseFlags([]string{"--cache-ttl", "3600"})
+	applyCacheOverrides(cmd)
+	if app.Config.Cache.Ttl != 3600 {
+		t.Errorf("expected 3600, got %d", app.Config.Cache.Ttl)
+	}
+}
+
+func TestApplyCheckOverrides_NothingChanged(t *testing.T) {
+	resetArgs()
+	resetConfig()
+	cmd := newCmdWithFlags()
+	cmd.ParseFlags([]string{})
+	applyCheckOverrides(cmd)
+	if app.Config.Check.Playlists.Timeout != 10000 {
+		t.Errorf("expected 10000, got %d", app.Config.Check.Playlists.Timeout)
+	}
+}
+
+func TestApplyCheckOverrides_PlaylistsTimeout(t *testing.T) {
+	resetArgs()
+	resetConfig()
+	cmd := newCmdWithFlags()
+	cmd.ParseFlags([]string{"--playlists-timeout", "5"})
+	applyCheckOverrides(cmd)
+	if app.Config.Check.Playlists.Timeout != 5000 {
+		t.Errorf("expected 5000 ms from 5 s, got %d", app.Config.Check.Playlists.Timeout)
+	}
+}
+
+func TestApplyCheckOverrides_PlaylistsAllCooldown(t *testing.T) {
+	resetArgs()
+	resetConfig()
+	cmd := newCmdWithFlags()
+	cmd.ParseFlags([]string{"--playlists-all-cooldown", "3"})
+	applyCheckOverrides(cmd)
+	if app.Config.Check.Playlists.AllCooldown.Min != 3000 || app.Config.Check.Playlists.AllCooldown.Max != 3000 {
+		t.Errorf("expected [3000,3000] ms from 3 s, got [%d,%d]",
+			app.Config.Check.Playlists.AllCooldown.Min, app.Config.Check.Playlists.AllCooldown.Max)
+	}
+}
+
+func TestApplyCheckOverrides_PlaylistsMaxRoutines(t *testing.T) {
+	resetArgs()
+	resetConfig()
+	cmd := newCmdWithFlags()
+	cmd.ParseFlags([]string{"--playlists-max-routines", "10"})
+	applyCheckOverrides(cmd)
+	if app.Config.Check.Playlists.MaxRoutines != 10 {
+		t.Errorf("expected 10, got %d", app.Config.Check.Playlists.MaxRoutines)
+	}
+}
+
+func TestApplyCheckOverrides_PlaylistsUserAgent(t *testing.T) {
+	resetArgs()
+	resetConfig()
+	cmd := newCmdWithFlags()
+	cmd.ParseFlags([]string{"--playlists-user-agent", "UA1,UA2"})
+	applyCheckOverrides(cmd)
+	if len(app.Config.Check.Playlists.UserAgent) != 2 {
+		t.Fatalf("expected 2 user-agents, got %d", len(app.Config.Check.Playlists.UserAgent))
+	}
+	if app.Config.Check.Playlists.UserAgent[0] != "UA1" || app.Config.Check.Playlists.UserAgent[1] != "UA2" {
+		t.Errorf("unexpected user-agents: %v", app.Config.Check.Playlists.UserAgent)
+	}
+}
+
+func TestApplyCheckOverrides_ChannelsTimeout(t *testing.T) {
+	resetArgs()
+	resetConfig()
+	cmd := newCmdWithFlags()
+	cmd.ParseFlags([]string{"--channels-timeout", "8"})
+	applyCheckOverrides(cmd)
+	if app.Config.Check.Channels.Timeout != 8000 {
+		t.Errorf("expected 8000 ms from 8 s, got %d", app.Config.Check.Channels.Timeout)
+	}
+}
+
+func TestApplyCheckOverrides_ChannelsByteRange(t *testing.T) {
+	resetArgs()
+	resetConfig()
+	cmd := newCmdWithFlags()
+	cmd.ParseFlags([]string{"--channels-byte-range", "1024"})
+	applyCheckOverrides(cmd)
+	if app.Config.Check.Channels.ByteRange != 1024 {
+		t.Errorf("expected 1024, got %d", app.Config.Check.Channels.ByteRange)
+	}
+}
+
+func TestApplyCheckOverrides_ChannelsCooldown(t *testing.T) {
+	resetArgs()
+	resetConfig()
+	cmd := newCmdWithFlags()
+	cmd.ParseFlags([]string{"--channels-cooldown", "1"})
+	applyCheckOverrides(cmd)
+	if app.Config.Check.Channels.Cooldown.Min != 1000 || app.Config.Check.Channels.Cooldown.Max != 1000 {
+		t.Errorf("expected [1000,1000] ms from 1 s, got [%d,%d]",
+			app.Config.Check.Channels.Cooldown.Min, app.Config.Check.Channels.Cooldown.Max)
+	}
+}
+
+func TestApplyCheckOverrides_ChannelsMaxRoutines(t *testing.T) {
+	resetArgs()
+	resetConfig()
+	cmd := newCmdWithFlags()
+	cmd.ParseFlags([]string{"--channels-max-routines", "100"})
+	applyCheckOverrides(cmd)
+	if app.Config.Check.Channels.MaxRoutines != 100 {
+		t.Errorf("expected 100, got %d", app.Config.Check.Channels.MaxRoutines)
+	}
+}
+
+func TestApplyCheckOverrides_ChannelsUserAgent(t *testing.T) {
+	resetArgs()
+	resetConfig()
+	cmd := newCmdWithFlags()
+	cmd.ParseFlags([]string{"--channels-user-agent", "VLC"})
+	applyCheckOverrides(cmd)
+	if len(app.Config.Check.Channels.UserAgent) != 1 || app.Config.Check.Channels.UserAgent[0] != "VLC" {
+		t.Errorf("expected [VLC], got %v", app.Config.Check.Channels.UserAgent)
+	}
+}