init

at-web/vendor/github.com/caixw/lib.go/assert/assertion.go

@@ -0,0 +1,114 @@
+// 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
+}