diff --git a/flyteadmin/auth/handlers.go b/flyteadmin/auth/handlers.go index 764b562650..368d601e27 100644 --- a/flyteadmin/auth/handlers.go +++ b/flyteadmin/auth/handlers.go @@ -450,6 +450,7 @@ func GetUserInfoForwardResponseHandler() UserInfoForwardResponseHandler { info, ok := m.(*service.UserInfoResponse) if ok { w.Header().Set("X-User-Subject", info.Subject) + w.Header().Set("X-User-Name", info.Name) } return nil } diff --git a/flyteadmin/auth/handlers_test.go b/flyteadmin/auth/handlers_test.go index b7cc636e5b..1a42030014 100644 --- a/flyteadmin/auth/handlers_test.go +++ b/flyteadmin/auth/handlers_test.go @@ -302,13 +302,18 @@ func TestUserInfoForwardResponseHander(t *testing.T) { w := httptest.NewRecorder() resp := service.UserInfoResponse{ Subject: "user-id", + Name: "User Name", } assert.NoError(t, handler(ctx, w, &resp)) assert.Contains(t, w.Result().Header, "X-User-Subject") assert.Equal(t, w.Result().Header["X-User-Subject"], []string{"user-id"}) + assert.Contains(t, w.Result().Header, "X-User-Name") + assert.Equal(t, w.Result().Header["X-User-Name"], []string{"User Name"}) + w = httptest.NewRecorder() unrelatedResp := service.OAuth2MetadataResponse{} assert.NoError(t, handler(ctx, w, &unrelatedResp)) assert.NotContains(t, w.Result().Header, "X-User-Subject") + assert.NotContains(t, w.Result().Header, "X-User-Name") }