110 lines
2.4 KiB
Go
110 lines
2.4 KiB
Go
package integration
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/caddyserver/caddy/v2/caddytest"
|
|
)
|
|
|
|
func TestIntercept(t *testing.T) {
|
|
tester := caddytest.NewTester(t)
|
|
tester.InitServer(`{
|
|
skip_install_trust
|
|
admin localhost:2999
|
|
http_port 9080
|
|
https_port 9443
|
|
grace_period 1ns
|
|
}
|
|
|
|
localhost:9080 {
|
|
respond /intercept "I'm a teapot" 408
|
|
header /intercept To-Intercept ok
|
|
respond /no-intercept "I'm not a teapot"
|
|
|
|
intercept {
|
|
@teapot status 408
|
|
handle_response @teapot {
|
|
header /intercept intercepted {resp.header.To-Intercept}
|
|
respond /intercept "I'm a combined coffee/tea pot that is temporarily out of coffee" 503
|
|
}
|
|
}
|
|
}
|
|
`, "caddyfile")
|
|
|
|
r, _ := tester.AssertGetResponse("http://localhost:9080/intercept", 503, "I'm a combined coffee/tea pot that is temporarily out of coffee")
|
|
if r.Header.Get("intercepted") != "ok" {
|
|
t.Fatalf(`header "intercepted" value is not "ok": %s`, r.Header.Get("intercepted"))
|
|
}
|
|
|
|
tester.AssertGetResponse("http://localhost:9080/no-intercept", 200, "I'm not a teapot")
|
|
}
|
|
|
|
func TestInterceptReplaceStatusWithMatcher(t *testing.T) {
|
|
tester := caddytest.NewTester(t)
|
|
tester.InitServer(`{
|
|
skip_install_trust
|
|
admin localhost:2999
|
|
http_port 9080
|
|
https_port 9443
|
|
grace_period 1ns
|
|
}
|
|
|
|
localhost:9080 {
|
|
respond /error "boom" 500
|
|
|
|
intercept {
|
|
@err status 5xx
|
|
replace_status @err 200
|
|
}
|
|
}
|
|
`, "caddyfile")
|
|
|
|
tester.AssertGetResponse("http://localhost:9080/error", 200, "boom")
|
|
}
|
|
|
|
func TestInterceptReplaceStatusWithoutMatcher(t *testing.T) {
|
|
tester := caddytest.NewTester(t)
|
|
tester.InitServer(`{
|
|
skip_install_trust
|
|
admin localhost:2999
|
|
http_port 9080
|
|
https_port 9443
|
|
grace_period 1ns
|
|
}
|
|
|
|
localhost:9080 {
|
|
respond /forbidden "denied" 403
|
|
|
|
intercept {
|
|
replace_status 200
|
|
}
|
|
}
|
|
`, "caddyfile")
|
|
|
|
tester.AssertGetResponse("http://localhost:9080/forbidden", 200, "denied")
|
|
}
|
|
|
|
func TestInterceptReplaceStatusNotMatched(t *testing.T) {
|
|
tester := caddytest.NewTester(t)
|
|
tester.InitServer(`{
|
|
skip_install_trust
|
|
admin localhost:2999
|
|
http_port 9080
|
|
https_port 9443
|
|
grace_period 1ns
|
|
}
|
|
|
|
localhost:9080 {
|
|
respond /ok "all good" 200
|
|
|
|
intercept {
|
|
@err status 5xx
|
|
replace_status @err 503
|
|
}
|
|
}
|
|
`, "caddyfile")
|
|
|
|
// 200 does not match @err (5xx), so status should pass through unchanged
|
|
tester.AssertGetResponse("http://localhost:9080/ok", 200, "all good")
|
|
}
|