summaryrefslogtreecommitdiff
path: root/vendor/github.com/onsi/gomega/internal/testingtsupport/testing_t_support_test.go
blob: 8fd8f0a6c27325aacf6a091a18220b1ed505b823 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
package testingtsupport_test

import (
	"regexp"
	"time"

	"github.com/onsi/gomega/internal/testingtsupport"

	. "github.com/onsi/gomega"

	"fmt"
	"testing"
)

func TestTestingT(t *testing.T) {
	RegisterTestingT(t)
	Ω(true).Should(BeTrue())
}

type FakeTWithHelper struct {
	LastFatal string
}

func (f *FakeTWithHelper) Fatalf(format string, args ...interface{}) {
	f.LastFatal = fmt.Sprintf(format, args...)
}

func TestGomegaWithTWithoutHelper(t *testing.T) {
	g := NewGomegaWithT(t)

	testingtsupport.StackTracePruneRE = regexp.MustCompile(`\/ginkgo\/`)

	f := &FakeTWithHelper{}
	testG := NewGomegaWithT(f)

	testG.Expect("foo").To(Equal("foo"))
	g.Expect(f.LastFatal).To(BeZero())

	testG.Expect("foo").To(Equal("bar"))
	g.Expect(f.LastFatal).To(ContainSubstring("<string>: foo"))
	g.Expect(f.LastFatal).To(ContainSubstring("testingtsupport_test"), "It should include a stacktrace")

	testG.Eventually("foo2", time.Millisecond).Should(Equal("bar"))
	g.Expect(f.LastFatal).To(ContainSubstring("<string>: foo2"))

	testG.Consistently("foo3", time.Millisecond).Should(Equal("bar"))
	g.Expect(f.LastFatal).To(ContainSubstring("<string>: foo3"))
}

type FakeTWithoutHelper struct {
	LastFatal   string
	HelperCount int
}

func (f *FakeTWithoutHelper) Fatalf(format string, args ...interface{}) {
	f.LastFatal = fmt.Sprintf(format, args...)
}

func (f *FakeTWithoutHelper) Helper() {
	f.HelperCount += 1
}

func (f *FakeTWithoutHelper) ResetHelper() {
	f.HelperCount = 0
}

func TestGomegaWithTWithHelper(t *testing.T) {
	g := NewGomegaWithT(t)

	f := &FakeTWithoutHelper{}
	testG := NewGomegaWithT(f)

	testG.Expect("foo").To(Equal("foo"))
	g.Expect(f.LastFatal).To(BeZero())
	g.Expect(f.HelperCount).To(BeNumerically(">", 0))
	f.ResetHelper()

	testG.Expect("foo").To(Equal("bar"))
	g.Expect(f.LastFatal).To(ContainSubstring("<string>: foo"))
	g.Expect(f.LastFatal).NotTo(ContainSubstring("testingtsupport_test"), "It should _not_ include a stacktrace")
	g.Expect(f.HelperCount).To(BeNumerically(">", 0))
	f.ResetHelper()

	testG.Eventually("foo2", time.Millisecond).Should(Equal("bar"))
	g.Expect(f.LastFatal).To(ContainSubstring("<string>: foo2"))
	g.Expect(f.HelperCount).To(BeNumerically(">", 0))
	f.ResetHelper()

	testG.Consistently("foo3", time.Millisecond).Should(Equal("bar"))
	g.Expect(f.LastFatal).To(ContainSubstring("<string>: foo3"))
	g.Expect(f.HelperCount).To(BeNumerically(">", 0))
}