Fix 'Load image' context menu item

chrome/renderer/chrome_content_renderer_client_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/renderer/chrome_content_renderer_client.h"
 
 #include <stddef.h>
 
+#include <map>
+#include <string>
 #include <vector>
 
 #include "base/metrics/histogram_samples.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/test/histogram_tester.h"
 #include "build/build_config.h"
 #include "chrome/renderer/searchbox/search_bouncer.h"
+#include "components/data_reduction_proxy/core/common/data_reduction_proxy_headers.h"
 #include "content/public/common/webplugininfo.h"
 #include "testing/gtest/include/gtest/gtest.h"
+#include "third_party/WebKit/public/platform/WebString.h"
+#include "third_party/WebKit/public/platform/WebURLResponse.h"
 #include "url/gurl.h"
 
 #if defined(ENABLE_EXTENSIONS)
 #include "extensions/common/extension.h"
 #include "extensions/common/extension_builder.h"
 #include "extensions/common/manifest_constants.h"
 #endif
 
 #if !defined(DISABLE_NACL)
 #include "third_party/WebKit/public/platform/WebString.h"
@@ skipping 376 matching lines @@
   SearchBouncer::GetInstance()->OnSetSearchURLs(
       std::vector<GURL>(), GURL("http://example.com/n"));
   EXPECT_FALSE(client.ShouldSuppressErrorPage(nullptr,
                                               GURL("http://example.com")));
   EXPECT_TRUE(client.ShouldSuppressErrorPage(nullptr,
                                              GURL("http://example.com/n")));
   SearchBouncer::GetInstance()->OnSetSearchURLs(
       std::vector<GURL>(), GURL::EmptyGURL());
 }
 
+TEST_F(ChromeContentRendererClientTest, AddImageContextMenuProperties) {
+  ChromeContentRendererClient client;
+  blink::WebURLResponse web_url_response;
+  web_url_response.addHTTPHeaderField(
+      blink::WebString::fromUTF8(
+          data_reduction_proxy::chrome_proxy_content_transform_header()),
+      blink::WebString::fromUTF8(
+          data_reduction_proxy::empty_image_directive()));
+  std::map<std::string, std::string> properties;
+  client.AddImageContextMenuProperties(web_url_response, &properties);
+  EXPECT_EQ(
+      data_reduction_proxy::empty_image_directive(),
+      properties
+          [data_reduction_proxy::chrome_proxy_content_transform_header()]);
+}
+
 // These are tests that are common for both Android and desktop browsers.
 TEST_F(ChromeContentRendererClientTest, RewriteEmbedCommon) {
   struct TestData {
     std::string original;
     std::string expected;
   } test_data[] = {
       // { original, expected }
       {"youtube.com", ""},
       {"www.youtube.com", ""},
       {"http://www.youtube.com", ""},
@@ skipping 298 matching lines @@
   for (const auto& data : test_data) {
     ++total_count;
     GURL gurl = GURL(data);
     OverrideFlashEmbed(gurl);
     samples = GetHistogramSamples();
     EXPECT_EQ(total_count, samples->GetCount(internal::FAILURE_ENABLEJSAPI));
     EXPECT_EQ(total_count, samples->TotalCount());
   }
 }
 #endif