tests/tests/shortcutmanager/src/android/content/pm/cts/shortcutmanager/ShortcutManagerRequestPinTest.java - platform/cts - Git at Google

 /*
  * Copyright (C) 2017 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.content.pm.cts.shortcutmanager;

 import static com.android.server.pm.shortcutmanagertest.ShortcutManagerTestUtils.assertWith;
 import static com.android.server.pm.shortcutmanagertest.ShortcutManagerTestUtils.list;
 import static com.android.server.pm.shortcutmanagertest.ShortcutManagerTestUtils.retryUntil;
 import static com.android.server.pm.shortcutmanagertest.ShortcutManagerTestUtils.setDefaultLauncher;

 import android.content.ComponentName;
 import android.content.Intent;
 import android.content.pm.ActivityInfo;
 import android.content.pm.LauncherApps.ShortcutQuery;
 import android.content.pm.PackageManager;
 import android.content.pm.ResolveInfo;
 import android.content.pm.ShortcutInfo;
 import android.content.pm.ShortcutManager;
 import android.content.pm.cts.shortcutmanager.common.Constants;
 import android.content.pm.cts.shortcutmanager.common.ReplyUtil;
 import android.os.PersistableBundle;
 import android.util.Log;

 import com.android.compatibility.common.util.CddTest;

 import java.util.HashMap;
 import java.util.List;

 @CddTest(requirement="3.8.1/C-4-1")
 public class ShortcutManagerRequestPinTest extends ShortcutManagerCtsTestsBase {
     private static final String TAG = "ShortcutMRPT";

     private static final String SHORTCUT_ID = "s12345";
     private static final String HIDDEN_SHORTCUT_ID = "s24680";

     @CddTest(requirement="[3.8.1/C-2-1],[3.8.1/C-3-1]")
     public void testIsRequestPinShortcutSupported() {

         // Launcher 1 supports it.
         setDefaultLauncher(getInstrumentation(), mLauncherContext1);

         runWithCaller(mPackageContext1, () -> {
             assertTrue(getManager().isRequestPinShortcutSupported());
         });

         // Launcher 4 does *not* supports it.
         setDefaultLauncher(getInstrumentation(), mLauncherContext4);

         runWithCaller(mPackageContext1, () -> {
             assertFalse(getManager().isRequestPinShortcutSupported());
         });
     }

     /**
      * A test for {@link ShortcutManager#requestPinShortcut}, a very simple case.
      */
     public void testRequestPinShortcut() {
         Log.i(TAG, "Testing with launcher1.");

         setDefaultLauncher(getInstrumentation(), mLauncherContext1);

         runWithCallerWithStrictMode(mPackageContext1, () -> {
             assertTrue(getManager().isRequestPinShortcutSupported());

             ReplyUtil.invokeAndWaitForReply(getTestContext(), (replyAction) -> {
                 final PersistableBundle extras = new PersistableBundle();
                 extras.putString(Constants.EXTRA_REPLY_ACTION, replyAction);
                 extras.putString(Constants.LABEL, "label1");

                 final ShortcutInfo shortcut = makeShortcutBuilder(SHORTCUT_ID)
                         .setShortLabel("label1")
                         .setIntent(new Intent(Intent.ACTION_MAIN))
                         .setExtras(extras)
                         .build();

                 // Note: Because requestPinShortcut() won't update the shortcut, but we need to
                 // update the extras that contains the broadcast ID, we need to update the shortcut
                 // manually here before requestPinShortcut().

                 // This is only needed when a shortcut is already published with the same ID.
                 assertTrue(getManager().updateShortcuts(list(shortcut)));

                 Log.i(TAG, "Calling requestPinShortcut...");
                 assertTrue(getManager().requestPinShortcut(shortcut, /* intent sender */ null));
                 Log.i(TAG, "Done.");
             });
         });
         runWithCallerWithStrictMode(mLauncherContext1, () -> {
             final ShortcutQuery query = new ShortcutQuery()
                     .setPackage(mPackageContext1.getPackageName())
                     .setShortcutIds(list(SHORTCUT_ID))
                     .setQueryFlags(ShortcutQuery.FLAG_MATCH_DYNAMIC
                             | ShortcutQuery.FLAG_MATCH_PINNED | ShortcutQuery.FLAG_MATCH_MANIFEST);
             Log.i(TAG, "Waiting for shortcut to be visible to launcher...");
             retryUntil(() -> {
                 final List<ShortcutInfo> shortcuts = getLauncherApps().getShortcuts(query,
                         android.os.Process.myUserHandle());
                 if (shortcuts == null) {
                     // Launcher not responded yet.
                     return false;
                 }
                 assertWith(shortcuts)
                         .haveIds(SHORTCUT_ID)
                         .areAllPinned()
                         .areAllNotDynamic()
                         .areAllNotManifest();
                 return true;
             }, "Shortcut still not pinned");
         });
         runWithCaller(mPackageContext1, () -> {
             assertWith(getManager().getPinnedShortcuts())
                     .forShortcutWithId(SHORTCUT_ID, si -> {
                         assertEquals("label1", si.getShortLabel());
                     })
                     .areAllPinned()
                     .areAllNotDynamic()
                     .areAllNotManifest()
                     .areAllMutable()
                     ;
         });

         Log.i(TAG, "Done testing with launcher1.");
     }

     public void testRequestPinShortcut_multiLaunchers() {
         testRequestPinShortcut();

         Log.i(TAG, "Testing with launcher2.");

         setDefaultLauncher(getInstrumentation(), mLauncherContext2);

         runWithCallerWithStrictMode(mPackageContext1, () -> {
             ReplyUtil.invokeAndWaitForReply(getTestContext(), (replyAction) -> {
                 final PersistableBundle extras = new PersistableBundle();
                 extras.putString(Constants.EXTRA_REPLY_ACTION, replyAction);
                 extras.putString(Constants.LABEL, "label1");

                 final ShortcutInfo shortcut = makeShortcutBuilder(SHORTCUT_ID)
                         .setExtras(extras)
                         .build();

                 // Note: Because requestPinShortcut() won't update the shortcut, but we need to
                 // update the extras that contains the broadcast ID, we need to update the shortcut
                 // manually here before requestPinShortcut().
                 assertTrue(getManager().updateShortcuts(list(shortcut)));

                 Log.i(TAG, "Calling requestPinShortcut...");
                 assertTrue(getManager().requestPinShortcut(shortcut, /* intent sender */ null));
                 Log.i(TAG, "Done.");
             });
         });
         runWithCallerWithStrictMode(mLauncherContext2, () -> {
             final ShortcutQuery query = new ShortcutQuery()
                     .setPackage(mPackageContext1.getPackageName())
                     .setShortcutIds(list(SHORTCUT_ID))
                     .setQueryFlags(ShortcutQuery.FLAG_MATCH_DYNAMIC
                             | ShortcutQuery.FLAG_MATCH_PINNED | ShortcutQuery.FLAG_MATCH_MANIFEST);
             Log.i(TAG, "Waiting for shortcut to be visible to launcher...");
             retryUntil(() -> {
                 final List<ShortcutInfo> shortcuts = getLauncherApps().getShortcuts(query,
                         android.os.Process.myUserHandle());
                 if (shortcuts == null) {
                     // Launcher not responded yet.
                     return false;
                 }
                 assertWith(shortcuts)
                         .haveIds(SHORTCUT_ID)
                         .areAllPinned()
                         .areAllNotDynamic()
                         .areAllNotManifest();
                 return true;
             }, "Shortcut still not pinned");
         });
         Log.i(TAG, "Done testing with launcher2.");
     }

     public void testRequestPinShortcut_multiLaunchers_withDynamic() {
         setDefaultLauncher(getInstrumentation(), mLauncherContext1);

         // Publish as a dynamic shortcut first, then call requestPin.
         ShortcutInfo shortcut = makeShortcutBuilder(SHORTCUT_ID)
                 .setShortLabel("label1")
                 .setIntent(new Intent(Intent.ACTION_MAIN))
                 .build();
         assertTrue(getManager().setDynamicShortcuts(list(shortcut)));

         // ==============================================================
         Log.i(TAG, "Testing with launcher1.");

         assertTrue(getManager().isRequestPinShortcutSupported());

         runWithCallerWithStrictMode(mPackageContext1, () -> {
             ReplyUtil.invokeAndWaitForReply(getTestContext(), (replyAction) -> {
                 final PersistableBundle extras = new PersistableBundle();
                 extras.putString(Constants.EXTRA_REPLY_ACTION, replyAction);
                 extras.putString(Constants.LABEL, "label1");

                 final ShortcutInfo shortcut2 = makeShortcutBuilder(SHORTCUT_ID)
                         .setExtras(extras)
                         .build();

                 // Note: Because requestPinShortcut() won't update the shortcut, but we need to
                 // update the extras that contains the broadcast ID, we need to update the shortcut
                 // manually here before requestPinShortcut().

                 // This is only needed when a shortcut is already published with the same ID.
                 assertTrue(getManager().updateShortcuts(list(shortcut2)));

                 Log.i(TAG, "Calling requestPinShortcut...");
                 assertTrue(getManager().requestPinShortcut(shortcut2, /* intent sender */ null));
                 Log.i(TAG, "Done.");
             });
         });
         runWithCallerWithStrictMode(mLauncherContext1, () -> {
             final ShortcutQuery query = new ShortcutQuery()
                     .setPackage(mPackageContext1.getPackageName())
                     .setShortcutIds(list(SHORTCUT_ID))
                     .setQueryFlags(ShortcutQuery.FLAG_MATCH_DYNAMIC
                             | ShortcutQuery.FLAG_MATCH_PINNED | ShortcutQuery.FLAG_MATCH_MANIFEST);
             Log.i(TAG, "Waiting for shortcut to be visible to launcher...");
             retryUntil(() -> {
                 final List<ShortcutInfo> shortcuts = getLauncherApps().getShortcuts(query,
                         android.os.Process.myUserHandle());
                 if (shortcuts == null) {
                     // Launcher not responded yet.
                     return false;
                 }
                 assertWith(shortcuts)
                         .haveIds(SHORTCUT_ID)
                         .areAllPinned()
                         .areAllDynamic()
                         .areAllNotManifest();
                 return true;
             }, "Shortcut still not pinned");
         });
         // ==============================================================
         Log.i(TAG, "Testing with launcher2.");
         setDefaultLauncher(getInstrumentation(), mLauncherContext2);

         assertTrue(getManager().isRequestPinShortcutSupported());

         runWithCallerWithStrictMode(mPackageContext1, () -> {
             ReplyUtil.invokeAndWaitForReply(getTestContext(), (replyAction) -> {
                 final PersistableBundle extras = new PersistableBundle();
                 extras.putString(Constants.EXTRA_REPLY_ACTION, replyAction);
                 extras.putString(Constants.LABEL, "label1");

                 final ShortcutInfo shortcut2 = makeShortcutBuilder(SHORTCUT_ID)
                         .setExtras(extras)
                         .build();

                 // Note: Because requestPinShortcut() won't update the shortcut, but we need to
                 // update the extras that contains the broadcast ID, we need to update the shortcut
                 // manually here before requestPinShortcut().

                 // This is only needed when a shortcut is already published with the same ID.
                 assertTrue(getManager().updateShortcuts(list(shortcut2)));

                 Log.i(TAG, "Calling requestPinShortcut...");
                 assertTrue(getManager().requestPinShortcut(shortcut2, /* intent sender */ null));
                 Log.i(TAG, "Done.");
             });
         });
         runWithCallerWithStrictMode(mLauncherContext2, () -> {
             final ShortcutQuery query = new ShortcutQuery()
                     .setPackage(mPackageContext1.getPackageName())
                     .setShortcutIds(list(SHORTCUT_ID))
                     .setQueryFlags(ShortcutQuery.FLAG_MATCH_DYNAMIC
                             | ShortcutQuery.FLAG_MATCH_PINNED | ShortcutQuery.FLAG_MATCH_MANIFEST);
             Log.i(TAG, "Waiting for shortcut to be visible to launcher...");
             retryUntil(() -> {
                 final List<ShortcutInfo> shortcuts = getLauncherApps().getShortcuts(query,
                         android.os.Process.myUserHandle());
                 if (shortcuts == null) {
                     // Launcher not responded yet.
                     return false;
                 }
                 assertWith(shortcuts)
                         .haveIds(SHORTCUT_ID)
                         .areAllPinned()
                         .areAllDynamic()
                         .areAllNotManifest();
                 return true;
             }, "Shortcut still not pinned");
         });

         runWithCaller(mPackageContext1, () -> {
             assertWith(getManager().getPinnedShortcuts())
                     .forShortcutWithId(SHORTCUT_ID, si -> {
                         assertEquals("label1", si.getShortLabel());
                     })
                     .areAllPinned()
                     .areAllDynamic()
                     .areAllNotManifest()
                     .areAllMutable()
             ;
         });
     }

     /**
      * Same as {@link ShortcutManager#requestPinShortcut} except the app has no main activities.
      */
     public void testRequestPinShortcut_noMainActivity() {
         setDefaultLauncher(getInstrumentation(), mLauncherContext1);

         final PackageManager pm = getTestContext().getPackageManager();
         final HashMap<ComponentName, Integer> originalState = new HashMap<>();
         try {
             for (ResolveInfo ri : pm.queryIntentActivities(
                     new Intent().setPackage(mPackageContext1.getPackageName()), 0)) {
                 final ActivityInfo activityInfo = ri.activityInfo;
                 final ComponentName componentName =
                         new ComponentName(activityInfo.packageName, activityInfo.name);

                 originalState.put(componentName, pm.getComponentEnabledSetting(componentName));
                 Log.i(TAG, "Disabling " + componentName);
                 pm.setComponentEnabledSetting(componentName,
                         PackageManager.COMPONENT_ENABLED_STATE_DISABLED, PackageManager.DONT_KILL_APP);
             }

             testRequestPinShortcut();

             runWithCaller(mPackageContext1, () -> {
                 assertWith(getManager().getPinnedShortcuts())
                         .areAllPinned()
                         .areAllNotDynamic()
                         .areAllNotManifest()
                         .areAllMutable()
                         .areAllWithActivity(null)
                 ;
                 assertWith(getManager().getManifestShortcuts()).isEmpty();
                 assertWith(getManager().getDynamicShortcuts()).isEmpty();
             });
         } finally {
             // Restore the original state.
             for (HashMap.Entry<ComponentName, Integer> e : originalState.entrySet()) {
                 pm.setComponentEnabledSetting(e.getKey(), e.getValue()
                         , PackageManager.DONT_KILL_APP);
             }
         }
     }

     /**
      * Same as {@link ShortcutManager#requestPinShortcut} except the app has no main activities.
      */
     public void testRequestPinShortcutExcludedFromLauncher_ThrowsException() {
         setDefaultLauncher(getInstrumentation(), mLauncherContext1);

         runWithCallerWithStrictMode(mPackageContext1, () -> {
             final ShortcutInfo shortcut = makeShortcutBuilder(HIDDEN_SHORTCUT_ID)
                     .setExcludedFromSurfaces(ShortcutInfo.SURFACE_LAUNCHER)
                     .build();

             Log.i(TAG, "Calling requestPinShortcut...");
             boolean isIllegalArgumentExceptionThrown = false;
             try {
                 assertTrue(getManager().requestPinShortcut(shortcut, /* intent sender */ null));
             } catch (IllegalArgumentException e) {
                 isIllegalArgumentExceptionThrown = true;
             }
             assertTrue(isIllegalArgumentExceptionThrown);
             Log.i(TAG, "Done.");
         });
     }

     // TODO Various other cases (already pinned, etc)
     // TODO Various error cases (missing mandatory fields, etc)
 }
	/*
	* Copyright (C) 2017 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.content.pm.cts.shortcutmanager;

	import static com.android.server.pm.shortcutmanagertest.ShortcutManagerTestUtils.assertWith;
	import static com.android.server.pm.shortcutmanagertest.ShortcutManagerTestUtils.list;
	import static com.android.server.pm.shortcutmanagertest.ShortcutManagerTestUtils.retryUntil;
	import static com.android.server.pm.shortcutmanagertest.ShortcutManagerTestUtils.setDefaultLauncher;

	import android.content.ComponentName;
	import android.content.Intent;
	import android.content.pm.ActivityInfo;
	import android.content.pm.LauncherApps.ShortcutQuery;
	import android.content.pm.PackageManager;
	import android.content.pm.ResolveInfo;
	import android.content.pm.ShortcutInfo;
	import android.content.pm.ShortcutManager;
	import android.content.pm.cts.shortcutmanager.common.Constants;
	import android.content.pm.cts.shortcutmanager.common.ReplyUtil;
	import android.os.PersistableBundle;
	import android.util.Log;

	import com.android.compatibility.common.util.CddTest;

	import java.util.HashMap;
	import java.util.List;

	@CddTest(requirement="3.8.1/C-4-1")
	public class ShortcutManagerRequestPinTest extends ShortcutManagerCtsTestsBase {
	private static final String TAG = "ShortcutMRPT";

	private static final String SHORTCUT_ID = "s12345";
	private static final String HIDDEN_SHORTCUT_ID = "s24680";

	@CddTest(requirement="[3.8.1/C-2-1],[3.8.1/C-3-1]")
	public void testIsRequestPinShortcutSupported() {

	// Launcher 1 supports it.
	setDefaultLauncher(getInstrumentation(), mLauncherContext1);

	runWithCaller(mPackageContext1, () -> {
	assertTrue(getManager().isRequestPinShortcutSupported());
	});

	// Launcher 4 does not supports it.
	setDefaultLauncher(getInstrumentation(), mLauncherContext4);

	runWithCaller(mPackageContext1, () -> {
	assertFalse(getManager().isRequestPinShortcutSupported());
	});
	}

	/**
	* A test for {@link ShortcutManager#requestPinShortcut}, a very simple case.
	*/
	public void testRequestPinShortcut() {
	Log.i(TAG, "Testing with launcher1.");

	setDefaultLauncher(getInstrumentation(), mLauncherContext1);

	runWithCallerWithStrictMode(mPackageContext1, () -> {
	assertTrue(getManager().isRequestPinShortcutSupported());

	ReplyUtil.invokeAndWaitForReply(getTestContext(), (replyAction) -> {
	final PersistableBundle extras = new PersistableBundle();
	extras.putString(Constants.EXTRA_REPLY_ACTION, replyAction);
	extras.putString(Constants.LABEL, "label1");

	final ShortcutInfo shortcut = makeShortcutBuilder(SHORTCUT_ID)
	.setShortLabel("label1")
	.setIntent(new Intent(Intent.ACTION_MAIN))
	.setExtras(extras)
	.build();

	// Note: Because requestPinShortcut() won't update the shortcut, but we need to
	// update the extras that contains the broadcast ID, we need to update the shortcut
	// manually here before requestPinShortcut().

	// This is only needed when a shortcut is already published with the same ID.
	assertTrue(getManager().updateShortcuts(list(shortcut)));

	Log.i(TAG, "Calling requestPinShortcut...");
	assertTrue(getManager().requestPinShortcut(shortcut, /* intent sender */ null));
	Log.i(TAG, "Done.");
	});
	});
	runWithCallerWithStrictMode(mLauncherContext1, () -> {
	final ShortcutQuery query = new ShortcutQuery()
	.setPackage(mPackageContext1.getPackageName())
	.setShortcutIds(list(SHORTCUT_ID))
	.setQueryFlags(ShortcutQuery.FLAG_MATCH_DYNAMIC
	\| ShortcutQuery.FLAG_MATCH_PINNED \| ShortcutQuery.FLAG_MATCH_MANIFEST);
	Log.i(TAG, "Waiting for shortcut to be visible to launcher...");
	retryUntil(() -> {
	final List<ShortcutInfo> shortcuts = getLauncherApps().getShortcuts(query,
	android.os.Process.myUserHandle());
	if (shortcuts == null) {
	// Launcher not responded yet.
	return false;
	}
	assertWith(shortcuts)
	.haveIds(SHORTCUT_ID)
	.areAllPinned()
	.areAllNotDynamic()
	.areAllNotManifest();
	return true;
	}, "Shortcut still not pinned");
	});
	runWithCaller(mPackageContext1, () -> {
	assertWith(getManager().getPinnedShortcuts())
	.forShortcutWithId(SHORTCUT_ID, si -> {
	assertEquals("label1", si.getShortLabel());
	})
	.areAllPinned()
	.areAllNotDynamic()
	.areAllNotManifest()
	.areAllMutable()
	;
	});

	Log.i(TAG, "Done testing with launcher1.");
	}

	public void testRequestPinShortcut_multiLaunchers() {
	testRequestPinShortcut();

	Log.i(TAG, "Testing with launcher2.");

	setDefaultLauncher(getInstrumentation(), mLauncherContext2);

	runWithCallerWithStrictMode(mPackageContext1, () -> {
	ReplyUtil.invokeAndWaitForReply(getTestContext(), (replyAction) -> {
	final PersistableBundle extras = new PersistableBundle();
	extras.putString(Constants.EXTRA_REPLY_ACTION, replyAction);
	extras.putString(Constants.LABEL, "label1");

	final ShortcutInfo shortcut = makeShortcutBuilder(SHORTCUT_ID)
	.setExtras(extras)
	.build();

	// Note: Because requestPinShortcut() won't update the shortcut, but we need to
	// update the extras that contains the broadcast ID, we need to update the shortcut
	// manually here before requestPinShortcut().
	assertTrue(getManager().updateShortcuts(list(shortcut)));

	Log.i(TAG, "Calling requestPinShortcut...");
	assertTrue(getManager().requestPinShortcut(shortcut, /* intent sender */ null));
	Log.i(TAG, "Done.");
	});
	});
	runWithCallerWithStrictMode(mLauncherContext2, () -> {
	final ShortcutQuery query = new ShortcutQuery()
	.setPackage(mPackageContext1.getPackageName())
	.setShortcutIds(list(SHORTCUT_ID))
	.setQueryFlags(ShortcutQuery.FLAG_MATCH_DYNAMIC
	\| ShortcutQuery.FLAG_MATCH_PINNED \| ShortcutQuery.FLAG_MATCH_MANIFEST);
	Log.i(TAG, "Waiting for shortcut to be visible to launcher...");
	retryUntil(() -> {
	final List<ShortcutInfo> shortcuts = getLauncherApps().getShortcuts(query,
	android.os.Process.myUserHandle());
	if (shortcuts == null) {
	// Launcher not responded yet.
	return false;
	}
	assertWith(shortcuts)
	.haveIds(SHORTCUT_ID)
	.areAllPinned()
	.areAllNotDynamic()
	.areAllNotManifest();
	return true;
	}, "Shortcut still not pinned");
	});
	Log.i(TAG, "Done testing with launcher2.");
	}

	public void testRequestPinShortcut_multiLaunchers_withDynamic() {
	setDefaultLauncher(getInstrumentation(), mLauncherContext1);

	// Publish as a dynamic shortcut first, then call requestPin.
	ShortcutInfo shortcut = makeShortcutBuilder(SHORTCUT_ID)
	.setShortLabel("label1")
	.setIntent(new Intent(Intent.ACTION_MAIN))
	.build();
	assertTrue(getManager().setDynamicShortcuts(list(shortcut)));

	// ==============================================================
	Log.i(TAG, "Testing with launcher1.");

	assertTrue(getManager().isRequestPinShortcutSupported());

	runWithCallerWithStrictMode(mPackageContext1, () -> {
	ReplyUtil.invokeAndWaitForReply(getTestContext(), (replyAction) -> {
	final PersistableBundle extras = new PersistableBundle();
	extras.putString(Constants.EXTRA_REPLY_ACTION, replyAction);
	extras.putString(Constants.LABEL, "label1");

	final ShortcutInfo shortcut2 = makeShortcutBuilder(SHORTCUT_ID)
	.setExtras(extras)
	.build();

	// Note: Because requestPinShortcut() won't update the shortcut, but we need to
	// update the extras that contains the broadcast ID, we need to update the shortcut
	// manually here before requestPinShortcut().

	// This is only needed when a shortcut is already published with the same ID.
	assertTrue(getManager().updateShortcuts(list(shortcut2)));

	Log.i(TAG, "Calling requestPinShortcut...");
	assertTrue(getManager().requestPinShortcut(shortcut2, /* intent sender */ null));
	Log.i(TAG, "Done.");
	});
	});
	runWithCallerWithStrictMode(mLauncherContext1, () -> {
	final ShortcutQuery query = new ShortcutQuery()
	.setPackage(mPackageContext1.getPackageName())
	.setShortcutIds(list(SHORTCUT_ID))
	.setQueryFlags(ShortcutQuery.FLAG_MATCH_DYNAMIC
	\| ShortcutQuery.FLAG_MATCH_PINNED \| ShortcutQuery.FLAG_MATCH_MANIFEST);
	Log.i(TAG, "Waiting for shortcut to be visible to launcher...");
	retryUntil(() -> {
	final List<ShortcutInfo> shortcuts = getLauncherApps().getShortcuts(query,
	android.os.Process.myUserHandle());
	if (shortcuts == null) {
	// Launcher not responded yet.
	return false;
	}
	assertWith(shortcuts)
	.haveIds(SHORTCUT_ID)
	.areAllPinned()
	.areAllDynamic()
	.areAllNotManifest();
	return true;
	}, "Shortcut still not pinned");
	});
	// ==============================================================
	Log.i(TAG, "Testing with launcher2.");
	setDefaultLauncher(getInstrumentation(), mLauncherContext2);

	assertTrue(getManager().isRequestPinShortcutSupported());

	runWithCallerWithStrictMode(mPackageContext1, () -> {
	ReplyUtil.invokeAndWaitForReply(getTestContext(), (replyAction) -> {
	final PersistableBundle extras = new PersistableBundle();
	extras.putString(Constants.EXTRA_REPLY_ACTION, replyAction);
	extras.putString(Constants.LABEL, "label1");

	final ShortcutInfo shortcut2 = makeShortcutBuilder(SHORTCUT_ID)
	.setExtras(extras)
	.build();

	// Note: Because requestPinShortcut() won't update the shortcut, but we need to
	// update the extras that contains the broadcast ID, we need to update the shortcut
	// manually here before requestPinShortcut().

	// This is only needed when a shortcut is already published with the same ID.
	assertTrue(getManager().updateShortcuts(list(shortcut2)));

	Log.i(TAG, "Calling requestPinShortcut...");
	assertTrue(getManager().requestPinShortcut(shortcut2, /* intent sender */ null));
	Log.i(TAG, "Done.");
	});
	});
	runWithCallerWithStrictMode(mLauncherContext2, () -> {
	final ShortcutQuery query = new ShortcutQuery()
	.setPackage(mPackageContext1.getPackageName())
	.setShortcutIds(list(SHORTCUT_ID))
	.setQueryFlags(ShortcutQuery.FLAG_MATCH_DYNAMIC
	\| ShortcutQuery.FLAG_MATCH_PINNED \| ShortcutQuery.FLAG_MATCH_MANIFEST);
	Log.i(TAG, "Waiting for shortcut to be visible to launcher...");
	retryUntil(() -> {
	final List<ShortcutInfo> shortcuts = getLauncherApps().getShortcuts(query,
	android.os.Process.myUserHandle());
	if (shortcuts == null) {
	// Launcher not responded yet.
	return false;
	}
	assertWith(shortcuts)
	.haveIds(SHORTCUT_ID)
	.areAllPinned()
	.areAllDynamic()
	.areAllNotManifest();
	return true;
	}, "Shortcut still not pinned");
	});

	runWithCaller(mPackageContext1, () -> {
	assertWith(getManager().getPinnedShortcuts())
	.forShortcutWithId(SHORTCUT_ID, si -> {
	assertEquals("label1", si.getShortLabel());
	})
	.areAllPinned()
	.areAllDynamic()
	.areAllNotManifest()
	.areAllMutable()
	;
	});
	}

	/**
	* Same as {@link ShortcutManager#requestPinShortcut} except the app has no main activities.
	*/
	public void testRequestPinShortcut_noMainActivity() {
	setDefaultLauncher(getInstrumentation(), mLauncherContext1);

	final PackageManager pm = getTestContext().getPackageManager();
	final HashMap<ComponentName, Integer> originalState = new HashMap<>();
	try {
	for (ResolveInfo ri : pm.queryIntentActivities(
	new Intent().setPackage(mPackageContext1.getPackageName()), 0)) {
	final ActivityInfo activityInfo = ri.activityInfo;
	final ComponentName componentName =
	new ComponentName(activityInfo.packageName, activityInfo.name);

	originalState.put(componentName, pm.getComponentEnabledSetting(componentName));
	Log.i(TAG, "Disabling " + componentName);
	pm.setComponentEnabledSetting(componentName,
	PackageManager.COMPONENT_ENABLED_STATE_DISABLED, PackageManager.DONT_KILL_APP);
	}

	testRequestPinShortcut();

	runWithCaller(mPackageContext1, () -> {
	assertWith(getManager().getPinnedShortcuts())
	.areAllPinned()
	.areAllNotDynamic()
	.areAllNotManifest()
	.areAllMutable()
	.areAllWithActivity(null)
	;
	assertWith(getManager().getManifestShortcuts()).isEmpty();
	assertWith(getManager().getDynamicShortcuts()).isEmpty();
	});
	} finally {
	// Restore the original state.
	for (HashMap.Entry<ComponentName, Integer> e : originalState.entrySet()) {
	pm.setComponentEnabledSetting(e.getKey(), e.getValue()
	, PackageManager.DONT_KILL_APP);
	}
	}
	}

	/**
	* Same as {@link ShortcutManager#requestPinShortcut} except the app has no main activities.
	*/
	public void testRequestPinShortcutExcludedFromLauncher_ThrowsException() {
	setDefaultLauncher(getInstrumentation(), mLauncherContext1);

	runWithCallerWithStrictMode(mPackageContext1, () -> {
	final ShortcutInfo shortcut = makeShortcutBuilder(HIDDEN_SHORTCUT_ID)
	.setExcludedFromSurfaces(ShortcutInfo.SURFACE_LAUNCHER)
	.build();

	Log.i(TAG, "Calling requestPinShortcut...");
	boolean isIllegalArgumentExceptionThrown = false;
	try {
	assertTrue(getManager().requestPinShortcut(shortcut, /* intent sender */ null));
	} catch (IllegalArgumentException e) {
	isIllegalArgumentExceptionThrown = true;
	}
	assertTrue(isIllegalArgumentExceptionThrown);
	Log.i(TAG, "Done.");
	});
	}

	// TODO Various other cases (already pinned, etc)
	// TODO Various error cases (missing mandatory fields, etc)
	}