forest/helpers_test.go at master · emicklei/forest · GitHub

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
package forest

import (
	"fmt"
	"os"
	"testing"
	"text/template"
)

func TestDump(t *testing.T) {
	r := tsAPI.GET(t, NewConfig("/jsonarray"))
	Dump(t, r)
	// check if we can read it again
	Dump(t, r)
}

func TestDumpTemplate(t *testing.T) {
	// backup and restore
	old := DumpTemplate
	defer func() {
		DumpTemplate = old
	}()
	DumpTemplate = template.Must(template.New("dump").Parse("Method:{{.Request.Method}}"))
	r := tsAPI.GET(t, NewConfig("/jsonarray"))
	mock := new(mockedT)
	Dump(mock, r)
	if "Method:GET" != mock.logMessage {
		t.Errorf("got %q want %q", mock.logMessage, "Method:GET")
	}
}

type skipper struct {
	*testing.T
	skipped bool
	reason  string
}

func (s *skipper) Skipf(f string, args ...interface{}) {
	s.skipped = true
	s.reason = fmt.Sprintf(f, args...)
}

func TestSkipUnless(t *testing.T) {
	os.Setenv("LABELS", "test,check")
	for _, each := range []struct {
		s      *skipper
		labels []string
		skip   bool
	}{
		{
			&skipper{t, false, ""},
			[]string{"any", "pass"},
			true,
		},
		{
			&skipper{t, false, ""},
			[]string{"test", "pass"},
			false,
		},
		{
			&skipper{t, false, ""},
			[]string{"any", "check"},
			false,
		},
		{
			&skipper{t, false, ""},
			[]string{},
			true,
		},
		{
			&skipper{t, false, ""},
			[]string{"check"},
			false,
		},
	} {
		SkipUnless(each.s, each.labels...)
		if each.s.skipped != each.skip {
			t.Errorf("unexpected skip")
		}
	}
}