forked from grafana/grafana
-
Notifications
You must be signed in to change notification settings - Fork 0
/
test_info.go
100 lines (83 loc) · 2.95 KB
/
test_info.go
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
93
94
95
96
97
98
99
100
// Copyright 2011 Aaron Jacobs. All Rights Reserved.
// Author: aaronjjacobs@gmail.com (Aaron Jacobs)
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package ogletest
import (
"sync"
"github.com/smartystreets/goconvey/convey/assertions/oglemock"
)
// TestInfo represents information about a currently running or previously-run
// test.
type TestInfo struct {
// A mock controller that is set up to report errors to the ogletest test
// runner. This can be used for setting up mock expectations and handling
// mock calls. The Finish method should not be run by the user; ogletest will
// do that automatically after the test's TearDown method is run.
//
// Note that this feature is still experimental, and is subject to change.
MockController oglemock.Controller
// A mutex protecting shared state.
mutex sync.RWMutex
// A set of failure records that the test has produced.
failureRecords []*failureRecord // Protected by mutex
}
// currentlyRunningTest is the state for the currently running test, if any.
var currentlyRunningTest *TestInfo
// newTestInfo creates a valid but empty TestInfo struct.
func newTestInfo() *TestInfo {
info := &TestInfo{}
info.failureRecords = make([]*failureRecord, 0)
info.MockController = oglemock.NewController(&testInfoErrorReporter{info})
return info
}
// failureRecord represents a single failed expectation for a test.
type failureRecord struct {
// The file name within which the expectation failed, e.g. "foo_test.go".
FileName string
// The line number at which the expectation failed.
LineNumber int
// The error generated by the testing framework. For example:
//
// Expected: 17
// Actual: "taco", which is not numeric
//
GeneratedError string
// A user-specified string to print out with the error, if any.
UserError string
}
// testInfoErrorReporter is an oglemock.ErrorReporter that writes failure
// records into a test info struct.
type testInfoErrorReporter struct {
testInfo *TestInfo
}
func (r *testInfoErrorReporter) ReportError(
fileName string,
lineNumber int,
err error) {
r.testInfo.mutex.Lock()
defer r.testInfo.mutex.Unlock()
record := &failureRecord{
FileName: fileName,
LineNumber: lineNumber,
GeneratedError: err.Error(),
}
r.testInfo.failureRecords = append(r.testInfo.failureRecords, record)
}
func (r *testInfoErrorReporter) ReportFatalError(
fileName string,
lineNumber int,
err error) {
r.ReportError(fileName, lineNumber, err)
panic(&assertThatError{})
}