From 39b012a27666e6f92464aaaf96f04e2a63e0dc07 Mon Sep 17 00:00:00 2001 From: Mateusz Mandera Date: Tue, 28 Jan 2020 14:41:08 +0100 Subject: [PATCH] server_initialization: Set internal bots owners to themselves. --- zerver/lib/server_initialization.py | 5 +++++ zerver/tests/test_subs.py | 2 +- 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/zerver/lib/server_initialization.py b/zerver/lib/server_initialization.py index 7bcf1708e7..ab68a0571a 100644 --- a/zerver/lib/server_initialization.py +++ b/zerver/lib/server_initialization.py @@ -17,6 +17,11 @@ def create_internal_realm() -> None: internal_realm_bots = [(bot['name'], bot['email_template'] % (settings.INTERNAL_BOT_DOMAIN,)) for bot in settings.INTERNAL_BOTS] create_users(internal_realm, internal_realm_bots, bot_type=UserProfile.DEFAULT_BOT) + # Set the owners for these bots to the bots themselves + bots = UserProfile.objects.filter(email__in=[bot_info[1] for bot_info in internal_realm_bots]) + for bot in bots: + bot.bot_owner = bot + bot.save() # Initialize the email gateway bot as an API Super User email_gateway_bot = get_system_bot(settings.EMAIL_GATEWAY_BOT) diff --git a/zerver/tests/test_subs.py b/zerver/tests/test_subs.py index 06158ba3af..b36a665c9c 100644 --- a/zerver/tests/test_subs.py +++ b/zerver/tests/test_subs.py @@ -3183,7 +3183,7 @@ class SubscriptionAPITest(ZulipTestCase): dict(principals=ujson.dumps([user1.email, user2.email])), invite_only=True, ) - self.assert_length(queries, 38) + self.assert_length(queries, 40) # Test creating a public stream with announce when realm has a notification stream. notifications_stream = get_stream(self.streams[0], self.test_realm)