From 65592d0b5acb8a42e25234a2e6c88ddee94499dc Mon Sep 17 00:00:00 2001 From: Shivansh Vij Date: Thu, 26 May 2022 15:54:54 -0400 Subject: [PATCH] Updating test cases Fixes https://github.com/dexidp/dex/issues/2537 Signed-off-by: Shivansh Vij --- server/deviceflowhandlers_test.go | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/server/deviceflowhandlers_test.go b/server/deviceflowhandlers_test.go index c387af43..95ca46e0 100644 --- a/server/deviceflowhandlers_test.go +++ b/server/deviceflowhandlers_test.go @@ -52,6 +52,7 @@ func TestHandleDeviceCode(t *testing.T) { requestType string scopes []string expectedResponseCode int + expectedContentType string expectedServerResponse string }{ { @@ -60,6 +61,7 @@ func TestHandleDeviceCode(t *testing.T) { requestType: "POST", scopes: []string{"openid", "profile", "email"}, expectedResponseCode: http.StatusOK, + expectedContentType: "application/json", }, { testName: "Invalid request Type (GET)", @@ -67,6 +69,7 @@ func TestHandleDeviceCode(t *testing.T) { requestType: "GET", scopes: []string{"openid", "profile", "email"}, expectedResponseCode: http.StatusBadRequest, + expectedContentType: "application/json", }, } for _, tc := range tests { @@ -101,6 +104,10 @@ func TestHandleDeviceCode(t *testing.T) { t.Errorf("Unexpected Response Type. Expected %v got %v", tc.expectedResponseCode, rr.Code) } + if rr.Header().Get("content-type") != tc.expectedContentType { + t.Errorf("Unexpected Response Content Type. Expected %v got %v", tc.expectedContentType, rr.Header().Get("content-type")) + } + body, err := io.ReadAll(rr.Body) if err != nil { t.Errorf("Could read token response %v", err)