Added more tests for IncognitoModeAvailability.

chrome/browser/tab_contents/render_view_context_menu_unittest.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/render_view_context_menu.h"
 
+#include "chrome/app/chrome_command_ids.h"
+#include "chrome/browser/custom_handlers/protocol_handler_registry.h"
 #include "chrome/browser/extensions/extension_prefs.h"
+#include "chrome/browser/prefs/incognito_mode_prefs.h"
+#include "chrome/browser/prefs/pref_service.h"
+#include "chrome/browser/tab_contents/render_view_context_menu_test_util.h"
+#include "chrome/test/base/chrome_render_view_host_test_harness.h"
+#include "chrome/test/base/testing_profile.h"
+#include "content/public/browser/browser_thread.h"
+#include "content/public/browser/web_contents.h"
+#include "content/public/test/test_browser_thread.h"
 #include "extensions/common/url_pattern.h"
 #include "googleurl/src/gurl.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebContextMenuData.h"
 
 using extensions::MenuItem;
 using extensions::URLPatternSet;
 
 class RenderViewContextMenuTest : public testing::Test {
  public:
@@ skipping 212 matching lines @@
       MenuItem::SELECTION | MenuItem::IMAGE);
 
   MenuItem::ContextList contexts;
   contexts.Add(MenuItem::SELECTION);
   contexts.Add(MenuItem::IMAGE);
 
   URLPatternSet patterns = CreatePatternSet("*://test.none/*");
 
   EXPECT_TRUE(ExtensionContextAndPatternMatch(params, contexts, patterns));
 }
+
+class RenderViewContextMenuPrefsTest : public ChromeRenderViewHostTestHarness {
+ public:
+  RenderViewContextMenuPrefsTest()
+      : browser_thread_(content::BrowserThread::UI, &message_loop_),
+        registry_(profile(), NULL) {}
+
+  TestRenderViewContextMenu* CreateContextMenu() {
+    content::ContextMenuParams params = CreateParams(MenuItem::LINK);
+    params.unfiltered_link_url = params.link_url;
+    content::WebContents* wc = web_contents();
+    TestRenderViewContextMenu* menu = new TestRenderViewContextMenu(
+        wc, params);
+    // TestingProfile (returned by profile()) does not provide a protocol
+    // registry.
+    menu->protocol_handler_registry_ = &registry_;
+    menu->Init();
+    return menu;
+  }
+
+ private:
+  content::TestBrowserThread browser_thread_;
+  ProtocolHandlerRegistry registry_;
+
+  DISALLOW_COPY_AND_ASSIGN(RenderViewContextMenuPrefsTest);
+};
+
+// Verifies when Incognito Mode is not available (disabled by policy),
+// Open Link in Incognito Window link in the context menu is disabled.
+TEST_F(RenderViewContextMenuPrefsTest,
+       DisableOpenInIncognitoWindowWhenIncognitoIsDisabled) {
+  scoped_ptr<TestRenderViewContextMenu> menu(CreateContextMenu());
+
+  // Initially the Incognito mode is be enabled. So is the Open Link in
+  // Incognito Window link.
+  ASSERT_TRUE(menu->IsItemPresent(IDC_CONTENT_CONTEXT_OPENLINKOFFTHERECORD));
+  EXPECT_TRUE(
+      menu->IsCommandIdEnabled(IDC_CONTENT_CONTEXT_OPENLINKOFFTHERECORD));
+
+  // Disable Incognito mode.
+  IncognitoModePrefs::SetAvailability(profile()->GetPrefs(),
+                                      IncognitoModePrefs::DISABLED);
+  menu.reset(CreateContextMenu());
+  ASSERT_TRUE(menu->IsItemPresent(IDC_CONTENT_CONTEXT_OPENLINKOFFTHERECORD));
+  EXPECT_FALSE(
+      menu->IsCommandIdEnabled(IDC_CONTENT_CONTEXT_OPENLINKOFFTHERECORD));
+}