def test_unexpected_response(self): connection = MockPortalConnection( _simulate_get_all_contact_lists_with_unsupported_response, ) with assert_raises(Invalid): with connection: list(get_all_contact_lists(connection))
def test_getting_existing_contact_lists_single_page(self): contact_lists = [_STUB_CONTACT_LIST] connection = self._make_connection_with_contact_lists(contact_lists) with connection: retrieved_contact_lists = list(get_all_contact_lists(connection)) eq_(contact_lists, retrieved_contact_lists)
def test_static_lists(): with _get_portal_connection() as connection: new_contact_list_name = 'contactlisttest' created_static_list = \ create_static_contact_list(new_contact_list_name, connection) eq_(new_contact_list_name, created_static_list.name) all_contact_lists = get_all_contact_lists(connection) assert_in(created_static_list, all_contact_lists) all_contacts = get_all_contacts(connection) first_contact = next(all_contacts) added_contact_vids = add_contacts_to_list( created_static_list, [first_contact], connection, ) assert_in(first_contact.vid, added_contact_vids) contacts_in_list = list( get_all_contacts_from_list(connection, created_static_list) ) assert_in(first_contact, contacts_in_list) removed_contact_vids = remove_contacts_from_list( created_static_list, [first_contact], connection, ) assert_in(first_contact.vid, removed_contact_vids) contacts_in_list = list( get_all_contacts_from_list(connection, created_static_list) ) assert_not_in(first_contact, contacts_in_list) delete_contact_list(created_static_list.id, connection) all_contact_lists = get_all_contact_lists(connection) assert_not_in(created_static_list, all_contact_lists)
def test_getting_existing_contact_lists_multiple_pages(self): contact_lists = [] for index in range(0, BATCH_RETRIEVAL_SIZE_LIMIT + 1): contact_list = ContactList( index, 'list{}'.format(index), True, ) contact_lists.append(contact_list) connection = self._make_connection_with_contact_lists(contact_lists) with connection: retrieved_contact_lists = list(get_all_contact_lists(connection)) eq_(contact_lists, retrieved_contact_lists)
def test_no_contact_lists(self): with self._make_connection_with_contact_lists([]) as connection: contact_lists = list(get_all_contact_lists(connection)) eq_([], contact_lists)
def test_is_generator(self): connection = self._make_connection_with_contact_lists([]) contact_lists = get_all_contact_lists(connection) ok_(isgenerator(contact_lists))