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	return repo
35}
36
37func createFilledRepo(bugNumber int) repository.ClockedRepo {
38	repo := createRepo(false)
39
40	var seed int64 = 42
41	options := random_bugs.DefaultOptions()
42
43	options.BugNumber = bugNumber
44
45	random_bugs.CommitRandomBugsWithSeed(repo, options, seed)
46	return repo
47}
48
49func TestReadBugs(t *testing.T) {
50	repo := createFilledRepo(15)
51	bugs := bug.ReadAllLocalBugs(repo)
52	for b := range bugs {
53		if b.Err != nil {
54			t.Fatal(b.Err)
55		}
56	}
57}
58
59func benchmarkReadBugs(bugNumber int, t *testing.B) {
60	repo := createFilledRepo(bugNumber)
61	t.ResetTimer()
62
63	for n := 0; n < t.N; n++ {
64		bugs := bug.ReadAllLocalBugs(repo)
65		for b := range bugs {
66			if b.Err != nil {
67				t.Fatal(b.Err)
68			}
69		}
70	}
71}
72
73func BenchmarkReadBugs5(b *testing.B)   { benchmarkReadBugs(5, b) }
74func BenchmarkReadBugs25(b *testing.B)  { benchmarkReadBugs(25, b) }
75func BenchmarkReadBugs150(b *testing.B) { benchmarkReadBugs(150, b) }