From 7f06a7fa2a06670b0276da5480c81d34ae0fb68f Mon Sep 17 00:00:00 2001 From: Vishnu Ks Date: Mon, 22 May 2017 16:26:38 -0700 Subject: [PATCH] Add ZulipTestCase.mit_user() function. --- zerver/lib/test_classes.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/zerver/lib/test_classes.py b/zerver/lib/test_classes.py index 30c3791d66..c84b102ed5 100644 --- a/zerver/lib/test_classes.py +++ b/zerver/lib/test_classes.py @@ -218,11 +218,22 @@ class ZulipTestCase(TestCase): ZOE='ZOE@zulip.com', ) + mit_user_map = dict( + sipbtest="sipbtest@mit.edu", + starnine="starnine@mit.edu", + espuser="espuser@mit.edu", + ) + def example_user(self, name): # type: (str) -> UserProfile email = self.example_user_map[name] return get_user_profile_by_email(email) + def mit_user(self, name): + # type: (str) -> UserProfile + email = self.mit_user_map[name] + return get_user_profile_by_email(email) + def notification_bot(self): # type: () -> UserProfile return get_user_profile_by_email('notification-bot@zulip.com')