From 2cdaae681d34b8065d8defc61405ca7da357e6e3 Mon Sep 17 00:00:00 2001 From: Eeshan Garg Date: Tue, 30 Nov 2021 20:10:40 -0500 Subject: [PATCH] actions: Rename do_change_plan_type -> do change_realm_plan_type. We will soon be adding an equivalent function for RemoteZulipServer, so it makes sense to rename this function to be more descriptive. --- analytics/tests/test_support_views.py | 6 +++--- analytics/views/support.py | 4 ++-- corporate/lib/stripe.py | 12 ++++++------ zerver/lib/actions.py | 4 ++-- zerver/lib/import_realm.py | 6 +++--- zerver/tests/test_events.py | 4 ++-- zerver/tests/test_home.py | 20 ++++++++++---------- zerver/tests/test_import_export.py | 4 ++-- zerver/tests/test_message_edit.py | 8 +++++--- zerver/tests/test_realm.py | 20 ++++++++++---------- zerver/tests/test_subs.py | 10 +++++----- zerver/tests/test_upload.py | 6 +++--- 12 files changed, 53 insertions(+), 51 deletions(-) diff --git a/analytics/tests/test_support_views.py b/analytics/tests/test_support_views.py index 4498f020cd..d5fe08f70b 100644 --- a/analytics/tests/test_support_views.py +++ b/analytics/tests/test_support_views.py @@ -334,7 +334,7 @@ class TestSupportEndpoint(ZulipTestCase): ["Billing method of zulip updated to pay by invoice"], result ) - def test_change_plan_type(self) -> None: + def test_change_realm_plan_type(self) -> None: cordelia = self.example_user("cordelia") self.login_user(cordelia) @@ -347,7 +347,7 @@ class TestSupportEndpoint(ZulipTestCase): iago = self.example_user("iago") self.login_user(iago) - with mock.patch("analytics.views.support.do_change_plan_type") as m: + with mock.patch("analytics.views.support.do_change_realm_plan_type") as m: result = self.client_post( "/activity/support", {"realm_id": f"{iago.realm_id}", "plan_type": "2"} ) @@ -356,7 +356,7 @@ class TestSupportEndpoint(ZulipTestCase): ["Plan type of zulip changed from self hosted to limited"], result ) - with mock.patch("analytics.views.support.do_change_plan_type") as m: + with mock.patch("analytics.views.support.do_change_realm_plan_type") as m: result = self.client_post( "/activity/support", {"realm_id": f"{iago.realm_id}", "plan_type": "10"} ) diff --git a/analytics/views/support.py b/analytics/views/support.py index f50c66ac3c..e6012472b0 100644 --- a/analytics/views/support.py +++ b/analytics/views/support.py @@ -19,8 +19,8 @@ from confirmation.settings import STATUS_ACTIVE from zerver.decorator import require_server_admin from zerver.forms import check_subdomain_available from zerver.lib.actions import ( - do_change_plan_type, do_change_realm_org_type, + do_change_realm_plan_type, do_change_realm_subdomain, do_deactivate_realm, do_scrub_realm, @@ -169,7 +169,7 @@ def support( assert isinstance(acting_user, UserProfile) if plan_type is not None: current_plan_type = realm.plan_type - do_change_plan_type(realm, plan_type, acting_user=acting_user) + do_change_realm_plan_type(realm, plan_type, acting_user=acting_user) msg = f"Plan type of {realm.string_id} changed from {get_plan_name(current_plan_type)} to {get_plan_name(plan_type)} " context["success_message"] = msg elif org_type is not None: diff --git a/corporate/lib/stripe.py b/corporate/lib/stripe.py index 72fd1f059c..6e56a18808 100644 --- a/corporate/lib/stripe.py +++ b/corporate/lib/stripe.py @@ -714,9 +714,9 @@ def process_initial_upgrade( ) stripe.Invoice.finalize_invoice(stripe_invoice) - from zerver.lib.actions import do_change_plan_type + from zerver.lib.actions import do_change_realm_plan_type - do_change_plan_type(realm, Realm.PLAN_TYPE_STANDARD, acting_user=user) + do_change_realm_plan_type(realm, Realm.PLAN_TYPE_STANDARD, acting_user=user) def update_license_ledger_for_manual_plan( @@ -937,9 +937,9 @@ def update_sponsorship_status( def approve_sponsorship(realm: Realm, *, acting_user: Optional[UserProfile]) -> None: - from zerver.lib.actions import do_change_plan_type, internal_send_private_message + from zerver.lib.actions import do_change_realm_plan_type, internal_send_private_message - do_change_plan_type(realm, Realm.PLAN_TYPE_STANDARD_FREE, acting_user=acting_user) + do_change_realm_plan_type(realm, Realm.PLAN_TYPE_STANDARD_FREE, acting_user=acting_user) customer = get_customer_by_realm(realm) if customer is not None and customer.sponsorship_pending: customer.sponsorship_pending = False @@ -986,10 +986,10 @@ def do_change_plan_status(plan: CustomerPlan, status: int) -> None: def process_downgrade(plan: CustomerPlan) -> None: - from zerver.lib.actions import do_change_plan_type + from zerver.lib.actions import do_change_realm_plan_type assert plan.customer.realm is not None - do_change_plan_type(plan.customer.realm, Realm.PLAN_TYPE_LIMITED, acting_user=None) + do_change_realm_plan_type(plan.customer.realm, Realm.PLAN_TYPE_LIMITED, acting_user=None) plan.status = CustomerPlan.ENDED plan.save(update_fields=["status"]) diff --git a/zerver/lib/actions.py b/zerver/lib/actions.py index 7a48c71712..7ab0aef64b 100644 --- a/zerver/lib/actions.py +++ b/zerver/lib/actions.py @@ -4619,7 +4619,7 @@ def do_change_realm_org_type( ) -def do_change_plan_type( +def do_change_realm_plan_type( realm: Realm, plan_type: int, *, acting_user: Optional[UserProfile] ) -> None: old_value = realm.plan_type @@ -5148,7 +5148,7 @@ def do_create_realm( realm.save(update_fields=["notifications_stream", "signup_notifications_stream"]) if plan_type is None and settings.BILLING_ENABLED: - do_change_plan_type(realm, Realm.PLAN_TYPE_LIMITED, acting_user=None) + do_change_realm_plan_type(realm, Realm.PLAN_TYPE_LIMITED, acting_user=None) admin_realm = get_realm(settings.SYSTEM_BOT_REALM) sender = get_system_bot(settings.NOTIFICATION_BOT, admin_realm.id) diff --git a/zerver/lib/import_realm.py b/zerver/lib/import_realm.py index fcd811304d..918f799037 100644 --- a/zerver/lib/import_realm.py +++ b/zerver/lib/import_realm.py @@ -20,7 +20,7 @@ from zerver.lib.actions import ( UserMessageLite, bulk_insert_ums, do_change_avatar_fields, - do_change_plan_type, + do_change_realm_plan_type, ) from zerver.lib.avatar_hash import user_avatar_path_from_ids from zerver.lib.bulk_create import bulk_create_users, bulk_set_users_or_streams_recipient_fields @@ -1286,9 +1286,9 @@ def do_import_realm(import_dir: Path, subdomain: str, processes: int = 1) -> Rea import_analytics_data(realm=realm, import_dir=import_dir) if settings.BILLING_ENABLED: - do_change_plan_type(realm, Realm.PLAN_TYPE_LIMITED, acting_user=None) + do_change_realm_plan_type(realm, Realm.PLAN_TYPE_LIMITED, acting_user=None) else: - do_change_plan_type(realm, Realm.PLAN_TYPE_SELF_HOSTED, acting_user=None) + do_change_realm_plan_type(realm, Realm.PLAN_TYPE_SELF_HOSTED, acting_user=None) return realm diff --git a/zerver/tests/test_events.py b/zerver/tests/test_events.py index 03a9bb98a0..c7c59f745e 100644 --- a/zerver/tests/test_events.py +++ b/zerver/tests/test_events.py @@ -39,8 +39,8 @@ from zerver.lib.actions import ( do_change_full_name, do_change_icon_source, do_change_logo_source, - do_change_plan_type, do_change_realm_domain, + do_change_realm_plan_type, do_change_stream_description, do_change_stream_message_retention_days, do_change_stream_permission, @@ -1546,7 +1546,7 @@ class NormalActionsTest(BaseAction): self.assertEqual(state_data["zulip_plan_is_not_limited"], True) events = self.verify_action( - lambda: do_change_plan_type( + lambda: do_change_realm_plan_type( realm, Realm.PLAN_TYPE_LIMITED, acting_user=self.user_profile ) ) diff --git a/zerver/tests/test_home.py b/zerver/tests/test_home.py index eb7dbc56c2..c0e783424f 100644 --- a/zerver/tests/test_home.py +++ b/zerver/tests/test_home.py @@ -13,7 +13,7 @@ from django.test import override_settings from django.utils.timezone import now as timezone_now from corporate.models import Customer, CustomerPlan -from zerver.lib.actions import change_user_is_active, do_change_plan_type, do_create_user +from zerver.lib.actions import change_user_is_active, do_change_realm_plan_type, do_create_user from zerver.lib.compatibility import LAST_SERVER_UPGRADE_TIME, is_outdated_server from zerver.lib.home import ( get_billing_info, @@ -752,7 +752,7 @@ class HomeTest(ZulipTestCase): self.assertFalse(billing_info.show_plans) # realm owner, with inactive CustomerPlan and realm plan_type LIMITED -> show billing link and plans - do_change_plan_type(user.realm, Realm.PLAN_TYPE_LIMITED, acting_user=None) + do_change_realm_plan_type(user.realm, Realm.PLAN_TYPE_LIMITED, acting_user=None) with self.settings(CORPORATE_ENABLED=True): billing_info = get_billing_info(user) self.assertTrue(billing_info.show_billing) @@ -781,7 +781,7 @@ class HomeTest(ZulipTestCase): # billing admin, with CustomerPlan and realm plan_type STANDARD -> show only billing link user.role = UserProfile.ROLE_MEMBER user.is_billing_admin = True - do_change_plan_type(user.realm, Realm.PLAN_TYPE_STANDARD, acting_user=None) + do_change_realm_plan_type(user.realm, Realm.PLAN_TYPE_STANDARD, acting_user=None) user.save(update_fields=["role", "is_billing_admin"]) with self.settings(CORPORATE_ENABLED=True): billing_info = get_billing_info(user) @@ -789,7 +789,7 @@ class HomeTest(ZulipTestCase): self.assertFalse(billing_info.show_plans) # billing admin, with CustomerPlan and realm plan_type PLUS -> show only billing link - do_change_plan_type(user.realm, Realm.PLAN_TYPE_PLUS, acting_user=None) + do_change_realm_plan_type(user.realm, Realm.PLAN_TYPE_PLUS, acting_user=None) user.save(update_fields=["role", "is_billing_admin"]) with self.settings(CORPORATE_ENABLED=True): billing_info = get_billing_info(user) @@ -797,7 +797,7 @@ class HomeTest(ZulipTestCase): self.assertFalse(billing_info.show_plans) # member, with CustomerPlan and realm plan_type STANDARD -> neither billing link or plans - do_change_plan_type(user.realm, Realm.PLAN_TYPE_STANDARD, acting_user=None) + do_change_realm_plan_type(user.realm, Realm.PLAN_TYPE_STANDARD, acting_user=None) user.is_billing_admin = False user.save(update_fields=["is_billing_admin"]) with self.settings(CORPORATE_ENABLED=True): @@ -808,7 +808,7 @@ class HomeTest(ZulipTestCase): # guest, with CustomerPlan and realm plan_type SELF_HOSTED -> neither billing link or plans user.role = UserProfile.ROLE_GUEST user.save(update_fields=["role"]) - do_change_plan_type(user.realm, Realm.PLAN_TYPE_SELF_HOSTED, acting_user=None) + do_change_realm_plan_type(user.realm, Realm.PLAN_TYPE_SELF_HOSTED, acting_user=None) with self.settings(CORPORATE_ENABLED=True): billing_info = get_billing_info(user) self.assertFalse(billing_info.show_billing) @@ -842,7 +842,7 @@ class HomeTest(ZulipTestCase): def test_promote_sponsoring_zulip_in_realm(self) -> None: realm = get_realm("zulip") - do_change_plan_type(realm, Realm.PLAN_TYPE_STANDARD_FREE, acting_user=None) + do_change_realm_plan_type(realm, Realm.PLAN_TYPE_STANDARD_FREE, acting_user=None) promote_zulip = promote_sponsoring_zulip_in_realm(realm) self.assertTrue(promote_zulip) @@ -850,15 +850,15 @@ class HomeTest(ZulipTestCase): promote_zulip = promote_sponsoring_zulip_in_realm(realm) self.assertFalse(promote_zulip) - do_change_plan_type(realm, Realm.PLAN_TYPE_STANDARD_FREE, acting_user=None) + do_change_realm_plan_type(realm, Realm.PLAN_TYPE_STANDARD_FREE, acting_user=None) promote_zulip = promote_sponsoring_zulip_in_realm(realm) self.assertTrue(promote_zulip) - do_change_plan_type(realm, Realm.PLAN_TYPE_LIMITED, acting_user=None) + do_change_realm_plan_type(realm, Realm.PLAN_TYPE_LIMITED, acting_user=None) promote_zulip = promote_sponsoring_zulip_in_realm(realm) self.assertFalse(promote_zulip) - do_change_plan_type(realm, Realm.PLAN_TYPE_STANDARD, acting_user=None) + do_change_realm_plan_type(realm, Realm.PLAN_TYPE_STANDARD, acting_user=None) promote_zulip = promote_sponsoring_zulip_in_realm(realm) self.assertFalse(promote_zulip) diff --git a/zerver/tests/test_import_export.py b/zerver/tests/test_import_export.py index 4f3fc8eba0..c0eb00075c 100644 --- a/zerver/tests/test_import_export.py +++ b/zerver/tests/test_import_export.py @@ -14,7 +14,7 @@ from zerver.lib.actions import ( do_add_reaction, do_change_icon_source, do_change_logo_source, - do_change_plan_type, + do_change_realm_plan_type, do_create_user, do_deactivate_user, do_mute_user, @@ -1313,7 +1313,7 @@ class ImportExportTest(ZulipTestCase): def test_plan_type(self) -> None: realm = get_realm("zulip") - do_change_plan_type(realm, Realm.PLAN_TYPE_LIMITED, acting_user=None) + do_change_realm_plan_type(realm, Realm.PLAN_TYPE_LIMITED, acting_user=None) self._setup_export_files(realm) self._export_realm(realm) diff --git a/zerver/tests/test_message_edit.py b/zerver/tests/test_message_edit.py index 57d595e115..14cd36ef77 100644 --- a/zerver/tests/test_message_edit.py +++ b/zerver/tests/test_message_edit.py @@ -9,7 +9,7 @@ from django.http import HttpResponse from django.utils.timezone import now as timezone_now from zerver.lib.actions import ( - do_change_plan_type, + do_change_realm_plan_type, do_change_stream_post_policy, do_change_user_role, do_deactivate_stream, @@ -366,14 +366,16 @@ class EditMessageTest(EditMessageTestCase): self.assertEqual(result.json()["raw_content"], "web-public message") # Verify LIMITED plan type does not allow web-public access. - do_change_plan_type(user_profile.realm, Realm.PLAN_TYPE_LIMITED, acting_user=None) + do_change_realm_plan_type(user_profile.realm, Realm.PLAN_TYPE_LIMITED, acting_user=None) result = self.client_get("/json/messages/" + str(web_public_stream_msg_id)) self.assert_json_error( result, "Not logged in: API authentication or user session required", 401 ) # Verify works with STANDARD_FREE plan type too. - do_change_plan_type(user_profile.realm, Realm.PLAN_TYPE_STANDARD_FREE, acting_user=None) + do_change_realm_plan_type( + user_profile.realm, Realm.PLAN_TYPE_STANDARD_FREE, acting_user=None + ) result = self.client_get("/json/messages/" + str(web_public_stream_msg_id)) self.assert_json_success(result) self.assertEqual(result.json()["raw_content"], "web-public message") diff --git a/zerver/tests/test_realm.py b/zerver/tests/test_realm.py index 8680ba6202..2ce6610978 100644 --- a/zerver/tests/test_realm.py +++ b/zerver/tests/test_realm.py @@ -10,8 +10,8 @@ from django.utils.timezone import now as timezone_now from confirmation.models import Confirmation, create_confirmation_link from zerver.lib.actions import ( do_add_deactivated_redirect, - do_change_plan_type, do_change_realm_org_type, + do_change_realm_plan_type, do_change_realm_subdomain, do_create_realm, do_deactivate_realm, @@ -671,7 +671,7 @@ class RealmTest(ZulipTestCase): self.assertEqual(realm_audit_log.acting_user, iago) self.assertEqual(realm.org_type, Realm.ORG_TYPES["government"]["id"]) - def test_change_plan_type(self) -> None: + def test_change_realm_plan_type(self) -> None: realm = get_realm("zulip") iago = self.example_user("iago") self.assertEqual(realm.plan_type, Realm.PLAN_TYPE_SELF_HOSTED) @@ -679,7 +679,7 @@ class RealmTest(ZulipTestCase): self.assertEqual(realm.message_visibility_limit, None) self.assertEqual(realm.upload_quota_gb, None) - do_change_plan_type(realm, Realm.PLAN_TYPE_STANDARD, acting_user=iago) + do_change_realm_plan_type(realm, Realm.PLAN_TYPE_STANDARD, acting_user=iago) realm = get_realm("zulip") realm_audit_log = RealmAuditLog.objects.filter( event_type=RealmAuditLog.REALM_PLAN_TYPE_CHANGED @@ -696,29 +696,29 @@ class RealmTest(ZulipTestCase): self.assertEqual(realm.message_visibility_limit, None) self.assertEqual(realm.upload_quota_gb, Realm.UPLOAD_QUOTA_STANDARD) - do_change_plan_type(realm, Realm.PLAN_TYPE_LIMITED, acting_user=iago) + do_change_realm_plan_type(realm, Realm.PLAN_TYPE_LIMITED, acting_user=iago) realm = get_realm("zulip") self.assertEqual(realm.plan_type, Realm.PLAN_TYPE_LIMITED) self.assertEqual(realm.max_invites, settings.INVITES_DEFAULT_REALM_DAILY_MAX) self.assertEqual(realm.message_visibility_limit, Realm.MESSAGE_VISIBILITY_LIMITED) self.assertEqual(realm.upload_quota_gb, Realm.UPLOAD_QUOTA_LIMITED) - do_change_plan_type(realm, Realm.PLAN_TYPE_STANDARD_FREE, acting_user=iago) + do_change_realm_plan_type(realm, Realm.PLAN_TYPE_STANDARD_FREE, acting_user=iago) realm = get_realm("zulip") self.assertEqual(realm.plan_type, Realm.PLAN_TYPE_STANDARD_FREE) self.assertEqual(realm.max_invites, Realm.INVITES_STANDARD_REALM_DAILY_MAX) self.assertEqual(realm.message_visibility_limit, None) self.assertEqual(realm.upload_quota_gb, Realm.UPLOAD_QUOTA_STANDARD) - do_change_plan_type(realm, Realm.PLAN_TYPE_LIMITED, acting_user=iago) - do_change_plan_type(realm, Realm.PLAN_TYPE_PLUS, acting_user=iago) + do_change_realm_plan_type(realm, Realm.PLAN_TYPE_LIMITED, acting_user=iago) + do_change_realm_plan_type(realm, Realm.PLAN_TYPE_PLUS, acting_user=iago) realm = get_realm("zulip") self.assertEqual(realm.plan_type, Realm.PLAN_TYPE_PLUS) self.assertEqual(realm.max_invites, Realm.INVITES_STANDARD_REALM_DAILY_MAX) self.assertEqual(realm.message_visibility_limit, None) self.assertEqual(realm.upload_quota_gb, Realm.UPLOAD_QUOTA_STANDARD) - do_change_plan_type(realm, Realm.PLAN_TYPE_SELF_HOSTED, acting_user=iago) + do_change_realm_plan_type(realm, Realm.PLAN_TYPE_SELF_HOSTED, acting_user=iago) self.assertEqual(realm.plan_type, Realm.PLAN_TYPE_SELF_HOSTED) self.assertEqual(realm.max_invites, settings.INVITES_DEFAULT_REALM_DAILY_MAX) self.assertEqual(realm.message_visibility_limit, None) @@ -759,12 +759,12 @@ class RealmTest(ZulipTestCase): result = self.client_patch("/json/realm", req) self.assert_json_success(result) - do_change_plan_type(realm, Realm.PLAN_TYPE_LIMITED, acting_user=None) + do_change_realm_plan_type(realm, Realm.PLAN_TYPE_LIMITED, acting_user=None) req = dict(message_retention_days=orjson.dumps(10).decode()) result = self.client_patch("/json/realm", req) self.assert_json_error(result, "Available on Zulip Standard. Upgrade to access.") - do_change_plan_type(realm, Realm.PLAN_TYPE_STANDARD, acting_user=None) + do_change_realm_plan_type(realm, Realm.PLAN_TYPE_STANDARD, acting_user=None) req = dict(message_retention_days=orjson.dumps(10).decode()) result = self.client_patch("/json/realm", req) self.assert_json_success(result) diff --git a/zerver/tests/test_subs.py b/zerver/tests/test_subs.py index e4749abf09..d6f3755300 100644 --- a/zerver/tests/test_subs.py +++ b/zerver/tests/test_subs.py @@ -18,7 +18,7 @@ from zerver.lib.actions import ( do_add_streams_to_default_stream_group, do_change_default_stream_group_description, do_change_default_stream_group_name, - do_change_plan_type, + do_change_realm_plan_type, do_change_stream_post_policy, do_change_subscription_property, do_change_user_role, @@ -1276,7 +1276,7 @@ class StreamAdminTest(ZulipTestCase): user_profile = self.example_user("desdemona") self.login_user(user_profile) realm = user_profile.realm - do_change_plan_type(realm, Realm.PLAN_TYPE_LIMITED, acting_user=None) + do_change_realm_plan_type(realm, Realm.PLAN_TYPE_LIMITED, acting_user=None) stream = self.subscribe(user_profile, "stream_name1") result = self.client_patch( @@ -1284,7 +1284,7 @@ class StreamAdminTest(ZulipTestCase): ) self.assert_json_error(result, "Available on Zulip Standard. Upgrade to access.") - do_change_plan_type(realm, Realm.PLAN_TYPE_SELF_HOSTED, acting_user=None) + do_change_realm_plan_type(realm, Realm.PLAN_TYPE_SELF_HOSTED, acting_user=None) events: List[Mapping[str, Any]] = [] with self.tornado_redirected_to_list(events, expected_num_events=1): result = self.client_patch( @@ -1450,13 +1450,13 @@ class StreamAdminTest(ZulipTestCase): }, ] - do_change_plan_type(realm, Realm.PLAN_TYPE_LIMITED, acting_user=admin) + do_change_realm_plan_type(realm, Realm.PLAN_TYPE_LIMITED, acting_user=admin) with self.assertRaisesRegex( JsonableError, "Available on Zulip Standard. Upgrade to access." ): list_to_streams(streams_raw, owner, autocreate=True) - do_change_plan_type(realm, Realm.PLAN_TYPE_SELF_HOSTED, acting_user=admin) + do_change_realm_plan_type(realm, Realm.PLAN_TYPE_SELF_HOSTED, acting_user=admin) result = list_to_streams(streams_raw, owner, autocreate=True) self.assert_length(result[0], 0) self.assert_length(result[1], 3) diff --git a/zerver/tests/test_upload.py b/zerver/tests/test_upload.py index 2352337a51..c8ed1fc16b 100644 --- a/zerver/tests/test_upload.py +++ b/zerver/tests/test_upload.py @@ -21,7 +21,7 @@ import zerver.lib.upload from zerver.lib.actions import ( do_change_icon_source, do_change_logo_source, - do_change_plan_type, + do_change_realm_plan_type, do_create_realm, do_delete_old_unclaimed_attachments, do_set_realm_property, @@ -1544,7 +1544,7 @@ class RealmLogoTest(UploadSerializeMixin, ZulipTestCase): def test_upload_limited_plan_type(self) -> None: user_profile = self.example_user("iago") - do_change_plan_type(user_profile.realm, Realm.PLAN_TYPE_LIMITED, acting_user=None) + do_change_realm_plan_type(user_profile.realm, Realm.PLAN_TYPE_LIMITED, acting_user=None) self.login_user(user_profile) with get_test_image_file(self.correct_files[0][0]) as fp: result = self.client_post( @@ -1588,7 +1588,7 @@ class RealmLogoTest(UploadSerializeMixin, ZulipTestCase): f"/user_avatars/{realm.id}/realm/{file_name}?version=2&night={is_night_str}", ) - do_change_plan_type(realm, Realm.PLAN_TYPE_LIMITED, acting_user=user_profile) + do_change_realm_plan_type(realm, Realm.PLAN_TYPE_LIMITED, acting_user=user_profile) if self.night: self.assertEqual(realm.night_logo_source, Realm.LOGO_UPLOADED) else: