🔍 More SearchActivity tests

chrome/android/java/src/org/chromium/chrome/browser/locale/DefaultSearchEnginePromoDialog.java

 // Copyright 2017 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.locale;
 
 import android.content.Context;
 import android.content.DialogInterface;
 import android.os.Bundle;
 import android.support.annotation.Nullable;
 import android.widget.Button;
 
 import org.chromium.base.Callback;
+import org.chromium.base.VisibleForTesting;
 import org.chromium.base.library_loader.LibraryLoader;
 import org.chromium.chrome.R;
 import org.chromium.chrome.browser.locale.LocaleManager.SearchEnginePromoType;
 import org.chromium.chrome.browser.search_engines.TemplateUrlService;
 import org.chromium.chrome.browser.widget.PromoDialog;
 import org.chromium.chrome.browser.widget.RadioButtonLayout;
 
 /** A dialog that forces the user to choose a default search engine. */
 public class DefaultSearchEnginePromoDialog extends PromoDialog {
+    /** Notified about events happening to the dialog. */
+    public static interface DefaultSearchEnginePromoDialogObserver {
+        void onDialogShown(DefaultSearchEnginePromoDialog shownDialog);
+    }
+    private static DefaultSearchEnginePromoDialogObserver sObserver;
+
     /** Used to determine the promo dialog contents. */
     @SearchEnginePromoType
     private final int mDialogType;
 
     /** Called when the dialog is dismissed after the user has chosen a search engine. */
     private final Callback<Boolean> mOnDismissed;
 
     /** Encapsulates most of the logic for filling the dialog and handling clicks. */
     private DefaultSearchEngineDialogHelper mHelper;
 
@@ skipping 43 matching lines @@
     }
 
     @Override
     protected void onCreate(Bundle savedInstanceState) {
         super.onCreate(savedInstanceState);
 
         Button okButton = (Button) findViewById(R.id.button_primary);
         okButton.setEnabled(false);
 
         RadioButtonLayout radioButtons = new RadioButtonLayout(getContext());
+        radioButtons.setId(R.id.default_search_engine_dialog_options);
         addControl(radioButtons);
 
         Runnable dismissRunnable = new Runnable() {
             @Override
             public void run() {
                 dismiss();
             }
         };
         mHelper = new DefaultSearchEngineDialogHelper(
                 mDialogType, radioButtons, okButton, dismissRunnable);
     }
 
     @Override
+    public void show() {
+        super.show();
+        if (sObserver != null) sObserver.onDialogShown(this);
+    }
+
+    @Override
     public void onDismiss(DialogInterface dialog) {
         if (mHelper.getCurrentlySelectedKeyword() == null) {
             // This shouldn't happen, but in case it does, finish the Activity so that the user has
             // to respond to the dialog next time.
             if (getOwnerActivity() != null) getOwnerActivity().finish();
         }
 
         if (mOnDismissed != null) {
             mOnDismissed.onResult(mHelper.getCurrentlySelectedKeyword() != null);
         }
     }
+
+    /** See {@link #sObserver}. */
+    @VisibleForTesting
+    @Nullable
+    public static void setObserverForTests(DefaultSearchEnginePromoDialogObserver observer) {
+        sObserver = observer;
+    }
 }