Migrate to shared MockWebServer.

Bug: 4726601
Change-Id: Ibe537bd5c2a092dbf974360cd454751881f7f4ea

1 files changed, 167 insertions, 0 deletions

diff --git a/tests/src/com/android/providers/downloads/DownloadProviderFunctionalTest.java b/tests/src/com/android/providers/downloads/DownloadProviderFunctionalTest.java
new file mode 100644
index 00000000..bbc5c3e0
--- /dev/null
+++ b/tests/src/com/android/providers/downloads/DownloadProviderFunctionalTest.java
@@ -0,0 +1,167 @@
+/*
+ * 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 com.android.providers.downloads;
+
+import android.content.ContentValues;
+import android.database.Cursor;
+import android.net.ConnectivityManager;
+import android.net.Uri;
+import android.os.Environment;
+import android.provider.Downloads;
+import android.test.suitebuilder.annotation.LargeTest;
+import android.util.Log;
+
+import com.google.mockwebserver.MockWebServer;
+import com.google.mockwebserver.RecordedRequest;
+
+import java.io.InputStream;
+import java.net.MalformedURLException;
+import java.net.UnknownHostException;
+
+/**
+ * This test exercises the entire download manager working together -- it requests downloads through
+ * the {@link DownloadProvider}, just like a normal client would, and runs the
+ * {@link DownloadService} with start intents.  It sets up a {@link MockWebServer} running on the
+ * device to serve downloads.
+ */
+@LargeTest
+public class DownloadProviderFunctionalTest extends AbstractDownloadProviderFunctionalTest {
+    private static final String TAG = "DownloadManagerFunctionalTest";
+
+    public DownloadProviderFunctionalTest() {
+        super(new FakeSystemFacade());
+    }
+
+    public void testDownloadTextFile() throws Exception {
+        enqueueResponse(buildResponse(HTTP_OK, FILE_CONTENT));
+
+        String path = "/download_manager_test_path";
+        Uri downloadUri = requestDownload(path);
+        assertEquals(Downloads.Impl.STATUS_PENDING, getDownloadStatus(downloadUri));
+        assertTrue(mTestContext.mHasServiceBeenStarted);
+
+        runUntilStatus(downloadUri, Downloads.Impl.STATUS_SUCCESS);
+        RecordedRequest request = takeRequest();
+        assertEquals("GET", request.getMethod());
+        assertEquals(path, request.getPath());
+        assertEquals(FILE_CONTENT, getDownloadContents(downloadUri));
+        assertStartsWith(Environment.getExternalStorageDirectory().getPath(),
+                         getDownloadFilename(downloadUri));
+    }
+
+    public void testDownloadToCache() throws Exception {
+        enqueueResponse(buildResponse(HTTP_OK, FILE_CONTENT));
+
+        Uri downloadUri = requestDownload("/path");
+        updateDownload(downloadUri, Downloads.Impl.COLUMN_DESTINATION,
+                       Integer.toString(Downloads.Impl.DESTINATION_CACHE_PARTITION));
+        runUntilStatus(downloadUri, Downloads.Impl.STATUS_SUCCESS);
+        assertEquals(FILE_CONTENT, getDownloadContents(downloadUri));
+        assertStartsWith(getContext().getCacheDir().getAbsolutePath(),
+                         getDownloadFilename(downloadUri));
+    }
+
+    public void testRoaming() throws Exception {
+        enqueueResponse(buildResponse(HTTP_OK, FILE_CONTENT));
+        enqueueResponse(buildResponse(HTTP_OK, FILE_CONTENT));
+
+        mSystemFacade.mActiveNetworkType = ConnectivityManager.TYPE_MOBILE;
+        mSystemFacade.mIsRoaming = true;
+
+        // for a normal download, roaming is fine
+        Uri downloadUri = requestDownload("/path");
+        runUntilStatus(downloadUri, Downloads.Impl.STATUS_SUCCESS);
+
+        // when roaming is disallowed, the download should pause...
+        downloadUri = requestDownload("/path");
+        updateDownload(downloadUri, Downloads.Impl.COLUMN_DESTINATION,
+                       Integer.toString(Downloads.Impl.DESTINATION_CACHE_PARTITION_NOROAMING));
+        runUntilStatus(downloadUri, Downloads.Impl.STATUS_WAITING_FOR_NETWORK);
+
+        // ...and pick up when we're off roaming
+        mSystemFacade.mIsRoaming = false;
+        runUntilStatus(downloadUri, Downloads.Impl.STATUS_SUCCESS);
+    }
+
+    /**
+     * Read a downloaded file from disk.
+     */
+    private String getDownloadContents(Uri downloadUri) throws Exception {
+        InputStream inputStream = mResolver.openInputStream(downloadUri);
+        try {
+            return readStream(inputStream);
+        } finally {
+            inputStream.close();
+        }
+    }
+
+    private void runUntilStatus(Uri downloadUri, int status) throws Exception {
+        runService();
+        boolean done = false;
+        while (!done) {
+            int rslt = getDownloadStatus(downloadUri);
+            if (rslt == Downloads.Impl.STATUS_RUNNING || rslt == Downloads.Impl.STATUS_PENDING) {
+                Log.i(TAG, "status is: " + rslt + ", for: " + downloadUri);
+                DownloadHandler.getInstance().WaitUntilDownloadsTerminate();
+                Thread.sleep(100);
+            } else {
+                done = true;
+            }
+        }
+        assertEquals(status, getDownloadStatus(downloadUri));
+    }
+
+    protected int getDownloadStatus(Uri downloadUri) {
+        return Integer.valueOf(getDownloadField(downloadUri, Downloads.Impl.COLUMN_STATUS));
+    }
+
+    private String getDownloadFilename(Uri downloadUri) {
+        return getDownloadField(downloadUri, Downloads.Impl._DATA);
+    }
+
+    private String getDownloadField(Uri downloadUri, String column) {
+        final String[] columns = new String[] {column};
+        Cursor cursor = mResolver.query(downloadUri, columns, null, null, null);
+        try {
+            assertEquals(1, cursor.getCount());
+            cursor.moveToFirst();
+            return cursor.getString(0);
+        } finally {
+            cursor.close();
+        }
+    }
+
+    /**
+     * Request a download from the Download Manager.
+     */
+    private Uri requestDownload(String path) throws MalformedURLException, UnknownHostException {
+        ContentValues values = new ContentValues();
+        values.put(Downloads.Impl.COLUMN_URI, getServerUri(path));
+        values.put(Downloads.Impl.COLUMN_DESTINATION, Downloads.Impl.DESTINATION_EXTERNAL);
+        return mResolver.insert(Downloads.Impl.CONTENT_URI, values);
+    }
+
+    /**
+     * Update one field of a download in the provider.
+     */
+    private void updateDownload(Uri downloadUri, String column, String value) {
+        ContentValues values = new ContentValues();
+        values.put(column, value);
+        int numChanged = mResolver.update(downloadUri, values, null, null);
+        assertEquals(1, numChanged);
+    }
+}