diff --git a/corporate/tests/test_support_views.py b/corporate/tests/test_support_views.py index 50091a0ea1..5ed4294224 100644 --- a/corporate/tests/test_support_views.py +++ b/corporate/tests/test_support_views.py @@ -40,6 +40,8 @@ import uuid from zilencer.models import ( RemoteRealm, RemoteRealmAuditLog, + RemoteRealmBillingUser, + RemoteServerBillingUser, RemoteZulipServer, RemoteZulipServerAuditLog, ) @@ -159,6 +161,15 @@ class TestRemoteServerSupportEndpoint(ZulipTestCase): # without upgrade scheduled add_legacy_plan(name="realm-name-5", upgrade=False) + # Add billing users + remote_realm = RemoteRealm.objects.get(name="realm-name-3") + RemoteRealmBillingUser.objects.create( + remote_realm=remote_realm, email="realm-admin@example.com", user_uuid=uuid.uuid4() + ) + RemoteServerBillingUser.objects.create( + remote_server=remote_realm.server, email="server-admin@example.com" + ) + def test_search(self) -> None: def assert_server_details_in_response( html_response: "TestHttpResponse", hostname: str @@ -168,6 +179,7 @@ class TestRemoteServerSupportEndpoint(ZulipTestCase): 'remote server', f"