modules/setting/database_test.go - master - uzer_007/Rixea

modules/setting/database_test.go
74 строки2 KB
uzer_007
feat: первая версия Rixea
02 авг 2026, 22:16
02 авг 2026, 22:16d4f7504
// Copyright 2026 Uzer_007. All rights reserved.
// SPDX-License-Identifier: MIT

package setting

import (
	"strings"
	"testing"
	"time"

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

func TestLoadReindexerDatabaseSettings(t *testing.T) {
	original := Database
	t.Cleanup(func() { Database = original })
	cfg, err := NewConfigProviderFromData(`
[database]
DB_TYPE = reindexer
HOST = rx-1.example.com:6534,rx-2.example.com:6534
NAME = gitea
REINDEXER_POOL_SIZE = 24
REINDEXER_NET_COMPRESSION = true
REINDEXER_LOGIN_TIMEOUT = 7s
REINDEXER_REQUEST_TIMEOUT = 21s
REINDEXER_READ_RETRIES = 3
REINDEXER_WRITE_RETRIES = 0
REINDEXER_CREATE_DATABASE = false
REINDEXER_OBJECT_CACHE_BUDGET = 1048576
REINDEXER_SCHEMA_WORKERS = 12
`)
	require.NoError(t, err)

	loadDBSetting(cfg)

	assert.Equal(t, DatabaseTypeReindexer, Database.Type)
	assert.Equal(t, "rx-1.example.com:6534,rx-2.example.com:6534", Database.Host)
	assert.Equal(t, 24, Database.ReindexerPoolSize)
	assert.True(t, Database.ReindexerNetCompression)
	assert.Equal(t, 7*time.Second, Database.ReindexerLoginTimeout)
	assert.Equal(t, 21*time.Second, Database.ReindexerRequestTimeout)
	assert.Equal(t, 3, Database.ReindexerReadRetries)
	assert.Zero(t, Database.ReindexerWriteRetries)
	assert.False(t, Database.ReindexerCreateDatabase)
	assert.Equal(t, 1048576, Database.ReindexerObjectCacheBudget)
	assert.Equal(t, 12, Database.ReindexerSchemaWorkers)
}

func TestLoadReindexerDatabaseSettingsWarnsAboutSQLKeys(t *testing.T) {
	originalDatabase := Database
	originalProblems := append([]string(nil), StartupProblems...)
	t.Cleanup(func() {
		Database = originalDatabase
		StartupProblems = originalProblems
	})

	cfg, err := NewConfigProviderFromData(`
[database]
DB_TYPE = reindexer
LOG_SQL = true
MAX_OPEN_CONNS = 20
`)
	require.NoError(t, err)
	problemCount := len(StartupProblems)

	loadDBSetting(cfg)

	require.Len(t, StartupProblems, problemCount+1)
	message := StartupProblems[problemCount]
	assert.Contains(t, message, "[database].LOG_SQL")
	assert.Contains(t, message, "[database].MAX_OPEN_CONNS")
	assert.Contains(t, strings.ToLower(message), "reindexer")
}