ctrld/cmd/cli/intercept_mode_config_test.go

package cli

import (
	"testing"

	"github.com/Control-D-Inc/ctrld"
)

func TestUpdateConfigInterceptMode(t *testing.T) {
	tests := []struct {
		name        string
		current     string
		mode        string
		want        string
		wantUpdated bool
	}{
		{name: "empty flag preserves config", current: "dns", mode: "", want: "dns"},
		{name: "dns is persisted", mode: "dns", want: "dns", wantUpdated: true},
		{name: "hard is persisted", current: "dns", mode: "hard", want: "hard", wantUpdated: true},
		{name: "off clears persisted mode", current: "dns", mode: "off", want: "", wantUpdated: true},
		{name: "off is idempotent", mode: "off", want: ""},
		{name: "invalid flag preserves config", current: "hard", mode: "invalid", want: "hard"},
	}

	for _, tc := range tests {
		t.Run(tc.name, func(t *testing.T) {
			cfg := &ctrld.Config{}
			cfg.Service.InterceptMode = tc.current
			updated := updateConfigInterceptMode(cfg, tc.mode)
			if updated != tc.wantUpdated {
				t.Fatalf("updateConfigInterceptMode() updated = %v, want %v", updated, tc.wantUpdated)
			}
			if cfg.Service.InterceptMode != tc.want {
				t.Fatalf("service.intercept_mode = %q, want %q", cfg.Service.InterceptMode, tc.want)
			}
		})
	}
}

func TestConfiguredInterceptMode(t *testing.T) {
	oldInterceptMode := interceptMode
	t.Cleanup(func() { interceptMode = oldInterceptMode })

	p := &prog{cfg: &ctrld.Config{}}
	p.cfg.Service.InterceptMode = "dns"

	tests := []struct {
		name string
		flag string
		want string
	}{
		{name: "empty flag falls back to config", flag: "", want: "dns"},
		{name: "explicit off is final", flag: "off", want: "off"},
		{name: "explicit hard wins over config", flag: "hard", want: "hard"},
	}

	for _, tc := range tests {
		t.Run(tc.name, func(t *testing.T) {
			interceptMode = tc.flag
			if got := p.configuredInterceptMode(); got != tc.want {
				t.Fatalf("configuredInterceptMode() = %q, want %q", got, tc.want)
			}
		})
	}
}