From 0cc6bd82b4f757b7966abd5863fde9fe25de6fb4 Mon Sep 17 00:00:00 2001
From: Wenchong Hu <whu@lyft.com>
Date: Tue, 18 Mar 2025 01:01:28 -0700
Subject: [PATCH] Update get_jwks response data structure

---
 confidant_client/__init__.py               | 4 ++--
 tests/unit/confidant_client/client_test.py | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/confidant_client/__init__.py b/confidant_client/__init__.py
index 9c3f747..f6e70c0 100644
--- a/confidant_client/__init__.py
+++ b/confidant_client/__init__.py
@@ -529,7 +529,7 @@ def get_jwt(self, environment, resource_id, expiry):
         return ret
 
     def get_jwks(self, environment):
-        ret = {'result': False, 'keys': {}}
+        ret = {'result': False}
         url = '{0}/v1/jwks/public/{1}'.format(self.config['url'], environment)
 
         try:
@@ -538,7 +538,7 @@ def get_jwks(self, environment):
                 url,
             )
             data = response.json()
-            ret['keys'] = data
+            ret.update(data)
         except RequestExecutionError:
             logging.exception('Error with executing request')
             return ret
diff --git a/tests/unit/confidant_client/client_test.py b/tests/unit/confidant_client/client_test.py
index 6664c81..737f7ad 100644
--- a/tests/unit/confidant_client/client_test.py
+++ b/tests/unit/confidant_client/client_test.py
@@ -952,7 +952,7 @@ def test_get_jwks(self):
 
         self.assertEqual(
             client.get_jwks('development'),
-            {'result': True, 'keys': {}}
+            {'result': True}
         )
         client.request_session.request.assert_called_with(
             'GET',