From 46bd99ad98b0c0d98341d7f797e6d6cb1dc63eef Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Tue, 9 Jan 2024 13:03:22 +0530
Subject: [PATCH] (test) test deployed proxy keygen

---
 litellm/tests/test_deployed_proxy_keygen.py | 53 +++++++++++++++++++++
 1 file changed, 53 insertions(+)
 create mode 100644 litellm/tests/test_deployed_proxy_keygen.py

diff --git a/litellm/tests/test_deployed_proxy_keygen.py b/litellm/tests/test_deployed_proxy_keygen.py
new file mode 100644
index 0000000000..661393edab
--- /dev/null
+++ b/litellm/tests/test_deployed_proxy_keygen.py
@@ -0,0 +1,53 @@
+import sys, os, time
+import traceback
+from dotenv import load_dotenv
+
+load_dotenv()
+import os, io
+
+# this file is to test litellm/proxy
+
+sys.path.insert(
+    0, os.path.abspath("../..")
+)  # Adds the parent directory to the system path
+import pytest, logging, requests
+import litellm
+from litellm import embedding, completion, completion_cost, Timeout
+from litellm import RateLimitError
+
+
+def test_add_new_key():
+    try:
+        # Your test data
+        test_data = {
+            "models": ["gpt-3.5-turbo", "gpt-4", "claude-2", "azure-model"],
+            "aliases": {"mistral-7b": "gpt-3.5-turbo"},
+            "duration": "20m",
+        }
+        print("testing proxy server")
+        # Your bearer token
+        token = os.getenv("PROXY_MASTER_KEY")
+
+        headers = {"Authorization": f"Bearer {token}"}
+
+        staging_endpoint = "https://litellm-litellm-pr-1366.up.railway.app"
+
+        # Your bearer token
+        token = os.getenv("PROXY_MASTER_KEY")
+
+        headers = {"Authorization": f"Bearer {token}"}
+
+        # Make a request to the staging endpoint
+        response = requests.post(
+            staging_endpoint + "/key/generate", json=test_data, headers=headers
+        )
+
+        print(f"response: {response.text}")
+        assert response.status_code == 200
+        result = response.json()
+    except Exception as e:
+        print(traceback.format_exc())
+        pytest.fail(f"An error occurred {e}")
+
+
+# test_add_new_key()