123 lines
4.0 KiB
Go
123 lines
4.0 KiB
Go
// Copyright 2023 PingCAP, Inc.
|
|
//
|
|
// 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 intest_test
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/pingcap/errors"
|
|
"github.com/pingcap/tidb/pkg/util/intest"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
type foo struct{}
|
|
|
|
func TestAssert(t *testing.T) {
|
|
require.True(t, intest.InTest)
|
|
checkAssert(t, true, true)
|
|
checkAssert(t, false, false)
|
|
checkAssert(t, false, false, "assert failed, msg1", "msg1")
|
|
checkAssert(t, false, false, "assert failed, msg2 a b 1", "msg2 %s %s %d", "a", "b", 1)
|
|
checkAssert(t, false, false, "assert failed, 123", 123)
|
|
checkAssertNotNil(t, "", true)
|
|
checkAssertNotNil(t, "abc", true)
|
|
checkAssertNotNil(t, 0, true)
|
|
checkAssertNotNil(t, 123, true)
|
|
checkAssertNotNil(t, foo{}, true)
|
|
checkAssertNotNil(t, &foo{}, true)
|
|
checkAssertNotNil(t, nil, false)
|
|
checkAssertNotNil(t, true, true)
|
|
checkAssertNotNil(t, false, true)
|
|
var f *foo
|
|
checkAssertNotNil(t, f, false)
|
|
checkAssertNotNil(t, func() bool { return true }, true)
|
|
checkAssertNotNil(t, func() bool { return false }, true)
|
|
checkAssertNotNil(t, func(_ string) bool { return true }, true)
|
|
checkAssertNotNil(t, nil, false, "assert failed, msg1", "msg1")
|
|
checkAssertNotNil(t, nil, false, "assert failed, msg2 a b 1", "msg2 %s %s %d", "a", "b", 1)
|
|
checkFuncAssert(t, func() bool { panic("inner panic1") }, false, "inner panic1")
|
|
checkFuncAssert(t, func() bool { return true }, true)
|
|
checkFuncAssert(t, func() bool { return false }, false)
|
|
checkFuncAssert(t, func() bool { return false }, false, "assert failed, msg3", "msg3")
|
|
checkFuncAssert(t, func() bool { return false }, false, "assert failed, msg4 c d 2", "msg4 %s %s %d", "c", "d", 2)
|
|
checkFuncAssert(t, func() bool { panic("inner panic2") }, false, "inner panic2")
|
|
checkFuncAssert(t, nil, false)
|
|
checkAssertNoError(t, nil, true)
|
|
checkAssertNoError(t, errors.New("mock err1"), false, "assert failed, error is not nil: mock err1")
|
|
var err error
|
|
checkAssertNoError(t, err, true)
|
|
}
|
|
|
|
func checkFuncAssert(t *testing.T, fn func() bool, pass bool, msgAndArgs ...any) {
|
|
doCheckAssert(t, intest.AssertFunc, fn, pass, msgAndArgs...)
|
|
}
|
|
|
|
func checkAssert(t *testing.T, cond bool, pass bool, msgAndArgs ...any) {
|
|
doCheckAssert(t, intest.Assert, cond, pass, msgAndArgs...)
|
|
}
|
|
|
|
func checkAssertNotNil(t *testing.T, obj any, pass bool, msgAndArgs ...any) {
|
|
doCheckAssert(t, intest.AssertNotNil, obj, pass, msgAndArgs...)
|
|
}
|
|
|
|
func checkAssertNoError(t *testing.T, err error, pass bool, msgAndArgs ...any) {
|
|
doCheckAssert(t, intest.AssertNoError, err, pass, msgAndArgs...)
|
|
}
|
|
|
|
func doCheckAssert(t *testing.T, fn any, cond any, pass bool, msgAndArgs ...any) {
|
|
expectMsg := "assert failed"
|
|
if len(msgAndArgs) > 0 {
|
|
expectMsg = msgAndArgs[0].(string)
|
|
msgAndArgs = msgAndArgs[1:]
|
|
}
|
|
|
|
fail := ""
|
|
onlyCheckPrefix := false
|
|
if !pass {
|
|
defer func() {
|
|
if fail != "" {
|
|
require.FailNow(t, fail)
|
|
}
|
|
r := recover()
|
|
require.NotNil(t, r)
|
|
if onlyCheckPrefix {
|
|
require.True(t, strings.HasPrefix(r.(string), expectMsg), "%q\nshould have prefix \n%q", r, expectMsg)
|
|
require.Contains(t, r.(string), expectMsg)
|
|
} else {
|
|
require.Equal(t, expectMsg, r)
|
|
}
|
|
}()
|
|
}
|
|
|
|
switch assertFn := fn.(type) {
|
|
case func(bool, ...any):
|
|
assertFn(cond.(bool), msgAndArgs...)
|
|
case func(any, ...any):
|
|
assertFn(cond, msgAndArgs...)
|
|
case func(func() bool, ...any):
|
|
assertFn(cond.(func() bool), msgAndArgs...)
|
|
case func(error, ...any):
|
|
if cond == nil {
|
|
assertFn(nil, msgAndArgs...)
|
|
} else {
|
|
onlyCheckPrefix = true
|
|
assertFn(cond.(error), msgAndArgs...)
|
|
}
|
|
default:
|
|
fail = "invalid input assert function"
|
|
}
|
|
}
|