Home | History | Annotate | Download | only in cts 
      1 /*
      2  * Copyright (C) 2010 The Android Open Source Project
      3  *
      4  * Licensed under the Apache License, Version 2.0 (the "License");
      5  * you may not use this file except in compliance with the License.
      6  * You may obtain a copy of the License at
      7  *
      8  *      http://www.apache.org/licenses/LICENSE-2.0
      9  *
     10  * Unless required by applicable law or agreed to in writing, software
     11  * distributed under the License is distributed on an "AS IS" BASIS,
     12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     13  * See the License for the specific language governing permissions and
     14  * limitations under the License.
     15  */
     16 
     17 package android.provider.cts;
     18 
     19 import android.content.ContentProviderClient;
     20 import android.content.ContentResolver;
     21 import android.content.Context;
     22 import android.cts.util.FileUtils;
     23 import android.provider.ContactsContract;
     24 import android.provider.ContactsContract.CommonDataKinds.Photo;
     25 import android.provider.ContactsContract.Contacts;
     26 import android.provider.cts.ContactsContract_TestDataBuilder.TestContact;
     27 import android.provider.cts.ContactsContract_TestDataBuilder.TestData;
     28 import android.provider.cts.ContactsContract_TestDataBuilder.TestRawContact;
     29 import android.test.InstrumentationTestCase;
     30 
     31 import java.io.IOException;
     32 import java.io.InputStream;
     33 
     34 public class ContactsContract_PhotoTest extends InstrumentationTestCase {
     35     private ContactsContract_TestDataBuilder mBuilder;
     36 
     37     private static final byte[] EMPTY_TEST_PHOTO_DATA = "".getBytes();
     38 
     39     private Context mContext;
     40 
     41     private ContentResolver mResolver;
     42 
     43     @Override
     44     protected void setUp() throws Exception {
     45         super.setUp();
     46 
     47         mContext= getInstrumentation().getTargetContext();
     48         mResolver = mContext.getContentResolver();
     49         ContentProviderClient provider =
     50                 mResolver.acquireContentProviderClient(ContactsContract.AUTHORITY);
     51         mBuilder = new ContactsContract_TestDataBuilder(provider);
     52     }
     53 
     54     @Override
     55     protected void tearDown() throws Exception {
     56         super.tearDown();
     57         mBuilder.cleanup();
     58     }
     59 
     60     public void testAddPhoto() throws Exception {
     61         TestRawContact rawContact = mBuilder.newRawContact().insert().load();
     62         TestContact contact = rawContact.getContact().load();
     63 
     64         assertNull(Contacts.openContactPhotoInputStream(mResolver, contact.getUri()));
     65         assertNull(Contacts.openContactPhotoInputStream(mResolver, contact.getUri(), true));
     66         assertNull(Contacts.openContactPhotoInputStream(mResolver, contact.getUri(), false));
     67 
     68         TestData photoData = rawContact.newDataRow(Photo.CONTENT_ITEM_TYPE)
     69                 .with(Photo.PHOTO, PhotoUtil.getTestPhotoData(mContext))
     70                 .insert();
     71 
     72         photoData.load();
     73         photoData.assertColumn(Photo.RAW_CONTACT_ID, rawContact.getId());
     74         photoData.assertBlobColumnNotNull(Photo.PHOTO);
     75 
     76         assertPhotoStream(Contacts.openContactPhotoInputStream(mResolver, contact.getUri()));
     77         assertPhotoStream(Contacts.openContactPhotoInputStream(mResolver, contact.getUri(), true));
     78         assertPhotoStream(Contacts.openContactPhotoInputStream(mResolver, contact.getUri(), false));
     79     }
     80 
     81     public void testAddEmptyPhoto() throws Exception {
     82         TestRawContact rawContact = mBuilder.newRawContact().insert().load();
     83         TestContact contact = rawContact.getContact().load();
     84 
     85         assertNull(Contacts.openContactPhotoInputStream(mResolver, contact.getUri()));
     86         assertNull(Contacts.openContactPhotoInputStream(mResolver, contact.getUri(), true));
     87         assertNull(Contacts.openContactPhotoInputStream(mResolver, contact.getUri(), false));
     88 
     89         TestData photoData = rawContact.newDataRow(Photo.CONTENT_ITEM_TYPE)
     90                 .with(Photo.PHOTO, EMPTY_TEST_PHOTO_DATA)
     91                 .insert();
     92         assertNotNull(photoData.load());
     93 
     94         assertNull(Contacts.openContactPhotoInputStream(mResolver, contact.getUri()));
     95         assertNull(Contacts.openContactPhotoInputStream(mResolver, contact.getUri(), true));
     96         assertNull(Contacts.openContactPhotoInputStream(mResolver, contact.getUri(), false));
     97     }
     98 
     99     private void assertPhotoStream(InputStream photoStream) throws IOException {
    100         try {
    101             assertNotNull(photoStream);
    102             byte[] actualBytes = FileUtils.readInputStreamFully(photoStream);
    103             assertTrue(actualBytes.length > 0);
    104         } finally {
    105             if (photoStream != null) {
    106                 photoStream.close();
    107             }
    108         }
    109     }
    110 }
    111 
    112