gitea/cmd/web_test.go

// Copyright 2026 The Gitea Authors. All rights reserved.
// SPDX-License-Identifier: MIT

package cmd

import (
	"context"
	"net/http/httptest"
	"os"
	"path/filepath"
	"testing"

	user_model "code.gitea.io/gitea/models/user"
	"code.gitea.io/gitea/modules/setting"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

// start edit/add - by petru @ codex
func TestMissingInstalledSQLiteDatabase(t *testing.T) {
	oldDatabase := setting.Database
	oldInstallLock := setting.InstallLock
	oldCustomConf := setting.CustomConf
	defer func() {
		setting.Database = oldDatabase
		setting.InstallLock = oldInstallLock
		setting.CustomConf = oldCustomConf
	}()

	tmpDir := t.TempDir()
	setting.CustomConf = filepath.Join(tmpDir, "custom", "conf", "app.ini")
	setting.Database.Type = "sqlite3"
	setting.Database.Path = filepath.Join(tmpDir, "data", "gitea.db")
	setting.InstallLock = false // edit/add - by petru @ codex

	missing, err := missingInstalledSQLiteDatabase()
	require.NoError(t, err)
	assert.False(t, missing)

	setting.InstallLock = true
	missing, err = missingInstalledSQLiteDatabase()
	require.NoError(t, err)
	assert.True(t, missing)

	require.NoError(t, os.MkdirAll(filepath.Dir(setting.Database.Path), 0o755))
	require.NoError(t, os.WriteFile(setting.Database.Path, []byte("db"), 0o600))
	missing, err = missingInstalledSQLiteDatabase()
	require.NoError(t, err)
	assert.False(t, missing)

	require.NoError(t, os.Remove(setting.Database.Path))
	setting.InstallLock = false
	require.NoError(t, os.MkdirAll(filepath.Dir(setting.InstallSentinelPath()), 0o755))
	require.NoError(t, os.WriteFile(setting.InstallSentinelPath(), []byte("installed\n"), 0o600))
	missing, err = missingInstalledSQLiteDatabase()
	require.NoError(t, err)
	assert.True(t, missing)

	setting.Database.Type = "postgres"
	missing, err = missingInstalledSQLiteDatabase()
	require.NoError(t, err)
	assert.False(t, missing)
}

func TestBuildAutoRecoveryConfigFromInstalledState(t *testing.T) {
	adminUser, adminErr := user_model.GetAdminUser(context.Background())
	recoveryCfg, err := buildAutoRecoveryConfigFromInstalledState(context.Background())
	require.NoError(t, err)
	if adminErr != nil {
		require.True(t, user_model.IsErrUserNotExist(adminErr))
		assert.Nil(t, recoveryCfg)
		return
	}
	require.NotNil(t, adminUser)
	if adminUser.Email == "" {
		assert.Nil(t, recoveryCfg)
		return
	}
	require.NotNil(t, recoveryCfg)
	assert.True(t, recoveryCfg.Enabled)
	assert.Equal(t, adminUser.Email, recoveryCfg.AllowedEmails)
	assert.Equal(t, setting.Backup.Path, recoveryCfg.BackupPath)
	assert.Equal(t, setting.RepoRootPath, recoveryCfg.RepoRootPath)
	if setting.MailService != nil {
		assert.Equal(t, setting.MailService.SMTPAddr, recoveryCfg.SMTPAddr)
		assert.Equal(t, setting.MailService.SMTPPort, recoveryCfg.SMTPPort)
		assert.Equal(t, setting.MailService.From, recoveryCfg.SMTPFrom)
		assert.Equal(t, setting.MailService.User, recoveryCfg.SMTPUser)
		assert.Equal(t, setting.MailService.Passwd, recoveryCfg.SMTPPasswd)
	}
}

func TestShouldServeInstalledRecoveryWithMissingSQLite(t *testing.T) {
	oldDatabase := setting.Database
	oldInstallLock := setting.InstallLock
	oldCustomConf := setting.CustomConf
	defer func() {
		setting.Database = oldDatabase
		setting.InstallLock = oldInstallLock
		setting.CustomConf = oldCustomConf
	}()

	tmpDir := t.TempDir()
	setting.CustomConf = filepath.Join(tmpDir, "custom", "conf", "app.ini")
	setting.Database.Type = "sqlite3"
	setting.Database.Path = filepath.Join(tmpDir, "data", "gitea.db")
	setting.InstallLock = true
	require.NoError(t, os.MkdirAll(filepath.Dir(setting.InstallSentinelPath()), 0o755))
	require.NoError(t, os.WriteFile(setting.InstallSentinelPath(), []byte("installed\n"), 0o600))
	require.NoError(t, setting.SaveRecoveryConfig(&setting.RecoveryConfig{
		Enabled:       true,
		AllowedEmails: "admin@example.com",
		SMTPAddr:      "smtp.example.com",
		SMTPPort:      "587",
		SMTPFrom:      "Gitea <noreply@example.com>",
	}))

	recoveryCfg, shouldServeRecovery, err := shouldServeInstalledRecovery(context.Background())
	require.NoError(t, err)
	require.True(t, shouldServeRecovery)
	require.NotNil(t, recoveryCfg)
	assert.Equal(t, "admin@example.com", recoveryCfg.AllowedEmails)
}

func TestRecoveryRequestBaseURLStripsRecoveryQueryFromAppURL(t *testing.T) {
	// start edit/add - by petru @ codex
	oldInstallLock := setting.InstallLock
	oldAppURL := setting.AppURL
	oldAppSubURL := setting.AppSubURL
	defer func() {
		setting.InstallLock = oldInstallLock
		setting.AppURL = oldAppURL
		setting.AppSubURL = oldAppSubURL
	}()

	setting.InstallLock = true
	setting.AppURL = "http://example.com:3000/?recovery=1"
	setting.AppSubURL = ""

	req := httptest.NewRequest("GET", "http://example.com:3000/recovery", nil)
	assert.Equal(t, "http://example.com:3000", recoveryRequestBaseURL(req))
	// end edit/add - by petru @ codex
}

// end edit/add - by petru @ codex