mirror of
https://github.com/zulip/zulip.git
synced 2025-11-05 22:43:42 +00:00
d343f25cc6
We extract the entire operations of the management command to a function create_if_missing_realm_internal_bots in the zerver/lib/onboarding.py. The logic for determining if there are any realm internal bots which have not been created is extracted to a function missing_any_realm_internal_bots in actions.py.
26 lines
1017 B
Python
26 lines
1017 B
Python
# -*- coding: utf-8 -*-
|
|
|
|
from zerver.models import Realm, UserProfile
|
|
from zerver.lib.onboarding import create_if_missing_realm_internal_bots
|
|
from zerver.lib.test_classes import (
|
|
ZulipTestCase,
|
|
)
|
|
|
|
class TestRealmInternalBotCreation(ZulipTestCase):
|
|
def test_create_if_missing_realm_internal_bots(self) -> None:
|
|
realm_internal_bots_dict = [{'var_name': 'TEST_BOT',
|
|
'email_template': 'test-bot@%s',
|
|
'name': 'Test Bot'}]
|
|
|
|
def check_test_bot_exists() -> bool:
|
|
all_realms_count = Realm.objects.count()
|
|
all_test_bot_count = UserProfile.objects.filter(
|
|
email='test-bot@zulip.com'
|
|
).count()
|
|
return all_realms_count == all_test_bot_count
|
|
|
|
self.assertFalse(check_test_bot_exists())
|
|
with self.settings(REALM_INTERNAL_BOTS=realm_internal_bots_dict):
|
|
create_if_missing_realm_internal_bots()
|
|
self.assertTrue(check_test_bot_exists())
|