tests/src/com/android/exchange/provider/MockProviderTests.java

/*
 * Copyright (C) 2010 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.exchange.provider;

import android.content.ContentProviderOperation;
import android.content.ContentProviderResult;
import android.content.ContentUris;
import android.content.ContentValues;
import android.content.Context;
import android.content.OperationApplicationException;
import android.database.Cursor;
import android.net.Uri;
import android.os.RemoteException;
import android.test.ProviderTestCase2;
import android.test.mock.MockContentResolver;
import android.test.suitebuilder.annotation.SmallTest;

import java.util.ArrayList;

/**
 * You can run this entire test case with:
 *   runtest -c com.android.exchange.provider.MockProviderTests exchange
 */
@SmallTest
public class MockProviderTests extends ProviderTestCase2<MockProvider> {
    Context mMockContext;
    MockContentResolver mMockResolver;

    private static final String CANHAZ_AUTHORITY = "com.android.canhaz";
    private static final String PONY_TABLE = "pony";
    private static final String PONY_COLUMN_NAME = "name";
    private static final String PONY_COLUMN_TYPE = "type";
    private static final String PONY_COLUMN_LEGS= "legs";
    private static final String PONY_COLUMN_CAN_RIDE = "canRide";
    private static final String[] PONY_PROJECTION = {MockProvider.ID_COLUMN, PONY_COLUMN_NAME,
        PONY_COLUMN_TYPE, PONY_COLUMN_LEGS, PONY_COLUMN_CAN_RIDE};
    private static final int PONY_ID = 0;
    private static final int PONY_NAME = 1;
    private static final int PONY_TYPE = 2;
    private static final int PONY_LEGS = 3;
    private static final int PONY_CAN_RIDE = 4;

    public MockProviderTests() {
        super(MockProvider.class, MockProvider.AUTHORITY);
    }

    @Override
    public void setUp() throws Exception {
        super.setUp();
        mMockContext = getMockContext();
        mMockResolver = (MockContentResolver)mMockContext.getContentResolver();
        mMockResolver.addProvider(CANHAZ_AUTHORITY, new MockProvider(mMockContext));
    }

    @Override
    public void tearDown() throws Exception {
        super.tearDown();
    }

    private ContentValues ponyValues(String name, String type, int legs, boolean canRide) {
        ContentValues cv = new ContentValues();
        cv.put(PONY_COLUMN_NAME, name);
        cv.put(PONY_COLUMN_TYPE, type);
        cv.put(PONY_COLUMN_LEGS, legs);
        cv.put(PONY_COLUMN_CAN_RIDE, canRide ? 1 : 0);
        return cv;
    }

    private ContentProviderResult[] setupPonies() throws RemoteException,
            OperationApplicationException {
        // The Uri is content://com.android.canhaz/pony
        Uri uri = new Uri.Builder().scheme("content").authority(CANHAZ_AUTHORITY)
            .path(PONY_TABLE).build();
        // Our array of CPO's to be used with applyBatch
        ArrayList<ContentProviderOperation> ops = new ArrayList<ContentProviderOperation>();

        // Insert two ponies
        ContentValues pony1 = ponyValues("Flicka", "wayward", 4, true);
        ops.add(ContentProviderOperation.newInsert(uri).withValues(pony1).build());
        ContentValues pony2 = ponyValues("Elise", "dastardly", 3, false);
        ops.add(ContentProviderOperation.newInsert(uri).withValues(pony2).build());
        // Apply the batch with one insert operation
        return mMockResolver.applyBatch(MockProvider.AUTHORITY, ops);
    }

    private Uri getPonyUri() {
        return new Uri.Builder().scheme("content").authority(CANHAZ_AUTHORITY)
            .path(PONY_TABLE).build();
    }

    public void testInsertQueryandDelete() throws RemoteException, OperationApplicationException {
        // The Uri is content://com.android.canhaz/pony
        ContentProviderResult[] results = setupPonies();
        Uri uri = getPonyUri();

        // Check the results
        assertNotNull(results);
        assertEquals(2, results.length);
        // Make sure that we've created matcher entries for pony and pony/#
        assertEquals(MockProvider.TABLE, MockProvider.sURIMatcher.match(MockProvider.uri(uri)));
        assertEquals(MockProvider.RECORD,
                MockProvider.sURIMatcher.match(MockProvider.uri(results[0].uri)));
        Cursor c = mMockResolver.query(MockProvider.uri(uri), PONY_PROJECTION, null, null, null);
        assertNotNull(c);
        assertEquals(2, c.getCount());
        long eliseId = -1;
        long flickaId = -1;
        while (c.moveToNext()) {
            String name = c.getString(PONY_NAME);
            if ("Flicka".equals(name)) {
                assertEquals("Flicka", c.getString(PONY_NAME));
                assertEquals("wayward", c.getString(PONY_TYPE));
                assertEquals(4, c.getInt(PONY_LEGS));
                assertEquals(1, c.getInt(PONY_CAN_RIDE));
                flickaId = c.getLong(PONY_ID);
            } else if ("Elise".equals(name)) {
                assertEquals("dastardly", c.getString(PONY_TYPE));
                assertEquals(3, c.getInt(PONY_LEGS));
                assertEquals(0, c.getInt(PONY_CAN_RIDE));
                eliseId = c.getLong(PONY_ID);
            } else {
                fail("Wrong record: " + name);
            }
        }

        // eliseId and flickaId should have been set
        assertNotSame(-1, eliseId);
        assertNotSame(-1, flickaId);
        // Delete the elise record
        assertEquals(1, mMockResolver.delete(ContentUris.withAppendedId(MockProvider.uri(uri),
                eliseId), null, null));
        c = mMockResolver.query(MockProvider.uri(uri), PONY_PROJECTION, null, null, null);
        assertNotNull(c);
        // There should be one left (Flicka)
        assertEquals(1, c.getCount());
        assertTrue(c.moveToNext());
        assertEquals("Flicka", c.getString(PONY_NAME));
    }

    public void testUpdate() throws RemoteException, OperationApplicationException {
        // The Uri is content://com.android.canhaz/pony
        Uri uri = getPonyUri();
        setupPonies();
        Cursor c = mMockResolver.query(MockProvider.uri(uri), PONY_PROJECTION, null, null, null);
        assertNotNull(c);
        assertEquals(2, c.getCount());
        // Give all the ponies 5 legs
        ContentValues cv = new ContentValues();
        cv.put(PONY_COLUMN_LEGS, 5);
        assertEquals(2, mMockResolver.update(MockProvider.uri(uri), cv, null, null));
        c = mMockResolver.query(MockProvider.uri(uri), PONY_PROJECTION, null, null, null);
        assertNotNull(c);
        // We should still have two records, and each should have 5 legs, but otherwise be the same
        assertEquals(2, c.getCount());
        long eliseId = -1;
        long flickaId = -1;
        while (c.moveToNext()) {
            String name = c.getString(PONY_NAME);
            if ("Flicka".equals(name)) {
                assertEquals("Flicka", c.getString(PONY_NAME));
                assertEquals("wayward", c.getString(PONY_TYPE));
                assertEquals(5, c.getInt(PONY_LEGS));
                assertEquals(1, c.getInt(PONY_CAN_RIDE));
                flickaId = c.getLong(PONY_ID);
            } else if ("Elise".equals(name)) {
                assertEquals("dastardly", c.getString(PONY_TYPE));
                assertEquals(5, c.getInt(PONY_LEGS));
                assertEquals(0, c.getInt(PONY_CAN_RIDE));
                eliseId = c.getLong(PONY_ID);
            } else {
                fail("Wrong record: " + name);
            }
        }
        // eliseId and flickaId should have been set
        assertNotSame(-1, eliseId);
        assertNotSame(-1, flickaId);
    }
}