fix broken DownloadManager tests

one big change in this CL is addition of a new feature to MockWebServer.
It can now play a long response to the Downloading thread to keep it busy
while something - such as cancel/remove - can be done to that Download Request.

Also, added changes to FakeSystemFacade to start threads in normal fashion
instead of queuing them up and later running just their run() methods.

the following tests should work now
	packages/providers/DownloadProvider/tests/src/com/android/providers/downloads/
	  DownloadManagerFunctionalTest.java
	  PublicApiFunctionalTest.java
	  ThreadingTest.java
	  PublicApiAccessTest.java
	  DownloadProviderPermissionsTest.java

the following are flaky. they need to be split up into smaller tests.
	frameworks/base/core/tests/coretests/src/android/app/
	  DownloadManagerIntegrationTest.java
	  DownloadManagerStressTest.java

Change-Id: Ia0b11963f92e8f8365f701761dcbce467be3ee9b

1 files changed, 75 insertions, 0 deletions

diff --git a/tests/src/com/android/providers/downloads/AbstractPublicApiTest.java b/tests/src/com/android/providers/downloads/AbstractPublicApiTest.java
index ed443b01..c38c2f1d 100644
--- a/tests/src/com/android/providers/downloads/AbstractPublicApiTest.java
+++ b/tests/src/com/android/providers/downloads/AbstractPublicApiTest.java
@@ -20,6 +20,8 @@ import android.app.DownloadManager;
 import android.database.Cursor;
 import android.net.Uri;
 import android.os.ParcelFileDescriptor;
+import android.provider.Downloads;
+import android.util.Log;
 
 import java.io.FileInputStream;
 import java.io.InputStream;
@@ -41,6 +43,22 @@ public abstract class AbstractPublicApiTest extends AbstractDownloadManagerFunct
             return (int) getLongField(DownloadManager.COLUMN_STATUS);
         }
 
+        public int getStatusIfExists() {
+            Cursor cursor = mManager.query(new DownloadManager.Query().setFilterById(mId));
+            try {
+                if (cursor.getCount() > 0) {
+                    cursor.moveToFirst();
+                    return (int) cursor.getLong(cursor.getColumnIndexOrThrow(
+                            DownloadManager.COLUMN_STATUS));
+                } else {
+                    // the row doesn't exist
+                    return -1;
+                }
+            } finally {
+                cursor.close();
+            }
+        }
+
         String getStringField(String field) {
             Cursor cursor = mManager.query(new DownloadManager.Query().setFilterById(mId));
             try {
@@ -79,6 +97,63 @@ public abstract class AbstractPublicApiTest extends AbstractDownloadManagerFunct
             runService();
             assertEquals(status, getStatus());
         }
+
+        // max time to wait before giving up on the current download operation.
+        private static final int MAX_TIME_TO_WAIT_FOR_OPERATION = 5;
+        // while waiting for the above time period, sleep this long to yield to the
+        // download thread
+        private static final int TIME_TO_SLEEP = 1000;
+
+        int runUntilDone() throws InterruptedException {
+            int sleepCounter = MAX_TIME_TO_WAIT_FOR_OPERATION * 1000 / TIME_TO_SLEEP;
+            for (int i = 0; i < sleepCounter; i++) {
+                int status = getStatusIfExists();
+                if (status == -1 || Downloads.Impl.isStatusCompleted(getStatus())) {
+                    // row doesn't exist or the download is done
+                    return status;
+                }
+                // download not done yet. sleep a while and try again
+                Thread.sleep(TIME_TO_SLEEP);
+            }
+            return 0; // failed
+        }
+
+        // waits until progress_so_far is >= (progress)%
+        boolean runUntilProgress(int progress) throws InterruptedException {
+            int sleepCounter = MAX_TIME_TO_WAIT_FOR_OPERATION * 1000 / TIME_TO_SLEEP;
+            int numBytesReceivedSoFar = 0;
+            int totalBytes = 0;
+            for (int i = 0; i < sleepCounter; i++) {
+                Cursor cursor = mManager.query(new DownloadManager.Query().setFilterById(mId));
+                try {
+                    assertEquals(1, cursor.getCount());
+                    cursor.moveToFirst();
+                    numBytesReceivedSoFar = cursor.getInt(
+                            cursor.getColumnIndexOrThrow(
+                                    DownloadManager.COLUMN_BYTES_DOWNLOADED_SO_FAR));
+                    totalBytes = cursor.getInt(
+                            cursor.getColumnIndexOrThrow(DownloadManager.COLUMN_TOTAL_SIZE_BYTES));
+                } finally {
+                    cursor.close();
+                }
+                Log.i(LOG_TAG, "in runUntilProgress, numBytesReceivedSoFar: " +
+                        numBytesReceivedSoFar + ", totalBytes: " + totalBytes);
+                if (totalBytes == 0) {
+                    fail("total_bytes should not be zero");
+                    return false;
+                } else {
+                    if (numBytesReceivedSoFar * 100 / totalBytes >= progress) {
+                        // progress_so_far is >= progress%. we are done
+                        return true;
+                    }
+                }
+                // download not done yet. sleep a while and try again
+                Thread.sleep(TIME_TO_SLEEP);
+            }
+            Log.i(LOG_TAG, "FAILED in runUntilProgress, numBytesReceivedSoFar: " +
+                    numBytesReceivedSoFar + ", totalBytes: " + totalBytes);
+            return false; // failed
+        }
     }
 
     protected static final String PACKAGE_NAME = "my.package.name";