2018-05-23 05:40:28 +02:00
|
|
|
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(
|
python: Use trailing commas consistently.
Automatically generated by the following script, based on the output
of lint with flake8-comma:
import re
import sys
last_filename = None
last_row = None
lines = []
for msg in sys.stdin:
m = re.match(
r"\x1b\[35mflake8 \|\x1b\[0m \x1b\[1;31m(.+):(\d+):(\d+): (\w+)", msg
)
if m:
filename, row_str, col_str, err = m.groups()
row, col = int(row_str), int(col_str)
if filename == last_filename:
assert last_row != row
else:
if last_filename is not None:
with open(last_filename, "w") as f:
f.writelines(lines)
with open(filename) as f:
lines = f.readlines()
last_filename = filename
last_row = row
line = lines[row - 1]
if err in ["C812", "C815"]:
lines[row - 1] = line[: col - 1] + "," + line[col - 1 :]
elif err in ["C819"]:
assert line[col - 2] == ","
lines[row - 1] = line[: col - 2] + line[col - 1 :].lstrip(" ")
if last_filename is not None:
with open(last_filename, "w") as f:
f.writelines(lines)
Signed-off-by: Anders Kaseorg <anders@zulipchat.com>
2020-04-10 05:23:40 +02:00
|
|
|
email='test-bot@zulip.com',
|
2018-05-23 05:40:28 +02:00
|
|
|
).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())
|
