Tests for "Save Image As..." when Data Saver is enabled.

chrome/browser/renderer_context_menu/render_view_context_menu_unittest.cc

 // Copyright 2014 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/renderer_context_menu/render_view_context_menu.h"
 
+#include "base/prefs/pref_registry_simple.h"
 #include "base/prefs/pref_service.h"
+#include "base/run_loop.h"
 #include "base/strings/utf_string_conversions.h"
 #include "chrome/app/chrome_command_ids.h"
 #include "chrome/browser/custom_handlers/protocol_handler_registry.h"
 #include "chrome/browser/extensions/menu_manager.h"
 #include "chrome/browser/extensions/menu_manager_factory.h"
 #include "chrome/browser/extensions/test_extension_environment.h"
 #include "chrome/browser/extensions/test_extension_prefs.h"
+#include "chrome/browser/net/spdyproxy/data_reduction_proxy_chrome_settings.h"
+#include "chrome/browser/net/spdyproxy/data_reduction_proxy_chrome_settings_factory.h"
 #include "chrome/browser/prefs/incognito_mode_prefs.h"
 #include "chrome/browser/renderer_context_menu/render_view_context_menu_test_util.h"
+#include "chrome/common/pref_names.h"
 #include "chrome/test/base/chrome_render_view_host_test_harness.h"
 #include "chrome/test/base/testing_profile.h"
+#include "components/data_reduction_proxy/core/browser/data_reduction_proxy_test_utils.h"
+#include "components/data_reduction_proxy/core/common/data_reduction_proxy_params.h"
+#include "components/data_reduction_proxy/core/common/data_reduction_proxy_pref_names.h"
 #include "content/public/browser/web_contents.h"
+#include "content/public/test/web_contents_tester.h"
+
 #include "extensions/browser/extension_prefs.h"
 #include "extensions/common/url_pattern.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "third_party/WebKit/public/web/WebContextMenuData.h"
 #include "url/gurl.h"
 
 using extensions::Extension;
 using extensions::MenuItem;
 using extensions::MenuManager;
 using extensions::MenuManagerFactory;
@@ skipping 308 matching lines @@
     content::WebContents* wc = web_contents();
     TestRenderViewContextMenu* menu = new TestRenderViewContextMenu(
         wc->GetMainFrame(), params);
     // TestingProfile (returned by profile()) does not provide a protocol
     // registry.
     menu->protocol_handler_registry_ = registry_.get();
     menu->Init();
     return menu;
   }
 
+ protected:
+  scoped_ptr<data_reduction_proxy::DataReductionProxyTestContext>
+      drp_test_context_;
+
  private:
   scoped_ptr<ProtocolHandlerRegistry> registry_;
 };
 
 // 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());
 
@@ skipping 13 matching lines @@
 }
 
 // Make sure the checking custom command id that is not enabled will not
 // cause DCHECK failure.
 TEST_F(RenderViewContextMenuPrefsTest,
        IsCustomCommandIdEnabled) {
   scoped_ptr<TestRenderViewContextMenu> menu(CreateContextMenu());
 
   EXPECT_FALSE(menu->IsCommandIdEnabled(IDC_CONTENT_CONTEXT_CUSTOM_FIRST));
 }
+
+// Verify that request headers specify that data reduction proxy should return
+// the original non compressed resource when "Save Image As..." is used with
+// Data Saver enabled.
+TEST_F(RenderViewContextMenuPrefsTest, DataSaverEnabledSaveImageAs) {
+  drp_test_context_ =
+      data_reduction_proxy::DataReductionProxyTestContext::Builder()
+          .WithParamsFlags(
+               data_reduction_proxy::DataReductionProxyParams::kAllowed |
+               data_reduction_proxy::DataReductionProxyParams::
+                   kFallbackAllowed |
+               data_reduction_proxy::DataReductionProxyParams::kPromoAllowed)
+          .WithMockConfig()
+          .SkipSettingsInitialization()
+          .Build();
+
+  DataReductionProxyChromeSettings* settings =
+      DataReductionProxyChromeSettingsFactory::GetForBrowserContext(profile());
+
+  // TODO(bengr): Remove prefs::kProxy registration after M46. See
+  // http://crbug.com/445599.
+  PrefRegistrySimple* registry = drp_test_context_->pref_service()->registry();
+  registry->RegisterDictionaryPref(prefs::kProxy);
+
+  drp_test_context_->pref_service()->SetBoolean(
+      data_reduction_proxy::prefs::kDataReductionProxyEnabled, true);
+  drp_test_context_->InitSettings();
+  data_reduction_proxy::DataReductionProxyIOData* io_data =
+      drp_test_context_->io_data();
+
+  settings->InitDataReductionProxySettings(
+      io_data, drp_test_context_->pref_service(),
+      drp_test_context_->request_context_getter(),
+      base::MessageLoopProxy::current());
+
+  content::ContextMenuParams params = CreateParams(MenuItem::IMAGE);
+  params.unfiltered_link_url = params.link_url;
+  content::WebContents* wc = web_contents();
+  TestRenderViewContextMenu* menu =
+      new TestRenderViewContextMenu(wc->GetMainFrame(), params);
+
+  menu->ExecuteCommand(IDC_CONTENT_CONTEXT_SAVEIMAGEAS, 0);
+
+  const std::string& headers =
+      content::WebContentsTester::For(web_contents())->GetSaveFrameHeaders();
+  EXPECT_TRUE(headers.find("X-PSA-Client-Options: v=1,m=1") !=
+                  std::string::npos &&
+              headers.find("Cache-Control: no-cache") != std::string::npos);
+}
+
+// Verify that request headers do not specify pass through when "Save Image
+// As..." is used with Data Saver disabled.
+TEST_F(RenderViewContextMenuPrefsTest, DataSaverDisabledSaveImageAs) {
+  drp_test_context_ =
+      data_reduction_proxy::DataReductionProxyTestContext::Builder()
+          .WithParamsFlags(
+               data_reduction_proxy::DataReductionProxyParams::kAllowed |
+               data_reduction_proxy::DataReductionProxyParams::
+                   kFallbackAllowed |
+               data_reduction_proxy::DataReductionProxyParams::kPromoAllowed)
+          .WithMockConfig()
+          .SkipSettingsInitialization()
+          .Build();

[Avi (use Gerrit), 2015/05/11 20:52:29]

Can you abstract some of this duplicate code out to a helper function?

[Not at Google. Contact bengr, 2015/05/11 22:43:14]

Done.

+
+  DataReductionProxyChromeSettings* settings =
+      DataReductionProxyChromeSettingsFactory::GetForBrowserContext(profile());
+
+  // TODO(bengr): Remove prefs::kProxy registration after M46. See
+  // http://crbug.com/445599.
+  PrefRegistrySimple* registry = drp_test_context_->pref_service()->registry();
+  registry->RegisterDictionaryPref(prefs::kProxy);
+
+  drp_test_context_->pref_service()->SetBoolean(
+      data_reduction_proxy::prefs::kDataReductionProxyEnabled, false);
+  drp_test_context_->InitSettings();
+  data_reduction_proxy::DataReductionProxyIOData* io_data =
+      drp_test_context_->io_data();
+
+  settings->InitDataReductionProxySettings(
+      io_data, drp_test_context_->pref_service(),
+      drp_test_context_->request_context_getter(),
+      base::MessageLoopProxy::current());
+
+  content::ContextMenuParams params = CreateParams(MenuItem::IMAGE);
+  params.unfiltered_link_url = params.link_url;
+  content::WebContents* wc = web_contents();
+  TestRenderViewContextMenu* menu =
+      new TestRenderViewContextMenu(wc->GetMainFrame(), params);
+
+  menu->ExecuteCommand(IDC_CONTENT_CONTEXT_SAVEIMAGEAS, 0);
+
+  const std::string& headers =
+      content::WebContentsTester::For(web_contents())->GetSaveFrameHeaders();
+  EXPECT_TRUE(headers.find("X-PSA-Client-Options: v=1,m=1") ==
+                  std::string::npos &&
+              headers.find("Cache-Control: no-cache") == std::string::npos);
+}