Create tree_entry_test.go

Author: sapk

tree_entry_test.go

@@ -0,0 +1,66 @@
+// 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 git
+
+import (
+	"io/ioutil"
+	"os"
+	"testing"
+)
+
+func setupGitRepo(url string) string {
+	
+	dir, err := ioutil.TempDir("", "gitea-bench")
+	if err != nil {
+		panic(err)
+	}
+	/* Manual method
+	_, err = NewCommand("clone", url, dir).Run()
+	if err != nil {
+		log.Fatal(err)
+	}
+	*/
+	err = Clone(url, dir, CloneRepoOptions{Mirror: false, Bare: false, Quiet: true})
+	if err != nil {
+		panic(err)
+	}
+	return dir
+}
+
+func benchmarkGetCommitsInfo(url string, b *testing.B) {
+
+	// setup env
+	repoPath := setupGitRepo(url)
+	defer os.RemoveAll(repoPath)
+
+	repo, err := OpenRepository(repoPath)
+	if err != nil {
+		panic(err)
+	}
+
+	commit, err := repo.GetBranchCommit("master")
+	if err != nil {
+		panic(err)
+	}
+
+	entries, err := commit.Tree.ListEntries()
+	if err != nil {
+		panic(err)
+	}
+	entries.Sort()
+
+	// run the GetCommitsInfo function b.N times
+	for n := 0; n < b.N; n++ {
+		_, err = entries.GetCommitsInfo(commit, "")
+		if err != nil {
+			panic(err)
+		}
+	}
+}
+
+
+func BenchmarkGetCommitsInfoGitea(b *testing.B)  { benchmarkGetCommitsInfo("https://github.com/go-gitea/gitea.git", b) } //5k+ commits
+func BenchmarkGetCommitsInfoMoby(b *testing.B)  { benchmarkGetCommitsInfo("https://github.com/moby/moby.git", b) } //32k+ commits
+func BenchmarkGetCommitsInfoGo(b *testing.B)  { benchmarkGetCommitsInfo("https://github.com/golang/go.git", b) } //+32k commits