From 5a852ca6472b41f2f4f6d2b12ba29e35b8ce736b Mon Sep 17 00:00:00 2001
From: "Jugal D. Bhatt" <55304795+jugaldb@users.noreply.github.com>
Date: Thu, 3 Jul 2025 05:27:49 +0530
Subject: [PATCH] Add fix to tests (#12263)

---
 .../send_emails/test_base_email.py            | 20 +++++++++----------
 1 file changed, 10 insertions(+), 10 deletions(-)

diff --git a/tests/test_litellm/enterprise/enterprise_callbacks/send_emails/test_base_email.py b/tests/test_litellm/enterprise/enterprise_callbacks/send_emails/test_base_email.py
index af6411b2413..4fac979ce65 100644
--- a/tests/test_litellm/enterprise/enterprise_callbacks/send_emails/test_base_email.py
+++ b/tests/test_litellm/enterprise/enterprise_callbacks/send_emails/test_base_email.py
@@ -67,8 +67,8 @@ async def test_send_key_created_email(
     with mock.patch.dict(
         os.environ,
         {
-            "EMAIL_LOGO_URL": "https://test-logo.com",
-            "EMAIL_SUPPORT_CONTACT": "support@test.com",
+            "EMAIL_LOGO_URL": "https://litellm-listing.s3.amazonaws.com/litellm_logo.png",
+            "EMAIL_SUPPORT_CONTACT": "support@berri.ai",
             "PROXY_BASE_URL": "http://test.com",
         },
     ):
@@ -103,8 +103,8 @@ async def test_send_user_invitation_email(
     with mock.patch.dict(
         os.environ,
         {
-            "EMAIL_LOGO_URL": "https://test-logo.com",
-            "EMAIL_SUPPORT_CONTACT": "support@test.com",
+            "EMAIL_LOGO_URL": "https://litellm-listing.s3.amazonaws.com/litellm_logo.png",
+            "EMAIL_SUPPORT_CONTACT": "support@berri.ai",
             "PROXY_BASE_URL": "http://test.com",
         },
     ):
@@ -140,8 +140,8 @@ async def test_send_user_invitation_email_from_db(
     with mock.patch.dict(
         os.environ,
         {
-            "EMAIL_LOGO_URL": "https://test-logo.com",
-            "EMAIL_SUPPORT_CONTACT": "support@test.com",
+            "EMAIL_LOGO_URL": "https://litellm-listing.s3.amazonaws.com/litellm_logo.png",
+            "EMAIL_SUPPORT_CONTACT": "support@berri.ai",
             "PROXY_BASE_URL": "http://test.com",
         },
     ):
@@ -360,8 +360,8 @@ async def test_get_email_params_user_invitation(
     with mock.patch.dict(
         os.environ,
         {
-            "EMAIL_LOGO_URL": "https://test-logo.com",
-            "EMAIL_SUPPORT_CONTACT": "support@test.com",
+            "EMAIL_LOGO_URL": "https://litellm-listing.s3.amazonaws.com/litellm_logo.png",
+            "EMAIL_SUPPORT_CONTACT": "support@berri.ai",
             "PROXY_BASE_URL": "http://test.com",
         },
     ):
@@ -378,8 +378,8 @@ async def test_get_email_params_user_invitation(
                 user_email="test@example.com",
             )
 
-            assert result.logo_url == "https://test-logo.com"
-            assert result.support_contact == "support@test.com"
+            assert result.logo_url == "https://litellm-listing.s3.amazonaws.com/litellm_logo.png"
+            assert result.support_contact == "support@berri.ai"
             assert result.base_url == "http://test.com/ui?invitation_id=test-id"
             assert result.recipient_email == "test@example.com"