def test_contact_printing(self):
# Setup
contact_list = ContactList(MasterKey(), Settings())
contact_list.contacts = [
create_contact(n) for n in ['Alice', 'Bob', 'Charlie', 'David']
]
# Teardown
self.assertIsNone(contact_list.print_contacts())
def test_local_contact(self):
# Setup
contact1 = Contact('*****@*****.**', '*****@*****.**', 'Alice',
32 * b'\x01', 32 * b'\x02', True, True, True)
contact_l = ContactList(MasterKey(), Settings())
contact_l.contacts = [contact1]
o_get_tty_w = src.common.misc.get_tty_w
src.common.misc.get_tty_w = lambda x: 1
# Test
self.assertFalse(contact_l.has_local_contact())
contact_l.add_contact('local', 'local', 'local', 32 * b'\x03',
32 * b'\x04', True, True, True)
self.assertTrue(contact_l.has_local_contact())
self.assertIsNone(contact_l.print_contacts())
self.assertIsNone(contact_l.print_contacts(spacing=True))
# Teardown
src.common.misc.get_tty_w = o_get_tty_w
