mirror of
				https://codeberg.org/forgejo/forgejo.git
				synced 2025-11-04 00:11:04 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			90 lines
		
	
	
	
		
			3 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			90 lines
		
	
	
	
		
			3 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
// Copyright 2019 The Gitea Authors. All rights reserved.
 | 
						|
// SPDX-License-Identifier: MIT
 | 
						|
 | 
						|
package webhook
 | 
						|
 | 
						|
import (
 | 
						|
	"testing"
 | 
						|
 | 
						|
	"code.gitea.io/gitea/models/db"
 | 
						|
	repo_model "code.gitea.io/gitea/models/repo"
 | 
						|
	"code.gitea.io/gitea/models/unittest"
 | 
						|
	webhook_model "code.gitea.io/gitea/models/webhook"
 | 
						|
	api "code.gitea.io/gitea/modules/structs"
 | 
						|
	webhook_module "code.gitea.io/gitea/modules/webhook"
 | 
						|
 | 
						|
	"github.com/stretchr/testify/assert"
 | 
						|
)
 | 
						|
 | 
						|
func TestWebhook_GetSlackHook(t *testing.T) {
 | 
						|
	w := &webhook_model.Webhook{
 | 
						|
		Meta: `{"channel": "foo", "username": "username", "color": "blue"}`,
 | 
						|
	}
 | 
						|
	slackHook := GetSlackHook(w)
 | 
						|
	assert.Equal(t, *slackHook, SlackMeta{
 | 
						|
		Channel:  "foo",
 | 
						|
		Username: "username",
 | 
						|
		Color:    "blue",
 | 
						|
	})
 | 
						|
}
 | 
						|
 | 
						|
func activateWebhook(t *testing.T, hookID int64) {
 | 
						|
	t.Helper()
 | 
						|
	updated, err := db.GetEngine(db.DefaultContext).ID(hookID).Cols("is_active").Update(webhook_model.Webhook{IsActive: true})
 | 
						|
	assert.Equal(t, int64(1), updated)
 | 
						|
	assert.NoError(t, err)
 | 
						|
}
 | 
						|
 | 
						|
func TestPrepareWebhooks(t *testing.T) {
 | 
						|
	assert.NoError(t, unittest.PrepareTestDatabase())
 | 
						|
 | 
						|
	repo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 1})
 | 
						|
	activateWebhook(t, 1)
 | 
						|
 | 
						|
	hookTasks := []*webhook_model.HookTask{
 | 
						|
		{HookID: 1, EventType: webhook_module.HookEventPush},
 | 
						|
	}
 | 
						|
	for _, hookTask := range hookTasks {
 | 
						|
		unittest.AssertNotExistsBean(t, hookTask)
 | 
						|
	}
 | 
						|
	assert.NoError(t, PrepareWebhooks(db.DefaultContext, EventSource{Repository: repo}, webhook_module.HookEventPush, &api.PushPayload{Commits: []*api.PayloadCommit{{}}}))
 | 
						|
	for _, hookTask := range hookTasks {
 | 
						|
		unittest.AssertExistsAndLoadBean(t, hookTask)
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func TestPrepareWebhooksBranchFilterMatch(t *testing.T) {
 | 
						|
	assert.NoError(t, unittest.PrepareTestDatabase())
 | 
						|
 | 
						|
	repo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 2})
 | 
						|
	activateWebhook(t, 4)
 | 
						|
 | 
						|
	hookTasks := []*webhook_model.HookTask{
 | 
						|
		{HookID: 4, EventType: webhook_module.HookEventPush},
 | 
						|
	}
 | 
						|
	for _, hookTask := range hookTasks {
 | 
						|
		unittest.AssertNotExistsBean(t, hookTask)
 | 
						|
	}
 | 
						|
	// this test also ensures that * doesn't handle / in any special way (like shell would)
 | 
						|
	assert.NoError(t, PrepareWebhooks(db.DefaultContext, EventSource{Repository: repo}, webhook_module.HookEventPush, &api.PushPayload{Ref: "refs/heads/feature/7791", Commits: []*api.PayloadCommit{{}}}))
 | 
						|
	for _, hookTask := range hookTasks {
 | 
						|
		unittest.AssertExistsAndLoadBean(t, hookTask)
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func TestPrepareWebhooksBranchFilterNoMatch(t *testing.T) {
 | 
						|
	assert.NoError(t, unittest.PrepareTestDatabase())
 | 
						|
 | 
						|
	repo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 2})
 | 
						|
	hookTasks := []*webhook_model.HookTask{
 | 
						|
		{HookID: 4, EventType: webhook_module.HookEventPush},
 | 
						|
	}
 | 
						|
	for _, hookTask := range hookTasks {
 | 
						|
		unittest.AssertNotExistsBean(t, hookTask)
 | 
						|
	}
 | 
						|
	assert.NoError(t, PrepareWebhooks(db.DefaultContext, EventSource{Repository: repo}, webhook_module.HookEventPush, &api.PushPayload{Ref: "refs/heads/fix_weird_bug"}))
 | 
						|
 | 
						|
	for _, hookTask := range hookTasks {
 | 
						|
		unittest.AssertNotExistsBean(t, hookTask)
 | 
						|
	}
 | 
						|
}
 |