Have testing.RunTests run the tests we find with their setup methods.

pull/43/head
Sean Talts 2014-03-07 12:14:37 -05:00
parent aff4bd8fb5
commit 5739ba4d16
1 changed files with 21 additions and 7 deletions

View File

@ -1,9 +1,9 @@
package suite
import (
"testing"
"reflect"
"regexp"
"testing"
)
// Suite is a basic testing suite with methods for storing and
@ -32,19 +32,33 @@ func Run(t *testing.T, suite TestingSuite) {
}
methodFinder := reflect.TypeOf(suite)
tests := []testing.InternalTest{}
for index := 0; index < methodFinder.NumMethod(); index++ {
method := methodFinder.Method(index)
if ok, _ := regexp.MatchString("^Test", method.Name); ok {
if setupTestSuite, ok := suite.(SetupTestSuite); ok {
setupTestSuite.SetupTest()
}
method.Func.Call([]reflect.Value{reflect.ValueOf(suite)})
if tearDownTestSuite, ok := suite.(TearDownTestSuite); ok {
tearDownTestSuite.TearDownTest()
test := testing.InternalTest{
Name: method.Name,
F: func(t *testing.T) {
suite.SetT(t)
if setupTestSuite, ok := suite.(SetupTestSuite); ok {
setupTestSuite.SetupTest()
}
method.Func.Call([]reflect.Value{reflect.ValueOf(suite)})
if tearDownTestSuite, ok := suite.(TearDownTestSuite); ok {
tearDownTestSuite.TearDownTest()
}
},
}
tests = append(tests, test)
}
}
if !testing.RunTests(func(pat, str string) (bool, error) {
return true, nil
}, tests) {
t.Fail()
}
if tearDownAllSuite, ok := suite.(TearDownAllSuite); ok {
tearDownAllSuite.TearDownSuite()
}