238 lines
8.1 KiB
Go
238 lines
8.1 KiB
Go
|
// Copyright 2023 The Gitea Authors. All rights reserved.
|
||
|
|
// SPDX-License-Identifier: MIT
|
||
|
|
|
||
|
|
package integration
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"net/http"
|
||
|
|
"net/http/httptest"
|
||
|
|
"net/url"
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
actions_model "code.gitea.io/gitea/models/actions"
|
||
|
|
"code.gitea.io/gitea/models/db"
|
||
|
|
repo_model "code.gitea.io/gitea/models/repo"
|
||
|
|
unit_model "code.gitea.io/gitea/models/unit"
|
||
|
|
"code.gitea.io/gitea/models/unittest"
|
||
|
|
user_model "code.gitea.io/gitea/models/user"
|
||
|
|
"code.gitea.io/gitea/modules/git"
|
||
|
|
"code.gitea.io/gitea/modules/test"
|
||
|
|
repo_service "code.gitea.io/gitea/services/repository"
|
||
|
|
files_service "code.gitea.io/gitea/services/repository/files"
|
||
|
|
"code.gitea.io/gitea/tests"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func assertBadge(t *testing.T, resp *httptest.ResponseRecorder, badge string) {
|
||
|
|
assert.Equal(t, fmt.Sprintf("https://img.shields.io/badge/%s", badge), test.RedirectURL(resp))
|
||
|
|
}
|
||
|
|
|
||
|
|
func createMinimalRepo(t *testing.T) func() {
|
||
|
|
user2 := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2})
|
||
|
|
|
||
|
|
// Create a new repository
|
||
|
|
repo, err := repo_service.CreateRepository(db.DefaultContext, user2, user2, repo_service.CreateRepoOptions{
|
||
|
|
Name: "minimal",
|
||
|
|
Description: "minimal repo for badge testing",
|
||
|
|
AutoInit: true,
|
||
|
|
Gitignores: "Go",
|
||
|
|
License: "MIT",
|
||
|
|
Readme: "Default",
|
||
|
|
DefaultBranch: "main",
|
||
|
|
IsPrivate: false,
|
||
|
|
})
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.NotEmpty(t, repo)
|
||
|
|
|
||
|
|
// Enable Actions, and disable Issues, PRs and Releases
|
||
|
|
err = repo_service.UpdateRepositoryUnits(db.DefaultContext, repo, []repo_model.RepoUnit{{
|
||
|
|
RepoID: repo.ID,
|
||
|
|
Type: unit_model.TypeActions,
|
||
|
|
}}, []unit_model.Type{unit_model.TypeIssues, unit_model.TypePullRequests, unit_model.TypeReleases})
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
return func() {
|
||
|
|
repo_service.DeleteRepository(db.DefaultContext, user2, repo, false)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func addWorkflow(t *testing.T) {
|
||
|
|
user2 := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2})
|
||
|
|
repo, err := repo_model.GetRepositoryByOwnerAndName(db.DefaultContext, "user2", "minimal")
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
// Add a workflow file to the repo
|
||
|
|
addWorkflowToBaseResp, err := files_service.ChangeRepoFiles(git.DefaultContext, repo, user2, &files_service.ChangeRepoFilesOptions{
|
||
|
|
Files: []*files_service.ChangeRepoFile{
|
||
|
|
{
|
||
|
|
Operation: "create",
|
||
|
|
TreePath: ".gitea/workflows/pr.yml",
|
||
|
|
ContentReader: strings.NewReader("name: test\non:\n push:\njobs:\n test:\n runs-on: ubuntu-latest\n steps:\n - run: echo helloworld\n"),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
Message: "add workflow",
|
||
|
|
OldBranch: "main",
|
||
|
|
NewBranch: "main",
|
||
|
|
Author: &files_service.IdentityOptions{
|
||
|
|
Name: user2.Name,
|
||
|
|
Email: user2.Email,
|
||
|
|
},
|
||
|
|
Committer: &files_service.IdentityOptions{
|
||
|
|
Name: user2.Name,
|
||
|
|
Email: user2.Email,
|
||
|
|
},
|
||
|
|
Dates: &files_service.CommitDateOptions{
|
||
|
|
Author: time.Now(),
|
||
|
|
Committer: time.Now(),
|
||
|
|
},
|
||
|
|
})
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.NotEmpty(t, addWorkflowToBaseResp)
|
||
|
|
|
||
|
|
assert.Equal(t, 1, unittest.GetCount(t, &actions_model.ActionRun{RepoID: repo.ID}))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestWorkflowBadges(t *testing.T) {
|
||
|
|
onGiteaRun(t, func(t *testing.T, u *url.URL) {
|
||
|
|
defer tests.PrintCurrentTest(t)()
|
||
|
|
defer createMinimalRepo(t)()
|
||
|
|
|
||
|
|
addWorkflow(t)
|
||
|
|
|
||
|
|
// Actions disabled
|
||
|
|
req := NewRequest(t, "GET", "/user2/repo1/badges/workflows/test.yaml/badge.svg")
|
||
|
|
resp := MakeRequest(t, req, http.StatusSeeOther)
|
||
|
|
assertBadge(t, resp, "test.yaml-Not%20found-crimson")
|
||
|
|
|
||
|
|
req = NewRequest(t, "GET", "/user2/repo1/badges/workflows/test.yaml/badge.svg?branch=no-such-branch")
|
||
|
|
resp = MakeRequest(t, req, http.StatusSeeOther)
|
||
|
|
assertBadge(t, resp, "test.yaml-Not%20found-crimson")
|
||
|
|
|
||
|
|
// Actions enabled
|
||
|
|
req = NewRequest(t, "GET", "/user2/minimal/badges/workflows/pr.yml/badge.svg")
|
||
|
|
resp = MakeRequest(t, req, http.StatusSeeOther)
|
||
|
|
assertBadge(t, resp, "pr.yml-waiting-lightgrey")
|
||
|
|
|
||
|
|
req = NewRequest(t, "GET", "/user2/minimal/badges/workflows/pr.yml/badge.svg?branch=main")
|
||
|
|
resp = MakeRequest(t, req, http.StatusSeeOther)
|
||
|
|
assertBadge(t, resp, "pr.yml-waiting-lightgrey")
|
||
|
|
|
||
|
|
req = NewRequest(t, "GET", "/user2/minimal/badges/workflows/pr.yml/badge.svg?branch=no-such-branch")
|
||
|
|
resp = MakeRequest(t, req, http.StatusSeeOther)
|
||
|
|
assertBadge(t, resp, "pr.yml-Not%20found-crimson")
|
||
|
|
|
||
|
|
req = NewRequest(t, "GET", "/user2/minimal/badges/workflows/pr.yml/badge.svg?event=cron")
|
||
|
|
resp = MakeRequest(t, req, http.StatusSeeOther)
|
||
|
|
assertBadge(t, resp, "pr.yml-Not%20found-crimson")
|
||
|
|
|
||
|
|
// GitHub compatibility
|
||
|
|
req = NewRequest(t, "GET", "/user2/minimal/actions/workflows/pr.yml/badge.svg")
|
||
|
|
resp = MakeRequest(t, req, http.StatusSeeOther)
|
||
|
|
assertBadge(t, resp, "pr.yml-waiting-lightgrey")
|
||
|
|
|
||
|
|
req = NewRequest(t, "GET", "/user2/minimal/actions/workflows/pr.yml/badge.svg?branch=main")
|
||
|
|
resp = MakeRequest(t, req, http.StatusSeeOther)
|
||
|
|
assertBadge(t, resp, "pr.yml-waiting-lightgrey")
|
||
|
|
|
||
|
|
req = NewRequest(t, "GET", "/user2/minimal/actions/workflows/pr.yml/badge.svg?branch=no-such-branch")
|
||
|
|
resp = MakeRequest(t, req, http.StatusSeeOther)
|
||
|
|
assertBadge(t, resp, "pr.yml-Not%20found-crimson")
|
||
|
|
|
||
|
|
req = NewRequest(t, "GET", "/user2/minimal/actions/workflows/pr.yml/badge.svg?event=cron")
|
||
|
|
resp = MakeRequest(t, req, http.StatusSeeOther)
|
||
|
|
assertBadge(t, resp, "pr.yml-Not%20found-crimson")
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestBadges(t *testing.T) {
|
||
|
|
defer tests.PrepareTestEnv(t)()
|
||
|
|
|
||
|
|
t.Run("Stars", func(t *testing.T) {
|
||
|
|
defer tests.PrintCurrentTest(t)()
|
||
|
|
|
||
|
|
req := NewRequest(t, "GET", "/user2/repo1/badges/stars.svg")
|
||
|
|
resp := MakeRequest(t, req, http.StatusSeeOther)
|
||
|
|
|
||
|
|
assertBadge(t, resp, "stars-0-blue")
|
||
|
|
})
|
||
|
|
|
||
|
|
t.Run("Issues", func(t *testing.T) {
|
||
|
|
defer tests.PrintCurrentTest(t)()
|
||
|
|
defer createMinimalRepo(t)()
|
||
|
|
|
||
|
|
// Issues enabled
|
||
|
|
req := NewRequest(t, "GET", "/user2/repo1/badges/issues.svg")
|
||
|
|
resp := MakeRequest(t, req, http.StatusSeeOther)
|
||
|
|
assertBadge(t, resp, "issues-2-blue")
|
||
|
|
|
||
|
|
req = NewRequest(t, "GET", "/user2/repo1/badges/issues/open.svg")
|
||
|
|
resp = MakeRequest(t, req, http.StatusSeeOther)
|
||
|
|
assertBadge(t, resp, "issues-1%20open-blue")
|
||
|
|
|
||
|
|
req = NewRequest(t, "GET", "/user2/repo1/badges/issues/closed.svg")
|
||
|
|
resp = MakeRequest(t, req, http.StatusSeeOther)
|
||
|
|
assertBadge(t, resp, "issues-1%20closed-blue")
|
||
|
|
|
||
|
|
// Issues disabled
|
||
|
|
req = NewRequest(t, "GET", "/user2/minimal/badges/issues.svg")
|
||
|
|
resp = MakeRequest(t, req, http.StatusSeeOther)
|
||
|
|
assertBadge(t, resp, "issues-Not%20found-crimson")
|
||
|
|
|
||
|
|
req = NewRequest(t, "GET", "/user2/minimal/badges/issues/open.svg")
|
||
|
|
resp = MakeRequest(t, req, http.StatusSeeOther)
|
||
|
|
assertBadge(t, resp, "issues-Not%20found-crimson")
|
||
|
|
|
||
|
|
req = NewRequest(t, "GET", "/user2/minimal/badges/issues/closed.svg")
|
||
|
|
resp = MakeRequest(t, req, http.StatusSeeOther)
|
||
|
|
assertBadge(t, resp, "issues-Not%20found-crimson")
|
||
|
|
})
|
||
|
|
|
||
|
|
t.Run("Pulls", func(t *testing.T) {
|
||
|
|
defer tests.PrintCurrentTest(t)()
|
||
|
|
defer createMinimalRepo(t)()
|
||
|
|
|
||
|
|
// Pull requests enabled
|
||
|
|
req := NewRequest(t, "GET", "/user2/repo1/badges/pulls.svg")
|
||
|
|
resp := MakeRequest(t, req, http.StatusSeeOther)
|
||
|
|
assertBadge(t, resp, "pulls-3-blue")
|
||
|
|
|
||
|
|
req = NewRequest(t, "GET", "/user2/repo1/badges/pulls/open.svg")
|
||
|
|
resp = MakeRequest(t, req, http.StatusSeeOther)
|
||
|
|
assertBadge(t, resp, "pulls-3%20open-blue")
|
||
|
|
|
||
|
|
req = NewRequest(t, "GET", "/user2/repo1/badges/pulls/closed.svg")
|
||
|
|
resp = MakeRequest(t, req, http.StatusSeeOther)
|
||
|
|
assertBadge(t, resp, "pulls-0%20closed-blue")
|
||
|
|
|
||
|
|
// Pull requests disabled
|
||
|
|
req = NewRequest(t, "GET", "/user2/minimal/badges/pulls.svg")
|
||
|
|
resp = MakeRequest(t, req, http.StatusSeeOther)
|
||
|
|
assertBadge(t, resp, "pulls-Not%20found-crimson")
|
||
|
|
|
||
|
|
req = NewRequest(t, "GET", "/user2/minimal/badges/pulls/open.svg")
|
||
|
|
resp = MakeRequest(t, req, http.StatusSeeOther)
|
||
|
|
assertBadge(t, resp, "pulls-Not%20found-crimson")
|
||
|
|
|
||
|
|
req = NewRequest(t, "GET", "/user2/minimal/badges/pulls/closed.svg")
|
||
|
|
resp = MakeRequest(t, req, http.StatusSeeOther)
|
||
|
|
assertBadge(t, resp, "pulls-Not%20found-crimson")
|
||
|
|
})
|
||
|
|
|
||
|
|
t.Run("Release", func(t *testing.T) {
|
||
|
|
defer tests.PrintCurrentTest(t)()
|
||
|
|
defer createMinimalRepo(t)()
|
||
|
|
|
||
|
|
req := NewRequest(t, "GET", "/user2/repo1/badges/release.svg")
|
||
|
|
resp := MakeRequest(t, req, http.StatusSeeOther)
|
||
|
|
assertBadge(t, resp, "release-v1.1-blue")
|
||
|
|
|
||
|
|
req = NewRequest(t, "GET", "/user2/minimal/badges/release.svg")
|
||
|
|
resp = MakeRequest(t, req, http.StatusSeeOther)
|
||
|
|
assertBadge(t, resp, "release-Not%20found-crimson")
|
||
|
|
})
|
||
|
|
}
|