added more test cases for set_response_headers/custom_log_fields in b…

…ackend/oauth2/backend
coupergateway · Feb 16, 2022 · 01bcd6b · 01bcd6b
1 parent e99ca67
commit 01bcd6b
Show file tree

Hide file tree

Showing 2 changed files with 63 additions and 10 deletions.
diff --git a/server/http_endpoints_test.go b/server/http_endpoints_test.go
@@ -28,7 +28,7 @@ import (
 
 const testdataPath = "testdata/endpoints"
 
-func TestBackend_BackendVariable_Request(t *testing.T) {
+func TestBackend_BackendVariable_RequestResponse(t *testing.T) {
 	client := newClient()
 	helper := test.New(t)
 
@@ -58,17 +58,32 @@ func TestBackend_BackendVariable_Request(t *testing.T) {
 			continue
 		}
 
+		responseHeaders := entry.Data["response"].(logging.Fields)["headers"].(map[string]string)
 		data := entry.Data["custom"].(logrus.Fields)
 
-		if data["x-from-request-json-body"] != float64(1) ||
-			data["x-from-request-header"] != "bar" ||
-			data["x-from-requests-json-body"] != float64(1) ||
-			data["x-from-requests-header"] != "bar" ||
-			data["x-from-response-header"] != "application/json" ||
-			data["x-from-response-json-body"] != "/anything" ||
-			data["x-from-responses-header"] != "application/json" ||
-			data["x-from-responses-json-body"] != "/anything" {
-			t.Errorf("Unexpected logs given: %#v", data)
+		url := entry.Data["url"]
+		if url == "http://localhost:8081/token" {
+			if data["x-from-request-body"] != "grant_type=client_credentials" ||
+				// data["x-from-request-form-body"] != "client_credentials" || // not yet
+				data["x-from-request-header"] != "Basic cXBlYjpiZW4=" ||
+				data["x-from-response-header"] != "60s" ||
+				data["x-from-response-body"] != `{"access_token":"the_access_token","expires_in":60}` ||
+				data["x-from-response-json-body"] != "the_access_token" ||
+				responseHeaders["location"] != "Basic cXBlYjpiZW4=||60s|" {
+				// responseHeaders["location"] != "Basic cXBlYjpiZW4=|client_credentials|60s|the_access_token" { // not yet
+				t.Errorf("Unexpected logs given: %#v", data)
+			}
+		} else {
+			if data["x-from-request-json-body"] != float64(1) ||
+				data["x-from-request-header"] != "bar" ||
+				data["x-from-requests-json-body"] != float64(1) ||
+				data["x-from-requests-header"] != "bar" ||
+				data["x-from-response-header"] != "application/json" ||
+				data["x-from-response-json-body"] != "/anything" ||
+				data["x-from-responses-header"] != "application/json" ||
+				data["x-from-responses-json-body"] != "/anything" {
+				t.Errorf("Unexpected logs given: %#v", data)
+			}
 		}
 	}
 }

diff --git a/server/testdata/integration/backend/01_couper.hcl b/server/testdata/integration/backend/01_couper.hcl
@@ -1,4 +1,5 @@
 server {
+  hosts = ["*:8080"]
   endpoint "/" {
     proxy "default" {
       url = "${env.COUPER_TEST_BACKEND_ADDR}/anything"
@@ -71,6 +72,26 @@ server {
           x-from-response-json-body = backend_response.json_body.Url
           x-from-responses-json-body = backend_responses.default.json_body.Url
         }
+        oauth2 {
+          grant_type = "client_credentials"
+          token_endpoint = "http://localhost:8081/token"
+          client_id = "qpeb"
+          client_secret = "ben"
+          backend {
+            set_response_headers = {
+              # use a response header field that is actually logged
+              location = "${backend_request.headers.authorization}|${backend_request.form_body.grant_type[0]}|${backend_response.headers.x-pires-in}|${backend_response.json_body.access_token}"
+            }
+            custom_log_fields = {
+              x-from-request-header = backend_request.headers.authorization
+              x-from-request-body = backend_request.body
+              x-from-request-form-body = backend_request.form_body.grant_type[0]
+              x-from-response-header = backend_response.headers.x-pires-in
+              x-from-response-body = backend_response.body
+              x-from-response-json-body = backend_response.json_body.access_token
+            }
+          }
+        }
       }
     }
   }
@@ -85,3 +106,20 @@ definitions {
     }
   }
 }
+
+server "as" {
+  hosts = ["*:8081"]
+  api {
+    endpoint "/token" {
+      response {
+        headers = {
+          x-pires-in = "60s"
+        }
+        json_body = {
+          access_token = "the_access_token"
+          expires_in = 60
+        }
+      }
+    }
+  }
+}