tests/src/com/android/dialer/database/VoicemailArchiveProviderTest.java - platform/packages/apps/Dialer - Gitiles

 /*
  * Copyright (C) 2016 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.dialer.database;

 import android.content.ContentUris;
 import android.content.ContentValues;
 import android.content.Context;
 import android.database.Cursor;
 import android.net.Uri;
 import android.test.ProviderTestCase2;
 import android.test.mock.MockContentResolver;
 import com.android.dialer.database.VoicemailArchiveContract.VoicemailArchive;

 import java.io.File;
 import java.io.OutputStream;

 /**
  * Tests for {@link VoicemailArchiveProvider}.
  */
 public class VoicemailArchiveProviderTest extends
         ProviderTestCase2<VoicemailArchiveProviderTest.TestVoicemailArchiveProvider> {
     private static final String TEST_MIME_TYPE = "audio/mp3";
     private static final String TEST_NUMBER = "+1412555555";
     private static final int TEST_ARCHIVED = 1;
     private static final String TEST_STRING = "TEST";

     private MockContentResolver mResolver;
     private Cursor mCursor;
     private Uri mVoicemailUri;

     public VoicemailArchiveProviderTest() {
         super(TestVoicemailArchiveProvider.class, VoicemailArchiveContract.AUTHORITY);
     }

     @Override
     protected void setUp() throws Exception {
         super.setUp();
         mResolver = getMockContentResolver();
     }

     @Override
     protected void tearDown() throws Exception {
         if (mCursor != null) {
             mCursor.close();
         }
         // Need to delete test.cache folder created by {@link android.test.ProviderTestCase2}
         deleteRecursive(getContext().getCacheDir());
         getProvider().closeDb();
         super.tearDown();
     }

     public void testInsert() {
         insertVoicemail();
         assertNotNull(mVoicemailUri);
         assertTrue(doesRowExist());
         long id = ContentUris.parseId(mVoicemailUri);
         assertTrue(id > 0);
     }

     public void testQuery_createFile() throws Exception {
         insertVoicemail();
         assertTrue(doesRowExist());
         assertFalse(doesFileExist());
         createFile();
         assertTrue(doesFileExist());
     }

     public void testQuery_createFileMimeTypeIsAMR() throws Exception {
         insertVoicemailWithMimeType("audio/amr");
         assertTrue(doesRowExist());
         assertFalse(doesFileExist());
         createFile();
         assertTrue(doesFileExist());
         assertEquals("amr", getFileExtension(getFilePath()));
     }

     public void testQuery_createFileMimeTypeIsMP3() throws Exception {
         insertVoicemailWithMimeType("audio/mpeg");
         assertTrue(doesRowExist());
         assertFalse(doesFileExist());
         createFile();
         assertTrue(doesFileExist());
         assertEquals("mp3", getFileExtension(getFilePath()));
     }

     public void testQuery_createFileMimeTypeNotExists() throws Exception {
         insertVoicemailWithMimeType(TEST_STRING);
         assertTrue(doesRowExist());
         assertFalse(doesFileExist());
         createFile();
         assertTrue(doesFileExist());
         assertEquals("", getFileExtension(getFilePath()));
     }

     public void testQuery() {
         insertVoicemail();
         updateCursor();
         assertCursorCount(1);
         assertContentValues();
     }

     public void testQuery_correctValuesSelection() {
         insertVoicemail();
         updateCursorWithIdQuery();
         assertCursorCount(1);
         assertContentValues();
     }

     public void testQuery_illegalUri() {
         try {
             mResolver.query(Uri.withAppendedPath(VoicemailArchive.CONTENT_URI, TEST_STRING),
                     null, null, null, null);
             fail("Expecting exception but none was thrown.");
         } catch (IllegalArgumentException e) {}
     }

     public void testUpdate() throws Exception {
         insertVoicemail();
         assertTrue(doesRowExist());
         ContentValues values = new ContentValues();
         values.put(VoicemailArchive.MIME_TYPE, TEST_STRING);
         values.put(VoicemailArchive.NUMBER, TEST_STRING);
         values.put(VoicemailArchive.ARCHIVED, false);
         mResolver.update(mVoicemailUri, values, null, null);
         updateCursor();
         assertEquals(TEST_STRING, mCursor.getString(mCursor.getColumnIndex(VoicemailArchive.MIME_TYPE)));
         assertEquals(TEST_STRING, mCursor.getString(mCursor.getColumnIndex(VoicemailArchive.NUMBER)));
         assertEquals(false, mCursor.getInt(
                 mCursor.getColumnIndex(VoicemailArchive.ARCHIVED)) == 1);
     }

     public void testUpdate_selection() throws Exception {
         insertVoicemail();
         assertTrue(doesRowExist());
         ContentValues values = new ContentValues();
         values.put(VoicemailArchive.MIME_TYPE, TEST_STRING);
         values.put(VoicemailArchive.NUMBER, TEST_STRING);
         values.put(VoicemailArchive.ARCHIVED, false);
         mResolver.update(VoicemailArchive.CONTENT_URI, values, getIdQuery(), null);
         updateCursor();
         assertEquals(TEST_STRING, mCursor.getString(mCursor.getColumnIndex(VoicemailArchive.MIME_TYPE)));
         assertEquals(TEST_STRING, mCursor.getString(mCursor.getColumnIndex(VoicemailArchive.NUMBER)));
         assertEquals(false, mCursor.getInt(
                 mCursor.getColumnIndex(VoicemailArchive.ARCHIVED)) == 1);
     }

     public void testUpdate_illegalUri() {
         try {
             mResolver.update(Uri.withAppendedPath(VoicemailArchive.CONTENT_URI, TEST_STRING),
                     null, null, null);
             fail("Expecting exception but none was thrown.");
         } catch (IllegalArgumentException e) {}
     }

     public void testDelete() throws Exception {
         insertVoicemail();
         createFile();
         assertTrue(doesRowExist());
         assertTrue(doesFileExist());
         mResolver.delete(mVoicemailUri, null, null);
         assertFalse(doesRowExist());
         assertFalse(doesFileExist());
     }

     public void testDelete_selection() throws Exception{
         insertVoicemail();
         createFile();
         assertTrue(doesRowExist());
         assertTrue(doesFileExist());
         mResolver.delete(VoicemailArchive.CONTENT_URI, getIdQuery(), null);
         assertFalse(doesRowExist());
         assertFalse(doesFileExist());
     }

     public void testDelete_illegalUri() {
         try {
             mResolver.delete(Uri.withAppendedPath(VoicemailArchive.CONTENT_URI, TEST_STRING),
                     null, null);
             fail("Expecting exception but none was thrown.");
         } catch (IllegalArgumentException e) {}
     }

     private boolean doesRowExist() {
         Cursor tempCursor = mResolver.query(mVoicemailUri, null, null, null, null);
         boolean rowExists = tempCursor != null && tempCursor.getCount() > 0;
         tempCursor.close();
         return rowExists;
     }

     private boolean doesFileExist() {
         File voicemailFile = new File(getFilePath());
         return voicemailFile.exists();
     }

     private static String getFileExtension(String filePath) {
         File file = new File(filePath);
         String fileName = file.getName();
         int index = fileName.lastIndexOf(".");
         return index > 0 ? fileName.substring(index + 1) : "";
     }

     private void assertCursorCount(int count) {
         assertEquals(count, mCursor.getCount());
     }

     private void assertContentValues() {
         assertEquals(TEST_MIME_TYPE, mCursor
                 .getString(mCursor.getColumnIndex(VoicemailArchive.MIME_TYPE)));
         assertEquals(TEST_NUMBER, mCursor
                 .getString(mCursor.getColumnIndex(VoicemailArchive.NUMBER)));
         assertEquals(TEST_ARCHIVED, mCursor
                 .getInt(mCursor.getColumnIndex(VoicemailArchive.ARCHIVED)));
     }

     private void insertVoicemail() {
         mVoicemailUri = mResolver.insert(VoicemailArchive.CONTENT_URI, getTestValues());
     }

     private void insertVoicemailWithMimeType(String mimeType) {
         ContentValues values = getTestValues();
         values.put(VoicemailArchive.MIME_TYPE, mimeType);
         mVoicemailUri = mResolver.insert(VoicemailArchive.CONTENT_URI, values);
     }

     private void updateCursor() {
         mCursor = mResolver.query(mVoicemailUri, null, null, null, null);
         assertEquals(true, mCursor.getCount() > 0);
         mCursor.moveToFirst();
     }

     private void updateCursorWithIdQuery() {
         mCursor = mResolver.query(mVoicemailUri, null, getIdQuery(), null, null);
         assertEquals(true, mCursor.getCount() > 0);
         mCursor.moveToFirst();
     }

     private void createFile() throws Exception {
         assertFalse(doesFileExist());
         // Opening output stream and closing it should create the file
         OutputStream outputStream = mResolver.openOutputStream(mVoicemailUri);
         outputStream.close();
     }

     private String getIdQuery() {
         return VoicemailArchive._ID + "=" + ContentUris.parseId(mVoicemailUri);
     }

     private String getFilePath() {
         if (mCursor == null) {
             updateCursor();
         }
         return mCursor.getString(mCursor.getColumnIndex(VoicemailArchive._DATA));
     }

     private static ContentValues getTestValues() {
         ContentValues values = new ContentValues();
         values.put(VoicemailArchive.NUMBER, TEST_NUMBER);
         values.put(VoicemailArchive.MIME_TYPE, TEST_MIME_TYPE);
         values.put(VoicemailArchive.ARCHIVED, TEST_ARCHIVED);
         return values;
     }

     private void deleteRecursive(File fileOrDirectory) {
         if (fileOrDirectory.isDirectory()) {
             for (File child : fileOrDirectory.listFiles()) {
                 deleteRecursive(child);
             }
         }
         fileOrDirectory.delete();
     }

     public static class TestVoicemailArchiveProvider extends VoicemailArchiveProvider {
         private DialerDatabaseHelper mDialerDatabaseHelper;

         @Override
         protected File getFilesDir() {
             return getContext().getCacheDir();
         }

         @Override
         protected DialerDatabaseHelper getDatabaseHelper(Context context) {
             if (mDialerDatabaseHelper == null) {
                 mDialerDatabaseHelper = DialerDatabaseHelper.getNewInstanceForTest(context);
             }
             return mDialerDatabaseHelper;
         }

         protected void closeDb() {
             mDialerDatabaseHelper.close();
         }
     }
 }
	/*
	* Copyright (C) 2016 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.dialer.database;

	import android.content.ContentUris;
	import android.content.ContentValues;
	import android.content.Context;
	import android.database.Cursor;
	import android.net.Uri;
	import android.test.ProviderTestCase2;
	import android.test.mock.MockContentResolver;
	import com.android.dialer.database.VoicemailArchiveContract.VoicemailArchive;

	import java.io.File;
	import java.io.OutputStream;

	/**
	* Tests for {@link VoicemailArchiveProvider}.
	*/
	public class VoicemailArchiveProviderTest extends
	ProviderTestCase2<VoicemailArchiveProviderTest.TestVoicemailArchiveProvider> {
	private static final String TEST_MIME_TYPE = "audio/mp3";
	private static final String TEST_NUMBER = "+1412555555";
	private static final int TEST_ARCHIVED = 1;
	private static final String TEST_STRING = "TEST";

	private MockContentResolver mResolver;
	private Cursor mCursor;
	private Uri mVoicemailUri;

	public VoicemailArchiveProviderTest() {
	super(TestVoicemailArchiveProvider.class, VoicemailArchiveContract.AUTHORITY);
	}

	@Override
	protected void setUp() throws Exception {
	super.setUp();
	mResolver = getMockContentResolver();
	}

	@Override
	protected void tearDown() throws Exception {
	if (mCursor != null) {
	mCursor.close();
	}
	// Need to delete test.cache folder created by {@link android.test.ProviderTestCase2}
	deleteRecursive(getContext().getCacheDir());
	getProvider().closeDb();
	super.tearDown();
	}

	public void testInsert() {
	insertVoicemail();
	assertNotNull(mVoicemailUri);
	assertTrue(doesRowExist());
	long id = ContentUris.parseId(mVoicemailUri);
	assertTrue(id > 0);
	}

	public void testQuery_createFile() throws Exception {
	insertVoicemail();
	assertTrue(doesRowExist());
	assertFalse(doesFileExist());
	createFile();
	assertTrue(doesFileExist());
	}

	public void testQuery_createFileMimeTypeIsAMR() throws Exception {
	insertVoicemailWithMimeType("audio/amr");
	assertTrue(doesRowExist());
	assertFalse(doesFileExist());
	createFile();
	assertTrue(doesFileExist());
	assertEquals("amr", getFileExtension(getFilePath()));
	}

	public void testQuery_createFileMimeTypeIsMP3() throws Exception {
	insertVoicemailWithMimeType("audio/mpeg");
	assertTrue(doesRowExist());
	assertFalse(doesFileExist());
	createFile();
	assertTrue(doesFileExist());
	assertEquals("mp3", getFileExtension(getFilePath()));
	}

	public void testQuery_createFileMimeTypeNotExists() throws Exception {
	insertVoicemailWithMimeType(TEST_STRING);
	assertTrue(doesRowExist());
	assertFalse(doesFileExist());
	createFile();
	assertTrue(doesFileExist());
	assertEquals("", getFileExtension(getFilePath()));
	}

	public void testQuery() {
	insertVoicemail();
	updateCursor();
	assertCursorCount(1);
	assertContentValues();
	}

	public void testQuery_correctValuesSelection() {
	insertVoicemail();
	updateCursorWithIdQuery();
	assertCursorCount(1);
	assertContentValues();
	}

	public void testQuery_illegalUri() {
	try {
	mResolver.query(Uri.withAppendedPath(VoicemailArchive.CONTENT_URI, TEST_STRING),
	null, null, null, null);
	fail("Expecting exception but none was thrown.");
	} catch (IllegalArgumentException e) {}
	}

	public void testUpdate() throws Exception {
	insertVoicemail();
	assertTrue(doesRowExist());
	ContentValues values = new ContentValues();
	values.put(VoicemailArchive.MIME_TYPE, TEST_STRING);
	values.put(VoicemailArchive.NUMBER, TEST_STRING);
	values.put(VoicemailArchive.ARCHIVED, false);
	mResolver.update(mVoicemailUri, values, null, null);
	updateCursor();
	assertEquals(TEST_STRING, mCursor.getString(mCursor.getColumnIndex(VoicemailArchive.MIME_TYPE)));
	assertEquals(TEST_STRING, mCursor.getString(mCursor.getColumnIndex(VoicemailArchive.NUMBER)));
	assertEquals(false, mCursor.getInt(
	mCursor.getColumnIndex(VoicemailArchive.ARCHIVED)) == 1);
	}

	public void testUpdate_selection() throws Exception {
	insertVoicemail();
	assertTrue(doesRowExist());
	ContentValues values = new ContentValues();
	values.put(VoicemailArchive.MIME_TYPE, TEST_STRING);
	values.put(VoicemailArchive.NUMBER, TEST_STRING);
	values.put(VoicemailArchive.ARCHIVED, false);
	mResolver.update(VoicemailArchive.CONTENT_URI, values, getIdQuery(), null);
	updateCursor();
	assertEquals(TEST_STRING, mCursor.getString(mCursor.getColumnIndex(VoicemailArchive.MIME_TYPE)));
	assertEquals(TEST_STRING, mCursor.getString(mCursor.getColumnIndex(VoicemailArchive.NUMBER)));
	assertEquals(false, mCursor.getInt(
	mCursor.getColumnIndex(VoicemailArchive.ARCHIVED)) == 1);
	}

	public void testUpdate_illegalUri() {
	try {
	mResolver.update(Uri.withAppendedPath(VoicemailArchive.CONTENT_URI, TEST_STRING),
	null, null, null);
	fail("Expecting exception but none was thrown.");
	} catch (IllegalArgumentException e) {}
	}

	public void testDelete() throws Exception {
	insertVoicemail();
	createFile();
	assertTrue(doesRowExist());
	assertTrue(doesFileExist());
	mResolver.delete(mVoicemailUri, null, null);
	assertFalse(doesRowExist());
	assertFalse(doesFileExist());
	}

	public void testDelete_selection() throws Exception{
	insertVoicemail();
	createFile();
	assertTrue(doesRowExist());
	assertTrue(doesFileExist());
	mResolver.delete(VoicemailArchive.CONTENT_URI, getIdQuery(), null);
	assertFalse(doesRowExist());
	assertFalse(doesFileExist());
	}

	public void testDelete_illegalUri() {
	try {
	mResolver.delete(Uri.withAppendedPath(VoicemailArchive.CONTENT_URI, TEST_STRING),
	null, null);
	fail("Expecting exception but none was thrown.");
	} catch (IllegalArgumentException e) {}
	}

	private boolean doesRowExist() {
	Cursor tempCursor = mResolver.query(mVoicemailUri, null, null, null, null);
	boolean rowExists = tempCursor != null && tempCursor.getCount() > 0;
	tempCursor.close();
	return rowExists;
	}

	private boolean doesFileExist() {
	File voicemailFile = new File(getFilePath());
	return voicemailFile.exists();
	}

	private static String getFileExtension(String filePath) {
	File file = new File(filePath);
	String fileName = file.getName();
	int index = fileName.lastIndexOf(".");
	return index > 0 ? fileName.substring(index + 1) : "";
	}

	private void assertCursorCount(int count) {
	assertEquals(count, mCursor.getCount());
	}

	private void assertContentValues() {
	assertEquals(TEST_MIME_TYPE, mCursor
	.getString(mCursor.getColumnIndex(VoicemailArchive.MIME_TYPE)));
	assertEquals(TEST_NUMBER, mCursor
	.getString(mCursor.getColumnIndex(VoicemailArchive.NUMBER)));
	assertEquals(TEST_ARCHIVED, mCursor
	.getInt(mCursor.getColumnIndex(VoicemailArchive.ARCHIVED)));
	}

	private void insertVoicemail() {
	mVoicemailUri = mResolver.insert(VoicemailArchive.CONTENT_URI, getTestValues());
	}

	private void insertVoicemailWithMimeType(String mimeType) {
	ContentValues values = getTestValues();
	values.put(VoicemailArchive.MIME_TYPE, mimeType);
	mVoicemailUri = mResolver.insert(VoicemailArchive.CONTENT_URI, values);
	}

	private void updateCursor() {
	mCursor = mResolver.query(mVoicemailUri, null, null, null, null);
	assertEquals(true, mCursor.getCount() > 0);
	mCursor.moveToFirst();
	}

	private void updateCursorWithIdQuery() {
	mCursor = mResolver.query(mVoicemailUri, null, getIdQuery(), null, null);
	assertEquals(true, mCursor.getCount() > 0);
	mCursor.moveToFirst();
	}

	private void createFile() throws Exception {
	assertFalse(doesFileExist());
	// Opening output stream and closing it should create the file
	OutputStream outputStream = mResolver.openOutputStream(mVoicemailUri);
	outputStream.close();
	}

	private String getIdQuery() {
	return VoicemailArchive._ID + "=" + ContentUris.parseId(mVoicemailUri);
	}

	private String getFilePath() {
	if (mCursor == null) {
	updateCursor();
	}
	return mCursor.getString(mCursor.getColumnIndex(VoicemailArchive._DATA));
	}

	private static ContentValues getTestValues() {
	ContentValues values = new ContentValues();
	values.put(VoicemailArchive.NUMBER, TEST_NUMBER);
	values.put(VoicemailArchive.MIME_TYPE, TEST_MIME_TYPE);
	values.put(VoicemailArchive.ARCHIVED, TEST_ARCHIVED);
	return values;
	}

	private void deleteRecursive(File fileOrDirectory) {
	if (fileOrDirectory.isDirectory()) {
	for (File child : fileOrDirectory.listFiles()) {
	deleteRecursive(child);
	}
	}
	fileOrDirectory.delete();
	}

	public static class TestVoicemailArchiveProvider extends VoicemailArchiveProvider {
	private DialerDatabaseHelper mDialerDatabaseHelper;

	@Override
	protected File getFilesDir() {
	return getContext().getCacheDir();
	}

	@Override
	protected DialerDatabaseHelper getDatabaseHelper(Context context) {
	if (mDialerDatabaseHelper == null) {
	mDialerDatabaseHelper = DialerDatabaseHelper.getNewInstanceForTest(context);
	}
	return mDialerDatabaseHelper;
	}

	protected void closeDb() {
	mDialerDatabaseHelper.close();
	}
	}
	}