Mirror
/
zulip
mirror of https://github.com/zulip/zulip.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

tests Add PM tests for test_export_realm_with_exportable_user_ids.

This commit is contained in:
Vishnu Ks 2019-05-21 17:21:30 +05:30 committed by Tim Abbott
parent 044d620ee0
commit 0a466914c4
1 changed files with 23 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
zerver/tests/test_import_export.py
View File

@ -443,25 +443,43 @@ class ImportExportTest(ZulipTestCase):
def test_export_realm_with_exportable_user_ids(self) -> None: def test_export_realm_with_exportable_user_ids(self) -> None:
realm = Realm.objects.get(string_id='zulip') realm = Realm.objects.get(string_id='zulip')
cordelia = self.example_user('cordelia') cordelia = self.example_user('iago')
hamlet = self.example_user('hamlet') hamlet = self.example_user('hamlet')
user_ids = set([cordelia.id, hamlet.id]) user_ids = set([cordelia.id, hamlet.id])
pm_a_msg_id = self.send_personal_message(self.example_email("AARON"), self.example_email("othello"))
pm_b_msg_id = self.send_personal_message(self.example_email("cordelia"), self.example_email("iago"))
pm_c_msg_id = self.send_personal_message(self.example_email("hamlet"), self.example_email("othello"))
pm_d_msg_id = self.send_personal_message(self.example_email("iago"), self.example_email("hamlet"))
realm_emoji = RealmEmoji.objects.get(realm=realm) realm_emoji = RealmEmoji.objects.get(realm=realm)
realm_emoji.delete() realm_emoji.delete()
full_data = self._export_realm(realm, exportable_user_ids=user_ids) full_data = self._export_realm(realm, exportable_user_ids=user_ids)
realm_emoji.save() realm_emoji.save()
data = full_data['realm'] data = full_data['realm']
exported_user_emails = self.get_set(data['zerver_userprofile'], 'email') exported_user_emails = self.get_set(data['zerver_userprofile'], 'email')
self.assertIn(self.example_email('cordelia'), exported_user_emails) self.assertIn(self.example_email('iago'), exported_user_emails)
self.assertIn(self.example_email('hamlet'), exported_user_emails) self.assertIn(self.example_email('hamlet'), exported_user_emails)
self.assertNotIn('default-bot@zulip.com', exported_user_emails) self.assertNotIn('default-bot@zulip.com', exported_user_emails)
self.assertNotIn(self.example_email('iago'), exported_user_emails) self.assertNotIn(self.example_email('cordelia'), exported_user_emails)
dummy_user_emails = self.get_set(data['zerver_userprofile_mirrordummy'], 'email') dummy_user_emails = self.get_set(data['zerver_userprofile_mirrordummy'], 'email')
self.assertIn(self.example_email('iago'), dummy_user_emails) self.assertIn(self.example_email('cordelia'), dummy_user_emails)
self.assertNotIn(self.example_email('cordelia'), dummy_user_emails) self.assertIn(self.example_email('othello'), dummy_user_emails)
self.assertIn('default-bot@zulip.com', dummy_user_emails)
self.assertIn('emailgateway@zulip.com', dummy_user_emails)
self.assertNotIn(self.example_email('iago'), dummy_user_emails)
self.assertNotIn(self.example_email('hamlet'), dummy_user_emails)
data = full_data['message']
exported_message_ids = self.get_set(data['zerver_message'], "id")
self.assertNotIn(pm_a_msg_id, exported_message_ids)
self.assertIn(pm_b_msg_id, exported_message_ids)
self.assertIn(pm_c_msg_id, exported_message_ids)
self.assertIn(pm_d_msg_id, exported_message_ids)
def test_export_realm_with_member_consent(self) -> None: def test_export_realm_with_member_consent(self) -> None:
realm = Realm.objects.get(string_id='zulip') realm = Realm.objects.get(string_id='zulip')