Mark flaky chrome_public tests with @RetryOnFailure Batch 3

chrome/android/javatests/src/org/chromium/chrome/browser/omnibox/UrlBarTest.java

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 package org.chromium.chrome.browser.omnibox;
 
 import android.content.ClipData;
 import android.content.ClipboardManager;
 import android.content.Context;
 import android.content.Intent;
 import android.test.suitebuilder.annotation.SmallTest;
 import android.text.Editable;
 import android.text.TextUtils;
 import android.view.KeyEvent;
 import android.view.inputmethod.BaseInputConnection;
 
 import org.chromium.base.ThreadUtils;
 import org.chromium.base.test.util.Feature;
+import org.chromium.base.test.util.RetryOnFailure;
 import org.chromium.chrome.R;
 import org.chromium.chrome.browser.ChromeActivity;
 import org.chromium.chrome.browser.profiles.Profile;
 import org.chromium.chrome.test.ChromeActivityTestCaseBase;
 import org.chromium.chrome.test.util.OmniboxTestUtils;
 import org.chromium.chrome.test.util.OmniboxTestUtils.StubAutocompleteController;
 import org.chromium.content.browser.test.util.CallbackHelper;
 import org.chromium.content.browser.test.util.Criteria;
 import org.chromium.content.browser.test.util.CriteriaHelper;
 import org.chromium.content.browser.test.util.KeyUtils;
@@ skipping 114 matching lines @@
         return getAutocompleteState(urlBar, new Runnable() {
             @Override
             public void run() {
                 urlBar.setSelection(selectionStart, selectionEnd);
             }
         });
     }
 
     @SmallTest
     @Feature({"Omnibox"})
+    @RetryOnFailure
     public void testRefocusing() throws InterruptedException {
         startMainActivityOnBlankPage();
         UrlBar urlBar = getUrlBar();
         assertFalse(OmniboxTestUtils.doesUrlBarHaveFocus(urlBar));
         OmniboxTestUtils.checkUrlBarRefocus(urlBar, 5);
     }
 
     @SmallTest
     @Feature({"Omnibox"})
+    @RetryOnFailure
     public void testAutocompleteUpdatedOnSetText() throws InterruptedException {
         startMainActivityOnBlankPage();
         stubLocationBarAutocomplete();
         final UrlBar urlBar = getUrlBar();
         OmniboxTestUtils.toggleUrlBarFocus(urlBar, true);
 
         // Verify that setting a new string will clear the autocomplete.
         setTextAndVerifyNoAutocomplete(urlBar, "test");
         setAutocomplete(urlBar, "test", "ing is fun");
         setTextAndVerifyNoAutocomplete(urlBar, "new string");
@@ skipping 62 matching lines @@
 
         autocompleteHelper.waitForCallback(0);
         assertEquals("Prevent inline autocomplete",
                 expectedPreventInline, didPreventInlineAutocomplete.get());
         assertEquals("Requested autocomplete text",
                 expectedRequestedAutocompleteText, requestedAutocompleteText.get());
     }
 
     @SmallTest
     @Feature({"Omnibox"})
+    @RetryOnFailure
     public void testAutocompleteUpdatedOnSelection()
             throws InterruptedException, TimeoutException {
         startMainActivityOnBlankPage();
         stubLocationBarAutocomplete();
 
         final UrlBar urlBar = getUrlBar();
         OmniboxTestUtils.toggleUrlBarFocus(urlBar, true);
 
         // Verify that setting a selection before the autocomplete clears it.
         verifySelectionState("test", "ing is fun", 1, 1, false, "test", "test", true, "test");
@@ skipping 89 matching lines @@
         assertFalse("Inline autocomplete incorrectly prevented.",
                 didPreventInlineAutocomplete.get());
     }
 
     /**
      * Ensure that if the user deletes just the inlined autocomplete text that the suggestions are
      * regenerated.
      */
     @SmallTest
     @Feature({"Omnibox"})
+    @RetryOnFailure
     public void testSuggestionsUpdatedWhenDeletingInlineAutocomplete()
             throws InterruptedException, TimeoutException {
         startMainActivityOnBlankPage();
 
         stubLocationBarAutocomplete();
         final UrlBar urlBar = getUrlBar();
         OmniboxTestUtils.toggleUrlBarFocus(urlBar, true);
 
         setTextAndVerifyNoAutocomplete(urlBar, "test");
         setAutocomplete(urlBar, "test", "ing");
@@ skipping 22 matching lines @@
             }
         }));
 
         autocompleteHelper.waitForCallback(0);
         assertTrue("Inline autocomplete incorrectly allowed after delete.",
                 didPreventInlineAutocomplete.get());
     }
 
     @SmallTest
     @Feature({"Omnibox"})
+    @RetryOnFailure
     public void testSelectionChangesIgnoredInBatchMode() throws InterruptedException {
         startMainActivityOnBlankPage();
         stubLocationBarAutocomplete();
         final UrlBar urlBar = getUrlBar();
         OmniboxTestUtils.toggleUrlBarFocus(urlBar, true);
 
         setTextAndVerifyNoAutocomplete(urlBar, "test");
         setAutocomplete(urlBar, "test", "ing is fun");
 
         ThreadUtils.runOnUiThreadBlocking(new Runnable() {
@@ skipping 17 matching lines @@
         // Ensure that after batch mode has ended that the autocomplete is cleared due to the
         // invalid selection range.
         state = getAutocompleteState(urlBar, null);
         assertFalse(state.hasAutocomplete);
         assertEquals("test", state.textWithoutAutocomplete);
         assertEquals("test", state.textWithAutocomplete);
     }
 
     @SmallTest
     @Feature({"Omnibox"})
+    @RetryOnFailure
     public void testBatchModeChangesTriggerCorrectSuggestions() throws InterruptedException {
         startMainActivityOnBlankPage();
 
         final AtomicReference<String> requestedAutocompleteText = new AtomicReference<String>();
         final StubAutocompleteController controller = new StubAutocompleteController() {
             @Override
             public void start(Profile profile, String url, String text, int cursorPosition,
                     boolean preventInlineAutocomplete) {
                 requestedAutocompleteText.set(text);
             }
@@ skipping 94 matching lines @@
                 urlBar.setText("googl");
                 urlBar.setSelection(5);
                 urlBar.endBatchEdit();
             }
         });
         assertFalse(state.hasAutocomplete);
     }
 
     @SmallTest
     @Feature("Omnibox")
+    @RetryOnFailure
     public void testAutocompleteSpanClearedOnNonMatchingCommitText() throws InterruptedException {
         startMainActivityOnBlankPage();
 
         stubLocationBarAutocomplete();
 
         final UrlBar urlBar = getUrlBar();
         OmniboxTestUtils.toggleUrlBarFocus(urlBar, true);
         OmniboxTestUtils.waitForFocusAndKeyboardActive(urlBar, true);
 
         setTextAndVerifyNoAutocomplete(urlBar, "a");
@@ skipping 12 matching lines @@
         CriteriaHelper.pollUiThread(Criteria.equals("al", new Callable<String>() {
             @Override
             public String call() {
                 return urlBar.getText().toString();
             }
         }));
     }
 
     @SmallTest
     @Feature({"Omnibox"})
+    @RetryOnFailure
     public void testAutocompleteUpdatedOnDefocus() throws InterruptedException {
         startMainActivityOnBlankPage();
         stubLocationBarAutocomplete();
         final UrlBar urlBar = getUrlBar();
         OmniboxTestUtils.toggleUrlBarFocus(urlBar, true);
 
         // Verify that defocusing the UrlBar clears the autocomplete.
         setTextAndVerifyNoAutocomplete(urlBar, "test");
         setAutocomplete(urlBar, "test", "ing is fun");
         OmniboxTestUtils.toggleUrlBarFocus(urlBar, false);
         AutocompleteState state = getAutocompleteState(urlBar, null);
         assertFalse(state.hasAutocomplete);
     }
 
     @SmallTest
     @Feature({"Omnibox"})
+    @RetryOnFailure
     public void testAutocompleteClearedOnComposition()
             throws InterruptedException, ExecutionException {
         startMainActivityOnBlankPage();
         stubLocationBarAutocomplete();
         final UrlBar urlBar = getUrlBar();
         OmniboxTestUtils.toggleUrlBarFocus(urlBar, true);
         OmniboxTestUtils.waitForFocusAndKeyboardActive(urlBar, true);
 
         setTextAndVerifyNoAutocomplete(urlBar, "test");
         setAutocomplete(urlBar, "test", "ing is fun");
 
         assertNotNull(urlBar.mInputConnection);
         AutocompleteState state = getAutocompleteState(urlBar, new Runnable() {
             @Override
             public void run() {
                 urlBar.mInputConnection.setComposingText("ing compose", 4);
             }
         });
         assertFalse(state.hasAutocomplete);
 
         Editable urlText = getUrlBarText(urlBar);
         assertEquals("testing compose", urlText.toString());
         // TODO(tedchoc): Investigate why this fails on x86.
         //assertEquals(4, BaseInputConnection.getComposingSpanStart(urlText));
         //assertEquals(15, BaseInputConnection.getComposingSpanEnd(urlText));
     }
 
     @SmallTest
     @Feature("Omnibox")
+    @RetryOnFailure
     public void testDelayedCompositionCorrectedWithAutocomplete()
             throws InterruptedException, ExecutionException {
         startMainActivityOnBlankPage();
         stubLocationBarAutocomplete();
 
         final UrlBar urlBar = getUrlBar();
         OmniboxTestUtils.toggleUrlBarFocus(urlBar, true);
         OmniboxTestUtils.waitForFocusAndKeyboardActive(urlBar, true);
 
         assertNotNull(urlBar.mInputConnection);
@@ skipping 94 matching lines @@
         assertEquals(BaseInputConnection.getComposingSpanStart(urlText), 10);
         assertEquals(BaseInputConnection.getComposingSpanEnd(urlText), 22);
     }
 
     /**
      * Test to verify the omnibox can take focus during startup before native libraries have
      * loaded.
      */
     @SmallTest
     @Feature({"Omnibox"})
+    @RetryOnFailure
     public void testFocusingOnStartup() throws InterruptedException {
         Intent intent = new Intent(Intent.ACTION_MAIN);
         intent.addCategory(Intent.CATEGORY_LAUNCHER);
         prepareUrlIntent(intent, "about:blank");
         startActivityCompletely(intent);
 
         UrlBar urlBar = getUrlBar();
         assertNotNull(urlBar);
         OmniboxTestUtils.toggleUrlBarFocus(urlBar, true);
         OmniboxTestUtils.waitForFocusAndKeyboardActive(urlBar, true);
     }
 
     @SmallTest
     @Feature({"Omnibox"})
+    @RetryOnFailure
     public void testCopyHuge() throws InterruptedException {
         startMainActivityWithURL(HUGE_URL);
         OmniboxTestUtils.toggleUrlBarFocus(getUrlBar(), true);
         assertEquals(HUGE_URL, copyUrlToClipboard(android.R.id.copy));
     }
 
     @SmallTest
     @Feature({"Omnibox"})
     public void testCutHuge() throws InterruptedException {
         startMainActivityWithURL(HUGE_URL);
@@ skipping 23 matching lines @@
         CharSequence text = (clip != null && clip.getItemCount() != 0)
                 ? clip.getItemAt(0).getText() : null;
         return text != null ? text.toString() : null;
     }
 
     @Override
     public void startMainActivity() throws InterruptedException {
         // Each test will start the activity.
     }
 }