Restore original test cases

Add `__getitem__` to `AuthenticationResponse` for backwards-compat.

neon_hana/auth/client_manager.py

@@ -183,9 +183,12 @@ def check_auth_request(self, client_id: str, username: str,
         @param origin_ip: Origin IP address of request
         @return: response tokens, permissions, and other metadata
         """
-        if client_id in self.authorized_clients:
-            print(f"Using cached client: {self.authorized_clients[client_id]}")
-            return self.authorized_clients[client_id]
+        # Caching does not work here because there is no guarantee that this
+        # instance knows the client's refresh token. One client may also want
+        # to generate multiple tokens.
+        # if client_id in self.authorized_clients:
+        #     print(f"Using cached client: {self.authorized_clients[client_id]}")
+        #     return self.authorized_clients[client_id]
 
         ratelimit_id = f"auth{origin_ip}"
         if not self.rate_limiter.get_all_buckets(ratelimit_id):

neon_hana/schema/auth_requests.py

@@ -65,6 +65,10 @@ class AuthenticationResponse(BaseModel):
                 "expiration": 1706045776.4168212
             }]}}
 
+    def __getitem__(self, item):
+        if hasattr(self, item):
+            return getattr(self, item)
+        raise KeyError(item)
 
 class RefreshRequest(BaseModel):
     access_token: str

tests/test_auth.py

@@ -49,16 +49,16 @@ def test_check_auth_request(self):
         auth_resp_1 = self.client_manager.check_auth_request(**request_1)
         self.assertEqual(self.client_manager.authorized_clients[client_1],
                          auth_resp_1)
-        self.assertEqual(auth_resp_1.username, 'guest')
-        self.assertEqual(auth_resp_1.client_id, client_1)
+        self.assertEqual(auth_resp_1['username'], 'guest')
+        self.assertEqual(auth_resp_1['client_id'], client_1)
 
         # Check auth from different client
         auth_resp_2 = self.client_manager.check_auth_request(**request_2)
         self.assertNotEquals(auth_resp_1, auth_resp_2)
         self.assertEqual(self.client_manager.authorized_clients[client_2],
                          auth_resp_2)
-        self.assertEqual(auth_resp_2.username, 'guest')
-        self.assertEqual(auth_resp_2.client_id, client_2)
+        self.assertEqual(auth_resp_2['username'], 'guest')
+        self.assertEqual(auth_resp_2['client_id'], client_2)
 
         # TODO: Test permissions