tests/src/com/android/providers/contacts/ContactsDatabaseHelperTest.java - platform/packages/providers/ContactsProvider - Git at Google

 /*
  * Copyright (C) 2011 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.database.sqlite.SQLiteDatabase;
 import android.test.MoreAsserts;
 import android.test.suitebuilder.annotation.LargeTest;
 import android.test.suitebuilder.annotation.SmallTest;

 import com.android.providers.contacts.ContactsDatabaseHelper.MimetypesColumns;
 import com.android.providers.contacts.ContactsDatabaseHelper.Tables;
 import com.google.android.collect.Sets;

 import java.util.HashSet;
 import java.util.Set;

 @SmallTest
 public class ContactsDatabaseHelperTest extends BaseContactsProvider2Test {
     private ContactsDatabaseHelper mDbHelper;
     private SQLiteDatabase mDb;

     @Override
     protected void setUp() throws Exception {
         super.setUp();
         mDbHelper = getContactsProvider().getDatabaseHelper(getContext());
         mDb = mDbHelper.getWritableDatabase();
     }

     public void testGetOrCreateAccountId() {
         final AccountWithDataSet a1 = null;
         final AccountWithDataSet a2 = new AccountWithDataSet("a", null, null);
         final AccountWithDataSet a3 = new AccountWithDataSet(null, "b", null);
         final AccountWithDataSet a4 = new AccountWithDataSet(null, null, "c");
         final AccountWithDataSet a5 = new AccountWithDataSet("a", "b", "c");

         // First, there's no accounts.  getAccountIdOrNull() always returns null.
         assertNull(mDbHelper.getAccountIdOrNull(a1));
         assertNull(mDbHelper.getAccountIdOrNull(a2));
         assertNull(mDbHelper.getAccountIdOrNull(a3));
         assertNull(mDbHelper.getAccountIdOrNull(a4));
         assertNull(mDbHelper.getAccountIdOrNull(a5));

         // getOrCreateAccountId should create accounts.
         final long a1id = mDbHelper.getOrCreateAccountIdInTransaction(a1);
         final long a2id = mDbHelper.getOrCreateAccountIdInTransaction(a2);
         final long a3id = mDbHelper.getOrCreateAccountIdInTransaction(a3);
         final long a4id = mDbHelper.getOrCreateAccountIdInTransaction(a4);
         final long a5id = mDbHelper.getOrCreateAccountIdInTransaction(a5);

         // The IDs should be all positive and unique.
         assertTrue(a1id > 0);
         assertTrue(a2id > 0);
         assertTrue(a3id > 0);
         assertTrue(a4id > 0);
         assertTrue(a5id > 0);

         final Set<Long> ids = Sets.newHashSet();
         ids.add(a1id);
         ids.add(a2id);
         ids.add(a3id);
         ids.add(a4id);
         ids.add(a5id);
         assertEquals(5, ids.size());

         // Second call: This time getOrCreateAccountId will return the existing IDs.
         assertEquals(a1id, mDbHelper.getOrCreateAccountIdInTransaction(a1));
         assertEquals(a2id, mDbHelper.getOrCreateAccountIdInTransaction(a2));
         assertEquals(a3id, mDbHelper.getOrCreateAccountIdInTransaction(a3));
         assertEquals(a4id, mDbHelper.getOrCreateAccountIdInTransaction(a4));
         assertEquals(a5id, mDbHelper.getOrCreateAccountIdInTransaction(a5));

         // Now getAccountIdOrNull() returns IDs too.
         assertEquals((Long) a1id, mDbHelper.getAccountIdOrNull(a1));
         assertEquals((Long) a2id, mDbHelper.getAccountIdOrNull(a2));
         assertEquals((Long) a3id, mDbHelper.getAccountIdOrNull(a3));
         assertEquals((Long) a4id, mDbHelper.getAccountIdOrNull(a4));
         assertEquals((Long) a5id, mDbHelper.getAccountIdOrNull(a5));

         // null and AccountWithDataSet.NULL should be treated as the same thing.
         assertEquals(a1id, mDbHelper.getOrCreateAccountIdInTransaction(AccountWithDataSet.LOCAL));
         assertEquals((Long) a1id, mDbHelper.getAccountIdOrNull(AccountWithDataSet.LOCAL));

         // Remove all accounts.
         mDbHelper.getWritableDatabase().execSQL("delete from " + Tables.ACCOUNTS);

         assertNull(mDbHelper.getAccountIdOrNull(AccountWithDataSet.LOCAL));
         assertNull(mDbHelper.getAccountIdOrNull(a1));
         assertNull(mDbHelper.getAccountIdOrNull(a2));
         assertNull(mDbHelper.getAccountIdOrNull(a3));
         assertNull(mDbHelper.getAccountIdOrNull(a4));
         assertNull(mDbHelper.getAccountIdOrNull(a5));

         // Logically same as a5, but physically different object.
         final AccountWithDataSet a5b = new AccountWithDataSet("a", "b", "c");
         // a5 and a5b should have the same ID.
         assertEquals(
                 mDbHelper.getOrCreateAccountIdInTransaction(a5),
                 mDbHelper.getOrCreateAccountIdInTransaction(a5b));
     }

     /**
      * Test for {@link ContactsDatabaseHelper#queryIdWithOneArg} and
      * {@link ContactsDatabaseHelper#insertWithOneArgAndReturnId}.
      */
     public void testQueryIdWithOneArg_insertWithOneArgAndReturnId() {
         final String query =
                 "SELECT " + MimetypesColumns._ID +
                         " FROM " + Tables.MIMETYPES +
                         " WHERE " + MimetypesColumns.MIMETYPE + "=?";

         final String insert =
                 "INSERT INTO " + Tables.MIMETYPES + "("
                         + MimetypesColumns.MIMETYPE +
                         ") VALUES (?)";

         // First, the table is empty.
         assertEquals(-1, ContactsDatabaseHelper.queryIdWithOneArg(mDb, query, "value1"));
         assertEquals(-1, ContactsDatabaseHelper.queryIdWithOneArg(mDb, query, "value2"));

         // Insert one value.
         final long id1 = ContactsDatabaseHelper.insertWithOneArgAndReturnId(mDb, insert, "value1");
         MoreAsserts.assertNotEqual(-1, id1);

         assertEquals(id1, ContactsDatabaseHelper.queryIdWithOneArg(mDb, query, "value1"));
         assertEquals(-1, ContactsDatabaseHelper.queryIdWithOneArg(mDb, query, "value2"));


         // Insert one value.
         final long id2 = ContactsDatabaseHelper.insertWithOneArgAndReturnId(mDb, insert, "value2");
         MoreAsserts.assertNotEqual(-1, id2);

         assertEquals(id1, ContactsDatabaseHelper.queryIdWithOneArg(mDb, query, "value1"));
         assertEquals(id2, ContactsDatabaseHelper.queryIdWithOneArg(mDb, query, "value2"));

         // Insert the same value and cause a conflict.
         assertEquals(-1, ContactsDatabaseHelper.insertWithOneArgAndReturnId(mDb, insert, "value2"));
     }

     /**
      * Test for {@link ContactsDatabaseHelper#getPackageId(String)} and
      * {@link ContactsDatabaseHelper#getMimeTypeId(String)}.
      *
      * We test them at the same time here, to make sure they're not mixing up the caches.
      */
     public void testGetPackageId_getMimeTypeId() {

         // Test for getPackageId.
         final long packageId1 = mDbHelper.getPackageId("value1");
         final long packageId2 = mDbHelper.getPackageId("value2");
         final long packageId3 = mDbHelper.getPackageId("value3");

         // Make sure they're all different.
         final HashSet<Long> set = new HashSet<>();
         set.add(packageId1);
         set.add(packageId2);
         set.add(packageId3);

         assertEquals(3, set.size());

         // Test for getMimeTypeId.
         final long mimetypeId1 = mDbHelper.getMimeTypeId("value1");
         final long mimetypeId2 = mDbHelper.getMimeTypeId("value2");
         final long mimetypeId3 = mDbHelper.getMimeTypeId("value3");

         // Make sure they're all different.
         set.clear();
         set.add(mimetypeId1);
         set.add(mimetypeId2);
         set.add(mimetypeId3);

         assertEquals(3, set.size());

         // Call with the same values and make sure they return the cached value.
         final long packageId1b = mDbHelper.getPackageId("value1");
         final long mimetypeId1b = mDbHelper.getMimeTypeId("value1");

         assertEquals(packageId1, packageId1b);
         assertEquals(mimetypeId1, mimetypeId1b);

         // Make sure the caches are also updated.
         assertEquals(packageId2, (long) mDbHelper.mPackageCache.get("value2"));
         assertEquals(mimetypeId2, (long) mDbHelper.mMimetypeCache.get("value2"));

         // Clear the cache, but they should still return the values, selecting from the database.
         mDbHelper.mPackageCache.clear();
         mDbHelper.mMimetypeCache.clear();
         assertEquals(packageId1, mDbHelper.getPackageId("value1"));
         assertEquals(mimetypeId1, mDbHelper.getMimeTypeId("value1"));

         // Empty the table
         mDb.execSQL("DELETE FROM " + Tables.MIMETYPES);

         // We should still have the cached value.
         assertEquals(mimetypeId1, mDbHelper.getMimeTypeId("value1"));
     }

     /**
      * Try to cause conflicts in getMimeTypeId() by calling it from multiple threads with
      * the current time as the argument and make sure it won't crash.
      *
      * We don't know from the test if there have actually been conflits, but if you look at
      * logcat you'll see a lot of conflict warnings.
      */
     @LargeTest
     public void testGetMimeTypeId_conflict() {

         final int NUM_THREADS = 4;
         final int DURATION_SECONDS = 5;

         final long finishTime = System.currentTimeMillis() + DURATION_SECONDS * 1000;

         final Runnable r = new Runnable() {
             @Override
             public void run() {
                 for (;;) {
                     final long now = System.currentTimeMillis();
                     if (now >= finishTime) {
                         return;
                     }
                     assertTrue(mDbHelper.getMimeTypeId(String.valueOf(now)) > 0);
                 }
             }
         };
         final Thread[] threads = new Thread[NUM_THREADS];
         for (int i = 0; i < threads.length; i++) {
             threads[i] = new Thread(r);
             threads[i].setDaemon(true);
         }
         for (int i = 0; i < threads.length; i++) {
             threads[i].start();
         }
         for (int i = 0; i < threads.length; i++) {
             try {
                 threads[i].join();
             } catch (InterruptedException ignore) {
             }
         }
     }
 }
	/*
	* Copyright (C) 2011 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.database.sqlite.SQLiteDatabase;
	import android.test.MoreAsserts;
	import android.test.suitebuilder.annotation.LargeTest;
	import android.test.suitebuilder.annotation.SmallTest;

	import com.android.providers.contacts.ContactsDatabaseHelper.MimetypesColumns;
	import com.android.providers.contacts.ContactsDatabaseHelper.Tables;
	import com.google.android.collect.Sets;

	import java.util.HashSet;
	import java.util.Set;

	@SmallTest
	public class ContactsDatabaseHelperTest extends BaseContactsProvider2Test {
	private ContactsDatabaseHelper mDbHelper;
	private SQLiteDatabase mDb;

	@Override
	protected void setUp() throws Exception {
	super.setUp();
	mDbHelper = getContactsProvider().getDatabaseHelper(getContext());
	mDb = mDbHelper.getWritableDatabase();
	}

	public void testGetOrCreateAccountId() {
	final AccountWithDataSet a1 = null;
	final AccountWithDataSet a2 = new AccountWithDataSet("a", null, null);
	final AccountWithDataSet a3 = new AccountWithDataSet(null, "b", null);
	final AccountWithDataSet a4 = new AccountWithDataSet(null, null, "c");
	final AccountWithDataSet a5 = new AccountWithDataSet("a", "b", "c");

	// First, there's no accounts. getAccountIdOrNull() always returns null.
	assertNull(mDbHelper.getAccountIdOrNull(a1));
	assertNull(mDbHelper.getAccountIdOrNull(a2));
	assertNull(mDbHelper.getAccountIdOrNull(a3));
	assertNull(mDbHelper.getAccountIdOrNull(a4));
	assertNull(mDbHelper.getAccountIdOrNull(a5));

	// getOrCreateAccountId should create accounts.
	final long a1id = mDbHelper.getOrCreateAccountIdInTransaction(a1);
	final long a2id = mDbHelper.getOrCreateAccountIdInTransaction(a2);
	final long a3id = mDbHelper.getOrCreateAccountIdInTransaction(a3);
	final long a4id = mDbHelper.getOrCreateAccountIdInTransaction(a4);
	final long a5id = mDbHelper.getOrCreateAccountIdInTransaction(a5);

	// The IDs should be all positive and unique.
	assertTrue(a1id > 0);
	assertTrue(a2id > 0);
	assertTrue(a3id > 0);
	assertTrue(a4id > 0);
	assertTrue(a5id > 0);

	final Set<Long> ids = Sets.newHashSet();
	ids.add(a1id);
	ids.add(a2id);
	ids.add(a3id);
	ids.add(a4id);
	ids.add(a5id);
	assertEquals(5, ids.size());

	// Second call: This time getOrCreateAccountId will return the existing IDs.
	assertEquals(a1id, mDbHelper.getOrCreateAccountIdInTransaction(a1));
	assertEquals(a2id, mDbHelper.getOrCreateAccountIdInTransaction(a2));
	assertEquals(a3id, mDbHelper.getOrCreateAccountIdInTransaction(a3));
	assertEquals(a4id, mDbHelper.getOrCreateAccountIdInTransaction(a4));
	assertEquals(a5id, mDbHelper.getOrCreateAccountIdInTransaction(a5));

	// Now getAccountIdOrNull() returns IDs too.
	assertEquals((Long) a1id, mDbHelper.getAccountIdOrNull(a1));
	assertEquals((Long) a2id, mDbHelper.getAccountIdOrNull(a2));
	assertEquals((Long) a3id, mDbHelper.getAccountIdOrNull(a3));
	assertEquals((Long) a4id, mDbHelper.getAccountIdOrNull(a4));
	assertEquals((Long) a5id, mDbHelper.getAccountIdOrNull(a5));

	// null and AccountWithDataSet.NULL should be treated as the same thing.
	assertEquals(a1id, mDbHelper.getOrCreateAccountIdInTransaction(AccountWithDataSet.LOCAL));
	assertEquals((Long) a1id, mDbHelper.getAccountIdOrNull(AccountWithDataSet.LOCAL));

	// Remove all accounts.
	mDbHelper.getWritableDatabase().execSQL("delete from " + Tables.ACCOUNTS);

	assertNull(mDbHelper.getAccountIdOrNull(AccountWithDataSet.LOCAL));
	assertNull(mDbHelper.getAccountIdOrNull(a1));
	assertNull(mDbHelper.getAccountIdOrNull(a2));
	assertNull(mDbHelper.getAccountIdOrNull(a3));
	assertNull(mDbHelper.getAccountIdOrNull(a4));
	assertNull(mDbHelper.getAccountIdOrNull(a5));

	// Logically same as a5, but physically different object.
	final AccountWithDataSet a5b = new AccountWithDataSet("a", "b", "c");
	// a5 and a5b should have the same ID.
	assertEquals(
	mDbHelper.getOrCreateAccountIdInTransaction(a5),
	mDbHelper.getOrCreateAccountIdInTransaction(a5b));
	}

	/**
	* Test for {@link ContactsDatabaseHelper#queryIdWithOneArg} and
	* {@link ContactsDatabaseHelper#insertWithOneArgAndReturnId}.
	*/
	public void testQueryIdWithOneArg_insertWithOneArgAndReturnId() {
	final String query =
	"SELECT " + MimetypesColumns._ID +
	" FROM " + Tables.MIMETYPES +
	" WHERE " + MimetypesColumns.MIMETYPE + "=?";

	final String insert =
	"INSERT INTO " + Tables.MIMETYPES + "("
	+ MimetypesColumns.MIMETYPE +
	") VALUES (?)";

	// First, the table is empty.
	assertEquals(-1, ContactsDatabaseHelper.queryIdWithOneArg(mDb, query, "value1"));
	assertEquals(-1, ContactsDatabaseHelper.queryIdWithOneArg(mDb, query, "value2"));

	// Insert one value.
	final long id1 = ContactsDatabaseHelper.insertWithOneArgAndReturnId(mDb, insert, "value1");
	MoreAsserts.assertNotEqual(-1, id1);

	assertEquals(id1, ContactsDatabaseHelper.queryIdWithOneArg(mDb, query, "value1"));
	assertEquals(-1, ContactsDatabaseHelper.queryIdWithOneArg(mDb, query, "value2"));


	// Insert one value.
	final long id2 = ContactsDatabaseHelper.insertWithOneArgAndReturnId(mDb, insert, "value2");
	MoreAsserts.assertNotEqual(-1, id2);

	assertEquals(id1, ContactsDatabaseHelper.queryIdWithOneArg(mDb, query, "value1"));
	assertEquals(id2, ContactsDatabaseHelper.queryIdWithOneArg(mDb, query, "value2"));

	// Insert the same value and cause a conflict.
	assertEquals(-1, ContactsDatabaseHelper.insertWithOneArgAndReturnId(mDb, insert, "value2"));
	}

	/**
	* Test for {@link ContactsDatabaseHelper#getPackageId(String)} and
	* {@link ContactsDatabaseHelper#getMimeTypeId(String)}.
	*
	* We test them at the same time here, to make sure they're not mixing up the caches.
	*/
	public void testGetPackageId_getMimeTypeId() {

	// Test for getPackageId.
	final long packageId1 = mDbHelper.getPackageId("value1");
	final long packageId2 = mDbHelper.getPackageId("value2");
	final long packageId3 = mDbHelper.getPackageId("value3");

	// Make sure they're all different.
	final HashSet<Long> set = new HashSet<>();
	set.add(packageId1);
	set.add(packageId2);
	set.add(packageId3);

	assertEquals(3, set.size());

	// Test for getMimeTypeId.
	final long mimetypeId1 = mDbHelper.getMimeTypeId("value1");
	final long mimetypeId2 = mDbHelper.getMimeTypeId("value2");
	final long mimetypeId3 = mDbHelper.getMimeTypeId("value3");

	// Make sure they're all different.
	set.clear();
	set.add(mimetypeId1);
	set.add(mimetypeId2);
	set.add(mimetypeId3);

	assertEquals(3, set.size());

	// Call with the same values and make sure they return the cached value.
	final long packageId1b = mDbHelper.getPackageId("value1");
	final long mimetypeId1b = mDbHelper.getMimeTypeId("value1");

	assertEquals(packageId1, packageId1b);
	assertEquals(mimetypeId1, mimetypeId1b);

	// Make sure the caches are also updated.
	assertEquals(packageId2, (long) mDbHelper.mPackageCache.get("value2"));
	assertEquals(mimetypeId2, (long) mDbHelper.mMimetypeCache.get("value2"));

	// Clear the cache, but they should still return the values, selecting from the database.
	mDbHelper.mPackageCache.clear();
	mDbHelper.mMimetypeCache.clear();
	assertEquals(packageId1, mDbHelper.getPackageId("value1"));
	assertEquals(mimetypeId1, mDbHelper.getMimeTypeId("value1"));

	// Empty the table
	mDb.execSQL("DELETE FROM " + Tables.MIMETYPES);

	// We should still have the cached value.
	assertEquals(mimetypeId1, mDbHelper.getMimeTypeId("value1"));
	}

	/**
	* Try to cause conflicts in getMimeTypeId() by calling it from multiple threads with
	* the current time as the argument and make sure it won't crash.
	*
	* We don't know from the test if there have actually been conflits, but if you look at
	* logcat you'll see a lot of conflict warnings.
	*/
	@LargeTest
	public void testGetMimeTypeId_conflict() {

	final int NUM_THREADS = 4;
	final int DURATION_SECONDS = 5;

	final long finishTime = System.currentTimeMillis() + DURATION_SECONDS * 1000;

	final Runnable r = new Runnable() {
	@Override
	public void run() {
	for (;;) {
	final long now = System.currentTimeMillis();
	if (now >= finishTime) {
	return;
	}
	assertTrue(mDbHelper.getMimeTypeId(String.valueOf(now)) > 0);
	}
	}
	};
	final Thread[] threads = new Thread[NUM_THREADS];
	for (int i = 0; i < threads.length; i++) {
	threads[i] = new Thread(r);
	threads[i].setDaemon(true);
	}
	for (int i = 0; i < threads.length; i++) {
	threads[i].start();
	}
	for (int i = 0; i < threads.length; i++) {
	try {
	threads[i].join();
	} catch (InterruptedException ignore) {
	}
	}
	}
	}