ContactProbeTest.java

/**
 * Funf: Open Sensing Framework
 * Copyright (C) 2010-2011 Nadav Aharony, Wei Pan, Alex Pentland. 
 * Acknowledgments: Alan Gardner
 * Contact: nadav@media.mit.edu
 * 
 * This file is part of Funf.
 * 
 * Funf is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as 
 * published by the Free Software Foundation, either version 3 of 
 * the License, or (at your option) any later version. 
 * 
 * Funf is distributed in the hope that it will be useful, but 
 * WITHOUT ANY WARRANTY; without even the implied warranty of 
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  
 * See the GNU Lesser General Public License for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public 
 * License along with Funf. If not, see <http://www.gnu.org/licenses/>.
 */
package edu.mit.media.funf.probe.builtin;

import java.util.ArrayList;

import junit.framework.AssertionFailedError;
import android.os.Bundle;
import android.os.Parcelable;
import android.provider.ContactsContract.Data;
import android.util.Log;
import edu.mit.media.funf.Utils;
import edu.mit.media.funf.probe.ProbeTestCase;
import edu.mit.media.funf.probe.Probe.Parameter;
import edu.mit.media.funf.probe.builtin.ProbeKeys.BaseProbeKeys;

public class ContactProbeTest extends ProbeTestCase<ContactProbe> {

	public ContactProbeTest() {
		super(ContactProbe.class);
	}

	public void testProbe() {
		Bundle params = new Bundle();
		params.putLong(Parameter.Builtin.PERIOD.name, 0L);
		params.putBoolean(ContactProbe.FULL_PARAM.getName(), true);
		startProbe(params);
		Bundle data = getData(30);
		assertNotNull(data.get(BaseProbeKeys.TIMESTAMP));
		ArrayList<Parcelable> contactData = data.getParcelableArrayList(ContactProbe.CONTACT_DATA);
		assertNotNull(contactData);
		int count = 1;
		while(data != null) {
			try {
				data = getData(5);
				count ++;
			} catch (AssertionFailedError e) {
				assertTrue(contactData.size() > 0);
				Log.i("ContactProbeTest", "Contact keys: " + Utils.join(Utils.getValues(data).keySet(), ", "));
				for (Parcelable dataRow : contactData) {
					Bundle b = (Bundle)dataRow;
					Log.i("ContactProbeTest", "Data keys: " + b.getString(Data.MIMETYPE) + ":" + String.valueOf(b.get(Data.DATA1)) + " - Others: " + Utils.join(Utils.getValues(b).keySet(), ", "));
				}
				data = null;
			}
		}
		Log.i("ContactProbeTest", "Count: " + String.valueOf(count));
	}
	
	public void testFullParameter() {
		// Run a full scan
		Bundle params = new Bundle();
		params.putLong(Parameter.Builtin.PERIOD.name, 0L);
		params.putBoolean(ContactProbe.FULL_PARAM.getName(), true);
		startProbe(params);
		Bundle data = getData(30);
		boolean hasAtLeastTwo = false;
		while (data != null) {
			// Get the rest of the data
			try {
				data = getData(5);
				hasAtLeastTwo = true;
			} catch (AssertionFailedError e) {
				data = null;
			}
		}
		assertTrue(hasAtLeastTwo);
		
		// Run a non-full scan
		params.putBoolean(ContactProbe.FULL_PARAM.getName(), false);
		startProbe(params);
		try {
			data = getData(30);  // Unfortunately we have to wait, since full scans take a while to return anything
			ArrayList<Parcelable> contactData = data.getParcelableArrayList(ContactProbe.CONTACT_DATA);
			Log.i(TAG, "Data: " + (contactData == null ? "<NONE>" : contactData.size()));
			fail("Should not get any contacts for a non-full scan after a full scan");
		} catch (AssertionFailedError e) {
			// Success
		}
	}
	
}