mirror of
https://github.com/valyala/fasthttp.git
synced 2026-06-15 16:07:51 +03:00
Added RequestURI helper to Request
This commit is contained in:
@@ -76,6 +76,15 @@ func (req *Request) SetRequestURIBytes(requestURI []byte) {
|
||||
req.parsedURI = false
|
||||
}
|
||||
|
||||
// RequestURI returns request's URI.
|
||||
func (req *Request) RequestURI() []byte {
|
||||
if req.parsedURI {
|
||||
requestURI := req.uri.RequestURI()
|
||||
req.SetRequestURIBytes(requestURI)
|
||||
}
|
||||
return req.Header.RequestURI()
|
||||
}
|
||||
|
||||
// StatusCode returns response status code.
|
||||
func (resp *Response) StatusCode() int {
|
||||
return resp.Header.StatusCode()
|
||||
|
||||
@@ -11,6 +11,34 @@ import (
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestRequestRequestURI(t *testing.T) {
|
||||
var r Request
|
||||
|
||||
// Set request uri via SetRequestURI()
|
||||
uri := "/foo/bar?baz"
|
||||
r.SetRequestURI(uri)
|
||||
if string(r.RequestURI()) != uri {
|
||||
t.Fatalf("unexpected request uri %q. Expecting %q", r.RequestURI(), uri)
|
||||
}
|
||||
|
||||
// Set request uri via Request.URI().Update()
|
||||
r.Reset()
|
||||
uri = "/aa/bbb?ccc=sdfsdf"
|
||||
r.URI().Update(uri)
|
||||
if string(r.RequestURI()) != uri {
|
||||
t.Fatalf("unexpected request uri %q. Expecting %q", r.RequestURI(), uri)
|
||||
}
|
||||
|
||||
// update query args in the request uri
|
||||
qa := r.URI().QueryArgs()
|
||||
qa.Reset()
|
||||
qa.Set("foo", "bar")
|
||||
uri = "/aa/bbb?foo=bar"
|
||||
if string(r.RequestURI()) != uri {
|
||||
t.Fatalf("unexpected request uri %q. Expecting %q", r.RequestURI(), uri)
|
||||
}
|
||||
}
|
||||
|
||||
func TestRequestUpdateURI(t *testing.T) {
|
||||
var r Request
|
||||
r.Header.SetHost("aaa.bbb")
|
||||
|
||||
Reference in New Issue
Block a user