Do screenshot capture async for share intents.

chrome/android/javatests/src/org/chromium/chrome/browser/util/ChromeFileProviderTest.java

Index: chrome/android/javatests/src/org/chromium/chrome/browser/util/ChromeFileProviderTest.java
diff --git a/chrome/android/javatests/src/org/chromium/chrome/browser/util/ChromeFileProviderTest.java b/chrome/android/javatests/src/org/chromium/chrome/browser/util/ChromeFileProviderTest.java
new file mode 100644
index 0000000000000000000000000000000000000000..6c7d9a010dca3198e45a521b14d2e8b9a9031b2c
--- /dev/null
+++ b/chrome/android/javatests/src/org/chromium/chrome/browser/util/ChromeFileProviderTest.java
@@ -0,0 +1,97 @@
+// Copyright 2016 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+package org.chromium.chrome.browser.util;
+
+import android.content.Context;
+import android.net.Uri;
+import android.os.AsyncTask;
+import android.os.ParcelFileDescriptor;
+import android.test.suitebuilder.annotation.LargeTest;
+import android.test.suitebuilder.annotation.SmallTest;
+
+import org.chromium.content.browser.test.NativeLibraryTestBase;
+
+import java.io.FileNotFoundException;
+
+/**
+ * Tests working of ChromeFileProvider.
+ *
+ * The openFile should be blocked till notify is called. These tests can timeout if the notify does
+ * not work correctly.
+ */
+public class ChromeFileProviderTest extends NativeLibraryTestBase {
+    private ParcelFileDescriptor openFileFromProvider(Uri uri) {
+        ChromeFileProvider provider = new ChromeFileProvider();
+        ParcelFileDescriptor file = null;
+        try {
+            provider.openFile(uri, "r");
+        } catch (FileNotFoundException e) {
+            assert false : "Failed to open file.";
+        }
+        return file;
+    }
+
+    @SmallTest
+    public void testOpenFileWhenReady() {
+        Uri uri = ChromeFileProvider.generateUriAndBlockAccess(getInstrumentation().getContext());
+        Uri fileUri = new Uri.Builder().path("1").build();
+        ChromeFileProvider.notifyFileReady(uri, fileUri);
+        Uri result = ChromeFileProvider.getFileUriWhenReady(uri);
+        assertEquals(result, fileUri);
+    }
+
+    @LargeTest
+    public void testOpenOnAsyncNotify() {
+        final Context context = getInstrumentation().getContext();
+        final Uri uri = ChromeFileProvider.generateUriAndBlockAccess(context);
+        AsyncTask.THREAD_POOL_EXECUTOR.execute(new Runnable() {
+            @Override
+            public void run() {
+                try {
+                    Thread.sleep(10);
+                } catch (InterruptedException e) {
+                    // Ignore exception.
+                }
+                ChromeFileProvider.notifyFileReady(uri, null);
+            }
+        });
+        ParcelFileDescriptor file = openFileFromProvider(uri);
+        // File should be null because the notify passes a null file uri.
+        assertNull(file);
+    }
+
+    @LargeTest
+    public void testFileChanged() {
+        final Context context = getInstrumentation().getContext();
+        Uri uri1 = ChromeFileProvider.generateUriAndBlockAccess(context);
+        final Uri uri2 = ChromeFileProvider.generateUriAndBlockAccess(context);
+        final Uri fileUri2 = new Uri.Builder().path("2").build();
+        AsyncTask.THREAD_POOL_EXECUTOR.execute(new Runnable() {
+            @Override
+            public void run() {
+                try {
+                    Thread.sleep(10);
+                } catch (InterruptedException e) {
+                    // Ignore exception.
+                }
+                ChromeFileProvider.notifyFileReady(uri2, fileUri2);
+            }
+        });
+
+        // This should not be blocked even without a notify since file was changed.
+        Uri file1 = ChromeFileProvider.getFileUriWhenReady(uri1);
+        // File should be null because the notify passes a null file uri.
+        assertNull(file1);
+
+        // This should be unblocked when the notify is called.
+        Uri file2 = ChromeFileProvider.getFileUriWhenReady(uri2);
+        assertEquals(fileUri2, file2);
+
+        // This should not be blocked even without a notify since file was changed.
+        file1 = ChromeFileProvider.getFileUriWhenReady(uri1);
+        // File should be null because the notify passes a null file uri.
+        assertNull(file1);
+    }
+}