chrome/android/javatests/src/org/chromium/chrome/browser/ShareIntentTest.java

// 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;

import android.content.ComponentName;
import android.content.Intent;
import android.content.pm.PackageManager;
import android.net.Uri;
import android.os.ParcelFileDescriptor;
import android.support.test.filters.LargeTest;

import org.junit.After;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
import org.junit.runner.RunWith;

import org.chromium.base.task.PostTask;
import org.chromium.base.task.TaskTraits;
import org.chromium.base.test.util.CommandLineFlags;
import org.chromium.base.test.util.RetryOnFailure;
import org.chromium.chrome.browser.share.ShareHelper;
import org.chromium.chrome.browser.share.ShareMenuActionHandler;
import org.chromium.chrome.browser.tab.Tab;
import org.chromium.chrome.browser.util.ChromeFileProvider;
import org.chromium.chrome.test.ChromeJUnit4ClassRunner;
import org.chromium.chrome.test.ChromeTabbedActivityTestRule;
import org.chromium.content_public.browser.test.util.TestThreadUtils;

import java.io.IOException;
import java.util.concurrent.ExecutionException;

/**
 * Instrumentation tests for Share intents.
 */
@RunWith(ChromeJUnit4ClassRunner.class)
@CommandLineFlags.Add({ChromeSwitches.DISABLE_FIRST_RUN_EXPERIENCE})
public class ShareIntentTest {
    @Rule
    public ChromeTabbedActivityTestRule mActivityTestRule = new ChromeTabbedActivityTestRule();

    private static final String TAG = "ShareIntentTest";

    /**
     * Mock activity class that overrides the startActivity and checks if the file passed in the
     * intent can be opened.
     *
     * This class is a wrapper around the actual activity of the test, while it also inherits from
     * activity and redirects the calls to the methods to the actual activity.
     */
    private static class MockChromeActivity extends ChromeTabbedActivity {
        private final Object mLock = new Object();
        private boolean mCheckCompleted;
        private ChromeActivity mActivity;

        public MockChromeActivity(ChromeActivity activity) {
            mActivity = activity;
            mCheckCompleted = false;
        }

        /**
         * Overrides startActivity and notifies check completed when the file from the uri of the
         * intent is opened.
         */
        @Override
        public void startActivity(Intent intent) {
            final Uri uri = intent.getClipData().getItemAt(0).getUri();
            PostTask.postTask(TaskTraits.BEST_EFFORT_MAY_BLOCK, () -> {
                ChromeFileProvider provider = new ChromeFileProvider();
                ParcelFileDescriptor file = null;
                try {
                    file = provider.openFile(uri, "r");
                    if (file != null) file.close();
                } catch (IOException e) {
                    assert false : "Error while opening the file";
                }
                synchronized (mLock) {
                    mCheckCompleted = true;
                    mLock.notify();
                }
            });
        }

        /**
         * Waits till the check for file opening is completed.
         */
        public void waitForFileCheck() throws InterruptedException {
            synchronized (mLock) {
                while (!mCheckCompleted) {
                    mLock.wait();
                }
            }
        }

        @Override
        public String getPackageName() {
            return mActivity.getPackageName();
        }

        @Override
        public Tab getActivityTab() {
            return mActivity.getActivityTab();
        }

        @Override
        public ChromeApplication getChromeApplication() {
            return mActivity.getChromeApplication();
        }

        @Override
        public PackageManager getPackageManager() {
            return mActivity.getPackageManager();
        }
    }

    @Test
    @LargeTest
    @RetryOnFailure
    public void testShareIntent() throws ExecutionException, InterruptedException {
        MockChromeActivity mockActivity = TestThreadUtils.runOnUiThreadBlocking(() -> {
            // Sets a test component as last shared and "shareDirectly" option is set so that
            // the share selector menu is not opened. The start activity is overriden, so the
            // package and class names do not matter.
            return new MockChromeActivity(mActivityTestRule.getActivity());
        });
        ShareHelper.setLastShareComponentName(
                new ComponentName("test.package", "test.activity"), null);
        // Skips the capture of screenshot and notifies with an empty file.
        ShareMenuActionHandler.setScreenshotCaptureSkippedForTesting(true);

        TestThreadUtils.runOnUiThreadBlocking(
                () -> mockActivity.onShareMenuItemSelected(
                                true /* shareDirectly */, false /* isIncognito */));

        mockActivity.waitForFileCheck();

        ShareHelper.setLastShareComponentName(new ComponentName("", ""), null);
    }

    @Before
    public void setUp() throws InterruptedException {
        mActivityTestRule.startMainActivityOnBlankPage();
    }

    @After
    public void tearDown() {
        ShareMenuActionHandler.setScreenshotCaptureSkippedForTesting(false);
    }
}