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')