95 lines
2.5 KiB
Go
95 lines
2.5 KiB
Go
// Package assert contains functions for making assertions in unit tests
|
|
package assert
|
|
|
|
import (
|
|
"fmt"
|
|
"path/filepath"
|
|
"reflect"
|
|
"runtime"
|
|
"strings"
|
|
)
|
|
|
|
// TestingT is an interface which defines the methods of testing.T that are
|
|
// required by this package
|
|
type TestingT interface {
|
|
Fatalf(string, ...interface{})
|
|
}
|
|
|
|
// Equal compare the actual value to the expected value and fails the test if
|
|
// they are not equal.
|
|
func Equal(t TestingT, actual, expected interface{}) {
|
|
if expected != actual {
|
|
fatal(t, "Expected '%v' (%T) got '%v' (%T)", expected, expected, actual, actual)
|
|
}
|
|
}
|
|
|
|
//EqualStringSlice compares two slices and fails the test if they do not contain
|
|
// the same items.
|
|
func EqualStringSlice(t TestingT, actual, expected []string) {
|
|
if len(actual) != len(expected) {
|
|
fatal(t, "Expected (length %d): %q\nActual (length %d): %q",
|
|
len(expected), expected, len(actual), actual)
|
|
}
|
|
for i, item := range actual {
|
|
if item != expected[i] {
|
|
fatal(t, "Slices differ at element %d, expected %q got %q",
|
|
i, expected[i], item)
|
|
}
|
|
}
|
|
}
|
|
|
|
// NilError asserts that the error is nil, otherwise it fails the test.
|
|
func NilError(t TestingT, err error) {
|
|
if err != nil {
|
|
fatal(t, "Expected no error, got: %s", err.Error())
|
|
}
|
|
}
|
|
|
|
// DeepEqual compare the actual value to the expected value and fails the test if
|
|
// they are not "deeply equal".
|
|
func DeepEqual(t TestingT, actual, expected interface{}) {
|
|
if !reflect.DeepEqual(actual, expected) {
|
|
fatal(t, "Expected '%v' (%T) got '%v' (%T)", expected, expected, actual, actual)
|
|
}
|
|
}
|
|
|
|
// Error asserts that error is not nil, and contains the expected text,
|
|
// otherwise it fails the test.
|
|
func Error(t TestingT, err error, contains string) {
|
|
if err == nil {
|
|
fatal(t, "Expected an error, but error was nil")
|
|
}
|
|
|
|
if !strings.Contains(err.Error(), contains) {
|
|
fatal(t, "Expected error to contain '%s', got '%s'", contains, err.Error())
|
|
}
|
|
}
|
|
|
|
// Contains asserts that the string contains a substring, otherwise it fails the
|
|
// test.
|
|
func Contains(t TestingT, actual, contains string) {
|
|
if !strings.Contains(actual, contains) {
|
|
fatal(t, "Expected '%s' to contain '%s'", actual, contains)
|
|
}
|
|
}
|
|
|
|
// NotNil fails the test if the object is nil
|
|
func NotNil(t TestingT, obj interface{}) {
|
|
if obj == nil {
|
|
fatal(t, "Expected non-nil value.")
|
|
}
|
|
}
|
|
|
|
func fatal(t TestingT, format string, args ...interface{}) {
|
|
t.Fatalf(errorSource()+format, args...)
|
|
}
|
|
|
|
// See testing.decorate()
|
|
func errorSource() string {
|
|
_, filename, line, ok := runtime.Caller(3)
|
|
if !ok {
|
|
return ""
|
|
}
|
|
return fmt.Sprintf("%s:%d: ", filepath.Base(filename), line)
|
|
}
|