paulmataruso/zulip
1
0
Fork 0
mirror of https://github.com/zulip/zulip.git synced 2025-11-04 22:13:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

test_decorator: Add explicit subdomains in tests.

Browse Source
This commit is contained in:
Tim Abbott
2017-08-28 22:40:56 -07:00
parent a8b9ffc020
commit 4a22316d90
3 changed files with 16 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
zerver/lib/test_classes.py
View File

@@ -273,12 +273,13 @@ class ZulipTestCase(TestCase):
# type: () -> UserProfile
return get_user('notification-bot@zulip.com', get_realm('zulip'))
def login_with_return(self, email, password=None):
# type: (Text, Optional[Text]) -> HttpResponse
def login_with_return(self, email, password=None, **kwargs):
# type: (Text, Optional[Text], **Any) -> HttpResponse
if password is None:
password = initial_password(email)
return self.client_post('/accounts/login/',
{'username': email, 'password': password})
{'username': email, 'password': password},
**kwargs)
def login(self, email, password=None, fails=False):
# type: (Text, Optional[Text], bool) -> HttpResponse

3
zerver/lib/test_helpers.py
View File

@@ -286,6 +286,9 @@ class HostRequestMock(object):
self.path = ''
self.user = user_profile
self.method = ''
self.body = ''
self.content_type = ''
self._email = ''
def get_host(self):
# type: () -> Text

21
zerver/tests/test_decorators.py
View File

@@ -219,22 +219,15 @@ class DecoratorTestCase(TestCase):
# type: (HttpRequest, UserProfile) -> None
raise Exception("raised by webhook function")
class Request(HostRequestMock):
GET = {} # type: Dict[str, str]
POST = {} # type: Dict[str, str]
COOKIES = {} # type: Dict[str, str]
META = {'PATH_INFO': ''}
webhook_bot_email = 'webhook-bot@zulip.com'
webhook_bot_realm = get_realm('zulip')
webhook_bot = get_user(webhook_bot_email, webhook_bot_realm)
webhook_bot_api_key = webhook_bot.api_key
webhook_client_name = "ZulipClientNameWebhook"
request = Request() # type: Any
request.host = settings.EXTERNAL_HOST
request = HostRequestMock()
request.POST['api_key'] = 'not_existing_api_key'
with self.assertRaisesRegex(JsonableError, "Invalid API key"):
my_webhook(request)
@@ -261,6 +254,7 @@ class DecoratorTestCase(TestCase):
"User {} attempted to access API on wrong "
"subdomain {}".format(webhook_bot_email, 'acme'))
request.host = "zulip.testserver"
# Test when content_type is application/json and request.body
# is valid JSON; exception raised in the webhook function
# should be re-raised
@@ -671,7 +665,8 @@ class DeactivatedRealmTest(ZulipTestCase):
"""
do_deactivate_realm(get_realm("zulip"))
result = self.login_with_return(self.example_email("hamlet"))
result = self.login_with_return(self.example_email("hamlet"),
subdomain="zulip")
self.assert_in_response("has been deactivated", result)
def test_webhook_deactivated_realm(self):
@@ -823,7 +818,7 @@ class InactiveUserTest(ZulipTestCase):
password = initial_password(user_profile.email)
request = mock.MagicMock()
request.get_host.return_value = 'testserver'
request.get_host.return_value = 'zulip.testserver'
# Test a mirror-dummy active user.
form = OurAuthenticationForm(request,
@@ -926,7 +921,9 @@ class TestValidateApiKey(ZulipTestCase):
def test_validate_api_key_if_profile_is_incoming_webhook_and_is_webhook_is_set(self):
# type: () -> None
profile = validate_api_key(HostRequestMock(), self.webhook_bot.email, self.webhook_bot.api_key, is_webhook=True)
profile = validate_api_key(HostRequestMock(host="zulip.testserver"),
self.webhook_bot.email, self.webhook_bot.api_key,
is_webhook=True)
self.assertEqual(profile.pk, self.webhook_bot.pk)
def test_valid_api_key_if_user_is_on_wrong_subdomain(self):