📰 Make the section dismissal range dynamic

chrome/android/junit/src/org/chromium/chrome/browser/ntp/cards/SuggestionsSectionTest.java

Index: chrome/android/junit/src/org/chromium/chrome/browser/ntp/cards/SuggestionsSectionTest.java
diff --git a/chrome/android/junit/src/org/chromium/chrome/browser/ntp/cards/SuggestionsSectionTest.java b/chrome/android/junit/src/org/chromium/chrome/browser/ntp/cards/SuggestionsSectionTest.java
index c2fe9dc509fb2dcefdc1d4703b09ce0e0f5f27a7..31302ff90b34fb29f3fac0c14997f1cc500a9264 100644
--- a/chrome/android/junit/src/org/chromium/chrome/browser/ntp/cards/SuggestionsSectionTest.java
+++ b/chrome/android/junit/src/org/chromium/chrome/browser/ntp/cards/SuggestionsSectionTest.java
@@ -118,6 +118,38 @@ public void testDismissSibling() {
 
     @Test
     @Feature({"Ntp"})
+    public void testGetDismissalGroupWithoutHeader() {
+        SuggestionsSection section = createSectionWithFetchAction(true);
+        section.setHeaderVisibility(false);
+
+        assertEquals(ItemViewType.STATUS, section.getItemViewType(0));
+        assertEquals(setOf(0, 1), section.getItemDismissalGroup(0));
+
+        assertEquals(ItemViewType.ACTION, section.getItemViewType(1));
+        assertEquals(setOf(0, 1), section.getItemDismissalGroup(1));
+    }
+
+    @Test
+    @Feature({"Ntp"})
+    public void testGetDismissalGroupWithoutAction() {
+        SuggestionsSection section = createSectionWithFetchAction(false);
+
+        assertEquals(ItemViewType.STATUS, section.getItemViewType(1));
+        assertEquals(Collections.singleton(1), section.getItemDismissalGroup(1));
+    }
+
+    @Test
+    @Feature({"Ntp"})
+    public void testGetDismissalGroupActionAndHeader() {
+        SuggestionsSection section = createSectionWithFetchAction(false);
+        section.setHeaderVisibility(false);
+
+        assertEquals(ItemViewType.STATUS, section.getItemViewType(0));
+        assertEquals(Collections.singleton(0), section.getItemDismissalGroup(0));
+    }
+
+    @Test
+    @Feature({"Ntp"})
     public void testAddSuggestionsNotification() {
         final int suggestionCount = 5;
         List<SnippetArticle> snippets = createDummySuggestions(suggestionCount,