115 lines
2.7 KiB
Go
115 lines
2.7 KiB
Go
// Copyright 2014 by caixw, All rights reserved.
|
|
// Use of this source code is governed by a MIT
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package assert
|
|
|
|
import (
|
|
"testing"
|
|
)
|
|
|
|
// Assertion是对testing.T进行了简单的封装。
|
|
// 可以以对象的方式调用包中的各个断言函数，
|
|
// 减少了参数t的传递。
|
|
type Assertion struct {
|
|
t *testing.T
|
|
}
|
|
|
|
// 返回Assertion对象。
|
|
func New(t *testing.T) *Assertion {
|
|
return &Assertion{t: t}
|
|
}
|
|
|
|
// 返回testing.T对象
|
|
func (a *Assertion) T() *testing.T {
|
|
return a.t
|
|
}
|
|
|
|
func (a *Assertion) True(expr bool, msg ...interface{}) *Assertion {
|
|
True(a.t, expr, msg...)
|
|
return a
|
|
}
|
|
|
|
func (a *Assertion) False(expr bool, msg ...interface{}) *Assertion {
|
|
False(a.t, expr, msg...)
|
|
return a
|
|
}
|
|
func (a *Assertion) Nil(expr interface{}, msg ...interface{}) *Assertion {
|
|
Nil(a.t, expr, msg...)
|
|
return a
|
|
}
|
|
|
|
func (a *Assertion) NotNil(expr interface{}, msg ...interface{}) *Assertion {
|
|
NotNil(a.t, expr, msg...)
|
|
return a
|
|
}
|
|
|
|
func (a *Assertion) Equal(v1, v2 interface{}, msg ...interface{}) *Assertion {
|
|
Equal(a.t, v1, v2, msg...)
|
|
return a
|
|
}
|
|
|
|
func (a *Assertion) NotEqual(v1, v2 interface{}, msg ...interface{}) *Assertion {
|
|
NotEqual(a.t, v1, v2, msg...)
|
|
return a
|
|
}
|
|
|
|
func (a *Assertion) Empty(expr interface{}, msg ...interface{}) *Assertion {
|
|
Empty(a.t, expr, msg...)
|
|
return a
|
|
}
|
|
func (a *Assertion) NotEmpty(expr interface{}, msg ...interface{}) *Assertion {
|
|
NotEmpty(a.t, expr, msg...)
|
|
return a
|
|
}
|
|
|
|
func (a *Assertion) Error(expr interface{}, msg ...interface{}) *Assertion {
|
|
Error(a.t, expr, msg...)
|
|
return a
|
|
}
|
|
|
|
func (a *Assertion) NotError(expr interface{}, msg ...interface{}) *Assertion {
|
|
NotError(a.t, expr, msg...)
|
|
return a
|
|
}
|
|
|
|
func (a *Assertion) FileExists(path string, msg ...interface{}) *Assertion {
|
|
FileExists(a.t, path, msg...)
|
|
return a
|
|
}
|
|
|
|
func (a *Assertion) FileNotExists(path string, msg ...interface{}) *Assertion {
|
|
FileNotExists(a.t, path, msg...)
|
|
return a
|
|
}
|
|
|
|
func (a *Assertion) Panic(fn func(), msg ...interface{}) *Assertion {
|
|
Panic(a.t, fn, msg...)
|
|
return a
|
|
}
|
|
|
|
func (a *Assertion) NotPanic(fn func(), msg ...interface{}) *Assertion {
|
|
NotPanic(a.t, fn, msg...)
|
|
return a
|
|
}
|
|
|
|
func (a *Assertion) Contains(container, item interface{}, msg ...interface{}) *Assertion {
|
|
Contains(a.t, container, item, msg...)
|
|
return a
|
|
}
|
|
|
|
func (a *Assertion) NotContains(container, item interface{}, msg ...interface{}) *Assertion {
|
|
NotContains(a.t, container, item, msg...)
|
|
return a
|
|
}
|
|
|
|
func (a *Assertion) StringEqual(s1, s2 string, style int, msg ...interface{}) *Assertion {
|
|
StringEqual(a.t, s1, s2, style, msg...)
|
|
return a
|
|
}
|
|
|
|
func (a *Assertion) StringNotEqual(s1, s2 string, style int, msg ...interface{}) *Assertion {
|
|
StringNotEqual(a.t, s1, s2, style, msg...)
|
|
return a
|
|
}
|