LegacyContactImporterPerformanceTest.java

/*
 * Copyright (C) 2009 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.android.providers.contacts;

import android.content.ContentProvider;
import android.content.Context;
import android.content.pm.ApplicationInfo;
import android.content.pm.PackageManager;
import android.content.res.Resources;
import android.os.Debug;
import android.provider.CallLog;
import android.provider.ContactsContract;
import android.test.AndroidTestCase;
import android.test.IsolatedContext;
import android.test.RenamingDelegatingContext;
import android.test.mock.MockContentResolver;
import android.test.mock.MockContext;
import android.test.suitebuilder.annotation.MediumTest;
import android.util.Log;

import java.io.File;

/**
 * Performance test for {@link ContactAggregator}. Run the test like this:
 * <code>
 * adb push <large contacts.db> \
 *         data/data/com.android.providers.contacts/databases/perf_imp.contacts.db
 * adb shell am instrument \
 *         -e class com.android.providers.contacts.LegacyContactImporterPerformanceTest \
 *         -w com.android.providers.contacts.tests/android.test.InstrumentationTestRunner
 * </code>
 *
 * Note that this SHOULD be a large test, but had to be bumped down to medium due to a bug in the
 * SQLite cleanup code.
 */
@MediumTest
public class LegacyContactImporterPerformanceTest extends AndroidTestCase {

    private static final String TAG = "LegacyContactImporterPerformanceTest";

    private static final boolean TRACE = false;

    @Override
    protected void setUp() throws Exception {
        super.setUp();
        SynchronousContactsProvider2.resetOpenHelper();
    }

    @Override
    protected void tearDown() throws Exception {
        super.tearDown();
        SynchronousContactsProvider2.resetOpenHelper();
    }

    public void testPerformance() {
        final Context targetContext = getContext();
        MockContentResolver resolver = new MockContentResolver();
        MockContext context = new MockContext() {
            @Override
            public Resources getResources() {
                return targetContext.getResources();
            }

            @Override
            public String getPackageName() {
                return "no.package";
            }

            @Override
            public ApplicationInfo getApplicationInfo() {
                ApplicationInfo ai = new ApplicationInfo();
                ai.packageName = "contactsTestPackage";
                return ai;
            }

            @Override
            public PackageManager getPackageManager() {
                return new ContactsMockPackageManager();
            }
        };

        RenamingDelegatingContext targetContextWrapper = new RenamingDelegatingContext(context,
                targetContext, "perf_imp.");
        targetContextWrapper.makeExistingFilesAndDbsAccessible();
        IsolatedContext providerContext = new IsolatedContext(resolver, targetContextWrapper) {
            @Override
            public File getFilesDir() {
                // TODO: Need to figure out something more graceful than this.
                return new File("/data/data/com.android.providers.contacts.tests/files");
            }
        };
        SynchronousContactsProvider2 provider = new SynchronousContactsProvider2();
        provider.setDataWipeEnabled(false);
        provider.attachInfo(providerContext, null);
        resolver.addProvider(ContactsContract.AUTHORITY, provider);

        ContentProvider callLogProvider = new TestCallLogProvider();
        callLogProvider.attachInfo(providerContext, null);
        resolver.addProvider(CallLog.AUTHORITY, callLogProvider);

        LegacyContactImporter importer = new LegacyContactImporter(providerContext, provider);

        provider.wipeData();

        long start = System.currentTimeMillis();
        if (TRACE) {
            Debug.startMethodTracing("import");
        }
        provider.importLegacyContacts(importer);
        if (TRACE) {
            Debug.stopMethodTracing();
        }
        long end = System.currentTimeMillis();
        long contactCount = provider.getRawContactCount();
        Log.i(TAG, String.format("Imported contacts in %d ms.\n"
                + "Contacts: %d\n"
                + "Per contact: %.3f",
                end - start,
                contactCount,
                ((double)(end - start) / contactCount)));
    }

    public static class TestCallLogProvider extends CallLogProvider {
        private static ContactsDatabaseHelper mDbHelper;

        @Override
        protected ContactsDatabaseHelper getDatabaseHelper(final Context context) {
            if (mDbHelper == null) {
                mDbHelper = new ContactsDatabaseHelper(context);
            }
            return mDbHelper;
        }
    }
}