1 /* 2 * Copyright (C) 2013 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License 15 */ 16 17 package com.android.providers.contacts.testutil; 18 19 import android.accounts.Account; 20 import android.content.ContentResolver; 21 import android.content.ContentUris; 22 import android.content.ContentValues; 23 import android.database.Cursor; 24 import android.net.Uri; 25 import android.provider.ContactsContract; 26 import android.test.mock.MockContentResolver; 27 28 import java.util.List; 29 30 /** 31 * Convenience methods for operating on the RawContacts table. 32 */ 33 public class RawContactUtil { 34 35 private static final Uri URI = ContactsContract.RawContacts.CONTENT_URI; 36 update(ContentResolver resolver, long rawContactId, ContentValues values)37 public static void update(ContentResolver resolver, long rawContactId, 38 ContentValues values) { 39 Uri uri = ContentUris.withAppendedId(URI, rawContactId); 40 resolver.update(uri, values, null, null); 41 } 42 queryByRawContactId(ContentResolver resolver, long rawContactId, String[] projection)43 public static String[] queryByRawContactId(ContentResolver resolver, 44 long rawContactId, String[] projection) { 45 Uri uri = ContentUris.withAppendedId(ContactsContract.RawContacts.CONTENT_URI, 46 rawContactId); 47 Cursor cursor = resolver.query(uri, projection, null, null, null); 48 return CommonDatabaseUtils.singleRecordToArray(cursor); 49 } 50 51 /** 52 * Returns a list of raw contact records. 53 * 54 * @return A list of records. Where each record is represented as an array of strings. 55 */ queryByContactId(ContentResolver resolver, long contactId, String[] projection)56 public static List<String[]> queryByContactId(ContentResolver resolver, long contactId, 57 String[] projection) { 58 Uri uri = ContentUris.withAppendedId(ContactsContract.RawContacts.CONTENT_URI, contactId); 59 Cursor cursor = resolver.query(uri, projection, null, null, null); 60 return CommonDatabaseUtils.multiRecordToArray(cursor); 61 } 62 delete(ContentResolver resolver, long rawContactId, boolean isSyncAdapter)63 public static void delete(ContentResolver resolver, long rawContactId, 64 boolean isSyncAdapter) { 65 Uri uri = ContentUris.withAppendedId(ContactsContract.RawContacts.CONTENT_URI, rawContactId) 66 .buildUpon() 67 .appendQueryParameter(ContactsContract.CALLER_IS_SYNCADAPTER, isSyncAdapter + "") 68 .build(); 69 resolver.delete(uri, null, null); 70 } 71 queryContactIdByRawContactId(ContentResolver resolver, long rawContactid)72 public static long queryContactIdByRawContactId(ContentResolver resolver, long rawContactid) { 73 String[] projection = new String[]{ 74 ContactsContract.RawContacts.CONTACT_ID 75 }; 76 String[] result = RawContactUtil.queryByRawContactId(resolver, rawContactid, 77 projection); 78 if (result == null) { 79 return CommonDatabaseUtils.NOT_FOUND; 80 } 81 return Long.parseLong(result[0]); 82 } 83 rawContactExistsById(ContentResolver resolver, long rawContactid)84 public static boolean rawContactExistsById(ContentResolver resolver, long rawContactid) { 85 long contactId = queryContactIdByRawContactId(resolver, rawContactid); 86 return contactId != CommonDatabaseUtils.NOT_FOUND; 87 } 88 createRawContact(ContentResolver resolver, Account account, String... extras)89 public static long createRawContact(ContentResolver resolver, Account account, 90 String... extras) { 91 ContentValues values = new ContentValues(); 92 CommonDatabaseUtils.extrasVarArgsToValues(values, extras); 93 final Uri uri = TestUtil.maybeAddAccountQueryParameters(ContactsContract.RawContacts.CONTENT_URI, account); 94 Uri contactUri = resolver.insert(uri, values); 95 return ContentUris.parseId(contactUri); 96 } 97 createRawContactWithName(ContentResolver resolver)98 public static long createRawContactWithName(ContentResolver resolver) { 99 return createRawContactWithName(resolver, null); 100 } 101 createRawContactWithName(ContentResolver resolver, Account account)102 public static long createRawContactWithName(ContentResolver resolver, Account account) { 103 return createRawContactWithName(resolver, "John", "Doe", account); 104 } 105 createRawContactWithName(ContentResolver resolver, String firstName, String lastName)106 public static long createRawContactWithName(ContentResolver resolver, String firstName, 107 String lastName) { 108 return createRawContactWithName(resolver, firstName, lastName, null); 109 } 110 createRawContactWithName(ContentResolver resolver, String firstName, String lastName, Account account)111 public static long createRawContactWithName(ContentResolver resolver, String firstName, 112 String lastName, Account account) { 113 long rawContactId = createRawContact(resolver, account); 114 DataUtil.insertStructuredName(resolver, rawContactId, firstName, lastName); 115 return rawContactId; 116 } 117 createRawContact(ContentResolver resolver)118 public static long createRawContact(ContentResolver resolver) { 119 return createRawContact(resolver, null); 120 } 121 } 122