Home | History | Annotate | Download | only in services 
      1 /*
      2  * Copyright (C) 2016 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 com.android.documentsui.services;
     18 
     19 import android.content.Context;
     20 import android.database.ContentObserver;
     21 import android.net.Uri;
     22 import android.provider.DocumentsContract;
     23 import android.test.mock.MockContentResolver;
     24 import android.util.Log;
     25 
     26 import java.util.concurrent.CountDownLatch;
     27 import java.util.concurrent.TimeUnit;
     28 import java.util.concurrent.TimeoutException;
     29 
     30 /**
     31  * A test resolver that enables this test suite to listen for notifications that mark when copy
     32  * operations are done.
     33  */
     34 class TestContentResolver extends MockContentResolver {
     35 
     36     private static final String TAG = "TestContextResolver";
     37 
     38     private CountDownLatch mReadySignal;
     39     private CountDownLatch mNotificationSignal;
     40     private Context mContext;
     41 
     42     public TestContentResolver(Context context) {
     43         mContext = context;
     44         mReadySignal = new CountDownLatch(1);
     45     }
     46 
     47     /**
     48      * Wait for the given number of files to be copied to destination. Times out after 1 sec.
     49      */
     50     public void waitForChanges(int count) throws Exception {
     51         // Wait for no more than 1 second by default.
     52         waitForChanges(count, 1000);
     53     }
     54 
     55     /**
     56      * Wait for files to be copied to destination.
     57      *
     58      * @param count Number of files to wait for.
     59      * @param timeOut Timeout in ms. TimeoutException will be thrown if this function times out.
     60      */
     61     public void waitForChanges(int count, int timeOut) throws Exception {
     62         mNotificationSignal = new CountDownLatch(count);
     63         // Signal that the test is now waiting for files.
     64         mReadySignal.countDown();
     65         if (!mNotificationSignal.await(timeOut, TimeUnit.MILLISECONDS)) {
     66             throw new TimeoutException("Timed out waiting for file operations to complete.");
     67         }
     68     }
     69 
     70     @Override
     71     public void notifyChange(Uri uri, ContentObserver observer, boolean syncToNetwork) {
     72         // Wait until the test is ready to receive file notifications.
     73         try {
     74             mReadySignal.await();
     75         } catch (InterruptedException e) {
     76             Log.d(TAG, "Interrupted while waiting for file copy readiness");
     77             Thread.currentThread().interrupt();
     78         }
     79         if (DocumentsContract.isDocumentUri(mContext, uri)) {
     80             Log.d(TAG, "Notification: " + uri);
     81             // Watch for document URI change notifications - this signifies the end of a copy.
     82             mNotificationSignal.countDown();
     83         }
     84     }
     85 }