From 1f1bc429ed7a4d591a7374f0e267b59747b177bd Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=94=B0=E6=AC=A7?= <thinkerou@gmail.com>
Date: Thu, 9 Aug 2018 17:20:06 +0800
Subject: [PATCH] chore: add test case for source/function of recovery.go
 (#1467)

---
 recovery_test.go | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/recovery_test.go b/recovery_test.go
index de3b62a..fa065b1 100644
--- a/recovery_test.go
+++ b/recovery_test.go
@@ -41,3 +41,23 @@ func TestPanicWithAbort(t *testing.T) {
 	// TEST
 	assert.Equal(t, 400, w.Code)
 }
+
+func TestSource(t *testing.T) {
+	bs := source(nil, 0)
+	assert.Equal(t, []byte("???"), bs)
+
+	in := [][]byte{
+		[]byte("Hello world."),
+		[]byte("Hi, gin.."),
+	}
+	bs = source(in, 10)
+	assert.Equal(t, []byte("???"), bs)
+
+	bs = source(in, 1)
+	assert.Equal(t, []byte("Hello world."), bs)
+}
+
+func TestFunction(t *testing.T) {
+	bs := function(1)
+	assert.Equal(t, []byte("???"), bs)
+}