mirror of
https://github.com/TeaOSLab/EdgeNode.git
synced 2025-11-06 18:10:26 +08:00
30 lines
784 B
Go
30 lines
784 B
Go
package checkpoints
|
|
|
|
import (
|
|
"bytes"
|
|
"github.com/TeaOSLab/EdgeNode/internal/waf/requests"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"testing"
|
|
)
|
|
|
|
func TestResponseBodyCheckpoint_ResponseValue(t *testing.T) {
|
|
resp := requests.NewResponse(new(http.Response))
|
|
resp.StatusCode = 200
|
|
resp.Header = http.Header{}
|
|
resp.Header.Set("Hello", "World")
|
|
resp.Body = ioutil.NopCloser(bytes.NewBuffer([]byte("Hello, World")))
|
|
|
|
checkpoint := new(ResponseBodyCheckpoint)
|
|
t.Log(checkpoint.ResponseValue(nil, resp, "", nil, 1))
|
|
t.Log(checkpoint.ResponseValue(nil, resp, "", nil, 1))
|
|
t.Log(checkpoint.ResponseValue(nil, resp, "", nil, 1))
|
|
t.Log(checkpoint.ResponseValue(nil, resp, "", nil, 1))
|
|
|
|
data, err := ioutil.ReadAll(resp.Body)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Log("after read:", string(data))
|
|
}
|