fix(test): mock Session.request instead of .post/.get in endpoint tests

The 401 tests mocked Session.post/get, but RunPodClient._request() calls
Session.request(). Mocks never intercepted, so tests hit the real API and
relied on getting 401 back. When the API started returning 404 for
non-existent endpoints, these tests broke. Also fix test_missing_api_key
to create a new Endpoint after nullifying the key, since the old endpoint
already had a valid client.

Author: deanq

tests/test_endpoint/test_runner.py

@@ -59,14 +59,14 @@ def test_client_custom_overrides_global(self):
         self.assertEqual(client.api_key, custom_key)
 
 
-    @patch.object(requests.Session, "post")
-    def test_post_with_401(self, mock_post):
+    @patch.object(requests.Session, "request")
+    def test_post_with_401(self, mock_request):
         """
         Tests RunPodClient.post with 401 status code
         """
         mock_response = Mock()
         mock_response.status_code = 401
-        mock_post.return_value = mock_response
+        mock_request.return_value = mock_response
 
         with self.assertRaises(RuntimeError):
             runpod.api_key = "MOCK_API_KEY"
@@ -89,14 +89,14 @@ def test_post(self, mock_post):
 
         self.assertEqual(response, {"id": "123"})
 
-    @patch.object(requests.Session, "get")
-    def test_get_with_401(self, mock_get):
+    @patch.object(requests.Session, "request")
+    def test_get_with_401(self, mock_request):
         """
         Tests RunPodClient.get with 401 status code
         """
         mock_response = Mock()
         mock_response.status_code = 401
-        mock_get.return_value = mock_response
+        mock_request.return_value = mock_response
 
         with self.assertRaises(RuntimeError):
             runpod.api_key = "MOCK_API_KEY"
@@ -211,16 +211,16 @@ def test_missing_api_key(self):
         """
         with self.assertRaises(RuntimeError):
             runpod.api_key = None
-            self.endpoint.run(self.MODEL_INPUT)
+            Endpoint("ENDPOINT_ID").run(self.MODEL_INPUT)
 
-    @patch.object(requests.Session, "post")
-    def test_run_with_401(self, mock_post):
+    @patch.object(requests.Session, "request")
+    def test_run_with_401(self, mock_request):
         """
         Tests Endpoint.run with 401 status code
         """
         mock_response = Mock()
         mock_response.status_code = 401
-        mock_post.return_value = mock_response
+        mock_request.return_value = mock_response
 
         endpoint = runpod.Endpoint("ENDPOINT_ID")
         request_data = {"YOUR_MODEL_INPUT_JSON": "YOUR_MODEL_INPUT_VALUE"}