read_bugs_test.go

 1package tests
 2
 3import (
 4	"testing"
 5
 6	"github.com/MichaelMure/git-bug/bug"
 7	"github.com/MichaelMure/git-bug/misc/random_bugs"
 8	"github.com/MichaelMure/git-bug/repository"
 9)
10
11func createFilledRepo(bugNumber int) repository.Repo {
12	repo := createRepo(false)
13
14	var seed int64 = 42
15	options := random_bugs.DefaultOptions()
16
17	options.BugNumber = bugNumber
18
19	random_bugs.CommitRandomBugsWithSeed(repo, options, seed)
20	return repo
21}
22
23func TestReadBugs(t *testing.T) {
24	repo := createFilledRepo(15)
25	bugs := bug.ReadAllLocalBugs(repo)
26	for b := range bugs {
27		if b.Err != nil {
28			t.Fatal(b.Err)
29		}
30	}
31}
32
33func benchmarkReadBugs(bugNumber int, t *testing.B) {
34	repo := createFilledRepo(bugNumber)
35	t.ResetTimer()
36
37	for n := 0; n < t.N; n++ {
38		bugs := bug.ReadAllLocalBugs(repo)
39		for b := range bugs {
40			if b.Err != nil {
41				t.Fatal(b.Err)
42			}
43		}
44	}
45}
46
47func BenchmarkReadBugs5(b *testing.B)   { benchmarkReadBugs(5, b) }
48func BenchmarkReadBugs25(b *testing.B)  { benchmarkReadBugs(25, b) }
49func BenchmarkReadBugs150(b *testing.B) { benchmarkReadBugs(150, b) }