1package tests
 2
 3import (
 4	"io/ioutil"
 5	"log"
 6	"testing"
 7
 8	"github.com/MichaelMure/git-bug/bug"
 9	"github.com/MichaelMure/git-bug/misc/random_bugs"
10	"github.com/MichaelMure/git-bug/repository"
11)
12
13func createRepo(bare bool) *repository.GitRepo {
14	dir, err := ioutil.TempDir("", "")
15	if err != nil {
16		log.Fatal(err)
17	}
18
19	// fmt.Println("Creating repo:", dir)
20
21	var creator func(string) (*repository.GitRepo, error)
22
23	if bare {
24		creator = repository.InitBareGitRepo
25	} else {
26		creator = repository.InitGitRepo
27	}
28
29	repo, err := creator(dir)
30	if err != nil {
31		log.Fatal(err)
32	}
33
34	if err := repo.StoreConfig("user.name", "testuser"); err != nil {
35		log.Fatal("failed to set user.name for test repository: ", err)
36	}
37	if err := repo.StoreConfig("user.email", "testuser@example.com"); err != nil {
38		log.Fatal("failed to set user.email for test repository: ", err)
39	}
40
41	return repo
42}
43
44func createFilledRepo(bugNumber int) repository.ClockedRepo {
45	repo := createRepo(false)
46
47	var seed int64 = 42
48	options := random_bugs.DefaultOptions()
49
50	options.BugNumber = bugNumber
51
52	random_bugs.CommitRandomBugsWithSeed(repo, options, seed)
53	return repo
54}
55
56func TestReadBugs(t *testing.T) {
57	repo := createFilledRepo(15)
58	bugs := bug.ReadAllLocalBugs(repo)
59	for b := range bugs {
60		if b.Err != nil {
61			t.Fatal(b.Err)
62		}
63	}
64}
65
66func benchmarkReadBugs(bugNumber int, t *testing.B) {
67	repo := createFilledRepo(bugNumber)
68	t.ResetTimer()
69
70	for n := 0; n < t.N; n++ {
71		bugs := bug.ReadAllLocalBugs(repo)
72		for b := range bugs {
73			if b.Err != nil {
74				t.Fatal(b.Err)
75			}
76		}
77	}
78}
79
80func BenchmarkReadBugs5(b *testing.B)   { benchmarkReadBugs(5, b) }
81func BenchmarkReadBugs25(b *testing.B)  { benchmarkReadBugs(25, b) }
82func BenchmarkReadBugs150(b *testing.B) { benchmarkReadBugs(150, b) }