mirror of https://github.com/0xERR0R/blocky.git
72 lines
1.5 KiB
Go
72 lines
1.5 KiB
Go
package helpertest
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"net/http"
|
|
"net/url"
|
|
"sync/atomic"
|
|
|
|
"github.com/onsi/ginkgo/v2"
|
|
)
|
|
|
|
type HTTPProxy struct {
|
|
Addr net.Addr
|
|
requestTarget atomic.Value // string: HTTP Host of latest request
|
|
}
|
|
|
|
// TestHTTPProxy returns a new HTTPProxy server.
|
|
//
|
|
// All requests return http.StatusNotImplemented.
|
|
func TestHTTPProxy() *HTTPProxy {
|
|
proxyListener, err := net.ListenTCP("tcp4", &net.TCPAddr{IP: net.ParseIP("127.0.0.1"), Port: 0})
|
|
if err != nil {
|
|
ginkgo.Fail(fmt.Sprintf("could not create HTTP proxy listener: %s", err))
|
|
}
|
|
|
|
proxy := &HTTPProxy{
|
|
Addr: proxyListener.Addr(),
|
|
}
|
|
|
|
proxySrv := http.Server{ //nolint:gosec
|
|
Addr: "127.0.0.1:0",
|
|
Handler: proxy,
|
|
}
|
|
|
|
go func() { _ = proxySrv.Serve(proxyListener) }()
|
|
ginkgo.DeferCleanup(proxySrv.Close)
|
|
|
|
return proxy
|
|
}
|
|
|
|
// URL returns the proxy's URL for use by clients.
|
|
func (p *HTTPProxy) URL() *url.URL {
|
|
return &url.URL{
|
|
Scheme: "http",
|
|
Host: p.Addr.String(),
|
|
}
|
|
}
|
|
|
|
// Check ReqURL has the right type signature for http.Transport.Proxy
|
|
var _ = http.Transport{Proxy: (*HTTPProxy)(nil).ReqURL}
|
|
|
|
func (p *HTTPProxy) ReqURL(*http.Request) (*url.URL, error) {
|
|
return p.URL(), nil
|
|
}
|
|
|
|
// RequestTarget returns the target of the last request.
|
|
func (p *HTTPProxy) RequestTarget() string {
|
|
val := p.requestTarget.Load()
|
|
if val == nil {
|
|
ginkgo.Fail(fmt.Sprintf("http proxy %s received no requests", p.Addr))
|
|
}
|
|
|
|
return val.(string)
|
|
}
|
|
|
|
func (p *HTTPProxy) ServeHTTP(w http.ResponseWriter, req *http.Request) {
|
|
p.requestTarget.Store(req.Host)
|
|
|
|
w.WriteHeader(http.StatusNotImplemented)
|
|
}
|