mirror of
https://github.com/TeaOSLab/EdgeNode.git
synced 2025-11-05 01:20:26 +08:00
33 lines
865 B
Go
33 lines
865 B
Go
|
package checkpoints
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"github.com/TeaOSLab/EdgeNode/internal/waf/requests"
|
||
|
|
"io/ioutil"
|
||
|
|
"net/http"
|
||
|
|
"net/url"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestRequestFormArgCheckpoint_RequestValue(t *testing.T) {
|
||
|
|
rawReq, err := http.NewRequest(http.MethodPost, "http://teaos.cn", bytes.NewBuffer([]byte("name=lu&age=20&encoded="+url.QueryEscape("<strong>ENCODED STRING</strong>"))))
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
req := requests.NewRequest(rawReq)
|
||
|
|
req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
|
||
|
|
|
||
|
|
checkpoint := new(RequestFormArgCheckpoint)
|
||
|
|
t.Log(checkpoint.RequestValue(req, "name", nil))
|
||
|
|
t.Log(checkpoint.RequestValue(req, "age", nil))
|
||
|
|
t.Log(checkpoint.RequestValue(req, "Hello", nil))
|
||
|
|
t.Log(checkpoint.RequestValue(req, "encoded", nil))
|
||
|
|
|
||
|
|
body, err := ioutil.ReadAll(req.Body)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
t.Log(string(body))
|
||
|
|
}
|