Smail/backend/services/setup/finish_test.go

package setup

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

	"git.pcux.cn/xuans/Smail/config"
	"git.pcux.cn/xuans/Smail/db"
	"git.pcux.cn/xuans/Smail/models"
	"xorm.io/xorm"
)

func TestFinishPersistsServicePorts(t *testing.T) {
	prepareFinishConfigTest(t)
	t.Setenv("SMAIL_HTTP_PORT", "9001")
	t.Setenv("SMAIL_HTTPS_PORT", "9002")
	config.SetRuntimeSetupPort(18080)

	if err := config.WriteConfig(&config.Config{
		SetupState:   config.SetupStateBootstrap,
		HttpsEnabled: 1,
		Domain:       "example.com",
		WebDomain:    "mail.example.com",
		DbType:       config.DBTypeSQLite,
		DbDSN:        filepath.Join(config.ROOT_PATH, "data", "smail.db"),
	}); err != nil {
		t.Fatal(err)
	}
	if err := Finish(); err != nil {
		t.Fatal(err)
	}

	loaded, err := config.ReadConfig()
	if err != nil {
		t.Fatal(err)
	}
	if loaded.HttpPort != 9001 || loaded.HttpsPort != 9002 {
		t.Fatalf("persisted HTTP ports = %d/%d, want 9001/9002", loaded.HttpPort, loaded.HttpsPort)
	}
	if loaded.SetupState != config.SetupStateReady || !loaded.IsInit {
		t.Fatalf("setup state = %q, isInit = %t", loaded.SetupState, loaded.IsInit)
	}
	if !loaded.VerificationEnabled || !loaded.ExternalRecoveryEnabled || loaded.LoginPolicy != "new_device" {
		t.Fatalf("security defaults = verification %t, recovery %t, policy %q", loaded.VerificationEnabled, loaded.ExternalRecoveryEnabled, loaded.LoginPolicy)
	}
	for _, name := range []string{"config.json", "email.json", "smtp.json", "ssl.json", "plugins.json"} {
		if _, err := os.Stat(filepath.Join(config.ROOT_PATH, "config", name)); err != nil {
			t.Fatalf("partition %s: %v", name, err)
		}
	}
	if err := Finish(); err != nil {
		t.Fatalf("repeated completion failed: %v", err)
	}
}

func TestFinishPreservesExplicitlyDisabledSecurityOptions(t *testing.T) {
	prepareFinishConfigTest(t)
	if err := config.WriteConfig(&config.Config{
		SetupState:              config.SetupStateReady,
		IsInit:                  true,
		VerificationEnabled:     false,
		ExternalRecoveryEnabled: false,
		LoginPolicy:             "off",
	}); err != nil {
		t.Fatal(err)
	}

	if err := Finish(); err != nil {
		t.Fatal(err)
	}
	loaded, err := config.ReadConfig()
	if err != nil {
		t.Fatal(err)
	}
	if loaded.VerificationEnabled || loaded.ExternalRecoveryEnabled || loaded.LoginPolicy != "off" {
		t.Fatalf("explicit security options changed: verification %t, recovery %t, policy %q", loaded.VerificationEnabled, loaded.ExternalRecoveryEnabled, loaded.LoginPolicy)
	}
}

func TestFinishRepairsReadyConfigWithMissingPorts(t *testing.T) {
	prepareFinishConfigTest(t)
	t.Setenv("SMAIL_HTTP_PORT", "9001")
	t.Setenv("SMAIL_HTTPS_PORT", "9002")

	if err := config.WriteConfig(&config.Config{
		SetupState:   config.SetupStateReady,
		IsInit:       true,
		HttpsEnabled: 1,
		Domain:       "example.com",
		WebDomain:    "mail.example.com",
		DbType:       config.DBTypeSQLite,
		DbDSN:        filepath.Join(config.ROOT_PATH, "data", "smail.db"),
	}); err != nil {
		t.Fatal(err)
	}
	if err := Finish(); err != nil {
		t.Fatal(err)
	}

	loaded, err := config.ReadConfig()
	if err != nil {
		t.Fatal(err)
	}
	if loaded.HttpPort != 9001 || loaded.HttpsPort != 9002 {
		t.Fatalf("repaired HTTP ports = %d/%d, want 9001/9002", loaded.HttpPort, loaded.HttpsPort)
	}
}

func TestFinishKeepsInternalHTTPSDisabledInProxyMode(t *testing.T) {
	prepareFinishConfigTest(t)
	t.Setenv("SMAIL_HTTP_PORT", "9001")
	t.Setenv("SMAIL_HTTPS_PORT", "9002")

	if err := config.WriteConfig(&config.Config{
		SetupState:   config.SetupStateBootstrap,
		TLSMode:      config.HTTPSTLSModeProxy,
		HttpsPort:    9443,
		HttpsEnabled: 1,
		Domain:       "example.com",
		WebDomain:    "mail.example.com",
		DbType:       config.DBTypeSQLite,
		DbDSN:        filepath.Join(config.ROOT_PATH, "data", "smail.db"),
	}); err != nil {
		t.Fatal(err)
	}
	if err := Finish(); err != nil {
		t.Fatal(err)
	}

	loaded, err := config.ReadConfig()
	if err != nil {
		t.Fatal(err)
	}
	if loaded.HttpPort != 9001 || loaded.HttpsPort != 0 || loaded.HttpsEnabled != 2 {
		t.Fatalf("proxy config = HTTP %d, HTTPS %d, enabled %d", loaded.HttpPort, loaded.HttpsPort, loaded.HttpsEnabled)
	}
}

func TestFinishRejectsInvalidConfigBeforeReady(t *testing.T) {
	prepareFinishConfigTest(t)
	if err := config.WriteConfig(&config.Config{
		SetupState: config.SetupStateBootstrap,
		DbType:     config.DBTypeSQLite,
		DbDSN:      filepath.Join(config.ROOT_PATH, "data", "smail.db"),
	}); err != nil {
		t.Fatal(err)
	}
	if err := Finish(); err == nil {
		t.Fatal("missing domains must prevent setup completion")
	}
	loaded, err := config.ReadConfig()
	if err != nil {
		t.Fatal(err)
	}
	if loaded.IsSetupReady() || loaded.IsInit || config.IsSetupReady() {
		t.Fatal("failed completion published ready state")
	}
	loaded.Domain = "example.com"
	loaded.WebDomain = "mail.example.com"
	if err := config.WriteConfig(loaded); err != nil {
		t.Fatal(err)
	}
	if err := Finish(); err != nil {
		t.Fatalf("corrected setup cannot retry: %v", err)
	}
}

func prepareFinishConfigTest(t *testing.T) {
	t.Helper()
	previousRoot := config.ROOT_PATH
	previousInstance := config.Instance
	previousManager := config.CurrentManager()
	previousSetupPort := config.RuntimeSetupPort()
	previousInitState := config.IsInit
	previousDB := db.Instance
	engine, err := xorm.NewEngine("sqlite", filepath.Join(t.TempDir(), "finish.db"))
	if err != nil {
		t.Fatal(err)
	}
	t.Cleanup(func() {
		db.Instance = previousDB
		_ = engine.Close()
	})
	if err := engine.Sync2(new(models.User)); err != nil {
		t.Fatal(err)
	}
	if _, err := engine.Insert(&models.User{Account: "root", Name: "root", Password: "unused", Role: "super_admin", IsAdmin: 1}); err != nil {
		t.Fatal(err)
	}
	db.Instance = engine

	config.ROOT_PATH = t.TempDir() + string(os.PathSeparator)
	config.Instance = nil
	config.PublishConfigSnapshot(nil)
	config.SetRuntimeManager(nil)
	config.SetRuntimeSetupPort(0)
	config.SetInitState(false)

	t.Cleanup(func() {
		config.ROOT_PATH = previousRoot
		config.Instance = previousInstance
		config.PublishConfigSnapshot(previousInstance)
		config.SetRuntimeManager(previousManager)
		config.SetRuntimeSetupPort(previousSetupPort)
		config.SetInitState(previousInitState)
	})
}