mirror of
https://github.com/go-gitea/gitea.git
synced 2026-02-19 21:17:38 +01:00
5e9b9b33d1
This simplifies the Makefile by removing the whole-file wrapping that creates a tempdir introduced by https://github.com/go-gitea/gitea/pull/11126. REPO_TEST_DIR is removed as well. Also clean up a lot of legacy code: unnecessary XSS test, incorrect test env init, unused "_old_uid" hack, etc Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com> Co-authored-by: wxiaoguang <wxiaoguang@gmail.com>
78 lines
2.6 KiB
Go
78 lines
2.6 KiB
Go
// Copyright 2024 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package actions
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
|
|
actions_model "code.gitea.io/gitea/models/actions"
|
|
"code.gitea.io/gitea/models/db"
|
|
"code.gitea.io/gitea/models/unittest"
|
|
"code.gitea.io/gitea/modules/util"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestMain(m *testing.M) {
|
|
unittest.MainTest(m)
|
|
}
|
|
|
|
func TestInitToken(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
|
|
t.Run("NoToken", func(t *testing.T) {
|
|
_, _ = db.Exec(t.Context(), "DELETE FROM action_runner_token")
|
|
t.Setenv("GITEA_RUNNER_REGISTRATION_TOKEN", "")
|
|
t.Setenv("GITEA_RUNNER_REGISTRATION_TOKEN_FILE", "")
|
|
err := initGlobalRunnerToken(t.Context())
|
|
require.NoError(t, err)
|
|
notEmpty, err := db.IsTableNotEmpty(&actions_model.ActionRunnerToken{})
|
|
require.NoError(t, err)
|
|
assert.False(t, notEmpty)
|
|
})
|
|
|
|
t.Run("EnvToken", func(t *testing.T) {
|
|
tokenValue, _ := util.CryptoRandomString(32)
|
|
t.Setenv("GITEA_RUNNER_REGISTRATION_TOKEN", tokenValue)
|
|
t.Setenv("GITEA_RUNNER_REGISTRATION_TOKEN_FILE", "")
|
|
err := initGlobalRunnerToken(t.Context())
|
|
require.NoError(t, err)
|
|
token := unittest.AssertExistsAndLoadBean(t, &actions_model.ActionRunnerToken{Token: tokenValue})
|
|
assert.True(t, token.IsActive)
|
|
|
|
// init with the same token again, should not create a new token
|
|
err = initGlobalRunnerToken(t.Context())
|
|
require.NoError(t, err)
|
|
token2 := unittest.AssertExistsAndLoadBean(t, &actions_model.ActionRunnerToken{Token: tokenValue})
|
|
assert.Equal(t, token.ID, token2.ID)
|
|
assert.True(t, token.IsActive)
|
|
})
|
|
|
|
t.Run("EnvFileToken", func(t *testing.T) {
|
|
tokenValue, _ := util.CryptoRandomString(32)
|
|
f := t.TempDir() + "/token"
|
|
_ = os.WriteFile(f, []byte(tokenValue), 0o644)
|
|
t.Setenv("GITEA_RUNNER_REGISTRATION_TOKEN", "")
|
|
t.Setenv("GITEA_RUNNER_REGISTRATION_TOKEN_FILE", f)
|
|
err := initGlobalRunnerToken(t.Context())
|
|
require.NoError(t, err)
|
|
token := unittest.AssertExistsAndLoadBean(t, &actions_model.ActionRunnerToken{Token: tokenValue})
|
|
assert.True(t, token.IsActive)
|
|
|
|
// if the env token is invalidated by another new token, then it shouldn't be active anymore
|
|
_, err = actions_model.NewRunnerToken(t.Context(), 0, 0)
|
|
require.NoError(t, err)
|
|
token = unittest.AssertExistsAndLoadBean(t, &actions_model.ActionRunnerToken{Token: tokenValue})
|
|
assert.False(t, token.IsActive)
|
|
})
|
|
|
|
t.Run("InvalidToken", func(t *testing.T) {
|
|
t.Setenv("GITEA_RUNNER_REGISTRATION_TOKEN", "abc")
|
|
err := initGlobalRunnerToken(t.Context())
|
|
assert.ErrorContains(t, err, "must be at least")
|
|
})
|
|
}
|