77 lines
1.5 KiB
Go
77 lines
1.5 KiB
Go
package zlibUtil
|
||
|
||
import (
|
||
"compress/zlib"
|
||
"strings"
|
||
"testing"
|
||
)
|
||
|
||
var (
|
||
InitString = `{"Code":4,"Message":"IPNotAuthorized","Data":null}`
|
||
CompressBytes []byte
|
||
)
|
||
|
||
func TestCompress(t *testing.T) {
|
||
data := ([]byte)(InitString)
|
||
result, _ := Compress(data, zlib.DefaultCompression)
|
||
// if isEqual(result, CompressBytes) == false {
|
||
// t.Errorf("压缩失败，期待%v，实际%v", InitBytes, result)
|
||
// }
|
||
|
||
CompressBytes = result
|
||
}
|
||
|
||
func TestDecompress(t *testing.T) {
|
||
data, _ := Decompress(CompressBytes)
|
||
result := string(data)
|
||
if result != InitString {
|
||
t.Errorf("解压缩失败，期待%s，实际%s", InitString, result)
|
||
}
|
||
}
|
||
|
||
func isEqual(a, b []byte) bool {
|
||
if a == nil && b == nil {
|
||
return true
|
||
} else if a == nil || b == nil {
|
||
return false
|
||
}
|
||
|
||
if len(a) != len(b) {
|
||
return false
|
||
}
|
||
|
||
for i := 0; i < len(a); i++ {
|
||
if a[i] != b[i] {
|
||
return false
|
||
}
|
||
}
|
||
|
||
return true
|
||
}
|
||
|
||
//BenchmarkCompress-12 10989 107509 ns/op
|
||
func BenchmarkCompress(b *testing.B) {
|
||
toCompress := []byte(strings.Repeat(InitString, 100))
|
||
b.ResetTimer()
|
||
for i := 0; i < b.N; i++ {
|
||
_,err :=Compress(toCompress, zlib.DefaultCompression)
|
||
if err!=nil{
|
||
b.Log(err)
|
||
}
|
||
}
|
||
b.StopTimer()
|
||
}
|
||
|
||
//BenchmarkDeCompress-12 99153 10802 ns/op
|
||
func BenchmarkDeCompress(b *testing.B) {
|
||
toCompress := []byte(strings.Repeat(InitString, 100))
|
||
compressedData,_ := Compress(toCompress, zlib.DefaultCompression)
|
||
b.ResetTimer()
|
||
for i := 0; i < b.N; i++ {
|
||
_,err := Decompress(compressedData)
|
||
if err!=nil{
|
||
b.Log(err)
|
||
}
|
||
}
|
||
b.StopTimer()
|
||
} |