Get rid of the EnableClose() infrastructure in Widget.

chrome/browser/ui/views/first_run_search_engine_view.h

 // Copyright (c) 2011 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.
 
 #ifndef CHROME_BROWSER_UI_VIEWS_FIRST_RUN_SEARCH_ENGINE_VIEW_H_
 #define CHROME_BROWSER_UI_VIEWS_FIRST_RUN_SEARCH_ENGINE_VIEW_H_
 #pragma once
 
 #include <vector>
 
@@ skipping 69 matching lines @@
  public:
   // |profile| allows us to get the set of imported search engines.
   // |randomize| is true if logos are to be displayed in random order.
   FirstRunSearchEngineView(Profile* profile, bool randomize);
 
   virtual ~FirstRunSearchEngineView();
 
   // Overridden from views::WidgetDelegateView:
   virtual string16 GetWindowTitle() const OVERRIDE;
   virtual views::View* GetContentsView() OVERRIDE { return this; }
+  virtual void WindowClosing() OVERRIDE;
 
   // Overridden from views::ButtonListener:
   virtual void ButtonPressed(views::Button* sender,
                              const views::Event& event) OVERRIDE;
 
   // Overridden from views::View:
   virtual gfx::Size GetPreferredSize() OVERRIDE;
   virtual void Layout() OVERRIDE;
   virtual void ViewHierarchyChanged(bool is_add,
                                     View* parent,
                                     View* child) OVERRIDE;
   virtual void GetAccessibleState(ui::AccessibleViewState* state) OVERRIDE;
 
   // Override from views::View so we can draw the gray background at dialog top.
   virtual void OnPaint(gfx::Canvas* canvas) OVERRIDE;
 
   // Overridden from TemplateURLServiceObserver. When the search engines have
   // loaded from the profile, we can populate the logos in the dialog box
   // to present to the user.
   virtual void OnTemplateURLServiceChanged() OVERRIDE;
 
+#if defined(UNIT_TEST)
+  void set_quit_on_closing(bool quit_on_closing) {
+    quit_on_closing_ = quit_on_closing;
+  }
+#endif
+
  private:
   // Once the TemplateURLService has loaded and we're in a View hierarchy, it's
   // OK to add the search engines from the TemplateURLService.
   void AddSearchEnginesIfPossible();
 
+  // Sets the default search engine to the one represented by |choice|.
+  void ChooseSearchEngine(SearchEngineChoice* choice);
+
   // One for each search engine choice offered, either three or four.
   std::vector<SearchEngineChoice*> search_engine_choices_;
 
   // If logos are to be displayed in random order. Used for UX testing.
   bool randomize_;
 
   // Services associated with the current profile.
   TemplateURLService* template_url_service_;
   ThemeService* theme_service_;
 
   bool text_direction_is_rtl_;
 
-  bool template_url_service_loaded_;
   bool added_to_view_hierarchy_;
 
   // Image of browser search box with grey background and bubble arrow.
   views::ImageView* background_image_;
 
   // UI elements:
   views::Label* title_label_;
   views::Label* text_label_;
 
+  // True when the user has chosen a particular search engine. Defaults to
+  // false. When the user closes the window without choosing a search engine,
+  // the engine specified by |fallback_choice_| is chosen.
+  bool user_chosen_engine_;
+
+  // The engine to choose when the user closes the window without explicitly
+  // making a selection. Because of randomization functionality, we cannot
+  // reliably deduce this from slot order, so this value is saved prior to
+  // randomization.
+  SearchEngineChoice* fallback_choice_;
+
+  // Defaults to true. Indicates that the current message loop should be quit
+  // when the window is closed. This is false in tests when this dialog does not
+  // spin its own message loop.
+  bool quit_on_closing_;
+
   DISALLOW_COPY_AND_ASSIGN(FirstRunSearchEngineView);
 };
 
 #endif  // CHROME_BROWSER_UI_VIEWS_FIRST_RUN_SEARCH_ENGINE_VIEW_H_