mirror of
https://gitee.com/johng/gf.git
synced 2024-12-03 04:37:49 +08:00
74 lines
1.7 KiB
Go
74 lines
1.7 KiB
Go
// Copyright GoFrame Author(https://goframe.org). All Rights Reserved.
|
|
//
|
|
// This Source Code Form is subject to the terms of the MIT License.
|
|
// If a copy of the MIT was not distributed with this file,
|
|
// You can obtain one at https://github.com/gogf/gf.
|
|
|
|
package utils_test
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"testing"
|
|
|
|
"github.com/gogf/gf/v2/internal/utils"
|
|
"github.com/gogf/gf/v2/test/gtest"
|
|
)
|
|
|
|
func Test_ReadCloser(t *testing.T) {
|
|
gtest.C(t, func(t *gtest.T) {
|
|
var (
|
|
n int
|
|
b = make([]byte, 3)
|
|
body = utils.NewReadCloser([]byte{1, 2, 3, 4}, false)
|
|
)
|
|
n, _ = body.Read(b)
|
|
t.Assert(b[:n], []byte{1, 2, 3})
|
|
n, _ = body.Read(b)
|
|
t.Assert(b[:n], []byte{4})
|
|
|
|
n, _ = body.Read(b)
|
|
t.Assert(b[:n], []byte{})
|
|
n, _ = body.Read(b)
|
|
t.Assert(b[:n], []byte{})
|
|
})
|
|
gtest.C(t, func(t *gtest.T) {
|
|
var (
|
|
r []byte
|
|
body = utils.NewReadCloser([]byte{1, 2, 3, 4}, false)
|
|
)
|
|
r, _ = ioutil.ReadAll(body)
|
|
t.Assert(r, []byte{1, 2, 3, 4})
|
|
r, _ = ioutil.ReadAll(body)
|
|
t.Assert(r, []byte{})
|
|
})
|
|
gtest.C(t, func(t *gtest.T) {
|
|
var (
|
|
n int
|
|
r []byte
|
|
b = make([]byte, 3)
|
|
body = utils.NewReadCloser([]byte{1, 2, 3, 4}, true)
|
|
)
|
|
n, _ = body.Read(b)
|
|
t.Assert(b[:n], []byte{1, 2, 3})
|
|
n, _ = body.Read(b)
|
|
t.Assert(b[:n], []byte{4})
|
|
|
|
n, _ = body.Read(b)
|
|
t.Assert(b[:n], []byte{1, 2, 3})
|
|
n, _ = body.Read(b)
|
|
t.Assert(b[:n], []byte{4})
|
|
|
|
r, _ = ioutil.ReadAll(body)
|
|
t.Assert(r, []byte{1, 2, 3, 4})
|
|
r, _ = ioutil.ReadAll(body)
|
|
t.Assert(r, []byte{1, 2, 3, 4})
|
|
})
|
|
}
|
|
|
|
func Test_RemoveSymbols(t *testing.T) {
|
|
gtest.C(t, func(t *gtest.T) {
|
|
t.Assert(utils.RemoveSymbols(`-a-b._a c1!@#$%^&*()_+:";'.,'01`), `abac101`)
|
|
t.Assert(utils.RemoveSymbols(`-a-b我._a c1!@#$%^&*是()_+:帅";'.,哥'01`), `ab我ac1是帅哥01`)
|
|
})
|
|
}
|