1package reporters
 2
 3import (
 4	"github.com/onsi/ginkgo/config"
 5	"github.com/onsi/ginkgo/types"
 6)
 7
 8//FakeReporter is useful for testing purposes
 9type FakeReporter struct {
10	Config config.GinkgoConfigType
11
12	BeginSummary         *types.SuiteSummary
13	BeforeSuiteSummary   *types.SetupSummary
14	SpecWillRunSummaries []*types.SpecSummary
15	SpecSummaries        []*types.SpecSummary
16	AfterSuiteSummary    *types.SetupSummary
17	EndSummary           *types.SuiteSummary
18
19	SpecWillRunStub     func(specSummary *types.SpecSummary)
20	SpecDidCompleteStub func(specSummary *types.SpecSummary)
21}
22
23func NewFakeReporter() *FakeReporter {
24	return &FakeReporter{
25		SpecWillRunSummaries: make([]*types.SpecSummary, 0),
26		SpecSummaries:        make([]*types.SpecSummary, 0),
27	}
28}
29
30func (fakeR *FakeReporter) SpecSuiteWillBegin(config config.GinkgoConfigType, summary *types.SuiteSummary) {
31	fakeR.Config = config
32	fakeR.BeginSummary = summary
33}
34
35func (fakeR *FakeReporter) BeforeSuiteDidRun(setupSummary *types.SetupSummary) {
36	fakeR.BeforeSuiteSummary = setupSummary
37}
38
39func (fakeR *FakeReporter) SpecWillRun(specSummary *types.SpecSummary) {
40	if fakeR.SpecWillRunStub != nil {
41		fakeR.SpecWillRunStub(specSummary)
42	}
43	fakeR.SpecWillRunSummaries = append(fakeR.SpecWillRunSummaries, specSummary)
44}
45
46func (fakeR *FakeReporter) SpecDidComplete(specSummary *types.SpecSummary) {
47	if fakeR.SpecDidCompleteStub != nil {
48		fakeR.SpecDidCompleteStub(specSummary)
49	}
50	fakeR.SpecSummaries = append(fakeR.SpecSummaries, specSummary)
51}
52
53func (fakeR *FakeReporter) AfterSuiteDidRun(setupSummary *types.SetupSummary) {
54	fakeR.AfterSuiteSummary = setupSummary
55}
56
57func (fakeR *FakeReporter) SpecSuiteDidEnd(summary *types.SuiteSummary) {
58	fakeR.EndSummary = summary
59}