Merge "Recover print job ID from intent extras" into pi-dev

tests/unit/src/com/android/settings/print/PrintJobSettingsActivityTest.java

@@ -0,0 +1,142 @@
+/*
+ * Copyright (C) 2018 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.settings.print;
+
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assume.assumeTrue;
+
+import android.app.Activity;
+import android.content.Intent;
+import android.content.pm.PackageManager;
+import android.net.Uri;
+import android.os.Bundle;
+import android.os.CancellationSignal;
+import android.os.ParcelFileDescriptor;
+import android.print.PageRange;
+import android.print.PrintAttributes;
+import android.print.PrintDocumentAdapter;
+import android.print.PrintDocumentInfo;
+import android.print.PrintJob;
+import android.print.PrintManager;
+import android.support.annotation.NonNull;
+import android.support.test.InstrumentationRegistry;
+import android.support.test.filters.LargeTest;
+import android.support.test.rule.ActivityTestRule;
+import android.support.test.runner.AndroidJUnit4;
+import android.support.test.uiautomator.By;
+import android.support.test.uiautomator.UiDevice;
+import android.support.test.uiautomator.UiObject2;
+import android.support.test.uiautomator.Until;
+import android.util.Log;
+
+import com.android.settings.Settings;
+
+import org.junit.Before;
+import org.junit.Rule;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+
+import java.io.FileInputStream;
+import java.io.IOException;
+import java.util.UUID;
+
+@RunWith(AndroidJUnit4.class)
+public class PrintJobSettingsActivityTest {
+    private static final String EXTRA_PRINT_JOB_ID = "EXTRA_PRINT_JOB_ID";
+    private static final String LOG_TAG = PrintJobSettingsActivityTest.class.getSimpleName();
+
+    // Any activity is fine
+    @Rule
+    public final ActivityTestRule<Settings.PrintSettingsActivity> mActivityRule =
+            new ActivityTestRule<>(Settings.PrintSettingsActivity.class, true);
+
+    public static void runShellCommand(@NonNull String cmd) throws IOException {
+        ParcelFileDescriptor stdOut =
+                InstrumentationRegistry.getInstrumentation().getUiAutomation().executeShellCommand(
+                        cmd);
+
+        try (FileInputStream fis = new ParcelFileDescriptor.AutoCloseInputStream(stdOut)) {
+            byte[] buf = new byte[512];
+            while (fis.read(buf) != -1) {
+                // keep reading
+            }
+        }
+    }
+
+    @Before
+    public void requirePrintFeature() {
+        assumeTrue(InstrumentationRegistry.getTargetContext().getPackageManager().hasSystemFeature(
+                PackageManager.FEATURE_PRINTING));
+    }
+
+    @Before
+    public void wakeUpScreen() throws Exception {
+        runShellCommand("input keyevent KEYCODE_WAKEUP");
+    }
+
+    @Test
+    @LargeTest
+    public void viewPrintJobSettings() throws Exception {
+        UUID uuid = UUID.randomUUID();
+        Object isWriteCalled = new Object();
+
+        // Create adapter that is good enough to start a print preview
+        PrintDocumentAdapter adapter = new PrintDocumentAdapter() {
+            @Override
+            public void onLayout(PrintAttributes oldAttributes, PrintAttributes newAttributes,
+                    CancellationSignal cancellationSignal,
+                    LayoutResultCallback callback, Bundle extras) {
+                callback.onLayoutFinished(new PrintDocumentInfo.Builder(uuid.toString()).build(),
+                        true);
+            }
+
+            @Override
+            public void onWrite(PageRange[] pages, ParcelFileDescriptor destination,
+                    CancellationSignal cancellationSignal,
+                    WriteResultCallback callback) {
+                synchronized (isWriteCalled) {
+                    isWriteCalled.notify();
+                }
+                callback.onWriteFailed(null);
+            }
+        };
+
+        Activity activity =  mActivityRule.getActivity();
+        PrintManager pm = mActivityRule.getActivity().getSystemService(PrintManager.class);
+
+        // Start printing
+        PrintJob printJob = pm.print(uuid.toString(), adapter, null);
+
+        // Wait until print preview is up
+        synchronized (isWriteCalled) {
+            isWriteCalled.wait();
+        }
+
+        // Start print job settings
+        Intent intent = new Intent(android.provider.Settings.ACTION_PRINT_SETTINGS);
+        intent.putExtra(EXTRA_PRINT_JOB_ID, printJob.getId().flattenToString());
+        intent.setData(Uri.fromParts("printjob", printJob.getId().flattenToString(), null));
+        activity.startActivity(intent);
+
+        UiDevice uiDevice = UiDevice.getInstance(InstrumentationRegistry.getInstrumentation());
+        UiObject2 printPrefTitle = uiDevice.wait(Until.findObject(By.text("Configuring "
+                + uuid.toString())), 5000);
+        assertNotNull(printPrefTitle);
+
+        Log.i(LOG_TAG, "Found " + printPrefTitle.getText());
+    }
+}