Test that multiple bundles are created correctly (08392d26) · Commits · e / os / android_frameworks_base

packages/SystemUI/tests/src/com/android/systemui/statusbar/notification/collection/ShadeListBuilderTest.java

+34 −1

Original line number	Diff line number	Diff line
		@@ -967,6 +967,35 @@ public class ShadeListBuilderTest extends SysuiTestCase {
		);
		}

		@Test
		@EnableFlags(NotificationBundleUi.FLAG_NAME)
		public void testBundle_multipleBundles() {
		mListBuilder.setBundler(TestBundler.INSTANCE);

		addNotif(0, PACKAGE_1, BUNDLE_1);
		addNotif(1, PACKAGE_1, BUNDLE_2);
		addGroupChild(2, PACKAGE_2, GROUP_1, BUNDLE_2);
		addGroupChild(3, PACKAGE_2, GROUP_1, BUNDLE_2);
		addGroupSummary(4, PACKAGE_2, GROUP_1, BUNDLE_2);
		dispatchBuild();

		verifyBuiltList(
		bundle(
		BUNDLE_2,
		notif(1),
		group(
		summary(4),
		child(2),
		child(3)
		)
		),
		bundle(
		BUNDLE_1,
		notif(0)
		)
		);
		}

		@Test
		@EnableFlags(NotificationBundleUi.FLAG_NAME)
		public void testBundle_notifNotPromoted() {
		@@ -3194,13 +3223,17 @@ public class ShadeListBuilderTest extends SysuiTestCase {
		private static final String GROUP_1 = "group_1";
		private static final String GROUP_2 = "group_2";
		private static final String BUNDLE_1 = "bundle_1";
		private static final String BUNDLE_2 = "bundle_2";
		}

		class TestBundler extends NotifBundler {

		public static final TestBundler INSTANCE = new TestBundler();

		List<BundleSpec> mBundleSpecs = List.of(new BundleSpec("bundle_1",0, 0, 0));
		List<BundleSpec> mBundleSpecs = List.of(
		new BundleSpec("bundle_1", 0, 0, 0),
		new BundleSpec("bundle_2", 0, 0, 0)
		);

		List<String> mBundleIds = this.mBundleSpecs.stream()
		.map(BundleSpec::getKey)