add test case for source

This commit is contained in:
thinkerou 2018-08-09 12:59:06 +08:00
parent 9666ba6738
commit d1db18da52

View File

@ -41,3 +41,18 @@ func TestPanicWithAbort(t *testing.T) {
// TEST // TEST
assert.Equal(t, 400, w.Code) 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)
}