mirror of
https://github.com/yusing/godoxy.git
synced 2026-03-23 09:31:02 +01:00
1ec2872f3d
- Replace template syntax ({{ .Request.Method }}) with $-prefixed variables ($req_method)
- Implement custom variable parser with static ($req_method, $status_code) and dynamic ($header(), $arg(), $form()) variables
- Replace templateOrStr interface with templateString struct and ExpandVars methods
- Add parser improvements for reliable quote handling
- Add new error types: ErrUnterminatedParenthesis, ErrUnexpectedVar, ErrExpectOneOrTwoArgs
- Update all tests and help text to use new variable syntax
- Add comprehensive unit and benchmark tests for variable expansion
29 lines
1.2 KiB
Go
29 lines
1.2 KiB
Go
package rules
|
|
|
|
import (
|
|
"io"
|
|
"net/http/httptest"
|
|
"net/url"
|
|
"testing"
|
|
)
|
|
|
|
func BenchmarkExpandVars(b *testing.B) {
|
|
testResponseModifier := NewResponseModifier(httptest.NewRecorder())
|
|
testResponseModifier.WriteHeader(200)
|
|
testResponseModifier.Write([]byte("Hello, world!"))
|
|
testRequest := httptest.NewRequest("GET", "/", nil)
|
|
testRequest.Header.Set("User-Agent", "test-agent/1.0")
|
|
testRequest.Header.Set("X-Custom", "value1,value2")
|
|
testRequest.ContentLength = 12345
|
|
testRequest.RemoteAddr = "192.168.1.100:54321"
|
|
testRequest.Form = url.Values{"param1": {"value1"}, "param2": {"value2"}}
|
|
testRequest.PostForm = url.Values{"param3": {"value3"}, "param4": {"value4"}}
|
|
|
|
for b.Loop() {
|
|
err := ExpandVars(testResponseModifier, testRequest, "$req_method $req_path $req_query $req_url $req_uri $req_host $req_port $req_addr $req_content_type $req_content_length $remote_host $remote_port $remote_addr $status_code $resp_content_type $resp_content_length $header(User-Agent) $header(X-Custom, 0) $header(X-Custom, 1) $arg(param1) $arg(param2) $arg(param3) $arg(param4) $form(param1) $form(param2) $form(param3) $form(param4) $postform(param1) $postform(param2) $postform(param3) $postform(param4)", io.Discard)
|
|
if err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
}
|
|
}
|