Add ZulipTestCase.mit_user() function.

shreya16aug · May 23, 2017 · 7f06a7f · 7f06a7f
1 parent 0595107
commit 7f06a7f
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/zerver/lib/test_classes.py b/zerver/lib/test_classes.py
@@ -218,11 +218,22 @@ def client_get(self, url, info={}, **kwargs):
         ZOE='[email protected]',
     )
 
+    mit_user_map = dict(
+        sipbtest="[email protected]",
+        starnine="[email protected]",
+        espuser="[email protected]",
+    )
+
     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('[email protected]')