0c64171e85
This allows the use of a built-in provider to do things like mark a node as ready once all the controllers are spun up. The e2e tests now use this instead of waiting on the pod that the vk provider is deployed in to be marked ready (this was waiting on /stats/summary to be serving, which is racey).
127 lines
3.2 KiB
Go
127 lines
3.2 KiB
Go
package suite
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"gotest.tools/assert"
|
|
is "gotest.tools/assert/cmp"
|
|
)
|
|
|
|
type basicTestSuite struct {
|
|
setupCount int
|
|
testFooCount int
|
|
testBarCount int
|
|
bazCount int
|
|
testFooBarCount int
|
|
testFooBazCount int
|
|
testBarBazCount int
|
|
teardownCount int
|
|
testsRan []string
|
|
}
|
|
|
|
func (bts *basicTestSuite) Setup(t *testing.T) {
|
|
bts.setupCount++
|
|
}
|
|
|
|
func (bts *basicTestSuite) Teardown() {
|
|
bts.teardownCount++
|
|
}
|
|
|
|
func (bts *basicTestSuite) TestFoo(t *testing.T) {
|
|
bts.testFooCount++
|
|
bts.testsRan = append(bts.testsRan, t.Name())
|
|
}
|
|
|
|
func (bts *basicTestSuite) TestBar(t *testing.T) {
|
|
bts.testBarCount++
|
|
bts.testsRan = append(bts.testsRan, t.Name())
|
|
}
|
|
|
|
// Baz should not be executed by the test suite
|
|
// because it does not have the prefix 'Test'
|
|
func (bts *basicTestSuite) Baz(t *testing.T) {
|
|
bts.bazCount++
|
|
bts.testsRan = append(bts.testsRan, t.Name())
|
|
}
|
|
|
|
// TestFooBar should not be executed by the test suite
|
|
// because the number of function input is not 2 (*basicTestSuite and *testing.T)
|
|
func (bts *basicTestSuite) TestFooBar() {
|
|
bts.testFooBarCount++
|
|
bts.testsRan = append(bts.testsRan, "TestFooBar")
|
|
}
|
|
|
|
// TestFooBaz should not be executed by the test suite
|
|
// because the number of function output is not 0
|
|
func (bts *basicTestSuite) TestFooBaz(t *testing.T) error {
|
|
bts.testFooBazCount++
|
|
bts.testsRan = append(bts.testsRan, t.Name())
|
|
return nil
|
|
}
|
|
|
|
// TestBarBaz should not be executed by the test suite
|
|
// because the type of the function input is not *testing.T
|
|
func (bts *basicTestSuite) TestBarBaz(t string) {
|
|
bts.testBarBazCount++
|
|
bts.testsRan = append(bts.testsRan, "TestBarBaz")
|
|
}
|
|
|
|
func TestBasicTestSuite(t *testing.T) {
|
|
bts := new(basicTestSuite)
|
|
Run(t, bts)
|
|
|
|
assert.Equal(t, bts.setupCount, 1)
|
|
assert.Equal(t, bts.testFooCount, 1)
|
|
assert.Equal(t, bts.testBarCount, 1)
|
|
assert.Equal(t, bts.teardownCount, 1)
|
|
assert.Assert(t, is.Len(bts.testsRan, 2))
|
|
assertTestsRan(t, bts.testsRan)
|
|
assertNonTests(t, bts)
|
|
}
|
|
|
|
type skipTestSuite struct {
|
|
basicTestSuite
|
|
skippedTestCount int
|
|
}
|
|
|
|
func (sts *skipTestSuite) ShouldSkipTest(testName string) bool {
|
|
if testName == "TestBar" {
|
|
sts.skippedTestCount++
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
func TestSkipTest(t *testing.T) {
|
|
sts := new(skipTestSuite)
|
|
Run(t, sts)
|
|
|
|
assert.Equal(t, sts.setupCount, 1)
|
|
assert.Equal(t, sts.testFooCount, 1)
|
|
assert.Equal(t, sts.testBarCount, 0)
|
|
assert.Equal(t, sts.teardownCount, 1)
|
|
assert.Equal(t, sts.skippedTestCount, 1)
|
|
assert.Assert(t, is.Len(sts.testsRan, 1))
|
|
assertTestsRan(t, sts.testsRan)
|
|
assertNonTests(t, &sts.basicTestSuite)
|
|
}
|
|
|
|
func assertTestsRan(t *testing.T, testsRan []string) {
|
|
for _, testRan := range testsRan {
|
|
parts := strings.Split(testRan, "/")
|
|
// Make sure that the name of the test has exactly one parent name and one subtest name
|
|
assert.Assert(t, is.Len(parts, 2))
|
|
// Check the parent test's name
|
|
assert.Equal(t, parts[0], t.Name())
|
|
}
|
|
}
|
|
|
|
// assertNonTests ensures that any malformed test functions are not run by the test suite
|
|
func assertNonTests(t *testing.T, bts *basicTestSuite) {
|
|
assert.Equal(t, bts.bazCount, 0)
|
|
assert.Equal(t, bts.testFooBarCount, 0)
|
|
assert.Equal(t, bts.testFooBazCount, 0)
|
|
assert.Equal(t, bts.testBarBazCount, 0)
|
|
}
|