forgejo/models/actions/forgejo_test.go

// SPDX-License-Identifier: MIT

package actions

import (
	"crypto/subtle"
	"testing"

	auth_model "code.gitea.io/gitea/models/auth"
	"code.gitea.io/gitea/models/db"
	"code.gitea.io/gitea/models/unittest"

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

func TestActions_RegisterRunner_Token(t *testing.T) {
	assert.NoError(t, unittest.PrepareTestDatabase())
	ownerID := int64(0)
	repoID := int64(0)
	token := "0123456789012345678901234567890123456789"
	labels := []string{}
	name := "runner"
	version := "v1.2.3"
	runner, err := RegisterRunner(db.DefaultContext, ownerID, repoID, token, &labels, name, version)
	assert.NoError(t, err)
	assert.EqualValues(t, name, runner.Name)

	assert.EqualValues(t, 1, subtle.ConstantTimeCompare([]byte(runner.TokenHash), []byte(auth_model.HashToken(token, runner.TokenSalt))), "the token cannot be verified with the same method as routers/api/actions/runner/interceptor.go as of 8228751c55d6a4263f0fec2932ca16181c09c97d")
}

func TestActions_RegisterRunner_CreateWithLabels(t *testing.T) {
	assert.NoError(t, unittest.PrepareTestDatabase())
	ownerID := int64(0)
	repoID := int64(0)
	token := "0123456789012345678901234567890123456789"
	name := "runner"
	version := "v1.2.3"
	labels := []string{"woop", "doop"}
	labelsCopy := labels // labels may be affected by the tested function so we copy them

	runner, err := RegisterRunner(db.DefaultContext, ownerID, repoID, token, &labels, name, version)
	assert.NoError(t, err)

	// Check that the returned record has been updated, except for the labels
	assert.EqualValues(t, ownerID, runner.OwnerID)
	assert.EqualValues(t, repoID, runner.RepoID)
	assert.EqualValues(t, name, runner.Name)
	assert.EqualValues(t, version, runner.Version)
	assert.EqualValues(t, labelsCopy, runner.AgentLabels)

	// Check that whatever is in the DB has been updated, except for the labels
	after := unittest.AssertExistsAndLoadBean(t, &ActionRunner{ID: runner.ID})
	assert.EqualValues(t, ownerID, after.OwnerID)
	assert.EqualValues(t, repoID, after.RepoID)
	assert.EqualValues(t, name, after.Name)
	assert.EqualValues(t, version, after.Version)
	assert.EqualValues(t, labelsCopy, after.AgentLabels)
}

func TestActions_RegisterRunner_CreateWithoutLabels(t *testing.T) {
	assert.NoError(t, unittest.PrepareTestDatabase())
	ownerID := int64(0)
	repoID := int64(0)
	token := "0123456789012345678901234567890123456789"
	name := "runner"
	version := "v1.2.3"

	runner, err := RegisterRunner(db.DefaultContext, ownerID, repoID, token, nil, name, version)
	assert.NoError(t, err)

	// Check that the returned record has been updated, except for the labels
	assert.EqualValues(t, ownerID, runner.OwnerID)
	assert.EqualValues(t, repoID, runner.RepoID)
	assert.EqualValues(t, name, runner.Name)
	assert.EqualValues(t, version, runner.Version)
	assert.EqualValues(t, []string{}, runner.AgentLabels)

	// Check that whatever is in the DB has been updated, except for the labels
	after := unittest.AssertExistsAndLoadBean(t, &ActionRunner{ID: runner.ID})
	assert.EqualValues(t, ownerID, after.OwnerID)
	assert.EqualValues(t, repoID, after.RepoID)
	assert.EqualValues(t, name, after.Name)
	assert.EqualValues(t, version, after.Version)
	assert.EqualValues(t, []string{}, after.AgentLabels)
}

func TestActions_RegisterRunner_UpdateWithLabels(t *testing.T) {
	const recordID = 12345678
	token := "7e577e577e577e57feedfacefeedfacefeedface"
	assert.NoError(t, unittest.PrepareTestDatabase())
	unittest.AssertExistsAndLoadBean(t, &ActionRunner{ID: recordID})

	newOwnerID := int64(1)
	newRepoID := int64(1)
	newName := "rennur"
	newVersion := "v4.5.6"
	newLabels := []string{"warp", "darp"}
	labelsCopy := newLabels // labels may be affected by the tested function so we copy them

	runner, err := RegisterRunner(db.DefaultContext, newOwnerID, newRepoID, token, &newLabels, newName, newVersion)
	assert.NoError(t, err)

	// Check that the returned record has been updated
	assert.EqualValues(t, newOwnerID, runner.OwnerID)
	assert.EqualValues(t, newRepoID, runner.RepoID)
	assert.EqualValues(t, newName, runner.Name)
	assert.EqualValues(t, newVersion, runner.Version)
	assert.EqualValues(t, labelsCopy, runner.AgentLabels)

	// Check that whatever is in the DB has been updated
	after := unittest.AssertExistsAndLoadBean(t, &ActionRunner{ID: recordID})
	assert.EqualValues(t, newOwnerID, after.OwnerID)
	assert.EqualValues(t, newRepoID, after.RepoID)
	assert.EqualValues(t, newName, after.Name)
	assert.EqualValues(t, newVersion, after.Version)
	assert.EqualValues(t, labelsCopy, after.AgentLabels)
}

func TestActions_RegisterRunner_UpdateWithoutLabels(t *testing.T) {
	const recordID = 12345678
	token := "7e577e577e577e57feedfacefeedfacefeedface"
	assert.NoError(t, unittest.PrepareTestDatabase())
	before := unittest.AssertExistsAndLoadBean(t, &ActionRunner{ID: recordID})

	newOwnerID := int64(1)
	newRepoID := int64(1)
	newName := "rennur"
	newVersion := "v4.5.6"

	runner, err := RegisterRunner(db.DefaultContext, newOwnerID, newRepoID, token, nil, newName, newVersion)
	assert.NoError(t, err)

	// Check that the returned record has been updated, except for the labels
	assert.EqualValues(t, newOwnerID, runner.OwnerID)
	assert.EqualValues(t, newRepoID, runner.RepoID)
	assert.EqualValues(t, newName, runner.Name)
	assert.EqualValues(t, newVersion, runner.Version)
	assert.EqualValues(t, before.AgentLabels, runner.AgentLabels)

	// Check that whatever is in the DB has been updated, except for the labels
	after := unittest.AssertExistsAndLoadBean(t, &ActionRunner{ID: recordID})
	assert.EqualValues(t, newOwnerID, after.OwnerID)
	assert.EqualValues(t, newRepoID, after.RepoID)
	assert.EqualValues(t, newName, after.Name)
	assert.EqualValues(t, newVersion, after.Version)
	assert.EqualValues(t, before.AgentLabels, after.AgentLabels)
}