Skip to content

Commit

Permalink
Fix tests
Browse files Browse the repository at this point in the history
  • Loading branch information
buger committed Mar 8, 2018
1 parent b1d04e6 commit 119b1ae
Show file tree
Hide file tree
Showing 2 changed files with 15 additions and 21 deletions.
1 change: 1 addition & 0 deletions helpers_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -127,6 +127,7 @@ func testHttpHandler() *mux.Router {
return
}
r.URL.Opaque = r.URL.RawPath
w.Header().Set("X-Tyk-Mock", "1")
err := json.NewEncoder(w).Encode(testHttpResponse{
Method: r.Method,
Url: r.URL.String(),
Expand Down
35 changes: 14 additions & 21 deletions res_handler_header_injector_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -12,51 +12,44 @@ import (
func TestResponseHeaderInjection(t *testing.T) {
ts := newTykTestServer()
defer ts.Close()

buildAndLoadAPI(func(spec *APISpec) {
spec.UseKeylessAccess = true
spec.Proxy.ListenPath = "/"
spec.UseKeylessAccess = true
spec.Proxy.ListenPath = "/"
spec.OrgID = "default"
updateAPIVersion(spec, "v1", func(v *apidef.VersionInfo) {
v.UseExtendedPaths = true
json.Unmarshal([]byte(`[
{
"delete_headers": [
"delete-this"
],
"add_headers": {
"add-this": "header"
},
"delete_headers": ["X-Tyk-Mock"],
"add_headers": {"X-Test": "test"},
"path": "/test-with-slash",
"method": "GET",
"act_on": false
},
{
"delete_headers": [
"delete-this"
],
"add_headers": {
"add-this": "header"
},
"delete_headers": ["X-Tyk-Mock"],
"add_headers": {"X-Test": "test"},
"path": "test-no-slash",
"method": "GET",
"act_on": false
}
]`), &v.ExtendedPaths.TransformResponseHeader)
})
spec.ResponseProcessors = []apidef.ResponseProcessor{{Name: "header_injector"}}
})

session := createStandardSession()
session.AccessRights = map[string]user.AccessDefinition{"test": {APIID: "test", Versions: []string{"v1"}}}

addHeaders := make(map[string]string)
deleteHeaders := make(map[string]string)
addHeaders["add-this"] = "header"
deleteHeaders["delete-this"] = "header"
addHeaders["X-Test"] = "test"
deleteHeaders["X-Tyk-Mock"] = "1"

ts.Run(t, []test.TestCase{
// Create base auth based key
// Create base auth based key
{Method: "GET", Path: "/test-with-slash", HeadersMatch: addHeaders, HeadersNotMatch: deleteHeaders},
{Method: "GET", Path: "/test-no-slash", HeadersMatch: addHeaders, HeadersNotMatch: deleteHeaders},
}...)
}...)
}

0 comments on commit 119b1ae

Please sign in to comment.