Use a check item instead of changing its text.

chrome/browser/tab_contents/spelling_menu_observer_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 "chrome/browser/tab_contents/spelling_menu_observer.h"
 
 #include <vector>
 
 #include "base/utf_string_conversions.h"
 #include "chrome/app/chrome_command_ids.h"
+#include "chrome/browser/prefs/pref_service.h"
 #include "chrome/browser/tab_contents/render_view_context_menu.h"
 #include "chrome/browser/tab_contents/render_view_context_menu_observer.h"
+#include "chrome/common/pref_names.h"
 #include "chrome/test/base/in_process_browser_test.h"
 #include "chrome/test/base/testing_profile.h"
 
 namespace {
 
 // A mock context menu used in this test. This class overrides virtual methods
 // derived from the RenderViewContextMenuProxy class to monitor calls from the
 // SpellingMenuObserver class.
 class MockRenderViewContextMenu : public RenderViewContextMenuProxy {
  public:
   // A menu item used in this test. This test uses a vector of this struct to
   // hold menu items added by this test.
   struct MockMenuItem {
+    MockMenuItem()
+        : command_id(0),
+          enabled(false),
+          checked(false),
+          hidden(true) {
+    }
     int command_id;
     bool enabled;
+    bool checked;
     bool hidden;
     string16 title;
   };
 
   MockRenderViewContextMenu();
   virtual ~MockRenderViewContextMenu();
 
   // RenderViewContextMenuProxy implementation.
   virtual void AddMenuItem(int command_id, const string16& title) OVERRIDE;
+  virtual void AddCheckItem(int command_id, const string16& title) OVERRIDE;
   virtual void AddSeparator() OVERRIDE;
   virtual void AddSubMenu(int command_id,
                           const string16& label,
                           ui::MenuModel* model) OVERRIDE;
   virtual void UpdateMenuItem(int command_id,
                               bool enabled,
                               bool hidden,
                               const string16& title) OVERRIDE;
   virtual RenderViewHost* GetRenderViewHost() const OVERRIDE;
   virtual Profile* GetProfile() const OVERRIDE;
@@ skipping 31 matching lines @@
 }
 
 MockRenderViewContextMenu::~MockRenderViewContextMenu() {
 }
 
 void MockRenderViewContextMenu::AddMenuItem(int command_id,
                                             const string16& title) {
   MockMenuItem item;
   item.command_id = command_id;
   item.enabled = observer_->IsCommandIdEnabled(command_id);
+  item.checked = false;
+  item.hidden = false;
+  item.title = title;
+  items_.push_back(item);
+}
+
+void MockRenderViewContextMenu::AddCheckItem(int command_id,
+                                             const string16& title) {
+  MockMenuItem item;
+  item.command_id = command_id;
+  item.enabled = observer_->IsCommandIdEnabled(command_id);
+  item.checked = observer_->IsCommandIdChecked(command_id);
   item.hidden = false;
   item.title = title;
   items_.push_back(item);
 }
 
 void MockRenderViewContextMenu::AddSeparator() {
   MockMenuItem item;
   item.command_id = -1;
   item.enabled = false;
+  item.checked = false;
   item.hidden = false;
   items_.push_back(item);
 }
 
 void MockRenderViewContextMenu::AddSubMenu(int command_id,
                                            const string16& label,
                                            ui::MenuModel* model) {
   MockMenuItem item;
   item.command_id = -1;
   item.enabled = false;
+  item.checked = false;
   item.hidden = false;
   items_.push_back(item);
 }
 
 void MockRenderViewContextMenu::UpdateMenuItem(int command_id,
                                                bool enabled,
                                                bool hidden,
                                                const string16& title) {
   for (std::vector<MockMenuItem>::iterator it = items_.begin();
        it != items_.end(); ++it) {
@@ skipping 21 matching lines @@
 size_t MockRenderViewContextMenu::GetMenuSize() const {
   return items_.size();
 }
 
 bool MockRenderViewContextMenu::GetMenuItem(size_t i,
                                             MockMenuItem* item) const {
   if (i >= items_.size())
     return false;
   item->command_id = items_[i].command_id;
   item->enabled = items_[i].enabled;
+  item->checked = items_[i].checked;
   item->hidden = items_[i].hidden;
   item->title = items_[i].title;
   return true;
 }
 
 void MockRenderViewContextMenu::SetObserver(
     RenderViewContextMenuObserver* observer) {
   observer_ = observer;
 }
 
@@ skipping 57 matching lines @@
   EXPECT_EQ(IDC_CONTENT_CONTEXT_NO_SPELLING_SUGGESTIONS, item.command_id);
   EXPECT_FALSE(item.enabled);
   EXPECT_FALSE(item.hidden);
   menu->GetMenuItem(1, &item);
   EXPECT_EQ(IDC_SPELLCHECK_ADD_TO_DICTIONARY, item.command_id);
   EXPECT_TRUE(item.enabled);
   EXPECT_FALSE(item.hidden);
   menu->GetMenuItem(2, &item);
   EXPECT_EQ(IDC_CONTENT_CONTEXT_SPELLING_TOGGLE, item.command_id);
   EXPECT_TRUE(item.enabled);
+  EXPECT_FALSE(item.checked);
   EXPECT_FALSE(item.hidden);
   menu->GetMenuItem(3, &item);
   EXPECT_EQ(-1, item.command_id);
   EXPECT_FALSE(item.enabled);
   EXPECT_FALSE(item.hidden);
 }
+
+// Tests that right-clicking a misspelled word when we enable spelling-service
+// integration to verify an item "Ask Google for suggestions" is checked. (This
+// test does not actually send JSON-RPC requests to the service because it makes
+// this test flaky.)
+IN_PROC_BROWSER_TEST_F(SpellingMenuObserverTest, EnableSpellingService) {
+  scoped_ptr<MockRenderViewContextMenu> menu(new MockRenderViewContextMenu);
+  scoped_ptr<SpellingMenuObserver> observer(
+      new SpellingMenuObserver(menu.get()));
+  menu->SetObserver(observer.get());
+  menu->GetPrefs()->SetBoolean(prefs::kSpellCheckUseSpellingService, true);
+
+  content::ContextMenuParams params;
+  params.is_editable = true;
+  params.misspelled_word = ASCIIToUTF16("wiimode");
+  observer->InitMenu(params);
+  EXPECT_EQ(static_cast<size_t>(4), menu->GetMenuSize());
+
+  // To avoid duplicates, this test reads only the "Ask Google for suggestions"
+  // item and verifies it is enabled and checked.
+  MockRenderViewContextMenu::MockMenuItem item;
+  menu->GetMenuItem(2, &item);
+  EXPECT_EQ(IDC_CONTENT_CONTEXT_SPELLING_TOGGLE, item.command_id);
+  EXPECT_TRUE(item.enabled);
+  EXPECT_TRUE(item.checked);
+  EXPECT_FALSE(item.hidden);
+}