mirror of
https://github.com/hanzoai/deploy.git
synced 2026-10-07 08:44:50 +00:00
zeekayandHanzo Dev
1ab6e799c1
go.mod still declared `module github.com/hanzoai/deploy` after the repo was renamed. A Go module path is not cosmetic — it is the resolvable location. So `go get github.com/hanzoai/cd` could not work, and anything importing this module had to spell a name that no longer exists at that address. The build log of v3.6.0 is where it surfaced: the compiler was still emitting github.com/hanzoai/deploy/cmd/cd/commands. 3541 import paths across 690 .go, 19 .proto, 4 .yaml, 3 .sh, 2 .mod, Procfile and Makefile. Plus the 62 ghcr.io/hanzoai/deploy image references, moved to ghcr.io/hanzoai/cd to match hanzo.yml's build target — a distinct string from the module path, replaced separately so neither could clobber the other. go mod tidy clean, CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build ./... clean. Co-authored-by: Hanzo Dev <[email protected]>
549 lines
16 KiB
Go
549 lines
16 KiB
Go
package pull_request
|
|
|
|
import (
|
|
"crypto/x509"
|
|
"encoding/pem"
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/hanzoai/cd/pkg/apis/application/v1alpha1"
|
|
)
|
|
|
|
func defaultHandler(t *testing.T) func(http.ResponseWriter, *http.Request) {
|
|
t.Helper()
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
w.Header().Set("Content-Type", "application/json")
|
|
var err error
|
|
switch r.RequestURI {
|
|
case "/rest/api/1.0/projects/PROJECT/repos/REPO/pull-requests?limit=100":
|
|
_, err = io.WriteString(w, `{
|
|
"size": 1,
|
|
"limit": 100,
|
|
"isLastPage": true,
|
|
"values": [
|
|
{
|
|
"id": 101,
|
|
"title": "feat(ABC) : 123",
|
|
"toRef": {
|
|
"latestCommit": "5b766e3564a3453808f3cd3dd3f2e5fad8ef0e7a",
|
|
"displayId": "master",
|
|
"id": "refs/heads/master"
|
|
},
|
|
"fromRef": {
|
|
"id": "refs/heads/feature-ABC-123",
|
|
"displayId": "feature-ABC-123",
|
|
"latestCommit": "cb3cf2e4d1517c83e720d2585b9402dbef71f992"
|
|
},
|
|
"author": {
|
|
"user": {
|
|
"name": "testName"
|
|
}
|
|
}
|
|
}
|
|
],
|
|
"start": 0
|
|
}`)
|
|
default:
|
|
t.Fail()
|
|
}
|
|
if err != nil {
|
|
t.Fail()
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestListPullRequestNoAuth(t *testing.T) {
|
|
t.Parallel()
|
|
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
assert.Empty(t, r.Header.Get("Authorization"))
|
|
defaultHandler(t)(w, r)
|
|
}))
|
|
defer ts.Close()
|
|
svc, err := NewBitbucketServiceNoAuth(t.Context(), ts.URL, "PROJECT", "REPO", "", false, nil, "", "")
|
|
require.NoError(t, err)
|
|
pullRequests, err := ListPullRequests(t.Context(), svc, []v1alpha1.PullRequestGeneratorFilter{})
|
|
require.NoError(t, err)
|
|
assert.Len(t, pullRequests, 1)
|
|
assert.Equal(t, int64(101), pullRequests[0].Number)
|
|
assert.Equal(t, "feat(ABC) : 123", pullRequests[0].Title)
|
|
assert.Equal(t, "feature-ABC-123", pullRequests[0].Branch)
|
|
assert.Equal(t, "master", pullRequests[0].TargetBranch)
|
|
assert.Equal(t, "cb3cf2e4d1517c83e720d2585b9402dbef71f992", pullRequests[0].HeadSHA)
|
|
assert.Equal(t, "testName", pullRequests[0].Author)
|
|
}
|
|
|
|
func TestListPullRequestPagination(t *testing.T) {
|
|
t.Parallel()
|
|
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
w.Header().Set("Content-Type", "application/json")
|
|
var err error
|
|
switch r.RequestURI {
|
|
case "/rest/api/1.0/projects/PROJECT/repos/REPO/pull-requests?limit=100":
|
|
_, err = io.WriteString(w, `{
|
|
"size": 2,
|
|
"limit": 2,
|
|
"isLastPage": false,
|
|
"values": [
|
|
{
|
|
"id": 101,
|
|
"title": "feat(101)",
|
|
"toRef": {
|
|
"latestCommit": "5b766e3564a3453808f3cd3dd3f2e5fad8ef0e7a",
|
|
"displayId": "master",
|
|
"id": "refs/heads/master"
|
|
},
|
|
"fromRef": {
|
|
"id": "refs/heads/feature-101",
|
|
"displayId": "feature-101",
|
|
"latestCommit": "ab3cf2e4d1517c83e720d2585b9402dbef71f992"
|
|
},
|
|
"author": {
|
|
"user": {
|
|
"name": "testName"
|
|
}
|
|
}
|
|
},
|
|
{
|
|
"id": 102,
|
|
"title": "feat(102)",
|
|
"toRef": {
|
|
"latestCommit": "5b766e3564a3453808f3cd3dd3f2e5fad8ef0e7a",
|
|
"displayId": "branch",
|
|
"id": "refs/heads/branch"
|
|
},
|
|
"fromRef": {
|
|
"id": "refs/heads/feature-102",
|
|
"displayId": "feature-102",
|
|
"latestCommit": "bb3cf2e4d1517c83e720d2585b9402dbef71f992"
|
|
},
|
|
"author": {
|
|
"user": {
|
|
"name": "testName"
|
|
}
|
|
}
|
|
}
|
|
],
|
|
"nextPageStart": 200
|
|
}`)
|
|
case "/rest/api/1.0/projects/PROJECT/repos/REPO/pull-requests?limit=100&start=200":
|
|
_, err = io.WriteString(w, `{
|
|
"size": 1,
|
|
"limit": 2,
|
|
"isLastPage": true,
|
|
"values": [
|
|
{
|
|
"id": 200,
|
|
"title": "feat(200)",
|
|
"toRef": {
|
|
"latestCommit": "5b766e3564a3453808f3cd3dd3f2e5fad8ef0e7a",
|
|
"displayId": "master",
|
|
"id": "refs/heads/master"
|
|
},
|
|
"fromRef": {
|
|
"id": "refs/heads/feature-200",
|
|
"displayId": "feature-200",
|
|
"latestCommit": "cb3cf2e4d1517c83e720d2585b9402dbef71f992"
|
|
},
|
|
"author": {
|
|
"user": {
|
|
"name": "testName"
|
|
}
|
|
}
|
|
}
|
|
],
|
|
"start": 200
|
|
}`)
|
|
default:
|
|
t.Fail()
|
|
}
|
|
if err != nil {
|
|
t.Fail()
|
|
}
|
|
}))
|
|
defer ts.Close()
|
|
svc, err := NewBitbucketServiceNoAuth(t.Context(), ts.URL, "PROJECT", "REPO", "", false, nil, "", "")
|
|
require.NoError(t, err)
|
|
pullRequests, err := ListPullRequests(t.Context(), svc, []v1alpha1.PullRequestGeneratorFilter{})
|
|
require.NoError(t, err)
|
|
assert.Len(t, pullRequests, 3)
|
|
assert.Equal(t, PullRequest{
|
|
Number: 101,
|
|
Title: "feat(101)",
|
|
Branch: "feature-101",
|
|
TargetBranch: "master",
|
|
HeadSHA: "ab3cf2e4d1517c83e720d2585b9402dbef71f992",
|
|
Labels: []string{},
|
|
Author: "testName",
|
|
}, *pullRequests[0])
|
|
assert.Equal(t, PullRequest{
|
|
Number: 102,
|
|
Title: "feat(102)",
|
|
Branch: "feature-102",
|
|
TargetBranch: "branch",
|
|
HeadSHA: "bb3cf2e4d1517c83e720d2585b9402dbef71f992",
|
|
Labels: []string{},
|
|
Author: "testName",
|
|
}, *pullRequests[1])
|
|
assert.Equal(t, PullRequest{
|
|
Number: 200,
|
|
Title: "feat(200)",
|
|
Branch: "feature-200",
|
|
TargetBranch: "master",
|
|
HeadSHA: "cb3cf2e4d1517c83e720d2585b9402dbef71f992",
|
|
Labels: []string{},
|
|
Author: "testName",
|
|
}, *pullRequests[2])
|
|
}
|
|
|
|
func TestListPullRequestBasicAuth(t *testing.T) {
|
|
t.Parallel()
|
|
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
// base64(user:password)
|
|
assert.Equal(t, "Basic dXNlcjpwYXNzd29yZA==", r.Header.Get("Authorization"))
|
|
assert.Equal(t, "no-check", r.Header.Get("X-Atlassian-Token"))
|
|
defaultHandler(t)(w, r)
|
|
}))
|
|
defer ts.Close()
|
|
svc, err := NewBitbucketServiceBasicAuth(t.Context(), "user", "password", ts.URL, "PROJECT", "REPO", "", false, nil, "", "")
|
|
require.NoError(t, err)
|
|
pullRequests, err := ListPullRequests(t.Context(), svc, []v1alpha1.PullRequestGeneratorFilter{})
|
|
require.NoError(t, err)
|
|
assert.Len(t, pullRequests, 1)
|
|
assert.Equal(t, int64(101), pullRequests[0].Number)
|
|
assert.Equal(t, "feature-ABC-123", pullRequests[0].Branch)
|
|
assert.Equal(t, "cb3cf2e4d1517c83e720d2585b9402dbef71f992", pullRequests[0].HeadSHA)
|
|
}
|
|
|
|
func TestListPullRequestBearerAuth(t *testing.T) {
|
|
t.Parallel()
|
|
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
assert.Equal(t, "Bearer tolkien", r.Header.Get("Authorization"))
|
|
assert.Equal(t, "no-check", r.Header.Get("X-Atlassian-Token"))
|
|
defaultHandler(t)(w, r)
|
|
}))
|
|
defer ts.Close()
|
|
svc, err := NewBitbucketServiceBearerToken(t.Context(), "tolkien", ts.URL, "PROJECT", "REPO", "", false, nil, "", "")
|
|
require.NoError(t, err)
|
|
pullRequests, err := ListPullRequests(t.Context(), svc, []v1alpha1.PullRequestGeneratorFilter{})
|
|
require.NoError(t, err)
|
|
assert.Len(t, pullRequests, 1)
|
|
assert.Equal(t, int64(101), pullRequests[0].Number)
|
|
assert.Equal(t, "feat(ABC) : 123", pullRequests[0].Title)
|
|
assert.Equal(t, "feature-ABC-123", pullRequests[0].Branch)
|
|
assert.Equal(t, "cb3cf2e4d1517c83e720d2585b9402dbef71f992", pullRequests[0].HeadSHA)
|
|
}
|
|
|
|
func TestListPullRequestTLS(t *testing.T) {
|
|
t.Parallel()
|
|
tests := []struct {
|
|
name string
|
|
tlsInsecure bool
|
|
passCerts bool
|
|
requireErr bool
|
|
}{
|
|
{
|
|
name: "TLS Insecure: true, No Certs",
|
|
tlsInsecure: true,
|
|
passCerts: false,
|
|
requireErr: false,
|
|
},
|
|
{
|
|
name: "TLS Insecure: true, With Certs",
|
|
tlsInsecure: true,
|
|
passCerts: true,
|
|
requireErr: false,
|
|
},
|
|
{
|
|
name: "TLS Insecure: false, With Certs",
|
|
tlsInsecure: false,
|
|
passCerts: true,
|
|
requireErr: false,
|
|
},
|
|
{
|
|
name: "TLS Insecure: false, No Certs",
|
|
tlsInsecure: false,
|
|
passCerts: false,
|
|
requireErr: true,
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
t.Run(test.name, func(t *testing.T) {
|
|
t.Parallel()
|
|
ts := httptest.NewTLSServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
defaultHandler(t)(w, r)
|
|
}))
|
|
defer ts.Close()
|
|
|
|
var certs []byte
|
|
if test.passCerts {
|
|
for _, cert := range ts.TLS.Certificates {
|
|
for _, c := range cert.Certificate {
|
|
parsedCert, err := x509.ParseCertificate(c)
|
|
require.NoError(t, err, "Failed to parse certificate")
|
|
certs = append(certs, pem.EncodeToMemory(&pem.Block{
|
|
Type: "CERTIFICATE",
|
|
Bytes: parsedCert.Raw,
|
|
})...)
|
|
}
|
|
}
|
|
}
|
|
|
|
svc, err := NewBitbucketServiceBasicAuth(t.Context(), "user", "password", ts.URL, "PROJECT", "REPO", "", test.tlsInsecure, certs, "", "")
|
|
require.NoError(t, err)
|
|
_, err = ListPullRequests(t.Context(), svc, []v1alpha1.PullRequestGeneratorFilter{})
|
|
if test.requireErr {
|
|
require.Error(t, err)
|
|
} else {
|
|
require.NoError(t, err)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestListResponseError(t *testing.T) {
|
|
t.Parallel()
|
|
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
}))
|
|
defer ts.Close()
|
|
svc, _ := NewBitbucketServiceNoAuth(t.Context(), ts.URL, "PROJECT", "REPO", "", false, nil, "", "")
|
|
_, err := ListPullRequests(t.Context(), svc, []v1alpha1.PullRequestGeneratorFilter{})
|
|
require.Error(t, err)
|
|
}
|
|
|
|
func TestListResponseMalformed(t *testing.T) {
|
|
t.Parallel()
|
|
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
w.Header().Set("Content-Type", "application/json")
|
|
switch r.RequestURI {
|
|
case "/rest/api/1.0/projects/PROJECT/repos/REPO/pull-requests?limit=100":
|
|
_, err := io.WriteString(w, `{
|
|
"size": 1,
|
|
"limit": 100,
|
|
"isLastPage": true,
|
|
"values": { "id": 101 },
|
|
"start": 0
|
|
}`)
|
|
if err != nil {
|
|
t.Fail()
|
|
}
|
|
default:
|
|
t.Fail()
|
|
}
|
|
}))
|
|
defer ts.Close()
|
|
svc, _ := NewBitbucketServiceNoAuth(t.Context(), ts.URL, "PROJECT", "REPO", "", false, nil, "", "")
|
|
_, err := ListPullRequests(t.Context(), svc, []v1alpha1.PullRequestGeneratorFilter{})
|
|
require.Error(t, err)
|
|
}
|
|
|
|
func TestListResponseEmpty(t *testing.T) {
|
|
t.Parallel()
|
|
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
w.Header().Set("Content-Type", "application/json")
|
|
switch r.RequestURI {
|
|
case "/rest/api/1.0/projects/PROJECT/repos/REPO/pull-requests?limit=100":
|
|
_, err := io.WriteString(w, `{
|
|
"size": 0,
|
|
"limit": 100,
|
|
"isLastPage": true,
|
|
"values": [],
|
|
"start": 0
|
|
}`)
|
|
if err != nil {
|
|
t.Fail()
|
|
}
|
|
default:
|
|
t.Fail()
|
|
}
|
|
}))
|
|
defer ts.Close()
|
|
svc, err := NewBitbucketServiceNoAuth(t.Context(), ts.URL, "PROJECT", "REPO", "", false, nil, "", "")
|
|
require.NoError(t, err)
|
|
pullRequests, err := ListPullRequests(t.Context(), svc, []v1alpha1.PullRequestGeneratorFilter{})
|
|
require.NoError(t, err)
|
|
assert.Empty(t, pullRequests)
|
|
}
|
|
|
|
func TestListPullRequestBranchMatch(t *testing.T) {
|
|
t.Parallel()
|
|
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
w.Header().Set("Content-Type", "application/json")
|
|
var err error
|
|
switch r.RequestURI {
|
|
case "/rest/api/1.0/projects/PROJECT/repos/REPO/pull-requests?limit=100":
|
|
_, err = io.WriteString(w, `{
|
|
"size": 2,
|
|
"limit": 2,
|
|
"isLastPage": false,
|
|
"values": [
|
|
{
|
|
"id": 101,
|
|
"title": "feat(101)",
|
|
"toRef": {
|
|
"latestCommit": "5b766e3564a3453808f3cd3dd3f2e5fad8ef0e7a",
|
|
"displayId": "master",
|
|
"id": "refs/heads/master"
|
|
},
|
|
"fromRef": {
|
|
"id": "refs/heads/feature-101",
|
|
"displayId": "feature-101",
|
|
"latestCommit": "ab3cf2e4d1517c83e720d2585b9402dbef71f992"
|
|
},
|
|
"author": {
|
|
"user": {
|
|
"name": "testName"
|
|
}
|
|
}
|
|
},
|
|
{
|
|
"id": 102,
|
|
"title": "feat(102)",
|
|
"toRef": {
|
|
"latestCommit": "5b766e3564a3453808f3cd3dd3f2e5fad8ef0e7a",
|
|
"displayId": "branch",
|
|
"id": "refs/heads/branch"
|
|
},
|
|
"fromRef": {
|
|
"id": "refs/heads/feature-102",
|
|
"displayId": "feature-102",
|
|
"latestCommit": "bb3cf2e4d1517c83e720d2585b9402dbef71f992"
|
|
},
|
|
"author": {
|
|
"user": {
|
|
"name": "testName"
|
|
}
|
|
}
|
|
}
|
|
],
|
|
"nextPageStart": 200
|
|
}`)
|
|
case "/rest/api/1.0/projects/PROJECT/repos/REPO/pull-requests?limit=100&start=200":
|
|
_, err = io.WriteString(w, `{
|
|
"size": 1,
|
|
"limit": 2,
|
|
"isLastPage": true,
|
|
"values": [
|
|
{
|
|
"id": 200,
|
|
"title": "feat(200)",
|
|
"toRef": {
|
|
"latestCommit": "5b766e3564a3453808f3cd3dd3f2e5fad8ef0e7a",
|
|
"displayId": "master",
|
|
"id": "refs/heads/master"
|
|
},
|
|
"fromRef": {
|
|
"id": "refs/heads/feature-200",
|
|
"displayId": "feature-200",
|
|
"latestCommit": "cb3cf2e4d1517c83e720d2585b9402dbef71f992"
|
|
},
|
|
"author": {
|
|
"user": {
|
|
"name": "testName"
|
|
}
|
|
}
|
|
}
|
|
],
|
|
"start": 200
|
|
}`)
|
|
default:
|
|
t.Fail()
|
|
}
|
|
if err != nil {
|
|
t.Fail()
|
|
}
|
|
}))
|
|
defer ts.Close()
|
|
regexp := `feature-1[\d]{2}`
|
|
svc, err := NewBitbucketServiceNoAuth(t.Context(), ts.URL, "PROJECT", "REPO", "", false, nil, "", "")
|
|
require.NoError(t, err)
|
|
pullRequests, err := ListPullRequests(t.Context(), svc, []v1alpha1.PullRequestGeneratorFilter{
|
|
{
|
|
BranchMatch: ®exp,
|
|
},
|
|
})
|
|
require.NoError(t, err)
|
|
assert.Len(t, pullRequests, 2)
|
|
assert.Equal(t, PullRequest{
|
|
Number: 101,
|
|
Title: "feat(101)",
|
|
Branch: "feature-101",
|
|
TargetBranch: "master",
|
|
HeadSHA: "ab3cf2e4d1517c83e720d2585b9402dbef71f992",
|
|
Labels: []string{},
|
|
Author: "testName",
|
|
}, *pullRequests[0])
|
|
assert.Equal(t, PullRequest{
|
|
Number: 102,
|
|
Title: "feat(102)",
|
|
Branch: "feature-102",
|
|
TargetBranch: "branch",
|
|
HeadSHA: "bb3cf2e4d1517c83e720d2585b9402dbef71f992",
|
|
Labels: []string{},
|
|
Author: "testName",
|
|
}, *pullRequests[1])
|
|
|
|
regexp = `.*2$`
|
|
svc, err = NewBitbucketServiceNoAuth(t.Context(), ts.URL, "PROJECT", "REPO", "", false, nil, "", "")
|
|
require.NoError(t, err)
|
|
pullRequests, err = ListPullRequests(t.Context(), svc, []v1alpha1.PullRequestGeneratorFilter{
|
|
{
|
|
BranchMatch: ®exp,
|
|
},
|
|
})
|
|
require.NoError(t, err)
|
|
assert.Len(t, pullRequests, 1)
|
|
assert.Equal(t, PullRequest{
|
|
Number: 102,
|
|
Title: "feat(102)",
|
|
Branch: "feature-102",
|
|
TargetBranch: "branch",
|
|
HeadSHA: "bb3cf2e4d1517c83e720d2585b9402dbef71f992",
|
|
Labels: []string{},
|
|
Author: "testName",
|
|
}, *pullRequests[0])
|
|
|
|
regexp = `[\d{2}`
|
|
svc, err = NewBitbucketServiceNoAuth(t.Context(), ts.URL, "PROJECT", "REPO", "", false, nil, "", "")
|
|
require.NoError(t, err)
|
|
_, err = ListPullRequests(t.Context(), svc, []v1alpha1.PullRequestGeneratorFilter{
|
|
{
|
|
BranchMatch: ®exp,
|
|
},
|
|
})
|
|
require.Error(t, err)
|
|
}
|
|
|
|
func TestBitbucketServerListReturnsRepositoryNotFoundError(t *testing.T) {
|
|
t.Parallel()
|
|
mux := http.NewServeMux()
|
|
server := httptest.NewServer(mux)
|
|
defer server.Close()
|
|
|
|
path := "/rest/api/1.0/projects/nonexistent/repos/nonexistent/pull-requests?limit=100"
|
|
|
|
mux.HandleFunc(path, func(w http.ResponseWriter, _ *http.Request) {
|
|
// Return 404 status to simulate repository not found
|
|
w.WriteHeader(http.StatusNotFound)
|
|
_, _ = w.Write([]byte(`{"message": "404 Project Not Found"}`))
|
|
})
|
|
|
|
svc, err := NewBitbucketServiceNoAuth(t.Context(), server.URL, "nonexistent", "nonexistent", "", false, nil, "", "")
|
|
require.NoError(t, err)
|
|
|
|
prs, err := svc.List(t.Context())
|
|
|
|
// Should return empty pull requests list
|
|
assert.Empty(t, prs)
|
|
|
|
// Should return RepositoryNotFoundError
|
|
require.Error(t, err)
|
|
assert.True(t, IsRepositoryNotFoundError(err), "Expected RepositoryNotFoundError but got: %v", err)
|
|
}
|