terror: improve test.

juniway · Nov 6, 2015 · f1a4be7 · f1a4be7
1 parent 807b897
commit f1a4be7
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/terror/terror_test.go b/terror/terror_test.go
@@ -14,11 +14,11 @@
 package terror
 
 import (
-	"fmt"
 	"testing"
 
 	"github.com/juju/errors"
 	. "github.com/pingcap/check"
+	"strings"
 )
 
 func TestT(t *testing.T) {
@@ -53,12 +53,22 @@ func (s *testTErrorSuite) TestTError(c *C) {
 var predefinedErr = ClassExecutor.New(ErrCode(123), "predefiend error")
 
 func example() error {
+	err := call()
+	return errors.Trace(err)
+}
+
+func call() error {
 	return predefinedErr.Gen("error message:%s", "abc")
 }
 
-func (s *testTErrorSuite) TestExample(c *C) {
+func (s *testTErrorSuite) TestTraceAndLocation(c *C) {
 	err := example()
-	fmt.Println(errors.ErrorStack(err))
+	stack := errors.ErrorStack(err)
+	lines := strings.Split(stack, "\n")
+	c.Assert(len(lines), Equals, 2)
+	for _, v := range lines {
+		c.Assert(strings.Contains(v, "terror_test.go"), IsTrue)
+	}
 }
 
 func (s *testTErrorSuite) TestErrorEqual(c *C) {