Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit ceef2de0 authored by Garfield Tan's avatar Garfield Tan
Browse files

Move common logic to TestDocumentClipper.

Change-Id: If96b0f11553c1c7fe71e328641d637365f742fd2
parent 3808de5d
Loading
Loading
Loading
Loading
+12 −0
Original line number Diff line number Diff line
@@ -16,6 +16,9 @@

package com.android.documentsui.testing;

import static junit.framework.Assert.assertNull;
import static junit.framework.Assert.assertSame;

import android.content.ClipData;
import android.net.Uri;

@@ -30,6 +33,8 @@ import java.util.function.Function;

public class TestDocumentClipper implements DocumentClipper {

    private ClipData mLastClipData;

    @Override
    public boolean hasItemsToPaste() {
        return false;
@@ -63,6 +68,7 @@ public class TestDocumentClipper implements DocumentClipper {
    @Override
    public void copyFromClipData(RootInfo root, DocumentInfo destination, ClipData clipData,
            Callback callback) {
        mLastClipData = clipData;
    }

    @Override
@@ -75,5 +81,11 @@ public class TestDocumentClipper implements DocumentClipper {
        return 0;
    }

    public void assertNoClipData() {
        assertNull(mLastClipData);
    }

    public void assertSameClipData(ClipData expect) {
        assertSame(expect, mLastClipData);
    }
}
+6 −25
Original line number Diff line number Diff line
@@ -72,8 +72,8 @@ public class ActionHandlerTest {
    private TestDialogController mDialogs;
    private TestConfirmationCallback mCallback;
    private ActionHandler<TestActivity> mHandler;
    private TestDocumentClipper mClipper;
    private boolean refreshAnswer = false;
    private ClipData mClipDataFromCallback;

    @Before
    public void setUp() {
@@ -84,14 +84,13 @@ public class ActionHandlerTest {
        mCallback = new TestConfirmationCallback();
        mEnv.roots.configurePm(mActivity.packageMgr);
        mEnv.injector.dialogs = mDialogs;
        mClipper = new TestDocumentClipper();

        mHandler = createHandler();

        mDialogs.confirmNext();

        mEnv.selectDocument(TestEnv.FILE_GIF);

        mClipDataFromCallback = null;
    }

    @Test
@@ -375,16 +374,7 @@ public class ActionHandlerTest {
                mEnv.searchViewManager,
                mEnv::lookupExecutor,
                mActionModeAddons,
                new TestDocumentClipper() {
                    @Override
                    public void copyFromClipData(
                            RootInfo root,
                            DocumentInfo destination,
                            ClipData clipData,
                            Callback callback) {
                        mClipDataFromCallback = clipData;
                    }
                },
                mClipper,
                null,
                mEnv.injector
        );
@@ -400,7 +390,7 @@ public class ActionHandlerTest {

        mEnv.beforeAsserts();

        assertSame(clipData, mClipDataFromCallback);
        mClipper.assertSameClipData(clipData);
    }

    @Test
@@ -413,16 +403,7 @@ public class ActionHandlerTest {
                mEnv.searchViewManager,
                mEnv::lookupExecutor,
                mActionModeAddons,
                new TestDocumentClipper() {
                    @Override
                    public void copyFromClipData(
                            RootInfo root,
                            DocumentInfo destination,
                            ClipData clipData,
                            Callback callback) {
                        mClipDataFromCallback = clipData;
                    }
                },
                mClipper,
                null,
                mEnv.injector
        );
@@ -436,7 +417,7 @@ public class ActionHandlerTest {

        mEnv.beforeAsserts();

        assertNull(mClipDataFromCallback);
        mClipper.assertNoClipData();
    }

    @Test