forgejo/models/system/setting_test.go

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

package system_test

import (
	"testing"

	"forgejo.org/models/db"
	"forgejo.org/models/system"
	"forgejo.org/models/unittest"

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

func TestSettings(t *testing.T) {
	keyName := "test.key"
	require.NoError(t, unittest.PrepareTestDatabase())

	require.NoError(t, db.TruncateBeans(db.DefaultContext, &system.Setting{}))

	rev, settings, err := system.GetAllSettings(db.DefaultContext)
	require.NoError(t, err)
	assert.EqualValues(t, 1, rev)
	assert.Len(t, settings, 1) // there is only one "revision" key

	err = system.SetSettings(db.DefaultContext, map[string]string{keyName: "true"})
	require.NoError(t, err)
	rev, settings, err = system.GetAllSettings(db.DefaultContext)
	require.NoError(t, err)
	assert.EqualValues(t, 2, rev)
	assert.Len(t, settings, 2)
	assert.EqualValues(t, "true", settings[keyName])

	err = system.SetSettings(db.DefaultContext, map[string]string{keyName: "false"})
	require.NoError(t, err)
	rev, settings, err = system.GetAllSettings(db.DefaultContext)
	require.NoError(t, err)
	assert.EqualValues(t, 3, rev)
	assert.Len(t, settings, 2)
	assert.EqualValues(t, "false", settings[keyName])

	// setting the same value should not trigger DuplicateKey error, and the "version" should be increased
	err = system.SetSettings(db.DefaultContext, map[string]string{keyName: "false"})
	require.NoError(t, err)

	rev, settings, err = system.GetAllSettings(db.DefaultContext)
	require.NoError(t, err)
	assert.Len(t, settings, 2)
	assert.EqualValues(t, 4, rev)
}