Merge "Add test size annotation." into ub-launcher3-calgary

tests/src/com/android/launcher3/BindWidgetTest.java

@@ -19,6 +19,7 @@ import android.os.ParcelFileDescriptor;
 import android.support.test.uiautomator.UiDevice;
 import android.support.test.uiautomator.UiSelector;
 import android.test.InstrumentationTestCase;
+import android.test.suitebuilder.annotation.LargeTest;
 
 import com.android.launcher3.compat.AppWidgetManagerCompat;
 import com.android.launcher3.compat.PackageInstallerCompat;
@@ -38,6 +39,7 @@ import java.util.concurrent.atomic.AtomicReference;
  *
  * Note running these tests will clear the workspace on the device.
  */
+@LargeTest
 @TargetApi(Build.VERSION_CODES.LOLLIPOP)
 public class BindWidgetTest extends InstrumentationTestCase {
 

tests/src/com/android/launcher3/QuickAddWidgetTest.java

@@ -9,6 +9,7 @@ import android.support.test.uiautomator.UiDevice;
 import android.support.test.uiautomator.UiObject2;
 import android.support.test.uiautomator.Until;
 import android.test.InstrumentationTestCase;
+import android.test.suitebuilder.annotation.LargeTest;
 import android.view.MotionEvent;
 import android.view.ViewConfiguration;
 
@@ -17,6 +18,7 @@ import java.util.List;
 /**
  * Add an arbitrary widget from the widget picker very quickly to test potential race conditions.
  */
+@LargeTest
 public class QuickAddWidgetTest extends InstrumentationTestCase {
     // Disabled because it's flaky and not particularly useful. But this class could still be useful
     // as an example if we want other UI tests in the future.

tests/src/com/android/launcher3/RotationPreferenceTest.java

@@ -8,10 +8,12 @@ import android.support.test.uiautomator.UiDevice;
 import android.support.test.uiautomator.UiObject;
 import android.support.test.uiautomator.UiSelector;
 import android.test.InstrumentationTestCase;
+import android.test.suitebuilder.annotation.MediumTest;
 
 /**
  * Test for auto rotate preference.
  */
+@MediumTest
 public class RotationPreferenceTest extends InstrumentationTestCase {
 
     private UiDevice mDevice;

tests/src/com/android/launcher3/model/GridSizeMigrationTaskTest.java

@@ -5,6 +5,7 @@ import android.content.Intent;
 import android.database.Cursor;
 import android.graphics.Point;
 import android.test.ProviderTestCase2;
+import android.test.suitebuilder.annotation.MediumTest;
 
 import com.android.launcher3.InvariantDeviceProfile;
 import com.android.launcher3.LauncherModel;
@@ -19,6 +20,7 @@ import java.util.HashSet;
 /**
  * Unit tests for {@link GridSizeMigrationTask}
  */
+@MediumTest
 public class GridSizeMigrationTaskTest extends ProviderTestCase2<TestLauncherProvider> {
 
     private static final long DESKTOP = LauncherSettings.Favorites.CONTAINER_DESKTOP;