tests/tests/print/src/android/print/cts/CustomPrintOptionsTest.java - platform/cts - Git at Google

 /*
  * Copyright (C) 2016 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 android.print.cts;

 import static android.print.test.Utils.eventually;

 import android.os.ParcelFileDescriptor;
 import android.platform.test.annotations.AppModeFull;
 import android.print.PageRange;
 import android.print.PrintAttributes;
 import android.print.PrintAttributes.Margins;
 import android.print.PrintAttributes.MediaSize;
 import android.print.PrintAttributes.Resolution;
 import android.print.PrintDocumentAdapter;
 import android.print.PrintDocumentAdapter.LayoutResultCallback;
 import android.print.PrintDocumentAdapter.WriteResultCallback;
 import android.print.PrintDocumentInfo;
 import android.print.PrintJobInfo;
 import android.print.PrinterCapabilitiesInfo;
 import android.print.PrinterId;
 import android.print.PrinterInfo;
 import android.print.test.BasePrintTest;
 import android.print.test.services.CustomPrintOptionsActivity;
 import android.print.test.services.FirstPrintService;
 import android.print.test.services.PrintServiceCallbacks;
 import android.print.test.services.PrinterDiscoverySessionCallbacks;
 import android.print.test.services.SecondPrintService;
 import android.print.test.services.StubbablePrinterDiscoverySession;
 import android.support.test.runner.AndroidJUnit4;
 import android.support.test.uiautomator.By;
 import android.support.test.uiautomator.UiObject;
 import android.support.test.uiautomator.UiObjectNotFoundException;
 import android.support.test.uiautomator.UiSelector;
 import android.util.Log;

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

 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.List;
 import java.util.concurrent.TimeoutException;

 /**
  * This test verifies changes to the printer capabilities are applied correctly.
  */
 @AppModeFull(reason = "Print UI cannot resolve custom print options activity in a instant app")
 @RunWith(AndroidJUnit4.class)
 public class CustomPrintOptionsTest extends BasePrintTest {
     private final static String LOG_TAG = "CustomPrintOptionsTest";
     private static final String PRINTER_NAME = "Test printer";
     private static final int MAX_TRIES = 10;

     // Default settings
     private final PageRange[] DEFAULT_PAGES = new PageRange[] { new PageRange(0, 0) };
     private final MediaSize DEFAULT_MEDIA_SIZE = MediaSize.ISO_A0;
     private final int DEFAULT_COLOR_MODE = PrintAttributes.COLOR_MODE_COLOR;
     private final int DEFAULT_DUPLEX_MODE = PrintAttributes.DUPLEX_MODE_LONG_EDGE;
     private final Resolution DEFAULT_RESOLUTION = new Resolution("300x300", "300x300", 300, 300);
     private final Margins DEFAULT_MARGINS = new Margins(0, 0, 0, 0);

     // All settings that are tested
     private final PageRange[][] PAGESS = { DEFAULT_PAGES, new PageRange[] { new PageRange(1, 1) },
             new PageRange[] { new PageRange(0, 2) }
     };
     private final MediaSize[] MEDIA_SIZES = { DEFAULT_MEDIA_SIZE, MediaSize.ISO_B0 };
     private final Integer[] COLOR_MODES = { DEFAULT_COLOR_MODE,
             PrintAttributes.COLOR_MODE_MONOCHROME
     };
     private final Integer[] DUPLEX_MODES = { DEFAULT_DUPLEX_MODE, PrintAttributes.DUPLEX_MODE_NONE
     };
     private final Resolution[] RESOLUTIONS = { DEFAULT_RESOLUTION,
             new Resolution("600x600", "600x600", 600, 600)
     };

     private PrintAttributes mLayoutAttributes;
     private PrintDocumentAdapter mAdapter;
     private static boolean sHasDefaultPrinterSet;

     /**
      * Get the page ranges currently selected as described in the UI.
      *
      * @return Only page ranges from {@link #PAGESS} are detected correctly.
      *
      * @throws Exception If something was unexpected
      */
     private PageRange[] getPages() throws Exception {
         if (getUiDevice().hasObject(By.text("All 3"))) {
             return PAGESS[2];
         }

         try {
             UiObject pagesEditText = getUiDevice().findObject(new UiSelector().resourceId(
                     "com.android.printspooler:id/page_range_edittext"));

             if (pagesEditText.getText().equals("2")) {
                 return PAGESS[1];
             }

             if (pagesEditText.getText().equals("1")) {
                 return PAGESS[0];
             }

             return null;
         } catch (UiObjectNotFoundException e) {
             dumpWindowHierarchy();
             throw e;
         }
     }

     @Before
     public void setUpServicesAndAdapter() {
         final PrinterDiscoverySessionCallbacks firstSessionCallbacks =
                 createMockPrinterDiscoverySessionCallbacks(invocation -> {
                     StubbablePrinterDiscoverySession session =
                             ((PrinterDiscoverySessionCallbacks) invocation.getMock())
                                     .getSession();
                     PrinterId printerId = session.getService().generatePrinterId(PRINTER_NAME);
                     List<PrinterInfo> printers = new ArrayList<>(1);
                     PrinterCapabilitiesInfo.Builder builder =
                             new PrinterCapabilitiesInfo.Builder(printerId);

                     builder.setMinMargins(DEFAULT_MARGINS)
                             .setColorModes(COLOR_MODES[0] | COLOR_MODES[1],
                                     DEFAULT_COLOR_MODE)
                             .setDuplexModes(DUPLEX_MODES[0] | DUPLEX_MODES[1],
                                     DEFAULT_DUPLEX_MODE)
                             .addMediaSize(DEFAULT_MEDIA_SIZE, true)
                             .addMediaSize(MEDIA_SIZES[1], false)
                             .addResolution(DEFAULT_RESOLUTION, true)
                             .addResolution(RESOLUTIONS[1], false);

                     printers.add(new PrinterInfo.Builder(printerId, PRINTER_NAME,
                             PrinterInfo.STATUS_IDLE).setCapabilities(builder.build()).build());

                     session.addPrinters(printers);
                     return null;
                 }, null, null, null, null, null, invocation -> {
                     onPrinterDiscoverySessionDestroyCalled();
                     return null;
                 });

         mAdapter = createMockPrintDocumentAdapter(
                 invocation -> {
                     LayoutResultCallback callback = (LayoutResultCallback) invocation
                             .getArguments()[3];
                     PrintDocumentInfo info = new PrintDocumentInfo.Builder(PRINT_JOB_NAME)
                             .setContentType(PrintDocumentInfo.CONTENT_TYPE_DOCUMENT)
                             .setPageCount(3)
                             .build();

                     synchronized (CustomPrintOptionsTest.this) {
                         mLayoutAttributes = (PrintAttributes) invocation.getArguments()[1];

                         CustomPrintOptionsTest.this.notifyAll();
                     }

                     callback.onLayoutFinished(info, true);
                     return null;
                 },
                 invocation -> {
                     Object[] args = invocation.getArguments();
                     ParcelFileDescriptor fd = (ParcelFileDescriptor) args[1];
                     WriteResultCallback callback = (WriteResultCallback) args[3];

                     PageRange[] writtenPages = (PageRange[]) args[0];

                     writeBlankPages(mLayoutAttributes, fd, writtenPages[0].getStart(),
                             writtenPages[0].getEnd());
                     fd.close();

                     callback.onWriteFinished(writtenPages);

                     onWriteCalled();

                     return null;
                 }, null);

         // Create the service callbacks for the first print service.
         PrintServiceCallbacks firstServiceCallbacks = createMockPrintServiceCallbacks(
                 invocation -> firstSessionCallbacks, null, null);

         // Configure the print services.
         FirstPrintService.setCallbacks(firstServiceCallbacks);
         SecondPrintService.setCallbacks(createMockPrintServiceCallbacks(null, null, null));

         // Set default printer
         if (!sHasDefaultPrinterSet) {
             // This is the first print test that runs. If this is run after other tests these other
             // test can still cause memory pressure and make the printactivity to go through a
             // destroy-create cycle. In this case we have to retry to operation.
             int tries = 0;
             while (true) {
                 try {
                     resetCounters();
                     makeDefaultPrinter(mAdapter, PRINTER_NAME);
                     break;
                 } catch (Throwable e) {
                     if (getActivityDestroyCallbackCallCount() > 0 && tries < MAX_TRIES) {
                         Log.e(LOG_TAG, "Activity was destroyed during test, retrying", e);

                         tries++;
                         continue;
                     }

                     throw new RuntimeException(e);
                 }
             }

             sHasDefaultPrinterSet = true;
         }
     }

     /**
      * Test that we can switch to a specific set of settings via the custom print options activity
      *
      * @param copyFromOriginal If the print job info should be copied from the original
      * @param numCopies        The copies to print
      * @param pages            The page ranges to print
      * @param mediaSize        The media size to use
      * @param isPortrait       If the mediaSize is portrait
      * @param colorMode        The color mode to use
      * @param duplexMode       The duplex mode to use
      * @param resolution       The resolution to use
      *
      * @throws Exception If anything is unexpected
      */
     private void testCase(final boolean copyFromOriginal, final Integer numCopies,
             final PageRange[] pages, final MediaSize mediaSize, final boolean isPortrait,
             final Integer colorMode, final Integer duplexMode, final Resolution resolution)
             throws Throwable {
         final PrintAttributes.Builder additionalAttributesBuilder = new PrintAttributes.Builder();
         final PrintAttributes.Builder newAttributesBuilder = new PrintAttributes.Builder();

         newAttributesBuilder.setMinMargins(DEFAULT_MARGINS);

         if (mediaSize != null) {
             if (isPortrait) {
                 additionalAttributesBuilder.setMediaSize(mediaSize.asPortrait());
                 newAttributesBuilder.setMediaSize(mediaSize.asPortrait());
             } else {
                 additionalAttributesBuilder.setMediaSize(mediaSize.asLandscape());
                 newAttributesBuilder.setMediaSize(mediaSize.asLandscape());
             }
         } else {
             newAttributesBuilder.setMediaSize(DEFAULT_MEDIA_SIZE);
         }

         if (colorMode != null) {
             additionalAttributesBuilder.setColorMode(colorMode);
             newAttributesBuilder.setColorMode(colorMode);
         } else {
             newAttributesBuilder.setColorMode(DEFAULT_COLOR_MODE);
         }

         if (duplexMode != null) {
             additionalAttributesBuilder.setDuplexMode(duplexMode);
             newAttributesBuilder.setDuplexMode(duplexMode);
         } else {
             newAttributesBuilder.setDuplexMode(DEFAULT_DUPLEX_MODE);
         }

         if (resolution != null) {
             additionalAttributesBuilder.setResolution(resolution);
             newAttributesBuilder.setResolution(resolution);
         } else {
             newAttributesBuilder.setResolution(DEFAULT_RESOLUTION);
         }

         CustomPrintOptionsActivity.setCallBack(
                 (printJob, printer) -> {
                     PrintJobInfo.Builder printJobBuilder;

                     if (copyFromOriginal) {
                         printJobBuilder = new PrintJobInfo.Builder(printJob);
                     } else {
                         printJobBuilder = new PrintJobInfo.Builder(null);
                     }

                     if (numCopies != null) {
                         printJobBuilder.setCopies(numCopies);
                     }

                     if (pages != null) {
                         printJobBuilder.setPages(pages);
                     }

                     if (mediaSize != null || colorMode != null || duplexMode != null
                             || resolution != null) {
                         printJobBuilder.setAttributes(additionalAttributesBuilder.build());
                     }

                     return printJobBuilder.build();
                 });

         // Check that the attributes were send to the print service
         PrintAttributes newAttributes = newAttributesBuilder.build();
         Log.i(LOG_TAG, "Change to attributes: " + newAttributes + ", copies: " + numCopies +
                 ", pages: " + Arrays.toString(pages) + ", copyFromOriginal: " + copyFromOriginal);

         // This is the first print test that runs. If this is run after other tests these other test
         // can still cause memory pressure and make the printactivity to go through a destroy-create
         // cycle. In this case we have to retry to operation.
         int tries = 0;
         while (true) {
             try {
                 resetCounters();

                 // Start printing
                 print(mAdapter);

                 // Wait for write.
                 waitForWriteAdapterCallback(1);

                 // Open the print options.
                 openPrintOptions();

                 // Apply options by executing callback above
                 Log.d(LOG_TAG, "Apply changes");
                 openCustomPrintOptions();

                 Log.d(LOG_TAG, "Check attributes");
                 long endTime = System.currentTimeMillis() + OPERATION_TIMEOUT_MILLIS;
                 synchronized (this) {
                     while (mLayoutAttributes == null ||
                             !mLayoutAttributes.equals(newAttributes)) {
                         wait(Math.max(1, endTime - System.currentTimeMillis()));

                         if (endTime < System.currentTimeMillis()) {
                             throw new TimeoutException(
                                     "Print attributes did not change to " + newAttributes + " in " +
                                             OPERATION_TIMEOUT_MILLIS + " ms. Current attributes"
                                             + mLayoutAttributes);
                         }
                     }
                 }

                 PageRange[] newPages;

                 if (pages == null) {
                     newPages = new PageRange[] { new PageRange(0, 2) };
                 } else {
                     newPages = pages;
                 }

                 Log.d(LOG_TAG, "Check pages");
                 eventually(() -> {
                     PageRange[] actualPages = getPages();
                     if (!Arrays.equals(newPages, actualPages)) {
                         throw new AssertionError(
                                 "Expected " + Arrays.toString(newPages) + ", actual " +
                                         Arrays.toString(actualPages));
                     }
                 });

                 break;
             } catch (Throwable e) {
                 if (getActivityDestroyCallbackCallCount() > 0 && tries < MAX_TRIES) {
                     Log.e(LOG_TAG, "Activity was destroyed during test, retrying", e);

                     tries++;
                     continue;
                 }

                 throw e;
             }
         }

         // Abort printing
         getUiDevice().pressBack();
         getUiDevice().pressBack();
         getUiDevice().pressBack();

         waitForPrinterDiscoverySessionDestroyCallbackCalled(1);
     }

     @Test
     public void changeToChangeEveryThingButPages() throws Throwable {
         testCase(false, 2, null, MEDIA_SIZES[1], false, COLOR_MODES[1], DUPLEX_MODES[1],
                 RESOLUTIONS[1]);
     }

     @Test
     public void changeToAttributes() throws Throwable {
         testCase(false, null, null, MEDIA_SIZES[1], false, COLOR_MODES[1], DUPLEX_MODES[1],
                 RESOLUTIONS[1]);
     }

     @Test
     public void changeToNonAttributes() throws Throwable {
         testCase(false, 2, PAGESS[1], null, true, null, null, null);
     }

     @Test
     public void changeToAttributesNoCopy() throws Throwable {
         testCase(true, null, null, MEDIA_SIZES[1], false, COLOR_MODES[1], DUPLEX_MODES[1],
                 RESOLUTIONS[1]);
     }

     @Test
     public void changeToNonAttributesNoCopy() throws Throwable {
         testCase(true, 2, PAGESS[1], null, true, null, null, null);
     }

     @Test
     public void changeToDefault() throws Throwable {
         testCase(false, 1, DEFAULT_PAGES, DEFAULT_MEDIA_SIZE, DEFAULT_MEDIA_SIZE.isPortrait(),
                 DEFAULT_COLOR_MODE, DEFAULT_DUPLEX_MODE, DEFAULT_RESOLUTION);
     }

     @Test
     public void changeToDefaultNoCopy() throws Throwable {
         testCase(true, 1, DEFAULT_PAGES, DEFAULT_MEDIA_SIZE, DEFAULT_MEDIA_SIZE.isPortrait(),
                 DEFAULT_COLOR_MODE, DEFAULT_DUPLEX_MODE, DEFAULT_RESOLUTION);
     }

     @Test
     public void changeToNothing() throws Throwable {
         testCase(false, null, null, null, true, null, null, null);
     }

     @Test
     public void testChangeToNothingNoCopy() throws Throwable {
         testCase(true, null, null, null, true, null, null, null);
     }

     @Test
     public void changeToAllPages() throws Throwable {
         testCase(false, null, PAGESS[2], null, true, null, null, null);
     }

     @Test
     public void changeToSomePages() throws Throwable {
         testCase(false, null, PAGESS[1], null, true, null, null, null);
     }
 }
	/*
	* Copyright (C) 2016 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 android.print.cts;

	import static android.print.test.Utils.eventually;

	import android.os.ParcelFileDescriptor;
	import android.platform.test.annotations.AppModeFull;
	import android.print.PageRange;
	import android.print.PrintAttributes;
	import android.print.PrintAttributes.Margins;
	import android.print.PrintAttributes.MediaSize;
	import android.print.PrintAttributes.Resolution;
	import android.print.PrintDocumentAdapter;
	import android.print.PrintDocumentAdapter.LayoutResultCallback;
	import android.print.PrintDocumentAdapter.WriteResultCallback;
	import android.print.PrintDocumentInfo;
	import android.print.PrintJobInfo;
	import android.print.PrinterCapabilitiesInfo;
	import android.print.PrinterId;
	import android.print.PrinterInfo;
	import android.print.test.BasePrintTest;
	import android.print.test.services.CustomPrintOptionsActivity;
	import android.print.test.services.FirstPrintService;
	import android.print.test.services.PrintServiceCallbacks;
	import android.print.test.services.PrinterDiscoverySessionCallbacks;
	import android.print.test.services.SecondPrintService;
	import android.print.test.services.StubbablePrinterDiscoverySession;
	import android.support.test.runner.AndroidJUnit4;
	import android.support.test.uiautomator.By;
	import android.support.test.uiautomator.UiObject;
	import android.support.test.uiautomator.UiObjectNotFoundException;
	import android.support.test.uiautomator.UiSelector;
	import android.util.Log;

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

	import java.util.ArrayList;
	import java.util.Arrays;
	import java.util.List;
	import java.util.concurrent.TimeoutException;

	/**
	* This test verifies changes to the printer capabilities are applied correctly.
	*/
	@AppModeFull(reason = "Print UI cannot resolve custom print options activity in a instant app")
	@RunWith(AndroidJUnit4.class)
	public class CustomPrintOptionsTest extends BasePrintTest {
	private final static String LOG_TAG = "CustomPrintOptionsTest";
	private static final String PRINTER_NAME = "Test printer";
	private static final int MAX_TRIES = 10;

	// Default settings
	private final PageRange[] DEFAULT_PAGES = new PageRange[] { new PageRange(0, 0) };
	private final MediaSize DEFAULT_MEDIA_SIZE = MediaSize.ISO_A0;
	private final int DEFAULT_COLOR_MODE = PrintAttributes.COLOR_MODE_COLOR;
	private final int DEFAULT_DUPLEX_MODE = PrintAttributes.DUPLEX_MODE_LONG_EDGE;
	private final Resolution DEFAULT_RESOLUTION = new Resolution("300x300", "300x300", 300, 300);
	private final Margins DEFAULT_MARGINS = new Margins(0, 0, 0, 0);

	// All settings that are tested
	private final PageRange[][] PAGESS = { DEFAULT_PAGES, new PageRange[] { new PageRange(1, 1) },
	new PageRange[] { new PageRange(0, 2) }
	};
	private final MediaSize[] MEDIA_SIZES = { DEFAULT_MEDIA_SIZE, MediaSize.ISO_B0 };
	private final Integer[] COLOR_MODES = { DEFAULT_COLOR_MODE,
	PrintAttributes.COLOR_MODE_MONOCHROME
	};
	private final Integer[] DUPLEX_MODES = { DEFAULT_DUPLEX_MODE, PrintAttributes.DUPLEX_MODE_NONE
	};
	private final Resolution[] RESOLUTIONS = { DEFAULT_RESOLUTION,
	new Resolution("600x600", "600x600", 600, 600)
	};

	private PrintAttributes mLayoutAttributes;
	private PrintDocumentAdapter mAdapter;
	private static boolean sHasDefaultPrinterSet;

	/**
	* Get the page ranges currently selected as described in the UI.
	*
	* @return Only page ranges from {@link #PAGESS} are detected correctly.
	*
	* @throws Exception If something was unexpected
	*/
	private PageRange[] getPages() throws Exception {
	if (getUiDevice().hasObject(By.text("All 3"))) {
	return PAGESS[2];
	}

	try {
	UiObject pagesEditText = getUiDevice().findObject(new UiSelector().resourceId(
	"com.android.printspooler:id/page_range_edittext"));

	if (pagesEditText.getText().equals("2")) {
	return PAGESS[1];
	}

	if (pagesEditText.getText().equals("1")) {
	return PAGESS[0];
	}

	return null;
	} catch (UiObjectNotFoundException e) {
	dumpWindowHierarchy();
	throw e;
	}
	}

	@Before
	public void setUpServicesAndAdapter() {
	final PrinterDiscoverySessionCallbacks firstSessionCallbacks =
	createMockPrinterDiscoverySessionCallbacks(invocation -> {
	StubbablePrinterDiscoverySession session =
	((PrinterDiscoverySessionCallbacks) invocation.getMock())
	.getSession();
	PrinterId printerId = session.getService().generatePrinterId(PRINTER_NAME);
	List<PrinterInfo> printers = new ArrayList<>(1);
	PrinterCapabilitiesInfo.Builder builder =
	new PrinterCapabilitiesInfo.Builder(printerId);

	builder.setMinMargins(DEFAULT_MARGINS)
	.setColorModes(COLOR_MODES[0] \| COLOR_MODES[1],
	DEFAULT_COLOR_MODE)
	.setDuplexModes(DUPLEX_MODES[0] \| DUPLEX_MODES[1],
	DEFAULT_DUPLEX_MODE)
	.addMediaSize(DEFAULT_MEDIA_SIZE, true)
	.addMediaSize(MEDIA_SIZES[1], false)
	.addResolution(DEFAULT_RESOLUTION, true)
	.addResolution(RESOLUTIONS[1], false);

	printers.add(new PrinterInfo.Builder(printerId, PRINTER_NAME,
	PrinterInfo.STATUS_IDLE).setCapabilities(builder.build()).build());

	session.addPrinters(printers);
	return null;
	}, null, null, null, null, null, invocation -> {
	onPrinterDiscoverySessionDestroyCalled();
	return null;
	});

	mAdapter = createMockPrintDocumentAdapter(
	invocation -> {
	LayoutResultCallback callback = (LayoutResultCallback) invocation
	.getArguments()[3];
	PrintDocumentInfo info = new PrintDocumentInfo.Builder(PRINT_JOB_NAME)
	.setContentType(PrintDocumentInfo.CONTENT_TYPE_DOCUMENT)
	.setPageCount(3)
	.build();

	synchronized (CustomPrintOptionsTest.this) {
	mLayoutAttributes = (PrintAttributes) invocation.getArguments()[1];

	CustomPrintOptionsTest.this.notifyAll();
	}

	callback.onLayoutFinished(info, true);
	return null;
	},
	invocation -> {
	Object[] args = invocation.getArguments();
	ParcelFileDescriptor fd = (ParcelFileDescriptor) args[1];
	WriteResultCallback callback = (WriteResultCallback) args[3];

	PageRange[] writtenPages = (PageRange[]) args[0];

	writeBlankPages(mLayoutAttributes, fd, writtenPages[0].getStart(),
	writtenPages[0].getEnd());
	fd.close();

	callback.onWriteFinished(writtenPages);

	onWriteCalled();

	return null;
	}, null);

	// Create the service callbacks for the first print service.
	PrintServiceCallbacks firstServiceCallbacks = createMockPrintServiceCallbacks(
	invocation -> firstSessionCallbacks, null, null);

	// Configure the print services.
	FirstPrintService.setCallbacks(firstServiceCallbacks);
	SecondPrintService.setCallbacks(createMockPrintServiceCallbacks(null, null, null));

	// Set default printer
	if (!sHasDefaultPrinterSet) {
	// This is the first print test that runs. If this is run after other tests these other
	// test can still cause memory pressure and make the printactivity to go through a
	// destroy-create cycle. In this case we have to retry to operation.
	int tries = 0;
	while (true) {
	try {
	resetCounters();
	makeDefaultPrinter(mAdapter, PRINTER_NAME);
	break;
	} catch (Throwable e) {
	if (getActivityDestroyCallbackCallCount() > 0 && tries < MAX_TRIES) {
	Log.e(LOG_TAG, "Activity was destroyed during test, retrying", e);

	tries++;
	continue;
	}

	throw new RuntimeException(e);
	}
	}

	sHasDefaultPrinterSet = true;
	}
	}

	/**
	* Test that we can switch to a specific set of settings via the custom print options activity
	*
	* @param copyFromOriginal If the print job info should be copied from the original
	* @param numCopies The copies to print
	* @param pages The page ranges to print
	* @param mediaSize The media size to use
	* @param isPortrait If the mediaSize is portrait
	* @param colorMode The color mode to use
	* @param duplexMode The duplex mode to use
	* @param resolution The resolution to use
	*
	* @throws Exception If anything is unexpected
	*/
	private void testCase(final boolean copyFromOriginal, final Integer numCopies,
	final PageRange[] pages, final MediaSize mediaSize, final boolean isPortrait,
	final Integer colorMode, final Integer duplexMode, final Resolution resolution)
	throws Throwable {
	final PrintAttributes.Builder additionalAttributesBuilder = new PrintAttributes.Builder();
	final PrintAttributes.Builder newAttributesBuilder = new PrintAttributes.Builder();

	newAttributesBuilder.setMinMargins(DEFAULT_MARGINS);

	if (mediaSize != null) {
	if (isPortrait) {
	additionalAttributesBuilder.setMediaSize(mediaSize.asPortrait());
	newAttributesBuilder.setMediaSize(mediaSize.asPortrait());
	} else {
	additionalAttributesBuilder.setMediaSize(mediaSize.asLandscape());
	newAttributesBuilder.setMediaSize(mediaSize.asLandscape());
	}
	} else {
	newAttributesBuilder.setMediaSize(DEFAULT_MEDIA_SIZE);
	}

	if (colorMode != null) {
	additionalAttributesBuilder.setColorMode(colorMode);
	newAttributesBuilder.setColorMode(colorMode);
	} else {
	newAttributesBuilder.setColorMode(DEFAULT_COLOR_MODE);
	}

	if (duplexMode != null) {
	additionalAttributesBuilder.setDuplexMode(duplexMode);
	newAttributesBuilder.setDuplexMode(duplexMode);
	} else {
	newAttributesBuilder.setDuplexMode(DEFAULT_DUPLEX_MODE);
	}

	if (resolution != null) {
	additionalAttributesBuilder.setResolution(resolution);
	newAttributesBuilder.setResolution(resolution);
	} else {
	newAttributesBuilder.setResolution(DEFAULT_RESOLUTION);
	}

	CustomPrintOptionsActivity.setCallBack(
	(printJob, printer) -> {
	PrintJobInfo.Builder printJobBuilder;

	if (copyFromOriginal) {
	printJobBuilder = new PrintJobInfo.Builder(printJob);
	} else {
	printJobBuilder = new PrintJobInfo.Builder(null);
	}

	if (numCopies != null) {
	printJobBuilder.setCopies(numCopies);
	}

	if (pages != null) {
	printJobBuilder.setPages(pages);
	}

	if (mediaSize != null \|\| colorMode != null \|\| duplexMode != null
	\|\| resolution != null) {
	printJobBuilder.setAttributes(additionalAttributesBuilder.build());
	}

	return printJobBuilder.build();
	});

	// Check that the attributes were send to the print service
	PrintAttributes newAttributes = newAttributesBuilder.build();
	Log.i(LOG_TAG, "Change to attributes: " + newAttributes + ", copies: " + numCopies +
	", pages: " + Arrays.toString(pages) + ", copyFromOriginal: " + copyFromOriginal);

	// This is the first print test that runs. If this is run after other tests these other test
	// can still cause memory pressure and make the printactivity to go through a destroy-create
	// cycle. In this case we have to retry to operation.
	int tries = 0;
	while (true) {
	try {
	resetCounters();

	// Start printing
	print(mAdapter);

	// Wait for write.
	waitForWriteAdapterCallback(1);

	// Open the print options.
	openPrintOptions();

	// Apply options by executing callback above
	Log.d(LOG_TAG, "Apply changes");
	openCustomPrintOptions();

	Log.d(LOG_TAG, "Check attributes");
	long endTime = System.currentTimeMillis() + OPERATION_TIMEOUT_MILLIS;
	synchronized (this) {
	while (mLayoutAttributes == null \|\|
	!mLayoutAttributes.equals(newAttributes)) {
	wait(Math.max(1, endTime - System.currentTimeMillis()));

	if (endTime < System.currentTimeMillis()) {
	throw new TimeoutException(
	"Print attributes did not change to " + newAttributes + " in " +
	OPERATION_TIMEOUT_MILLIS + " ms. Current attributes"
	+ mLayoutAttributes);
	}
	}
	}

	PageRange[] newPages;

	if (pages == null) {
	newPages = new PageRange[] { new PageRange(0, 2) };
	} else {
	newPages = pages;
	}

	Log.d(LOG_TAG, "Check pages");
	eventually(() -> {
	PageRange[] actualPages = getPages();
	if (!Arrays.equals(newPages, actualPages)) {
	throw new AssertionError(
	"Expected " + Arrays.toString(newPages) + ", actual " +
	Arrays.toString(actualPages));
	}
	});

	break;
	} catch (Throwable e) {
	if (getActivityDestroyCallbackCallCount() > 0 && tries < MAX_TRIES) {
	Log.e(LOG_TAG, "Activity was destroyed during test, retrying", e);

	tries++;
	continue;
	}

	throw e;
	}
	}

	// Abort printing
	getUiDevice().pressBack();
	getUiDevice().pressBack();
	getUiDevice().pressBack();

	waitForPrinterDiscoverySessionDestroyCallbackCalled(1);
	}

	@Test
	public void changeToChangeEveryThingButPages() throws Throwable {
	testCase(false, 2, null, MEDIA_SIZES[1], false, COLOR_MODES[1], DUPLEX_MODES[1],
	RESOLUTIONS[1]);
	}

	@Test
	public void changeToAttributes() throws Throwable {
	testCase(false, null, null, MEDIA_SIZES[1], false, COLOR_MODES[1], DUPLEX_MODES[1],
	RESOLUTIONS[1]);
	}

	@Test
	public void changeToNonAttributes() throws Throwable {
	testCase(false, 2, PAGESS[1], null, true, null, null, null);
	}

	@Test
	public void changeToAttributesNoCopy() throws Throwable {
	testCase(true, null, null, MEDIA_SIZES[1], false, COLOR_MODES[1], DUPLEX_MODES[1],
	RESOLUTIONS[1]);
	}

	@Test
	public void changeToNonAttributesNoCopy() throws Throwable {
	testCase(true, 2, PAGESS[1], null, true, null, null, null);
	}

	@Test
	public void changeToDefault() throws Throwable {
	testCase(false, 1, DEFAULT_PAGES, DEFAULT_MEDIA_SIZE, DEFAULT_MEDIA_SIZE.isPortrait(),
	DEFAULT_COLOR_MODE, DEFAULT_DUPLEX_MODE, DEFAULT_RESOLUTION);
	}

	@Test
	public void changeToDefaultNoCopy() throws Throwable {
	testCase(true, 1, DEFAULT_PAGES, DEFAULT_MEDIA_SIZE, DEFAULT_MEDIA_SIZE.isPortrait(),
	DEFAULT_COLOR_MODE, DEFAULT_DUPLEX_MODE, DEFAULT_RESOLUTION);
	}

	@Test
	public void changeToNothing() throws Throwable {
	testCase(false, null, null, null, true, null, null, null);
	}

	@Test
	public void testChangeToNothingNoCopy() throws Throwable {
	testCase(true, null, null, null, true, null, null, null);
	}

	@Test
	public void changeToAllPages() throws Throwable {
	testCase(false, null, PAGESS[2], null, true, null, null, null);
	}

	@Test
	public void changeToSomePages() throws Throwable {
	testCase(false, null, PAGESS[1], null, true, null, null, null);
	}
	}