class TestContacts(unittest.TestCase): def setUp(self): self.contacts = LegacyContactsAPI() def test_create_add_get_contact(self): jid = '*****@*****.**' account = "account" contact = self.contacts.create_contact(jid=jid, account=account) self.contacts.add_contact(account, contact) retrieved_contact = self.contacts.get_contact(account, jid) self.assertEqual(contact, retrieved_contact, "Contact must be known") self.contacts.remove_contact(account, contact) retrieved_contact = self.contacts.get_contact(account, jid) self.assertNotEqual(contact, retrieved_contact, msg="Contact must not be known any longer") def test_copy_contact(self): jid = '*****@*****.**' account = "account" contact = self.contacts.create_contact(jid=jid, account=account) copy = self.contacts.copy_contact(contact) self.assertFalse(contact is copy, msg="Must not be the same") # Not yet implemented to remain backwart compatible # self.assertEqual(contact, copy, msg="Must be equal") def test_legacy_accounts_handling(self): self.contacts.add_account("one") self.contacts.add_account("two") self.contacts.change_account_name("two", "old") self.contacts.remove_account("one") self.assertEqual(["old"], self.contacts.get_accounts()) def test_legacy_contacts_from_groups(self): jid1 = "*****@*****.**" jid2 = "*****@*****.**" account = "account" group = "GroupA" contact1 = self.contacts.create_contact(jid=jid1, account=account, groups=[group]) self.contacts.add_contact(account, contact1) contact2 = self.contacts.create_contact(jid=jid2, account=account, groups=[group]) self.contacts.add_contact(account, contact2) self.assertEqual(2, len(self.contacts.get_contacts_from_group(account, group))) self.assertEqual(0, len(self.contacts.get_contacts_from_group(account, '')))
class TestContacts(unittest.TestCase): def setUp(self): self.contacts = LegacyContactsAPI() def test_create_add_get_contact(self): jid = '*****@*****.**' account = "account" contact = self.contacts.create_contact(jid=jid, account=account) self.contacts.add_contact(account, contact) retrieved_contact = self.contacts.get_contact(account, jid) self.assertEqual(contact, retrieved_contact, "Contact must be known") self.contacts.remove_contact(account, contact) retrieved_contact = self.contacts.get_contact(account, jid) self.assertNotEqual(contact, retrieved_contact, msg="Contact must not be known any longer") def test_copy_contact(self): jid = '*****@*****.**' account = "account" contact = self.contacts.create_contact(jid=jid, account=account) copy = self.contacts.copy_contact(contact) self.assertFalse(contact is copy, msg="Must not be the same") # Not yet implemented to remain backwart compatible # self.assertEqual(contact, copy, msg="Must be equal") def test_legacy_accounts_handling(self): self.contacts.add_account("one") self.contacts.add_account("two") self.contacts.change_account_name("two", "old") self.contacts.remove_account("one") self.assertEqual(["old"], self.contacts.get_accounts()) def test_legacy_contacts_from_groups(self): jid1 = "*****@*****.**" jid2 = "*****@*****.**" account = "account" group = "GroupA" contact1 = self.contacts.create_contact(jid=jid1, account=account, groups=[group]) self.contacts.add_contact(account, contact1) contact2 = self.contacts.create_contact(jid=jid2, account=account, groups=[group]) self.contacts.add_contact(account, contact2) self.assertEqual( 2, len(self.contacts.get_contacts_from_group(account, group))) self.assertEqual( 0, len(self.contacts.get_contacts_from_group(account, '')))