Mac: Omnibox: Retain the "SelectAll" state after a navigation.

chrome/browser/ui/omnibox/omnibox_view_browsertest.cc

 // Copyright (c) 2012 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.
 
 #include <stdio.h>
 
 #include "base/command_line.h"
 #include "base/scoped_observer.h"
 #include "base/strings/string16.h"
 #include "base/strings/string_util.h"
@@ skipping 1848 matching lines @@
   browser()->toolbar_model()->set_url_replacement_enabled(false);
   test_toolbar_model->set_text(url_b);
   omnibox_view->Update();
   EXPECT_EQ(url_a, omnibox_view->GetText());
   omnibox_view->RevertAll();
   EXPECT_EQ(url_b, omnibox_view->GetText());
   test_toolbar_model->set_text(url_c);
   omnibox_view->Update();
   EXPECT_EQ(url_c, omnibox_view->GetText());
 }
+
+// Test that if the Omnibox has focus, and had everything selected before a
+// non-user-initiated update, then it retains the selection after the update.
+IN_PROC_BROWSER_TEST_F(OmniboxViewTest, SelectAllStaysAfterUpdate) {
+  OmniboxView* omnibox_view = nullptr;
+  ASSERT_NO_FATAL_FAILURE(GetOmniboxView(&omnibox_view));
+  TestToolbarModel* test_toolbar_model = new TestToolbarModel;
+  scoped_ptr<ToolbarModel> toolbar_model(test_toolbar_model);
+  browser()->swap_toolbar_models(&toolbar_model);
+
+  base::string16 url_a(ASCIIToUTF16("http://www.a.com/"));
+  base::string16 url_b(ASCIIToUTF16("http://www.b.com/"));
+  chrome::FocusLocationBar(browser());
+
+  test_toolbar_model->set_text(url_a);
+  omnibox_view->Update();
+  EXPECT_EQ(url_a, omnibox_view->GetText());
+  EXPECT_TRUE(omnibox_view->IsSelectAll());
+
+  // Updating while selected should retain SelectAll().
+  test_toolbar_model->set_text(url_b);
+  omnibox_view->Update();
+  EXPECT_EQ(url_b, omnibox_view->GetText());
+  EXPECT_TRUE(omnibox_view->IsSelectAll());
+
+  // Select nothing, then switch back. Shouldn't gain a selection.
+  ASSERT_NO_FATAL_FAILURE(SendKey(ui::VKEY_RIGHT, 0));
+  test_toolbar_model->set_text(url_a);
+  omnibox_view->Update();
+  EXPECT_EQ(url_a, omnibox_view->GetText());
+  EXPECT_FALSE(omnibox_view->IsSelectAll());
+}