mirror of
https://github.com/go-gitea/gitea.git
synced 2024-12-15 17:49:20 +08:00
9e6e1dc950
The current implementation of checkBranchName is highly inefficient involving opening the repository, the listing all of the branch names checking them individually before then using using opened repo to get the tags. This PR avoids this by simply walking the references from show-ref instead of opening the repository (in the nogogit case). Signed-off-by: Andrew Thornton <art27@cantab.net>
57 lines
1.4 KiB
Go
57 lines
1.4 KiB
Go
// Copyright 2018 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 git
|
|
|
|
import (
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestRepository_GetBranches(t *testing.T) {
|
|
bareRepo1Path := filepath.Join(testReposDir, "repo1_bare")
|
|
bareRepo1, err := OpenRepository(bareRepo1Path)
|
|
assert.NoError(t, err)
|
|
defer bareRepo1.Close()
|
|
|
|
branches, countAll, err := bareRepo1.GetBranchNames(0, 2)
|
|
|
|
assert.NoError(t, err)
|
|
assert.Len(t, branches, 2)
|
|
assert.EqualValues(t, 3, countAll)
|
|
assert.ElementsMatch(t, []string{"branch1", "branch2"}, branches)
|
|
|
|
branches, countAll, err = bareRepo1.GetBranchNames(0, 0)
|
|
|
|
assert.NoError(t, err)
|
|
assert.Len(t, branches, 3)
|
|
assert.EqualValues(t, 3, countAll)
|
|
assert.ElementsMatch(t, []string{"branch1", "branch2", "master"}, branches)
|
|
|
|
branches, countAll, err = bareRepo1.GetBranchNames(5, 1)
|
|
|
|
assert.NoError(t, err)
|
|
assert.Len(t, branches, 0)
|
|
assert.EqualValues(t, 3, countAll)
|
|
assert.ElementsMatch(t, []string{}, branches)
|
|
}
|
|
|
|
func BenchmarkRepository_GetBranches(b *testing.B) {
|
|
bareRepo1Path := filepath.Join(testReposDir, "repo1_bare")
|
|
bareRepo1, err := OpenRepository(bareRepo1Path)
|
|
if err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
defer bareRepo1.Close()
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
_, _, err := bareRepo1.GetBranchNames(0, 0)
|
|
if err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
}
|
|
}
|