main_test.go

package main

import (
	"os"
	"path/filepath"
	"reflect"
	"testing"
)

func TestNormalizeSubdomain(t *testing.T) {
	t.Parallel()

	tests := []struct {
		name    string
		input   string
		domain  string
		want    string
		wantErr bool
	}{
		{name: "relative", input: "home", domain: "example.com", want: "home"},
		{name: "fqdn", input: "Home.Example.com.", domain: "example.com", want: "home"},
		{name: "root marker", input: "@", domain: "example.com", want: "@"},
		{name: "root fqdn", input: "example.com", domain: "example.com", want: "@"},
		{name: "nested", input: "vpn.eu", domain: "example.com", want: "vpn.eu"},
		{name: "invalid", input: "bad / name", domain: "example.com", wantErr: true},
	}

	for _, tt := range tests {
		tt := tt
		t.Run(tt.name, func(t *testing.T) {
			t.Parallel()

			got, err := normalizeSubdomain(tt.input, tt.domain)
			if tt.wantErr {
				if err == nil {
					t.Fatalf("expected error, got none")
				}
				return
			}
			if err != nil {
				t.Fatalf("normalizeSubdomain returned error: %v", err)
			}
			if got != tt.want {
				t.Fatalf("normalizeSubdomain = %q, want %q", got, tt.want)
			}
		})
	}
}

func TestConfigStoreBootstrapAndManualReload(t *testing.T) {
	t.Parallel()

	path := filepath.Join(t.TempDir(), ".runtime.env")
	store := newConfigStore(path, "example.com", managedCfg{
		Mode:       modeSelected,
		Subdomains: []string{"NAS.Example.com", "@", "home", "home"},
	})

	cfg, err := store.Load()
	if err != nil {
		t.Fatalf("Load() error = %v", err)
	}
	if cfg.Mode != modeSelected {
		t.Fatalf("Load() mode = %q, want %q", cfg.Mode, modeSelected)
	}
	if want := []string{"@", "home", "nas"}; !reflect.DeepEqual(cfg.Subdomains, want) {
		t.Fatalf("Load() subdomains = %#v, want %#v", cfg.Subdomains, want)
	}

	manual := "MODE=selected\nSUBDOMAINS=WWW.example.com,@,WWW.example.com\n"
	if err := os.WriteFile(path, []byte(manual), 0o600); err != nil {
		t.Fatalf("WriteFile() error = %v", err)
	}

	reloaded, err := store.Load()
	if err != nil {
		t.Fatalf("Load() after manual edit error = %v", err)
	}
	if want := []string{"@", "www"}; !reflect.DeepEqual(reloaded.Subdomains, want) {
		t.Fatalf("Load() after manual edit subdomains = %#v, want %#v", reloaded.Subdomains, want)
	}
}

func TestSelectedModeStaysSelectedWithEmptyList(t *testing.T) {
	t.Parallel()

	path := filepath.Join(t.TempDir(), ".runtime.env")
	store := newConfigStore(path, "example.com", managedCfg{
		Mode:       modeSelected,
		Subdomains: []string{"home"},
	})

	if _, err := store.Load(); err != nil {
		t.Fatalf("Load() error = %v", err)
	}

	cfg, err := store.Update(func(cfg *managedCfg) error {
		removeSubdomain(&cfg.Subdomains, "home")
		return nil
	})
	if err != nil {
		t.Fatalf("Update() error = %v", err)
	}
	if cfg.Mode != modeSelected {
		t.Fatalf("Update() mode = %q, want %q", cfg.Mode, modeSelected)
	}
	if len(cfg.Subdomains) != 0 {
		t.Fatalf("Update() subdomains = %#v, want empty list", cfg.Subdomains)
	}
}