paulmataruso/zulip
1
0
Fork 0
mirror of https://github.com/zulip/zulip.git synced 2025-11-06 15:03:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

zerver/lib/test_helpers.py: Fix string annotations.

Browse Source
This commit is contained in:
Eklavya Sharma
2016-06-29 00:22:27 +05:30
parent d9eb711e5e
commit 2cba2caa7f
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
zerver/lib/test_helpers.py
View File

@@ -236,17 +236,17 @@ class AuthedTestCase(TestCase):
# Helper because self.client.patch annoying requires you to urlencode
def client_patch(self, url, info={}, **kwargs):
# type: (str, Dict[str, Any], **Any) -> HttpResponse
# type: (text_type, Dict[str, Any], **Any) -> HttpResponse
encoded = urllib.parse.urlencode(info)
return self.client.patch(url, encoded, **kwargs)
def client_put(self, url, info={}, **kwargs):
# type: (str, Dict[str, Any], **Any) -> HttpResponse
# type: (text_type, Dict[str, Any], **Any) -> HttpResponse
encoded = urllib.parse.urlencode(info)
return self.client.put(url, encoded, **kwargs)
def client_delete(self, url, info={}, **kwargs):
# type: (str, Dict[str, Any], **Any) -> HttpResponse
# type: (text_type, Dict[str, Any], **Any) -> HttpResponse
encoded = urllib.parse.urlencode(info)
return self.client.delete(url, encoded, **kwargs)
@@ -311,7 +311,7 @@ class AuthedTestCase(TestCase):
def send_message(self, sender_name, raw_recipients, message_type,
content=u"test content", subject=u"test", **kwargs):
# type: (str, Union[text_type, List[text_type]], int, text_type, text_type, **Any) -> int
# type: (text_type, Union[text_type, List[text_type]], int, text_type, text_type, **Any) -> int
sender = get_user_profile_by_email(sender_name)
if message_type == Recipient.PERSONAL:
message_type_name = "private"
@@ -367,7 +367,7 @@ class AuthedTestCase(TestCase):
return json['msg']
def assert_json_error(self, result, msg, status_code=400):
# type: (HttpResponse, str, int) -> None
# type: (HttpResponse, text_type, int) -> None
"""
Invalid POSTs return an error status code and JSON of the form
{"result": "error", "msg": "reason"}.
@@ -384,7 +384,7 @@ class AuthedTestCase(TestCase):
"len(%s) == %s, > %s" % (queries, actual_count, count))
def assert_json_error_contains(self, result, msg_substring, status_code=400):
# type: (HttpResponse, str, int) -> None
# type: (HttpResponse, text_type, int) -> None
self.assertIn(msg_substring, self.get_json_error(result, status_code=status_code))
def fixture_data(self, type, action, file_type='json'):
@@ -405,7 +405,7 @@ class AuthedTestCase(TestCase):
# Subscribe to a stream by making an API request
def common_subscribe_to_streams(self, email, streams, extra_post_data={}, invite_only=False):
# type: (str, Iterable[text_type], Dict[str, Any], bool) -> HttpResponse
# type: (text_type, Iterable[text_type], Dict[str, Any], bool) -> HttpResponse
post_data = {'subscriptions': ujson.dumps([{"name": stream} for stream in streams]),
'invite_only': ujson.dumps(invite_only)}
post_data.update(extra_post_data)