gitea/services/audit/notifier_test.go

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

package audit

import (
	"testing"

	audit_model "gitea.dev/models/audit"
	issues_model "gitea.dev/models/issues"
	repo_model "gitea.dev/models/repo"
	"gitea.dev/models/unittest"
	user_model "gitea.dev/models/user"
	"gitea.dev/modules/reqctx"
	"gitea.dev/modules/setting"
	"gitea.dev/modules/test"

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

func TestAuditNotifier(t *testing.T) {
	assert.NoError(t, unittest.PrepareTestDatabase())
	defer test.MockVariableValue(&setting.Audit.RecordOutput, setting.AuditRecordOutputDatabase)()

	doer := &user_model.User{ID: 2, Name: "doer"}
	repo := &repo_model.Repository{ID: 1, OwnerName: "owner", Name: "repo"}
	ctx := reqctx.NewRequestContextForTest(t)
	SetRequestInfo(ctx, audit_model.OriginUI, "127.0.0.1")
	notifier := new(auditNotifier)

	for _, tc := range []struct {
		notify func()
		event  audit_model.Event
	}{
		{
			func() {
				notifier.NewIssue(ctx, &issues_model.Issue{ID: 10, Index: 5, Title: "Issue title", Poster: doer, Repo: repo}, nil)
			},
			audit_model.Event{Action: audit_model.IssueCreate, Origin: audit_model.OriginUI},
		},
		{
			func() {
				notifier.NewPullRequest(ctx, &issues_model.PullRequest{ID: 11, Issue: &issues_model.Issue{ID: 12, Index: 6, Title: "PR title", Poster: doer, Repo: repo, IsPull: true}}, nil)
			},
			audit_model.Event{Action: audit_model.PullRequestCreate, Origin: audit_model.OriginUI},
		},
		{
			func() { notifier.NewWikiPage(ctx, doer, repo, "Home", "") },
			audit_model.Event{Action: audit_model.WikiPageCreate, Origin: audit_model.OriginUI},
		},
		{
			func() { notifier.CreateRepository(ctx, doer, doer, repo) },
			audit_model.Event{Action: audit_model.RepositoryCreate, ActorID: doer.ID, Origin: audit_model.OriginUI},
		},
		{
			func() { notifier.TransferRepository(ctx, doer, repo, "previous_owner") },
			audit_model.Event{Action: audit_model.RepositoryTransferFinish, ActorID: doer.ID, Message: "Transferred repository owner/repo from previous_owner to owner."},
		},
		{
			func() { notifier.ChangeDefaultBranch(ctx, repo) },
			audit_model.Event{Action: audit_model.RepositoryBranchDefault},
		},
	} {
		tc.notify()
		tc.event.ScopeType = audit_model.ScopeRepository
		tc.event.ScopeID = repo.ID
		unittest.AssertExistsAndLoadBean(t, &tc.event)
	}
}