| /* |
| * 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 android.provider.cts; |
| |
| import android.content.ContentProviderClient; |
| import android.content.ContentResolver; |
| import android.content.Context; |
| import android.os.cts.FileUtils; |
| import android.provider.ContactsContract; |
| import android.provider.ContactsContract.CommonDataKinds.Photo; |
| import android.provider.ContactsContract.Contacts; |
| import android.provider.cts.ContactsContract_TestDataBuilder.TestContact; |
| import android.provider.cts.ContactsContract_TestDataBuilder.TestData; |
| import android.provider.cts.ContactsContract_TestDataBuilder.TestRawContact; |
| import android.test.InstrumentationTestCase; |
| |
| import java.io.IOException; |
| import java.io.InputStream; |
| |
| public class ContactsContract_PhotoTest extends InstrumentationTestCase { |
| private ContactsContract_TestDataBuilder mBuilder; |
| |
| private static final byte[] EMPTY_TEST_PHOTO_DATA = "".getBytes(); |
| |
| private Context mContext; |
| |
| private ContentResolver mResolver; |
| |
| @Override |
| protected void setUp() throws Exception { |
| super.setUp(); |
| |
| mContext= getInstrumentation().getTargetContext(); |
| mResolver = mContext.getContentResolver(); |
| ContentProviderClient provider = |
| mResolver.acquireContentProviderClient(ContactsContract.AUTHORITY); |
| mBuilder = new ContactsContract_TestDataBuilder(provider); |
| } |
| |
| @Override |
| protected void tearDown() throws Exception { |
| super.tearDown(); |
| mBuilder.cleanup(); |
| } |
| |
| public void testAddPhoto() throws Exception { |
| TestRawContact rawContact = mBuilder.newRawContact().insert().load(); |
| TestContact contact = rawContact.getContact().load(); |
| |
| assertNull(Contacts.openContactPhotoInputStream(mResolver, contact.getUri())); |
| assertNull(Contacts.openContactPhotoInputStream(mResolver, contact.getUri(), true)); |
| assertNull(Contacts.openContactPhotoInputStream(mResolver, contact.getUri(), false)); |
| |
| TestData photoData = rawContact.newDataRow(Photo.CONTENT_ITEM_TYPE) |
| .with(Photo.PHOTO, PhotoUtil.getTestPhotoData(mContext)) |
| .insert(); |
| |
| photoData.load(); |
| photoData.assertColumn(Photo.RAW_CONTACT_ID, rawContact.getId()); |
| photoData.assertBlobColumnNotNull(Photo.PHOTO); |
| |
| assertPhotoStream(Contacts.openContactPhotoInputStream(mResolver, contact.getUri())); |
| assertPhotoStream(Contacts.openContactPhotoInputStream(mResolver, contact.getUri(), true)); |
| assertPhotoStream(Contacts.openContactPhotoInputStream(mResolver, contact.getUri(), false)); |
| } |
| |
| public void testAddEmptyPhoto() throws Exception { |
| TestRawContact rawContact = mBuilder.newRawContact().insert().load(); |
| TestContact contact = rawContact.getContact().load(); |
| |
| assertNull(Contacts.openContactPhotoInputStream(mResolver, contact.getUri())); |
| assertNull(Contacts.openContactPhotoInputStream(mResolver, contact.getUri(), true)); |
| assertNull(Contacts.openContactPhotoInputStream(mResolver, contact.getUri(), false)); |
| |
| TestData photoData = rawContact.newDataRow(Photo.CONTENT_ITEM_TYPE) |
| .with(Photo.PHOTO, EMPTY_TEST_PHOTO_DATA) |
| .insert(); |
| assertNotNull(photoData.load()); |
| |
| assertNull(Contacts.openContactPhotoInputStream(mResolver, contact.getUri())); |
| assertNull(Contacts.openContactPhotoInputStream(mResolver, contact.getUri(), true)); |
| assertNull(Contacts.openContactPhotoInputStream(mResolver, contact.getUri(), false)); |
| } |
| |
| private void assertPhotoStream(InputStream photoStream) throws IOException { |
| try { |
| assertNotNull(photoStream); |
| byte[] actualBytes = FileUtils.readInputStreamFully(photoStream); |
| assertTrue(actualBytes.length > 0); |
| } finally { |
| if (photoStream != null) { |
| photoStream.close(); |
| } |
| } |
| } |
| } |
| |