BaseContactsProvider2Test.java revision ff2de103f7e3eeeff4665ef63f07460fef053d6d
1d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov/*
2d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov * Copyright (C) 2009 The Android Open Source Project
3d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov *
4d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov * Licensed under the Apache License, Version 2.0 (the "License");
5d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov * you may not use this file except in compliance with the License.
6d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov * You may obtain a copy of the License at
7d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov *
8d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov *      http://www.apache.org/licenses/LICENSE-2.0
9d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov *
10d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov * Unless required by applicable law or agreed to in writing, software
11d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov * distributed under the License is distributed on an "AS IS" BASIS,
12d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov * See the License for the specific language governing permissions and
14d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov * limitations under the License.
15d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov */
1628f8857b1b46bde18b85c6d3c2a63ac44c3c2e1cEvan Millarpackage com.android.providers.contacts;
17d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov
1828f8857b1b46bde18b85c6d3c2a63ac44c3c2e1cEvan Millarimport static com.android.providers.contacts.ContactsActor.PACKAGE_GREY;
19d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov
20e99988b266dd1263162583e81e2b408e7329b1c8Dmitri Plotnikovimport android.accounts.Account;
21bee1a6b2e7cbea778195890e442c9e50f2a4e6d9Dmitri Plotnikovimport android.content.ContentProvider;
22b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikovimport android.content.ContentResolver;
23d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikovimport android.content.ContentUris;
24d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikovimport android.content.ContentValues;
25e99988b266dd1263162583e81e2b408e7329b1c8Dmitri Plotnikovimport android.content.Context;
269261b2141aa90a4fed632fd6da03026d4c216280Fred Quintanaimport android.content.Entity;
27a908fb5f39aa2021662a6cc317cc7e4db2d8bfb0Dmitri Plotnikovimport android.content.res.Resources;
28d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikovimport android.database.Cursor;
29d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikovimport android.net.Uri;
30bee1a6b2e7cbea778195890e442c9e50f2a4e6d9Dmitri Plotnikovimport android.provider.ContactsContract;
31d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikovimport android.provider.ContactsContract.AggregationExceptions;
32e99988b266dd1263162583e81e2b408e7329b1c8Dmitri Plotnikovimport android.provider.ContactsContract.Contacts;
33d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikovimport android.provider.ContactsContract.Data;
349261b2141aa90a4fed632fd6da03026d4c216280Fred Quintanaimport android.provider.ContactsContract.Groups;
35bc5c799a52b5bde2f273efd118ebe2228c3d8f15Evan Millarimport android.provider.ContactsContract.Presence;
36e99988b266dd1263162583e81e2b408e7329b1c8Dmitri Plotnikovimport android.provider.ContactsContract.RawContacts;
37a5ad551e1753086825499f1aeb6415bb986f3588Dmitri Plotnikovimport android.provider.ContactsContract.CommonDataKinds.Email;
389261b2141aa90a4fed632fd6da03026d4c216280Fred Quintanaimport android.provider.ContactsContract.CommonDataKinds.GroupMembership;
39e99988b266dd1263162583e81e2b408e7329b1c8Dmitri Plotnikovimport android.provider.ContactsContract.CommonDataKinds.Im;
40a5ad551e1753086825499f1aeb6415bb986f3588Dmitri Plotnikovimport android.provider.ContactsContract.CommonDataKinds.Nickname;
41bf659107617a6291ba8bfeebc3f2e50138075ab5Dmitri Plotnikovimport android.provider.ContactsContract.CommonDataKinds.Phone;
42ae6ca1f34cf5458d79ec803411d4308879a91e92Evan Millarimport android.provider.ContactsContract.CommonDataKinds.Photo;
43e99988b266dd1263162583e81e2b408e7329b1c8Dmitri Plotnikovimport android.provider.ContactsContract.CommonDataKinds.StructuredName;
444a023070dab9a069be4cac5f5ba5554b66238484Dmitri Plotnikovimport android.provider.ContactsContract.CommonDataKinds.StructuredPostal;
45d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikovimport android.test.AndroidTestCase;
46d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikovimport android.test.mock.MockContentResolver;
47d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikovimport android.test.suitebuilder.annotation.LargeTest;
48b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikovimport android.util.Log;
49bee1a6b2e7cbea778195890e442c9e50f2a4e6d9Dmitri Plotnikov
50a908fb5f39aa2021662a6cc317cc7e4db2d8bfb0Dmitri Plotnikovimport java.io.ByteArrayOutputStream;
51a908fb5f39aa2021662a6cc317cc7e4db2d8bfb0Dmitri Plotnikovimport java.io.IOException;
52a908fb5f39aa2021662a6cc317cc7e4db2d8bfb0Dmitri Plotnikovimport java.io.InputStream;
539261b2141aa90a4fed632fd6da03026d4c216280Fred Quintanaimport java.util.ArrayList;
549261b2141aa90a4fed632fd6da03026d4c216280Fred Quintanaimport java.util.Arrays;
55e99988b266dd1263162583e81e2b408e7329b1c8Dmitri Plotnikovimport java.util.Comparator;
56ff2de103f7e3eeeff4665ef63f07460fef053d6dDmitri Plotnikovimport java.util.Iterator;
5720a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikovimport java.util.Map;
5820a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikovimport java.util.Set;
59ff2de103f7e3eeeff4665ef63f07460fef053d6dDmitri Plotnikovimport java.util.Map.Entry;
60d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov
61d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov/**
62d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov * A common superclass for {@link ContactsProvider2}-related tests.
63d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov */
64d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov@LargeTest
65d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikovpublic abstract class BaseContactsProvider2Test extends AndroidTestCase {
66d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov
67d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov    protected static final String PACKAGE = "ContactsProvider2Test";
68d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov
69bee1a6b2e7cbea778195890e442c9e50f2a4e6d9Dmitri Plotnikov    protected ContactsActor mActor;
70d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov    protected MockContentResolver mResolver;
719261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana    protected Account mAccount = new Account("account1", "account type1");
729261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana
732482dd0309794d57233d0af37a7c12fc647e7bf9Dmitri Plotnikov    private byte[] mTestPhoto;
742482dd0309794d57233d0af37a7c12fc647e7bf9Dmitri Plotnikov
759261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana    protected final static Long NO_LONG = new Long(0);
769261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana    protected final static String NO_STRING = new String("");
779261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana    protected final static Account NO_ACCOUNT = new Account("a", "b");
78d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov
79bee1a6b2e7cbea778195890e442c9e50f2a4e6d9Dmitri Plotnikov    protected Class<? extends ContentProvider> getProviderClass() {
80bee1a6b2e7cbea778195890e442c9e50f2a4e6d9Dmitri Plotnikov        return SynchronousContactsProvider2.class;
81bee1a6b2e7cbea778195890e442c9e50f2a4e6d9Dmitri Plotnikov    }
82bee1a6b2e7cbea778195890e442c9e50f2a4e6d9Dmitri Plotnikov
83bee1a6b2e7cbea778195890e442c9e50f2a4e6d9Dmitri Plotnikov    protected String getAuthority() {
84bee1a6b2e7cbea778195890e442c9e50f2a4e6d9Dmitri Plotnikov        return ContactsContract.AUTHORITY;
85bee1a6b2e7cbea778195890e442c9e50f2a4e6d9Dmitri Plotnikov    }
86bee1a6b2e7cbea778195890e442c9e50f2a4e6d9Dmitri Plotnikov
87d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov    @Override
88d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov    protected void setUp() throws Exception {
89d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov        super.setUp();
90d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov
91bee1a6b2e7cbea778195890e442c9e50f2a4e6d9Dmitri Plotnikov        mActor = new ContactsActor(getContext(), PACKAGE_GREY, getProviderClass(), getAuthority());
92d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov        mResolver = mActor.resolver;
93445ca81effbb0d61c7bc0033acf2d3dfd228fd4eDmitri Plotnikov        if (mActor.provider instanceof SynchronousContactsProvider2) {
94445ca81effbb0d61c7bc0033acf2d3dfd228fd4eDmitri Plotnikov            ((SynchronousContactsProvider2) mActor.provider)
95445ca81effbb0d61c7bc0033acf2d3dfd228fd4eDmitri Plotnikov                    .getOpenHelper(mActor.context).wipeData();
96445ca81effbb0d61c7bc0033acf2d3dfd228fd4eDmitri Plotnikov        }
979261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana    }
989261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana
99e99988b266dd1263162583e81e2b408e7329b1c8Dmitri Plotnikov    public Context getMockContext() {
100e99988b266dd1263162583e81e2b408e7329b1c8Dmitri Plotnikov        return mActor.context;
101e99988b266dd1263162583e81e2b408e7329b1c8Dmitri Plotnikov    }
102e99988b266dd1263162583e81e2b408e7329b1c8Dmitri Plotnikov
103e99988b266dd1263162583e81e2b408e7329b1c8Dmitri Plotnikov    public void addAuthority(String authority) {
104e99988b266dd1263162583e81e2b408e7329b1c8Dmitri Plotnikov        mActor.addAuthority(authority);
105e99988b266dd1263162583e81e2b408e7329b1c8Dmitri Plotnikov    }
106e99988b266dd1263162583e81e2b408e7329b1c8Dmitri Plotnikov
107e99988b266dd1263162583e81e2b408e7329b1c8Dmitri Plotnikov    public ContentProvider addProvider(Class<? extends ContentProvider> providerClass,
108e99988b266dd1263162583e81e2b408e7329b1c8Dmitri Plotnikov            String authority) throws Exception {
109e99988b266dd1263162583e81e2b408e7329b1c8Dmitri Plotnikov        return mActor.addProvider(providerClass, authority);
110e99988b266dd1263162583e81e2b408e7329b1c8Dmitri Plotnikov    }
111e99988b266dd1263162583e81e2b408e7329b1c8Dmitri Plotnikov
1123d8b043c3341a5b6c2e781b7eba9767d5cd13267Dmitri Plotnikov    public ContentProvider getProvider() {
1133d8b043c3341a5b6c2e781b7eba9767d5cd13267Dmitri Plotnikov        return mActor.provider;
1143d8b043c3341a5b6c2e781b7eba9767d5cd13267Dmitri Plotnikov    }
1153d8b043c3341a5b6c2e781b7eba9767d5cd13267Dmitri Plotnikov
1169261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana    protected Uri maybeAddAccountQueryParameters(Uri uri, Account account) {
1179261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        if (account == null) {
1189261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            return uri;
1199261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        }
1209261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        return uri.buildUpon()
121df9fd6b239de5829b04cb413e4dfa3e6da649c38Fred Quintana                .appendQueryParameter(RawContacts.ACCOUNT_NAME, account.name)
122df9fd6b239de5829b04cb413e4dfa3e6da649c38Fred Quintana                .appendQueryParameter(RawContacts.ACCOUNT_TYPE, account.type)
1239261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana                .build();
124d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov    }
125d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov
126d3fde755e73cd3912a488e7cb7d412d3c5f6ca94Dmitri Plotnikov    protected long createRawContact() {
127226c3dc6e93ca76a84c99100caa31045cba06cf6Fred Quintana        return createRawContact(null);
128d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov    }
129d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov
130c70dc0e38ff82c6e6d6b7458637c54fbdf446aacDmitri Plotnikov    protected long createRawContactWithName() {
1313cfe8d532d509fbbe605454e3a32b2361b7e1501Dmitri Plotnikov        return createRawContactWithName("John", "Doe");
1323cfe8d532d509fbbe605454e3a32b2361b7e1501Dmitri Plotnikov    }
1333cfe8d532d509fbbe605454e3a32b2361b7e1501Dmitri Plotnikov
1343cfe8d532d509fbbe605454e3a32b2361b7e1501Dmitri Plotnikov    protected long createRawContactWithName(String firstName, String lastName) {
135c70dc0e38ff82c6e6d6b7458637c54fbdf446aacDmitri Plotnikov        long rawContactId = createRawContact(null);
1363cfe8d532d509fbbe605454e3a32b2361b7e1501Dmitri Plotnikov        insertStructuredName(rawContactId, firstName, lastName);
137c70dc0e38ff82c6e6d6b7458637c54fbdf446aacDmitri Plotnikov        return rawContactId;
138c70dc0e38ff82c6e6d6b7458637c54fbdf446aacDmitri Plotnikov    }
139c70dc0e38ff82c6e6d6b7458637c54fbdf446aacDmitri Plotnikov
140226c3dc6e93ca76a84c99100caa31045cba06cf6Fred Quintana    protected long createRawContact(Account account, String... extras) {
1419261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        ContentValues values = new ContentValues();
142226c3dc6e93ca76a84c99100caa31045cba06cf6Fred Quintana        for (int i = 0; i < extras.length; ) {
143226c3dc6e93ca76a84c99100caa31045cba06cf6Fred Quintana            values.put(extras[i], extras[i + 1]);
144226c3dc6e93ca76a84c99100caa31045cba06cf6Fred Quintana            i += 2;
145226c3dc6e93ca76a84c99100caa31045cba06cf6Fred Quintana        }
1466cffee46a1334d2b3ed19f436b27638451541044Dmitri Plotnikov        final Uri uri = maybeAddAccountQueryParameters(RawContacts.CONTENT_URI, account);
1479261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        Uri contactUri = mResolver.insert(uri, values);
1489261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        return ContentUris.parseId(contactUri);
1499261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana    }
1509261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana
1519261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana    protected long createGroup(Account account, String sourceId, String title) {
1529261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        ContentValues values = new ContentValues();
1539261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        values.put(Groups.SOURCE_ID, sourceId);
1549261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        values.put(Groups.TITLE, title);
155a908fb5f39aa2021662a6cc317cc7e4db2d8bfb0Dmitri Plotnikov        values.put(Groups.GROUP_VISIBLE, 1);
1569261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        final Uri uri = maybeAddAccountQueryParameters(Groups.CONTENT_URI, account);
1579261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        return ContentUris.parseId(mResolver.insert(uri, values));
1589261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana    }
1599261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana
1605ef0401c311c62e53bde415b99cbb0ff83b0a9a2Dmitri Plotnikov    protected Uri insertStructuredName(long rawContactId, String givenName, String familyName) {
161d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov        ContentValues values = new ContentValues();
162d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov        StringBuilder sb = new StringBuilder();
163d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov        if (givenName != null) {
164d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov            sb.append(givenName);
165d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov        }
166d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov        if (givenName != null && familyName != null) {
167d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov            sb.append(" ");
168d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov        }
169d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov        if (familyName != null) {
170d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov            sb.append(familyName);
171d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov        }
172d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov        values.put(StructuredName.DISPLAY_NAME, sb.toString());
173d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov        values.put(StructuredName.GIVEN_NAME, givenName);
174d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov        values.put(StructuredName.FAMILY_NAME, familyName);
175d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov
1765ef0401c311c62e53bde415b99cbb0ff83b0a9a2Dmitri Plotnikov        return insertStructuredName(rawContactId, values);
1774097855e2d672b3f8e1c5c8a169efb80203bf53eDmitri Plotnikov    }
1784097855e2d672b3f8e1c5c8a169efb80203bf53eDmitri Plotnikov
1795ef0401c311c62e53bde415b99cbb0ff83b0a9a2Dmitri Plotnikov    protected Uri insertStructuredName(long rawContactId, ContentValues values) {
1805ef0401c311c62e53bde415b99cbb0ff83b0a9a2Dmitri Plotnikov        values.put(Data.RAW_CONTACT_ID, rawContactId);
1814097855e2d672b3f8e1c5c8a169efb80203bf53eDmitri Plotnikov        values.put(Data.MIMETYPE, StructuredName.CONTENT_ITEM_TYPE);
182d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov        Uri resultUri = mResolver.insert(Data.CONTENT_URI, values);
183d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov        return resultUri;
184d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov    }
185d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov
1865ef0401c311c62e53bde415b99cbb0ff83b0a9a2Dmitri Plotnikov    protected Uri insertPhoneNumber(long rawContactId, String phoneNumber) {
187bf659107617a6291ba8bfeebc3f2e50138075ab5Dmitri Plotnikov        ContentValues values = new ContentValues();
1885ef0401c311c62e53bde415b99cbb0ff83b0a9a2Dmitri Plotnikov        values.put(Data.RAW_CONTACT_ID, rawContactId);
189bf659107617a6291ba8bfeebc3f2e50138075ab5Dmitri Plotnikov        values.put(Data.MIMETYPE, Phone.CONTENT_ITEM_TYPE);
190bf659107617a6291ba8bfeebc3f2e50138075ab5Dmitri Plotnikov        values.put(Phone.NUMBER, phoneNumber);
1913cebbf7141252768d3e272e049e9c5b0cb9d710eDmitri Plotnikov        values.put(Phone.TYPE, Phone.TYPE_HOME);
192bf659107617a6291ba8bfeebc3f2e50138075ab5Dmitri Plotnikov
193bf659107617a6291ba8bfeebc3f2e50138075ab5Dmitri Plotnikov        Uri resultUri = mResolver.insert(Data.CONTENT_URI, values);
194bf659107617a6291ba8bfeebc3f2e50138075ab5Dmitri Plotnikov        return resultUri;
195bf659107617a6291ba8bfeebc3f2e50138075ab5Dmitri Plotnikov    }
196bf659107617a6291ba8bfeebc3f2e50138075ab5Dmitri Plotnikov
1975ef0401c311c62e53bde415b99cbb0ff83b0a9a2Dmitri Plotnikov    protected Uri insertEmail(long rawContactId, String email) {
198a5ad551e1753086825499f1aeb6415bb986f3588Dmitri Plotnikov        ContentValues values = new ContentValues();
1995ef0401c311c62e53bde415b99cbb0ff83b0a9a2Dmitri Plotnikov        values.put(Data.RAW_CONTACT_ID, rawContactId);
200a5ad551e1753086825499f1aeb6415bb986f3588Dmitri Plotnikov        values.put(Data.MIMETYPE, Email.CONTENT_ITEM_TYPE);
201a5ad551e1753086825499f1aeb6415bb986f3588Dmitri Plotnikov        values.put(Email.DATA, email);
2023cebbf7141252768d3e272e049e9c5b0cb9d710eDmitri Plotnikov        values.put(Email.TYPE, Email.TYPE_HOME);
203a5ad551e1753086825499f1aeb6415bb986f3588Dmitri Plotnikov
204a5ad551e1753086825499f1aeb6415bb986f3588Dmitri Plotnikov        Uri resultUri = mResolver.insert(Data.CONTENT_URI, values);
205a5ad551e1753086825499f1aeb6415bb986f3588Dmitri Plotnikov        return resultUri;
206a5ad551e1753086825499f1aeb6415bb986f3588Dmitri Plotnikov    }
207a5ad551e1753086825499f1aeb6415bb986f3588Dmitri Plotnikov
2085ef0401c311c62e53bde415b99cbb0ff83b0a9a2Dmitri Plotnikov    protected Uri insertNickname(long rawContactId, String nickname) {
209a5ad551e1753086825499f1aeb6415bb986f3588Dmitri Plotnikov        ContentValues values = new ContentValues();
2105ef0401c311c62e53bde415b99cbb0ff83b0a9a2Dmitri Plotnikov        values.put(Data.RAW_CONTACT_ID, rawContactId);
211a5ad551e1753086825499f1aeb6415bb986f3588Dmitri Plotnikov        values.put(Data.MIMETYPE, Nickname.CONTENT_ITEM_TYPE);
212a5ad551e1753086825499f1aeb6415bb986f3588Dmitri Plotnikov        values.put(Nickname.NAME, nickname);
2133cebbf7141252768d3e272e049e9c5b0cb9d710eDmitri Plotnikov        values.put(Nickname.TYPE, Nickname.TYPE_OTHER_NAME);
214a5ad551e1753086825499f1aeb6415bb986f3588Dmitri Plotnikov
215a5ad551e1753086825499f1aeb6415bb986f3588Dmitri Plotnikov        Uri resultUri = mResolver.insert(Data.CONTENT_URI, values);
216a5ad551e1753086825499f1aeb6415bb986f3588Dmitri Plotnikov        return resultUri;
217a5ad551e1753086825499f1aeb6415bb986f3588Dmitri Plotnikov    }
218a5ad551e1753086825499f1aeb6415bb986f3588Dmitri Plotnikov
2194a023070dab9a069be4cac5f5ba5554b66238484Dmitri Plotnikov    protected Uri insertPostalAddress(long rawContactId, String formattedAddress) {
2204a023070dab9a069be4cac5f5ba5554b66238484Dmitri Plotnikov        ContentValues values = new ContentValues();
2214a023070dab9a069be4cac5f5ba5554b66238484Dmitri Plotnikov        values.put(Data.RAW_CONTACT_ID, rawContactId);
2224a023070dab9a069be4cac5f5ba5554b66238484Dmitri Plotnikov        values.put(Data.MIMETYPE, StructuredPostal.CONTENT_ITEM_TYPE);
2234a023070dab9a069be4cac5f5ba5554b66238484Dmitri Plotnikov        values.put(StructuredPostal.FORMATTED_ADDRESS, formattedAddress);
2244a023070dab9a069be4cac5f5ba5554b66238484Dmitri Plotnikov
2254a023070dab9a069be4cac5f5ba5554b66238484Dmitri Plotnikov        Uri resultUri = mResolver.insert(Data.CONTENT_URI, values);
2264a023070dab9a069be4cac5f5ba5554b66238484Dmitri Plotnikov        return resultUri;
2274a023070dab9a069be4cac5f5ba5554b66238484Dmitri Plotnikov    }
2284a023070dab9a069be4cac5f5ba5554b66238484Dmitri Plotnikov
2295ef0401c311c62e53bde415b99cbb0ff83b0a9a2Dmitri Plotnikov    protected Uri insertPhoto(long rawContactId) {
230ae6ca1f34cf5458d79ec803411d4308879a91e92Evan Millar        ContentValues values = new ContentValues();
2315ef0401c311c62e53bde415b99cbb0ff83b0a9a2Dmitri Plotnikov        values.put(Data.RAW_CONTACT_ID, rawContactId);
232ae6ca1f34cf5458d79ec803411d4308879a91e92Evan Millar        values.put(Data.MIMETYPE, Photo.CONTENT_ITEM_TYPE);
2332482dd0309794d57233d0af37a7c12fc647e7bf9Dmitri Plotnikov        values.put(Photo.PHOTO, loadTestPhoto());
234ae6ca1f34cf5458d79ec803411d4308879a91e92Evan Millar        Uri resultUri = mResolver.insert(Data.CONTENT_URI, values);
235ae6ca1f34cf5458d79ec803411d4308879a91e92Evan Millar        return resultUri;
236ae6ca1f34cf5458d79ec803411d4308879a91e92Evan Millar    }
237ae6ca1f34cf5458d79ec803411d4308879a91e92Evan Millar
2385ef0401c311c62e53bde415b99cbb0ff83b0a9a2Dmitri Plotnikov    protected Uri insertGroupMembership(long rawContactId, String sourceId) {
2399261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        ContentValues values = new ContentValues();
2405ef0401c311c62e53bde415b99cbb0ff83b0a9a2Dmitri Plotnikov        values.put(Data.RAW_CONTACT_ID, rawContactId);
2419261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        values.put(Data.MIMETYPE, GroupMembership.CONTENT_ITEM_TYPE);
2429261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        values.put(GroupMembership.GROUP_SOURCE_ID, sourceId);
2439261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        return mResolver.insert(Data.CONTENT_URI, values);
2449261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana    }
2459261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana
2465ef0401c311c62e53bde415b99cbb0ff83b0a9a2Dmitri Plotnikov    protected Uri insertGroupMembership(long rawContactId, Long groupId) {
2479261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        ContentValues values = new ContentValues();
2485ef0401c311c62e53bde415b99cbb0ff83b0a9a2Dmitri Plotnikov        values.put(Data.RAW_CONTACT_ID, rawContactId);
2499261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        values.put(Data.MIMETYPE, GroupMembership.CONTENT_ITEM_TYPE);
2509261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        values.put(GroupMembership.GROUP_ROW_ID, groupId);
2519261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        return mResolver.insert(Data.CONTENT_URI, values);
2529261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana    }
2539261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana
2544dcd106ccc27dbbfaae86baf0cd57beb42c27cccDmitri Plotnikov    protected Uri insertPresence(int protocol, String customProtocol, String handle, int presence,
2554dcd106ccc27dbbfaae86baf0cd57beb42c27cccDmitri Plotnikov            String status) {
256bc5c799a52b5bde2f273efd118ebe2228c3d8f15Evan Millar        ContentValues values = new ContentValues();
2574dcd106ccc27dbbfaae86baf0cd57beb42c27cccDmitri Plotnikov        values.put(Presence.PROTOCOL, protocol);
2584dcd106ccc27dbbfaae86baf0cd57beb42c27cccDmitri Plotnikov        values.put(Presence.CUSTOM_PROTOCOL, customProtocol);
259bc5c799a52b5bde2f273efd118ebe2228c3d8f15Evan Millar        values.put(Presence.IM_HANDLE, handle);
260bc5c799a52b5bde2f273efd118ebe2228c3d8f15Evan Millar        values.put(Presence.PRESENCE_STATUS, presence);
261e46667e641cd1c60998e1ccab4b60531d5b12ef7Dmitri Plotnikov        values.put(Presence.PRESENCE_CUSTOM_STATUS, status);
262bc5c799a52b5bde2f273efd118ebe2228c3d8f15Evan Millar
263bc5c799a52b5bde2f273efd118ebe2228c3d8f15Evan Millar        Uri resultUri = mResolver.insert(Presence.CONTENT_URI, values);
264bc5c799a52b5bde2f273efd118ebe2228c3d8f15Evan Millar        return resultUri;
265bc5c799a52b5bde2f273efd118ebe2228c3d8f15Evan Millar    }
266bc5c799a52b5bde2f273efd118ebe2228c3d8f15Evan Millar
2674dcd106ccc27dbbfaae86baf0cd57beb42c27cccDmitri Plotnikov    protected Uri insertImHandle(long rawContactId, int protocol, String customProtocol,
2684dcd106ccc27dbbfaae86baf0cd57beb42c27cccDmitri Plotnikov            String handle) {
269bc5c799a52b5bde2f273efd118ebe2228c3d8f15Evan Millar        ContentValues values = new ContentValues();
2705ef0401c311c62e53bde415b99cbb0ff83b0a9a2Dmitri Plotnikov        values.put(Data.RAW_CONTACT_ID, rawContactId);
271bc5c799a52b5bde2f273efd118ebe2228c3d8f15Evan Millar        values.put(Data.MIMETYPE, Im.CONTENT_ITEM_TYPE);
272bc5c799a52b5bde2f273efd118ebe2228c3d8f15Evan Millar        values.put(Im.PROTOCOL, protocol);
2734dcd106ccc27dbbfaae86baf0cd57beb42c27cccDmitri Plotnikov        values.put(Im.CUSTOM_PROTOCOL, customProtocol);
274bc5c799a52b5bde2f273efd118ebe2228c3d8f15Evan Millar        values.put(Im.DATA, handle);
2753cebbf7141252768d3e272e049e9c5b0cb9d710eDmitri Plotnikov        values.put(Im.TYPE, Im.TYPE_HOME);
276bc5c799a52b5bde2f273efd118ebe2228c3d8f15Evan Millar
277bc5c799a52b5bde2f273efd118ebe2228c3d8f15Evan Millar        Uri resultUri = mResolver.insert(Data.CONTENT_URI, values);
278bc5c799a52b5bde2f273efd118ebe2228c3d8f15Evan Millar        return resultUri;
279bc5c799a52b5bde2f273efd118ebe2228c3d8f15Evan Millar    }
280bc5c799a52b5bde2f273efd118ebe2228c3d8f15Evan Millar
2815ef0401c311c62e53bde415b99cbb0ff83b0a9a2Dmitri Plotnikov    protected void setContactAccountName(long rawContactId, String accountName) {
282ae6ca1f34cf5458d79ec803411d4308879a91e92Evan Millar        ContentValues values = new ContentValues();
2836cffee46a1334d2b3ed19f436b27638451541044Dmitri Plotnikov        values.put(RawContacts.ACCOUNT_NAME, accountName);
284ae6ca1f34cf5458d79ec803411d4308879a91e92Evan Millar
285ae6ca1f34cf5458d79ec803411d4308879a91e92Evan Millar        mResolver.update(ContentUris.withAppendedId(
2865ef0401c311c62e53bde415b99cbb0ff83b0a9a2Dmitri Plotnikov                RawContacts.CONTENT_URI, rawContactId), values, null, null);
287ae6ca1f34cf5458d79ec803411d4308879a91e92Evan Millar    }
288ae6ca1f34cf5458d79ec803411d4308879a91e92Evan Millar
289d3fde755e73cd3912a488e7cb7d412d3c5f6ca94Dmitri Plotnikov    protected void setAggregationException(int type, long contactId, long rawContactId) {
290d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov        ContentValues values = new ContentValues();
291d3fde755e73cd3912a488e7cb7d412d3c5f6ca94Dmitri Plotnikov        values.put(AggregationExceptions.CONTACT_ID, contactId);
2925ef0401c311c62e53bde415b99cbb0ff83b0a9a2Dmitri Plotnikov        values.put(AggregationExceptions.RAW_CONTACT_ID, rawContactId);
293d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov        values.put(AggregationExceptions.TYPE, type);
2943d8b043c3341a5b6c2e781b7eba9767d5cd13267Dmitri Plotnikov        assertEquals(1, mResolver.update(AggregationExceptions.CONTENT_URI, values, null, null));
295d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov    }
296d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov
297d3fde755e73cd3912a488e7cb7d412d3c5f6ca94Dmitri Plotnikov    protected Cursor queryRawContact(long rawContactId) {
2985ef0401c311c62e53bde415b99cbb0ff83b0a9a2Dmitri Plotnikov        return mResolver.query(ContentUris.withAppendedId(RawContacts.CONTENT_URI, rawContactId), null,
299d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov                null, null, null);
300d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov    }
301d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov
302d3fde755e73cd3912a488e7cb7d412d3c5f6ca94Dmitri Plotnikov    protected Cursor queryContact(long contactId) {
303d3fde755e73cd3912a488e7cb7d412d3c5f6ca94Dmitri Plotnikov        return mResolver.query(ContentUris.withAppendedId(Contacts.CONTENT_URI, contactId),
304d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov                null, null, null, null);
305d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov    }
306d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov
307d3fde755e73cd3912a488e7cb7d412d3c5f6ca94Dmitri Plotnikov    protected Cursor queryContactSummary(long contactId, String[] projection) {
308d3fde755e73cd3912a488e7cb7d412d3c5f6ca94Dmitri Plotnikov        return mResolver.query(ContentUris.withAppendedId(Contacts.CONTENT_SUMMARY_URI,
309d3fde755e73cd3912a488e7cb7d412d3c5f6ca94Dmitri Plotnikov                contactId), projection, null, null, null);
310bc5c799a52b5bde2f273efd118ebe2228c3d8f15Evan Millar    }
311bc5c799a52b5bde2f273efd118ebe2228c3d8f15Evan Millar
312d3fde755e73cd3912a488e7cb7d412d3c5f6ca94Dmitri Plotnikov    protected Cursor queryContactSummary() {
313d3fde755e73cd3912a488e7cb7d412d3c5f6ca94Dmitri Plotnikov        return mResolver.query(Contacts.CONTENT_SUMMARY_URI, null, null, null, null);
314bc5c799a52b5bde2f273efd118ebe2228c3d8f15Evan Millar    }
315bc5c799a52b5bde2f273efd118ebe2228c3d8f15Evan Millar
316d3fde755e73cd3912a488e7cb7d412d3c5f6ca94Dmitri Plotnikov    protected long queryContactId(long rawContactId) {
317d3fde755e73cd3912a488e7cb7d412d3c5f6ca94Dmitri Plotnikov        Cursor c = queryRawContact(rawContactId);
318d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov        assertTrue(c.moveToFirst());
319d3fde755e73cd3912a488e7cb7d412d3c5f6ca94Dmitri Plotnikov        long contactId = c.getLong(c.getColumnIndex(RawContacts.CONTACT_ID));
320d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov        c.close();
321d3fde755e73cd3912a488e7cb7d412d3c5f6ca94Dmitri Plotnikov        return contactId;
322d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov    }
323d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov
324d3fde755e73cd3912a488e7cb7d412d3c5f6ca94Dmitri Plotnikov    protected long queryPhotoId(long contactId) {
325d3fde755e73cd3912a488e7cb7d412d3c5f6ca94Dmitri Plotnikov        Cursor c = queryContact(contactId);
326ae6ca1f34cf5458d79ec803411d4308879a91e92Evan Millar        assertTrue(c.moveToFirst());
327d3fde755e73cd3912a488e7cb7d412d3c5f6ca94Dmitri Plotnikov        long photoId = c.getInt(c.getColumnIndex(Contacts.PHOTO_ID));
328ae6ca1f34cf5458d79ec803411d4308879a91e92Evan Millar        c.close();
329ae6ca1f34cf5458d79ec803411d4308879a91e92Evan Millar        return photoId;
330ae6ca1f34cf5458d79ec803411d4308879a91e92Evan Millar    }
331ae6ca1f34cf5458d79ec803411d4308879a91e92Evan Millar
332d3fde755e73cd3912a488e7cb7d412d3c5f6ca94Dmitri Plotnikov    protected String queryDisplayName(long contactId) {
333d3fde755e73cd3912a488e7cb7d412d3c5f6ca94Dmitri Plotnikov        Cursor c = queryContact(contactId);
334d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov        assertTrue(c.moveToFirst());
335d3fde755e73cd3912a488e7cb7d412d3c5f6ca94Dmitri Plotnikov        String displayName = c.getString(c.getColumnIndex(Contacts.DISPLAY_NAME));
336d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov        c.close();
337d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov        return displayName;
338d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov    }
339d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov
340d3fde755e73cd3912a488e7cb7d412d3c5f6ca94Dmitri Plotnikov    protected void assertAggregated(long rawContactId1, long rawContactId2) {
341d3fde755e73cd3912a488e7cb7d412d3c5f6ca94Dmitri Plotnikov        long contactId1 = queryContactId(rawContactId1);
342d3fde755e73cd3912a488e7cb7d412d3c5f6ca94Dmitri Plotnikov        long contactId2 = queryContactId(rawContactId2);
343d3fde755e73cd3912a488e7cb7d412d3c5f6ca94Dmitri Plotnikov        assertTrue(contactId1 == contactId2);
344bf659107617a6291ba8bfeebc3f2e50138075ab5Dmitri Plotnikov    }
345bf659107617a6291ba8bfeebc3f2e50138075ab5Dmitri Plotnikov
346d3fde755e73cd3912a488e7cb7d412d3c5f6ca94Dmitri Plotnikov    protected void assertAggregated(long rawContactId1, long rawContactId2,
347d3fde755e73cd3912a488e7cb7d412d3c5f6ca94Dmitri Plotnikov            String expectedDisplayName) {
348d3fde755e73cd3912a488e7cb7d412d3c5f6ca94Dmitri Plotnikov        long contactId1 = queryContactId(rawContactId1);
349d3fde755e73cd3912a488e7cb7d412d3c5f6ca94Dmitri Plotnikov        long contactId2 = queryContactId(rawContactId2);
350d3fde755e73cd3912a488e7cb7d412d3c5f6ca94Dmitri Plotnikov        assertTrue(contactId1 == contactId2);
351d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov
352d3fde755e73cd3912a488e7cb7d412d3c5f6ca94Dmitri Plotnikov        String displayName = queryDisplayName(contactId1);
353d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov        assertEquals(expectedDisplayName, displayName);
354d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov    }
355d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov
356d3fde755e73cd3912a488e7cb7d412d3c5f6ca94Dmitri Plotnikov    protected void assertNotAggregated(long rawContactId1, long rawContactId2) {
357d3fde755e73cd3912a488e7cb7d412d3c5f6ca94Dmitri Plotnikov        long contactId1 = queryContactId(rawContactId1);
358d3fde755e73cd3912a488e7cb7d412d3c5f6ca94Dmitri Plotnikov        long contactId2 = queryContactId(rawContactId2);
359d3fde755e73cd3912a488e7cb7d412d3c5f6ca94Dmitri Plotnikov        assertTrue(contactId1 != contactId2);
360d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov    }
3614097855e2d672b3f8e1c5c8a169efb80203bf53eDmitri Plotnikov
3625ef0401c311c62e53bde415b99cbb0ff83b0a9a2Dmitri Plotnikov    protected void assertStructuredName(long rawContactId, String prefix, String givenName,
3634097855e2d672b3f8e1c5c8a169efb80203bf53eDmitri Plotnikov            String middleName, String familyName, String suffix) {
3646cffee46a1334d2b3ed19f436b27638451541044Dmitri Plotnikov        Uri uri =
3655ef0401c311c62e53bde415b99cbb0ff83b0a9a2Dmitri Plotnikov                Uri.withAppendedPath(ContentUris.withAppendedId(RawContacts.CONTENT_URI, rawContactId),
3666cffee46a1334d2b3ed19f436b27638451541044Dmitri Plotnikov                RawContacts.Data.CONTENT_DIRECTORY);
3674097855e2d672b3f8e1c5c8a169efb80203bf53eDmitri Plotnikov
3684097855e2d672b3f8e1c5c8a169efb80203bf53eDmitri Plotnikov        final String[] projection = new String[] {
3694097855e2d672b3f8e1c5c8a169efb80203bf53eDmitri Plotnikov                StructuredName.PREFIX, StructuredName.GIVEN_NAME, StructuredName.MIDDLE_NAME,
3704097855e2d672b3f8e1c5c8a169efb80203bf53eDmitri Plotnikov                StructuredName.FAMILY_NAME, StructuredName.SUFFIX
3714097855e2d672b3f8e1c5c8a169efb80203bf53eDmitri Plotnikov        };
3724097855e2d672b3f8e1c5c8a169efb80203bf53eDmitri Plotnikov
3734097855e2d672b3f8e1c5c8a169efb80203bf53eDmitri Plotnikov        Cursor c = mResolver.query(uri, projection, Data.MIMETYPE + "='"
3744097855e2d672b3f8e1c5c8a169efb80203bf53eDmitri Plotnikov                + StructuredName.CONTENT_ITEM_TYPE + "'", null, null);
3754097855e2d672b3f8e1c5c8a169efb80203bf53eDmitri Plotnikov
3764097855e2d672b3f8e1c5c8a169efb80203bf53eDmitri Plotnikov        assertTrue(c.moveToFirst());
3774097855e2d672b3f8e1c5c8a169efb80203bf53eDmitri Plotnikov        assertEquals(prefix, c.getString(0));
3784097855e2d672b3f8e1c5c8a169efb80203bf53eDmitri Plotnikov        assertEquals(givenName, c.getString(1));
3794097855e2d672b3f8e1c5c8a169efb80203bf53eDmitri Plotnikov        assertEquals(middleName, c.getString(2));
3804097855e2d672b3f8e1c5c8a169efb80203bf53eDmitri Plotnikov        assertEquals(familyName, c.getString(3));
3814097855e2d672b3f8e1c5c8a169efb80203bf53eDmitri Plotnikov        assertEquals(suffix, c.getString(4));
3824097855e2d672b3f8e1c5c8a169efb80203bf53eDmitri Plotnikov        c.close();
3834097855e2d672b3f8e1c5c8a169efb80203bf53eDmitri Plotnikov    }
3849261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana
3859261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana    protected long assertSingleGroup(Long rowId, Account account, String sourceId, String title) {
3869261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        Cursor c = mResolver.query(Groups.CONTENT_URI, null, null, null, null);
3879261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        try {
3889261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            assertTrue(c.moveToNext());
3899261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            long actualRowId = assertGroup(c, rowId, account, sourceId, title);
3909261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            assertFalse(c.moveToNext());
3919261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            return actualRowId;
3929261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        } finally {
3939261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            c.close();
3949261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        }
3959261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana    }
3969261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana
3975ef0401c311c62e53bde415b99cbb0ff83b0a9a2Dmitri Plotnikov    protected long assertSingleGroupMembership(Long rowId, Long rawContactId, Long groupRowId,
3989261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            String sourceId) {
3999261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        Cursor c = mResolver.query(ContactsContract.Data.CONTENT_URI, null, null, null, null);
4009261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        try {
4019261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            assertTrue(c.moveToNext());
4025ef0401c311c62e53bde415b99cbb0ff83b0a9a2Dmitri Plotnikov            long actualRowId = assertGroupMembership(c, rowId, rawContactId, groupRowId, sourceId);
4039261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            assertFalse(c.moveToNext());
4049261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            return actualRowId;
4059261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        } finally {
4069261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            c.close();
4079261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        }
4089261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana    }
4099261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana
4105ef0401c311c62e53bde415b99cbb0ff83b0a9a2Dmitri Plotnikov    protected long assertGroupMembership(Cursor c, Long rowId, Long rawContactId, Long groupRowId,
4119261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            String sourceId) {
4129261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        assertNullOrEquals(c, rowId, Data._ID);
4135ef0401c311c62e53bde415b99cbb0ff83b0a9a2Dmitri Plotnikov        assertNullOrEquals(c, rawContactId, GroupMembership.RAW_CONTACT_ID);
4149261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        assertNullOrEquals(c, groupRowId, GroupMembership.GROUP_ROW_ID);
4159261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        assertNullOrEquals(c, sourceId, GroupMembership.GROUP_SOURCE_ID);
4169261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        return c.getLong(c.getColumnIndexOrThrow("_id"));
4179261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana    }
4189261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana
4199261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana    protected long assertGroup(Cursor c, Long rowId, Account account, String sourceId, String title) {
4209261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        assertNullOrEquals(c, rowId, Groups._ID);
4219261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        assertNullOrEquals(c, account);
4229261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        assertNullOrEquals(c, sourceId, Groups.SOURCE_ID);
4239261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        assertNullOrEquals(c, title, Groups.TITLE);
4249261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        return c.getLong(c.getColumnIndexOrThrow("_id"));
4259261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana    }
4269261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana
4279261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana    private void assertNullOrEquals(Cursor c, Account account) {
4289261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        if (account == NO_ACCOUNT) {
4299261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            return;
4309261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        }
4319261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        if (account == null) {
4329261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            assertTrue(c.isNull(c.getColumnIndexOrThrow(Groups.ACCOUNT_NAME)));
4339261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            assertTrue(c.isNull(c.getColumnIndexOrThrow(Groups.ACCOUNT_TYPE)));
4349261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        } else {
435df9fd6b239de5829b04cb413e4dfa3e6da649c38Fred Quintana            assertEquals(account.name, c.getString(c.getColumnIndexOrThrow(Groups.ACCOUNT_NAME)));
436df9fd6b239de5829b04cb413e4dfa3e6da649c38Fred Quintana            assertEquals(account.type, c.getString(c.getColumnIndexOrThrow(Groups.ACCOUNT_TYPE)));
4379261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        }
4389261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana    }
4399261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana
4409261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana    private void assertNullOrEquals(Cursor c, Long value, String columnName) {
4419261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        if (value != NO_LONG) {
4429261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            if (value == null) assertTrue(c.isNull(c.getColumnIndexOrThrow(columnName)));
4439261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            else assertEquals((long) value, c.getLong(c.getColumnIndexOrThrow(columnName)));
4449261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        }
4459261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana    }
4469261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana
4479261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana    private void assertNullOrEquals(Cursor c, String value, String columnName) {
4489261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        if (value != NO_STRING) {
4499261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            if (value == null) assertTrue(c.isNull(c.getColumnIndexOrThrow(columnName)));
4509261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            else assertEquals(value, c.getString(c.getColumnIndexOrThrow(columnName)));
4519261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        }
4529261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana    }
4539261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana
4549261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana    protected void assertDataRow(ContentValues actual, String expectedMimetype,
4559261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            Object... expectedArguments) {
4569261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        assertEquals(actual.toString(), expectedMimetype, actual.getAsString(Data.MIMETYPE));
4579261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        for (int i = 0; i < expectedArguments.length; i += 2) {
4589261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            String columnName = (String) expectedArguments[i];
4599261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            Object expectedValue = expectedArguments[i + 1];
4609261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            if (expectedValue instanceof Uri) {
4619261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana                expectedValue = ContentUris.parseId((Uri) expectedValue);
4629261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            }
4639261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            if (expectedValue == null) {
4649261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana                assertNull(actual.toString(), actual.get(columnName));
4659261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            }
4669261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            if (expectedValue instanceof Long) {
467226c3dc6e93ca76a84c99100caa31045cba06cf6Fred Quintana                assertEquals("mismatch at " + columnName + " from " + actual.toString(),
468226c3dc6e93ca76a84c99100caa31045cba06cf6Fred Quintana                        expectedValue, actual.getAsLong(columnName));
4699261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            } else if (expectedValue instanceof Integer) {
470226c3dc6e93ca76a84c99100caa31045cba06cf6Fred Quintana                assertEquals("mismatch at " + columnName + " from " + actual.toString(),
471226c3dc6e93ca76a84c99100caa31045cba06cf6Fred Quintana                        expectedValue, actual.getAsInteger(columnName));
4729261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            } else if (expectedValue instanceof String) {
473226c3dc6e93ca76a84c99100caa31045cba06cf6Fred Quintana                assertEquals("mismatch at " + columnName + " from " + actual.toString(),
474226c3dc6e93ca76a84c99100caa31045cba06cf6Fred Quintana                        expectedValue, actual.getAsString(columnName));
4759261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            } else {
476226c3dc6e93ca76a84c99100caa31045cba06cf6Fred Quintana                assertEquals("mismatch at " + columnName + " from " + actual.toString(),
477226c3dc6e93ca76a84c99100caa31045cba06cf6Fred Quintana                        expectedValue, actual.get(columnName));
4789261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            }
4799261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        }
4809261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana    }
4819261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana
4829261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana    protected static class IdComparator implements Comparator<ContentValues> {
4839261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        public int compare(ContentValues o1, ContentValues o2) {
4849261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            long id1 = o1.getAsLong(ContactsContract.Data._ID);
4859261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            long id2 = o2.getAsLong(ContactsContract.Data._ID);
4869261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            if (id1 == id2) return 0;
4879261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            return (id1 < id2) ? -1 : 1;
4889261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        }
4899261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana    }
4909261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana
4919261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana    protected ContentValues[] asSortedContentValuesArray(
4929261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            ArrayList<Entity.NamedContentValues> subValues) {
4939261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        ContentValues[] result = new ContentValues[subValues.size()];
4949261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        int i = 0;
4959261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        for (Entity.NamedContentValues subValue : subValues) {
4969261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            result[i] = subValue.values;
4979261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana            i++;
4989261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        }
4999261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        Arrays.sort(result, new IdComparator());
5009261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        return result;
5019261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana    }
5029261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana
5039261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana    protected void assertDirty(Uri uri, boolean state) {
5049261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        Cursor c = mResolver.query(uri, new String[]{"dirty"}, null, null, null);
5059261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        assertTrue(c.moveToNext());
5069261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        assertEquals(state, c.getLong(0) != 0);
5079261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        assertFalse(c.moveToNext());
5083cfe8d532d509fbbe605454e3a32b2361b7e1501Dmitri Plotnikov        c.close();
5099261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana    }
5109261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana
5119261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana    protected long getVersion(Uri uri) {
5129261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        Cursor c = mResolver.query(uri, new String[]{"version"}, null, null, null);
5139261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        assertTrue(c.moveToNext());
5149261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        long version = c.getLong(0);
5159261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        assertFalse(c.moveToNext());
5163cfe8d532d509fbbe605454e3a32b2361b7e1501Dmitri Plotnikov        c.close();
5179261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        return version;
5189261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana    }
5199261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana
5209261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana    protected void clearDirty(Uri uri) {
5219261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        ContentValues values = new ContentValues();
5229261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        values.put("dirty", 0);
5239261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana        mResolver.update(uri, values, null, null);
5249261b2141aa90a4fed632fd6da03026d4c216280Fred Quintana    }
52520a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov
52620a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov    protected void assertStoredValues(Uri rowUri, String column, String expectedValue) {
52720a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov        String value = getStoredValue(rowUri, column);
52820a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov        assertEquals("Column value " + column, expectedValue, value);
52920a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov    }
53020a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov
53120a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov    protected String getStoredValue(Uri rowUri, String column) {
5328e45e5f2142db78941b095f7418cc05b71668094Dmitri Plotnikov        String value = null;
53320a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov        Cursor c = mResolver.query(rowUri, new String[] { column }, null, null, null);
53420a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov        try {
5358e45e5f2142db78941b095f7418cc05b71668094Dmitri Plotnikov            if (c.moveToFirst()) {
5368e45e5f2142db78941b095f7418cc05b71668094Dmitri Plotnikov                value = c.getString(c.getColumnIndex(column));
5378e45e5f2142db78941b095f7418cc05b71668094Dmitri Plotnikov            }
53820a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov        } finally {
53920a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov            c.close();
54020a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov        }
54120a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov        return value;
54220a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov    }
54320a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov
54420a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov    protected void assertStoredValues(Uri rowUri, ContentValues expectedValues) {
54520a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov        Cursor c = mResolver.query(rowUri, null, null, null, null);
54620a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov        try {
54720a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov            assertEquals("Record count", 1, c.getCount());
54820a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov            c.moveToFirst();
54920a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov            assertCursorValues(c, expectedValues);
55020a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov        } finally {
55120a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov            c.close();
55220a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov        }
55320a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov    }
55420a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov
555ff2de103f7e3eeeff4665ef63f07460fef053d6dDmitri Plotnikov    protected void assertStoredValuesWithProjection(Uri rowUri, ContentValues expectedValues) {
556ff2de103f7e3eeeff4665ef63f07460fef053d6dDmitri Plotnikov        Cursor c = mResolver.query(rowUri, buildProjection(expectedValues), null, null, null);
557ff2de103f7e3eeeff4665ef63f07460fef053d6dDmitri Plotnikov        try {
558ff2de103f7e3eeeff4665ef63f07460fef053d6dDmitri Plotnikov            assertEquals("Record count", 1, c.getCount());
559ff2de103f7e3eeeff4665ef63f07460fef053d6dDmitri Plotnikov            c.moveToFirst();
560ff2de103f7e3eeeff4665ef63f07460fef053d6dDmitri Plotnikov            assertCursorValues(c, expectedValues);
561ff2de103f7e3eeeff4665ef63f07460fef053d6dDmitri Plotnikov        } finally {
562ff2de103f7e3eeeff4665ef63f07460fef053d6dDmitri Plotnikov            c.close();
563ff2de103f7e3eeeff4665ef63f07460fef053d6dDmitri Plotnikov        }
564ff2de103f7e3eeeff4665ef63f07460fef053d6dDmitri Plotnikov    }
565ff2de103f7e3eeeff4665ef63f07460fef053d6dDmitri Plotnikov
56620a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov    /**
56720a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov     * Constructs a selection (where clause) out of all supplied values, uses it
56820a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov     * to query the provider and verifies that a single row is returned and it
56920a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov     * has the same values as requested.
57020a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov     */
57120a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov    protected void assertSelection(Uri uri, ContentValues values, String idColumn, long id) {
572ff2de103f7e3eeeff4665ef63f07460fef053d6dDmitri Plotnikov        assertSelection(uri, values, idColumn, id, null);
573ff2de103f7e3eeeff4665ef63f07460fef053d6dDmitri Plotnikov    }
574ff2de103f7e3eeeff4665ef63f07460fef053d6dDmitri Plotnikov
575ff2de103f7e3eeeff4665ef63f07460fef053d6dDmitri Plotnikov    public void assertSelectionWithProjection(Uri uri, ContentValues values, String idColumn,
576ff2de103f7e3eeeff4665ef63f07460fef053d6dDmitri Plotnikov            long id) {
577ff2de103f7e3eeeff4665ef63f07460fef053d6dDmitri Plotnikov        assertSelection(uri, values, idColumn, id, buildProjection(values));
578ff2de103f7e3eeeff4665ef63f07460fef053d6dDmitri Plotnikov    }
579ff2de103f7e3eeeff4665ef63f07460fef053d6dDmitri Plotnikov
580ff2de103f7e3eeeff4665ef63f07460fef053d6dDmitri Plotnikov    private void assertSelection(Uri uri, ContentValues values, String idColumn, long id,
581ff2de103f7e3eeeff4665ef63f07460fef053d6dDmitri Plotnikov            String[] projection) {
58220a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov        StringBuilder sb = new StringBuilder();
58320a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov        ArrayList<String> selectionArgs = new ArrayList<String>(values.size());
584c70dc0e38ff82c6e6d6b7458637c54fbdf446aacDmitri Plotnikov        if (idColumn != null) {
585c70dc0e38ff82c6e6d6b7458637c54fbdf446aacDmitri Plotnikov            sb.append(idColumn).append("=").append(id);
586c70dc0e38ff82c6e6d6b7458637c54fbdf446aacDmitri Plotnikov        }
58720a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov        Set<Map.Entry<String, Object>> entries = values.valueSet();
58820a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov        for (Map.Entry<String, Object> entry : entries) {
58920a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov            String column = entry.getKey();
59020a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov            Object value = entry.getValue();
591c70dc0e38ff82c6e6d6b7458637c54fbdf446aacDmitri Plotnikov            if (sb.length() != 0) {
592c70dc0e38ff82c6e6d6b7458637c54fbdf446aacDmitri Plotnikov                sb.append(" AND ");
593c70dc0e38ff82c6e6d6b7458637c54fbdf446aacDmitri Plotnikov            }
594c70dc0e38ff82c6e6d6b7458637c54fbdf446aacDmitri Plotnikov            sb.append(column);
59520a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov            if (value == null) {
59620a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov                sb.append(" IS NULL");
59720a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov            } else {
59820a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov                sb.append("=?");
59920a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov                selectionArgs.add(String.valueOf(value));
60020a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov            }
60120a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov        }
60220a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov
603ff2de103f7e3eeeff4665ef63f07460fef053d6dDmitri Plotnikov        Cursor c = mResolver.query(uri, projection, sb.toString(), selectionArgs.toArray(new String[0]),
60420a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov                null);
60520a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov        try {
60620a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov            assertEquals("Record count", 1, c.getCount());
60720a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov            c.moveToFirst();
60820a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov            assertCursorValues(c, values);
60920a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov        } finally {
61020a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov            c.close();
61120a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov        }
61220a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov    }
61320a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov
61420a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov    protected void assertCursorValues(Cursor cursor, ContentValues expectedValues) {
61520a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov        Set<Map.Entry<String, Object>> entries = expectedValues.valueSet();
61620a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov        for (Map.Entry<String, Object> entry : entries) {
61720a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov            String column = entry.getKey();
61820a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov            int index = cursor.getColumnIndex(column);
61920a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov            assertTrue("No such column: " + column, index != -1);
62020a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov            Object expectedValue = expectedValues.get(column);
62120a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov            String value;
62220a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov            if (expectedValue instanceof byte[]) {
62320a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov                expectedValue = Hex.encodeHex((byte[])expectedValue, false);
62420a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov                value = Hex.encodeHex(cursor.getBlob(index), false);
62520a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov            } else {
62620a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov                expectedValue = expectedValues.getAsString(column);
62720a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov                value = cursor.getString(index);
62820a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov            }
62920a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov            assertEquals("Column value " + column, expectedValue, value);
63020a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov        }
63120a94c86ede7380c8dd8df2f6a72b3c00ac1bed8Dmitri Plotnikov    }
63233b41fdb8d7c3c654cb070799c9d6e2b4ab16078Dmitri Plotnikov
633ff2de103f7e3eeeff4665ef63f07460fef053d6dDmitri Plotnikov    private String[] buildProjection(ContentValues values) {
634ff2de103f7e3eeeff4665ef63f07460fef053d6dDmitri Plotnikov        String[] projection = new String[values.size()];
635ff2de103f7e3eeeff4665ef63f07460fef053d6dDmitri Plotnikov        Iterator<Entry<String, Object>> iter = values.valueSet().iterator();
636ff2de103f7e3eeeff4665ef63f07460fef053d6dDmitri Plotnikov        for (int i = 0; i < projection.length; i++) {
637ff2de103f7e3eeeff4665ef63f07460fef053d6dDmitri Plotnikov            projection[i] = iter.next().getKey();
638ff2de103f7e3eeeff4665ef63f07460fef053d6dDmitri Plotnikov        }
639ff2de103f7e3eeeff4665ef63f07460fef053d6dDmitri Plotnikov        return projection;
640ff2de103f7e3eeeff4665ef63f07460fef053d6dDmitri Plotnikov    }
641ff2de103f7e3eeeff4665ef63f07460fef053d6dDmitri Plotnikov
64233b41fdb8d7c3c654cb070799c9d6e2b4ab16078Dmitri Plotnikov    protected int getCount(Uri uri, String selection, String[] selectionArgs) {
64333b41fdb8d7c3c654cb070799c9d6e2b4ab16078Dmitri Plotnikov        Cursor c = mResolver.query(uri, null, selection, selectionArgs, null);
64433b41fdb8d7c3c654cb070799c9d6e2b4ab16078Dmitri Plotnikov        try {
64533b41fdb8d7c3c654cb070799c9d6e2b4ab16078Dmitri Plotnikov            return c.getCount();
64633b41fdb8d7c3c654cb070799c9d6e2b4ab16078Dmitri Plotnikov        } finally {
64733b41fdb8d7c3c654cb070799c9d6e2b4ab16078Dmitri Plotnikov            c.close();
64833b41fdb8d7c3c654cb070799c9d6e2b4ab16078Dmitri Plotnikov        }
64933b41fdb8d7c3c654cb070799c9d6e2b4ab16078Dmitri Plotnikov    }
650a908fb5f39aa2021662a6cc317cc7e4db2d8bfb0Dmitri Plotnikov
6512482dd0309794d57233d0af37a7c12fc647e7bf9Dmitri Plotnikov    protected byte[] loadTestPhoto() {
6522482dd0309794d57233d0af37a7c12fc647e7bf9Dmitri Plotnikov        if (mTestPhoto == null) {
6532482dd0309794d57233d0af37a7c12fc647e7bf9Dmitri Plotnikov            final Resources resources = getContext().getResources();
6542482dd0309794d57233d0af37a7c12fc647e7bf9Dmitri Plotnikov            InputStream is = resources
6552482dd0309794d57233d0af37a7c12fc647e7bf9Dmitri Plotnikov                    .openRawResource(com.android.internal.R.drawable.ic_contact_picture);
6562482dd0309794d57233d0af37a7c12fc647e7bf9Dmitri Plotnikov            ByteArrayOutputStream os = new ByteArrayOutputStream();
6572482dd0309794d57233d0af37a7c12fc647e7bf9Dmitri Plotnikov            byte[] buffer = new byte[1000];
6582482dd0309794d57233d0af37a7c12fc647e7bf9Dmitri Plotnikov            int count;
6592482dd0309794d57233d0af37a7c12fc647e7bf9Dmitri Plotnikov            try {
6602482dd0309794d57233d0af37a7c12fc647e7bf9Dmitri Plotnikov                while ((count = is.read(buffer)) != -1) {
6612482dd0309794d57233d0af37a7c12fc647e7bf9Dmitri Plotnikov                    os.write(buffer, 0, count);
6622482dd0309794d57233d0af37a7c12fc647e7bf9Dmitri Plotnikov                }
6632482dd0309794d57233d0af37a7c12fc647e7bf9Dmitri Plotnikov            } catch (IOException e) {
6642482dd0309794d57233d0af37a7c12fc647e7bf9Dmitri Plotnikov                throw new RuntimeException(e);
6652482dd0309794d57233d0af37a7c12fc647e7bf9Dmitri Plotnikov            }
6662482dd0309794d57233d0af37a7c12fc647e7bf9Dmitri Plotnikov            mTestPhoto = os.toByteArray();
667a908fb5f39aa2021662a6cc317cc7e4db2d8bfb0Dmitri Plotnikov        }
6682482dd0309794d57233d0af37a7c12fc647e7bf9Dmitri Plotnikov        return mTestPhoto;
669a908fb5f39aa2021662a6cc317cc7e4db2d8bfb0Dmitri Plotnikov    }
670b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov
671b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov    public static void dump(ContentResolver resolver, boolean aggregatedOnly) {
672b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov        String[] projection = new String[] {
673b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov                Contacts._ID,
674b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov                Contacts.DISPLAY_NAME
675b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov        };
676b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov        String selection = null;
677b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov        if (aggregatedOnly) {
678b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov            selection = Contacts._ID
679b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov                    + " IN (SELECT contact_id" +
680b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov                    		" FROM raw_contacts GROUP BY contact_id HAVING count(*) > 1)";
681b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov        }
682b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov
683b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov        Cursor c = resolver.query(Contacts.CONTENT_URI, projection, selection, null,
684b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov                Contacts.DISPLAY_NAME);
685b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov        while(c.moveToNext()) {
686b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov            long contactId = c.getLong(0);
687b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov            Log.i("Contact   ", String.format("%5d %s", contactId, c.getString(1)));
688b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov            dumpRawContacts(resolver, contactId);
689b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov            Log.i("          ", ".");
690b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov        }
691b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov        c.close();
692b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov    }
693b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov
694b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov    private static void dumpRawContacts(ContentResolver resolver, long contactId) {
695b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov        String[] projection = new String[] {
696b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov                RawContacts._ID,
697b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov        };
698b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov        Cursor c = resolver.query(RawContacts.CONTENT_URI, projection, RawContacts.CONTACT_ID + "="
699b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov                + contactId, null, null);
700b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov        while(c.moveToNext()) {
701b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov            long rawContactId = c.getLong(0);
702b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov            Log.i("RawContact", String.format("      %-5d", rawContactId));
703b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov            dumpData(resolver, rawContactId);
704b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov        }
705b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov        c.close();
706b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov    }
707b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov
708b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov    private static void dumpData(ContentResolver resolver, long rawContactId) {
709b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov        String[] projection = new String[] {
710b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov                Data.MIMETYPE,
711b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov                Data.DATA1,
712b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov                Data.DATA2,
713b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov                Data.DATA3,
714b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov        };
715b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov        Cursor c = resolver.query(Data.CONTENT_URI, projection, Data.RAW_CONTACT_ID + "="
716b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov                + rawContactId, null, Data.MIMETYPE);
717b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov        while(c.moveToNext()) {
718b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov            String mimetype = c.getString(0);
719b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov            if (Photo.CONTENT_ITEM_TYPE.equals(mimetype)) {
720b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov                Log.i("Photo     ", "");
721b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov            } else {
722b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov                mimetype = mimetype.substring(mimetype.indexOf('/') + 1);
723b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov                Log.i("Data      ", String.format("            %-10s %s,%s,%s", mimetype,
724b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov                        c.getString(1), c.getString(2), c.getString(3)));
725b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov            }
726b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov        }
727b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov        c.close();
728b58058243bcd4405ad5ba5c3a465c46a35182ff7Dmitri Plotnikov    }
729d35d9c748af4c3182679c4c546137acfc11eb7a8Dmitri Plotnikov}
730