1/*
2 * Copyright (C) 2009 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
17package com.android.providers.contacts;
18
19import android.content.ContentProvider;
20import android.content.Context;
21import android.content.pm.ApplicationInfo;
22import android.content.pm.PackageManager;
23import android.content.res.Resources;
24import android.os.Debug;
25import android.provider.CallLog;
26import android.provider.ContactsContract;
27import android.test.AndroidTestCase;
28import android.test.IsolatedContext;
29import android.test.RenamingDelegatingContext;
30import android.test.mock.MockContentResolver;
31import android.test.mock.MockContext;
32import android.test.suitebuilder.annotation.MediumTest;
33import android.util.Log;
34
35import java.io.File;
36
37/**
38 * Performance test for {@link ContactAggregator}. Run the test like this:
39 * <code>
40 * adb push <large contacts.db> \
41 *         data/data/com.android.providers.contacts/databases/perf_imp.contacts.db
42 * adb shell am instrument \
43 *         -e class com.android.providers.contacts.LegacyContactImporterPerformanceTest \
44 *         -w com.android.providers.contacts.tests/android.test.InstrumentationTestRunner
45 * </code>
46 *
47 * Note that this SHOULD be a large test, but had to be bumped down to medium due to a bug in the
48 * SQLite cleanup code.
49 */
50@MediumTest
51public class LegacyContactImporterPerformanceTest extends AndroidTestCase {
52
53    private static final String TAG = "LegacyContactImporterPerformanceTest";
54
55    private static final boolean TRACE = false;
56
57    @Override
58    protected void setUp() throws Exception {
59        super.setUp();
60        SynchronousContactsProvider2.resetOpenHelper();
61    }
62
63    @Override
64    protected void tearDown() throws Exception {
65        super.tearDown();
66        SynchronousContactsProvider2.resetOpenHelper();
67    }
68
69    public void testPerformance() {
70        final Context targetContext = getContext();
71        MockContentResolver resolver = new MockContentResolver();
72        MockContext context = new MockContext() {
73            @Override
74            public Resources getResources() {
75                return targetContext.getResources();
76            }
77
78            @Override
79            public String getPackageName() {
80                return "no.package";
81            }
82
83            @Override
84            public ApplicationInfo getApplicationInfo() {
85                ApplicationInfo ai = new ApplicationInfo();
86                ai.packageName = "contactsTestPackage";
87                return ai;
88            }
89
90            @Override
91            public PackageManager getPackageManager() {
92                return new ContactsMockPackageManager();
93            }
94        };
95
96        RenamingDelegatingContext targetContextWrapper = new RenamingDelegatingContext(context,
97                targetContext, "perf_imp.");
98        targetContextWrapper.makeExistingFilesAndDbsAccessible();
99        IsolatedContext providerContext = new IsolatedContext(resolver, targetContextWrapper) {
100            @Override
101            public File getFilesDir() {
102                // TODO: Need to figure out something more graceful than this.
103                return new File("/data/data/com.android.providers.contacts.tests/files");
104            }
105        };
106        SynchronousContactsProvider2 provider = new SynchronousContactsProvider2();
107        provider.setDataWipeEnabled(false);
108        provider.attachInfo(providerContext, null);
109        resolver.addProvider(ContactsContract.AUTHORITY, provider);
110
111        ContentProvider callLogProvider = new TestCallLogProvider();
112        callLogProvider.attachInfo(providerContext, null);
113        resolver.addProvider(CallLog.AUTHORITY, callLogProvider);
114
115        LegacyContactImporter importer = new LegacyContactImporter(providerContext, provider);
116
117        provider.wipeData();
118
119        long start = System.currentTimeMillis();
120        if (TRACE) {
121            Debug.startMethodTracing("import");
122        }
123        provider.importLegacyContacts(importer);
124        if (TRACE) {
125            Debug.stopMethodTracing();
126        }
127        long end = System.currentTimeMillis();
128        long contactCount = provider.getRawContactCount();
129        Log.i(TAG, String.format("Imported contacts in %d ms.\n"
130                + "Contacts: %d\n"
131                + "Per contact: %.3f",
132                end - start,
133                contactCount,
134                ((double)(end - start) / contactCount)));
135    }
136
137    public static class TestCallLogProvider extends CallLogProvider {
138        private static ContactsDatabaseHelper mDbHelper;
139
140        @Override
141        protected ContactsDatabaseHelper getDatabaseHelper(final Context context) {
142            if (mDbHelper == null) {
143                mDbHelper = new ContactsDatabaseHelper(context);
144            }
145            return mDbHelper;
146        }
147    }
148}
149