extractor_test.go

package request

import (
"fmt"
"net/http"
"net/url"
"testing"
)

varextractorTestTokenA="A"
varextractorTestTokenB="B"

varextractorTestData= []struct {
namestring
extractorExtractor
headersmap[string]string
query url.Values
tokenstring
errerror
}{
 {
name: "simple header",
extractor: HeaderExtractor{"Foo"},
headers: map[string]string{"Foo": extractorTestTokenA},
query: nil,
token: extractorTestTokenA,
err: nil,
 },
 {
name: "simple argument",
extractor: ArgumentExtractor{"token"},
headers: map[string]string{},
query: url.Values{"token": {extractorTestTokenA}},
token: extractorTestTokenA,
err: nil,
 },
 {
name: "multiple extractors",
extractor: MultiExtractor{
HeaderExtractor{"Foo"},
ArgumentExtractor{"token"},
 },
headers: map[string]string{"Foo": extractorTestTokenA},
query: url.Values{"token": {extractorTestTokenB}},
token: extractorTestTokenA,
err: nil,
 },
 {
name: "simple miss",
extractor: HeaderExtractor{"This-Header-Is-Not-Set"},
headers: map[string]string{"Foo": extractorTestTokenA},
query: nil,
token: "",
err: ErrNoTokenInRequest,
 },
 {
name: "filter",
extractor: AuthorizationHeaderExtractor,
headers: map[string]string{"Authorization": "Bearer "+extractorTestTokenA},
query: nil,
token: extractorTestTokenA,
err: nil,
 },
}

funcTestExtractor(t*testing.T) {
// Bearer token request
for_, data:=rangeextractorTestData {
// Make request from test struct
r:=makeExampleRequest("GET", "/", data.headers, data.query)

// Test extractor
token, err:=data.extractor.ExtractToken(r)
iftoken!=data.token {
t.Errorf("[%v] Expected token '%v'. Got '%v'", data.name, data.token, token)
continue
 }
iferr!=data.err {
t.Errorf("[%v] Expected error '%v'. Got '%v'", data.name, data.err, err)
continue
 }
 }
}

funcmakeExampleRequest(method, pathstring, headersmap[string]string, urlArgs url.Values) *http.Request {
r, _:=http.NewRequest(method, fmt.Sprintf("%v?%v", path, urlArgs.Encode()), nil)
fork, v:=rangeheaders {
r.Header.Set(k, v)
 }
returnr
}

funcTestBearerExtractor(t*testing.T) {
request:=makeExampleRequest("POST", "https://example.com/", map[string]string{"Authorization": "Bearer ToKen"}, nil)
token, err:=BearerExtractor{}.ExtractToken(request)
iferr!=nil||token!="ToKen" {
t.Errorf("ExtractToken did not return token, returned: %v, %v", token, err)
 }

request=makeExampleRequest("POST", "https://example.com/", map[string]string{"Authorization": "Bearo ToKen"}, nil)
token, err=BearerExtractor{}.ExtractToken(request)
iferr==nil||token!="" {
t.Errorf("ExtractToken did not return error, returned: %v, %v", token, err)
 }

request=makeExampleRequest("POST", "https://example.com/", map[string]string{"Authorization": "BeArEr HeLO"}, nil)
token, err=BearerExtractor{}.ExtractToken(request)
iferr!=nil||token!="HeLO" {
t.Errorf("ExtractToken did not return token, returned: %v, %v", token, err)
 }
}