1 files changed, 22 insertions, 1 deletions
diff --git a/tests/src/com/android/providers/downloads/AbstractDownloadManagerFunctionalTest.java b/tests/src/com/android/providers/downloads/AbstractDownloadManagerFunctionalTest.java
index a06597f1..5283d425 100644
--- a/tests/src/com/android/providers/downloads/AbstractDownloadManagerFunctionalTest.java
+++ b/tests/src/com/android/providers/downloads/AbstractDownloadManagerFunctionalTest.java
@@ -59,6 +59,14 @@ public abstract class AbstractDownloadManagerFunctionalTest extends
     protected MockContentResolverWithNotify mResolver;
     protected TestContext mTestContext;
     protected FakeSystemFacade mSystemFacade;
+    protected static String STRING_1K;
+    static {
+        StringBuilder buff = new StringBuilder();
+        for (int i = 0; i < 1024; i++) {
+            buff.append("a" + i % 26);
+        }
+        STRING_1K = buff.toString();
+    }
 
     static class MockContentResolverWithNotify extends MockContentResolver {
         public boolean mNotifyWasCalled = false;
@@ -161,6 +169,7 @@ public abstract class AbstractDownloadManagerFunctionalTest extends
     @Override
     protected void tearDown() throws Exception {
         cleanUpDownloads();
+        mServer.shutdown();
         super.tearDown();
     }
 
@@ -205,7 +214,7 @@ public abstract class AbstractDownloadManagerFunctionalTest extends
     }
 
     /**
-     * Enqueue a response from the MockWebServer.
+     * Enqueue a String response from the MockWebServer.
      */
     MockResponse enqueueResponse(int status, String body) {
         MockResponse response = new MockResponse()
@@ -216,6 +225,18 @@ public abstract class AbstractDownloadManagerFunctionalTest extends
         mServer.enqueue(response);
         return response;
     }
+    /**
+     * Enqueue a byte[] response from the MockWebServer.
+     */
+    MockResponse enqueueResponse(int status, byte[] body) {
+        MockResponse response = new MockResponse()
+                                .setResponseCode(status)
+                                .setBody(body)
+                                .addHeader("Content-type", "text/plain")
+                                .setCloseConnectionAfter(true);
+        mServer.enqueue(response);
+        return response;
+    }
 
     MockResponse enqueueEmptyResponse(int status) {
         return enqueueResponse(status, "");