Matthias Andreas Benkard | 832a54e | 2019-01-29 09:27:38 +0100 | [diff] [blame] | 1 | package reporters |
| 2 | |
| 3 | import ( |
| 4 | "github.com/onsi/ginkgo/config" |
| 5 | "github.com/onsi/ginkgo/types" |
| 6 | ) |
| 7 | |
| 8 | //FakeReporter is useful for testing purposes |
| 9 | type 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 | |
| 23 | func NewFakeReporter() *FakeReporter { |
| 24 | return &FakeReporter{ |
| 25 | SpecWillRunSummaries: make([]*types.SpecSummary, 0), |
| 26 | SpecSummaries: make([]*types.SpecSummary, 0), |
| 27 | } |
| 28 | } |
| 29 | |
| 30 | func (fakeR *FakeReporter) SpecSuiteWillBegin(config config.GinkgoConfigType, summary *types.SuiteSummary) { |
| 31 | fakeR.Config = config |
| 32 | fakeR.BeginSummary = summary |
| 33 | } |
| 34 | |
| 35 | func (fakeR *FakeReporter) BeforeSuiteDidRun(setupSummary *types.SetupSummary) { |
| 36 | fakeR.BeforeSuiteSummary = setupSummary |
| 37 | } |
| 38 | |
| 39 | func (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 | |
| 46 | func (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 | |
| 53 | func (fakeR *FakeReporter) AfterSuiteDidRun(setupSummary *types.SetupSummary) { |
| 54 | fakeR.AfterSuiteSummary = setupSummary |
| 55 | } |
| 56 | |
| 57 | func (fakeR *FakeReporter) SpecSuiteDidEnd(summary *types.SuiteSummary) { |
| 58 | fakeR.EndSummary = summary |
| 59 | } |