From 4cbfd383e9df44295a5eb404619c7fd1e1780520 Mon Sep 17 00:00:00 2001
From: Earl Warren <contact@earl-warren.org>
Date: Wed, 29 May 2024 17:17:58 +0200
Subject: [PATCH] tests(api): POST /repos/{owner}/{repo}/push_mirrors coverage

(cherry picked from commit 166bb2861f91d2ac5d83906dd178d0cef3f1adbc)
---
 models/fixtures/push_mirror.yml           |   1 +
 tests/integration/api_push_mirror_test.go | 130 ++++++++++++++++++++++
 2 files changed, 131 insertions(+)
 create mode 100644 models/fixtures/push_mirror.yml
 create mode 100644 tests/integration/api_push_mirror_test.go

diff --git a/models/fixtures/push_mirror.yml b/models/fixtures/push_mirror.yml
new file mode 100644
index 0000000000..ca780a73aa
--- /dev/null
+++ b/models/fixtures/push_mirror.yml
@@ -0,0 +1 @@
+[] # empty
diff --git a/tests/integration/api_push_mirror_test.go b/tests/integration/api_push_mirror_test.go
new file mode 100644
index 0000000000..9b0497aa3b
--- /dev/null
+++ b/tests/integration/api_push_mirror_test.go
@@ -0,0 +1,130 @@
+// Copyright The Forgejo Authors
+// SPDX-License-Identifier: MIT
+
+package integration
+
+import (
+	"context"
+	"fmt"
+	"net/http"
+	"net/url"
+	"testing"
+
+	auth_model "code.gitea.io/gitea/models/auth"
+	"code.gitea.io/gitea/models/db"
+	repo_model "code.gitea.io/gitea/models/repo"
+	"code.gitea.io/gitea/models/unittest"
+	user_model "code.gitea.io/gitea/models/user"
+	"code.gitea.io/gitea/modules/setting"
+	api "code.gitea.io/gitea/modules/structs"
+	"code.gitea.io/gitea/modules/test"
+	"code.gitea.io/gitea/services/migrations"
+	mirror_service "code.gitea.io/gitea/services/mirror"
+	repo_service "code.gitea.io/gitea/services/repository"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestAPIPushMirror(t *testing.T) {
+	onGiteaRun(t, testAPIPushMirror)
+}
+
+func testAPIPushMirror(t *testing.T, u *url.URL) {
+	defer test.MockVariableValue(&setting.Migrations.AllowLocalNetworks, true)()
+	defer test.MockVariableValue(&setting.Mirror.Enabled, true)()
+	defer test.MockProtect(&mirror_service.AddPushMirrorRemote)()
+	defer test.MockProtect(&repo_model.DeletePushMirrors)()
+
+	assert.NoError(t, migrations.Init())
+
+	user := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 1})
+	srcRepo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 1})
+	owner := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: srcRepo.OwnerID})
+	session := loginUser(t, user.Name)
+	token := getTokenForLoggedInUser(t, session, auth_model.AccessTokenScopeAll)
+	urlStr := fmt.Sprintf("/api/v1/repos/%s/%s/push_mirrors", owner.Name, srcRepo.Name)
+
+	mirrorRepo, err := repo_service.CreateRepositoryDirectly(db.DefaultContext, user, user, repo_service.CreateRepoOptions{
+		Name: "test-push-mirror",
+	})
+	assert.NoError(t, err)
+	remoteAddress := fmt.Sprintf("%s%s/%s", u.String(), url.PathEscape(user.Name), url.PathEscape(mirrorRepo.Name))
+
+	deletePushMirrors := repo_model.DeletePushMirrors
+	deletePushMirrorsError := "deletePushMirrorsError"
+	deletePushMirrorsFail := func(ctx context.Context, opts repo_model.PushMirrorOptions) error {
+		return fmt.Errorf(deletePushMirrorsError)
+	}
+
+	addPushMirrorRemote := mirror_service.AddPushMirrorRemote
+	addPushMirrorRemoteError := "addPushMirrorRemoteError"
+	addPushMirrorRemoteFail := func(ctx context.Context, m *repo_model.PushMirror, addr string) error {
+		return fmt.Errorf(addPushMirrorRemoteError)
+	}
+
+	for _, testCase := range []struct {
+		name        string
+		message     string
+		status      int
+		mirrorCount int
+		setup       func()
+	}{
+		{
+			name:        "success",
+			status:      http.StatusOK,
+			mirrorCount: 1,
+			setup: func() {
+				mirror_service.AddPushMirrorRemote = addPushMirrorRemote
+				repo_model.DeletePushMirrors = deletePushMirrors
+			},
+		},
+		{
+			name:        "fail to add and delete",
+			message:     deletePushMirrorsError,
+			status:      http.StatusInternalServerError,
+			mirrorCount: 1,
+			setup: func() {
+				mirror_service.AddPushMirrorRemote = addPushMirrorRemoteFail
+				repo_model.DeletePushMirrors = deletePushMirrorsFail
+			},
+		},
+		{
+			name:        "fail to add",
+			message:     addPushMirrorRemoteError,
+			status:      http.StatusInternalServerError,
+			mirrorCount: 0,
+			setup: func() {
+				mirror_service.AddPushMirrorRemote = addPushMirrorRemoteFail
+				repo_model.DeletePushMirrors = deletePushMirrors
+			},
+		},
+	} {
+		t.Run(testCase.name, func(t *testing.T) {
+			testCase.setup()
+			req := NewRequestWithJSON(t, "POST", urlStr, &api.CreatePushMirrorOption{
+				RemoteAddress: remoteAddress,
+				Interval:      "8h",
+			}).AddTokenAuth(token)
+
+			resp := MakeRequest(t, req, testCase.status)
+			if testCase.message != "" {
+				err := api.APIError{}
+				DecodeJSON(t, resp, &err)
+				assert.EqualValues(t, testCase.message, err.Message)
+			}
+
+			req = NewRequest(t, "GET", urlStr).AddTokenAuth(token)
+			resp = MakeRequest(t, req, http.StatusOK)
+			var pushMirrors []*api.PushMirror
+			DecodeJSON(t, resp, &pushMirrors)
+			if assert.Len(t, pushMirrors, testCase.mirrorCount) && testCase.mirrorCount > 0 {
+				pushMirror := pushMirrors[0]
+				assert.EqualValues(t, remoteAddress, pushMirror.RemoteAddress)
+
+				repo_model.DeletePushMirrors = deletePushMirrors
+				req = NewRequest(t, "DELETE", fmt.Sprintf("%s/%s", urlStr, pushMirror.RemoteName)).AddTokenAuth(token)
+				MakeRequest(t, req, http.StatusNoContent)
+			}
+		})
+	}
+}