mirror of
https://gitee.com/IrisVega/frp.git
synced 2024-11-01 22:31:29 +08:00
74 lines
1.6 KiB
Go
74 lines
1.6 KiB
Go
package beego
|
|
|
|
import (
|
|
"bytes"
|
|
"compress/gzip"
|
|
"compress/zlib"
|
|
"io"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
)
|
|
|
|
var currentWorkDir, _ = os.Getwd()
|
|
var licenseFile = filepath.Join(currentWorkDir, "LICENSE")
|
|
|
|
func testOpenFile(encoding string, content []byte, t *testing.T) {
|
|
fi, _ := os.Stat(licenseFile)
|
|
b, n, sch, err := openFile(licenseFile, fi, encoding)
|
|
if err != nil {
|
|
t.Log(err)
|
|
t.Fail()
|
|
}
|
|
|
|
t.Log("open static file encoding "+n, b)
|
|
|
|
assetOpenFileAndContent(sch, content, t)
|
|
}
|
|
func TestOpenStaticFile_1(t *testing.T) {
|
|
file, _ := os.Open(licenseFile)
|
|
content, _ := ioutil.ReadAll(file)
|
|
testOpenFile("", content, t)
|
|
}
|
|
|
|
func TestOpenStaticFileGzip_1(t *testing.T) {
|
|
file, _ := os.Open(licenseFile)
|
|
var zipBuf bytes.Buffer
|
|
fileWriter, _ := gzip.NewWriterLevel(&zipBuf, gzip.BestCompression)
|
|
io.Copy(fileWriter, file)
|
|
fileWriter.Close()
|
|
content, _ := ioutil.ReadAll(&zipBuf)
|
|
|
|
testOpenFile("gzip", content, t)
|
|
}
|
|
func TestOpenStaticFileDeflate_1(t *testing.T) {
|
|
file, _ := os.Open(licenseFile)
|
|
var zipBuf bytes.Buffer
|
|
fileWriter, _ := zlib.NewWriterLevel(&zipBuf, zlib.BestCompression)
|
|
io.Copy(fileWriter, file)
|
|
fileWriter.Close()
|
|
content, _ := ioutil.ReadAll(&zipBuf)
|
|
|
|
testOpenFile("deflate", content, t)
|
|
}
|
|
|
|
func assetOpenFileAndContent(sch *serveContentHolder, content []byte, t *testing.T) {
|
|
t.Log(sch.size, len(content))
|
|
if sch.size != int64(len(content)) {
|
|
t.Log("static content file size not same")
|
|
t.Fail()
|
|
}
|
|
bs, _ := ioutil.ReadAll(sch)
|
|
for i, v := range content {
|
|
if v != bs[i] {
|
|
t.Log("content not same")
|
|
t.Fail()
|
|
}
|
|
}
|
|
if len(staticFileMap) == 0 {
|
|
t.Log("men map is empty")
|
|
t.Fail()
|
|
}
|
|
}
|