forked from gitea/gitea
45 lines
1.3 KiB
Go
45 lines
1.3 KiB
Go
// Copyright 2017 The Gitea Authors. All rights reserved.
|
|
// Use of this source code is governed by a MIT-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package integrations
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"net/url"
|
|
"testing"
|
|
|
|
"code.gitea.io/gitea/models"
|
|
"code.gitea.io/gitea/modules/setting"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func assertProtectedBranch(t *testing.T, repoID int64, branchName string, isErr, canPush bool) {
|
|
reqURL := fmt.Sprintf("/api/internal/branch/%d/%s", repoID, url.QueryEscape(branchName))
|
|
req := NewRequest(t, "GET", reqURL)
|
|
t.Log(reqURL)
|
|
req.Header.Add("Authorization", fmt.Sprintf("Bearer %s", setting.InternalToken))
|
|
|
|
resp := MakeRequest(t, req, NoExpectedStatus)
|
|
if isErr {
|
|
assert.EqualValues(t, http.StatusInternalServerError, resp.Code)
|
|
} else {
|
|
assert.EqualValues(t, http.StatusOK, resp.Code)
|
|
var branch models.ProtectedBranch
|
|
t.Log(resp.Body.String())
|
|
assert.NoError(t, json.Unmarshal(resp.Body.Bytes(), &branch))
|
|
assert.Equal(t, canPush, !branch.IsProtected())
|
|
}
|
|
}
|
|
|
|
func TestInternal_GetProtectedBranch(t *testing.T) {
|
|
prepareTestEnv(t)
|
|
|
|
assertProtectedBranch(t, 1, "master", false, true)
|
|
assertProtectedBranch(t, 1, "dev", false, true)
|
|
assertProtectedBranch(t, 1, "lunny/dev", false, true)
|
|
}
|