Adding tests for WebContent.DownloadImage()

content/browser/web_contents/web_contents_impl_browsertest.cc

 // Copyright 2013 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 "base/macros.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/values.h"
 #include "build/build_config.h"
 #include "content/browser/frame_host/navigation_entry_impl.h"
 #include "content/browser/renderer_host/render_widget_host_impl.h"
@@ skipping 1149 matching lines @@
 
   // Check that pre-warmed process isn't used.
   EXPECT_NE(renderer_id, web_contents->GetRenderProcessHost()->GetID());
   EXPECT_EQ(1, web_contents->GetController().GetEntryCount());
   NavigationEntry* entry =
       web_contents->GetController().GetLastCommittedEntry();
   ASSERT_TRUE(entry);
   EXPECT_EQ(web_ui_url, entry->GetURL());
 }
 
+namespace {
+
+class DownloadImageObserver :
+      public base::RefCountedThreadSafe<DownloadImageObserver> {

[pkotwicz, 2016/07/04 17:17:01]

Why is DownloadImageObserver ref counted?

The test times out if the callback is not called

[Zhiqiang Zhang (Slow), 2016/07/04 19:34:27]

Oops, I forget using base::Unretained in base::Bind, so that require some
refcounting. Now problem solved.
Added loop_runner_->Quit() when unmatch.

+ public:
+  MOCK_METHOD5(OnFinishDownloadImage, void(
+      int id,
+      int status_code,
+      const GURL& image_url,
+      const std::vector<SkBitmap>& bitmap,
+      const std::vector<gfx::Size>& sizes));
+
+ private:
+  friend class base::RefCountedThreadSafe<DownloadImageObserver>;
+  ~DownloadImageObserver() = default;
+};
+
+void DownloadImageTestInternal(Shell* shell,
+                               const GURL &page_url,
+                               const GURL &image_url,
+                               int expected_http_status) {
+  using ::testing::_;
+  using ::testing::InvokeWithoutArgs;
+
+  // Set up everything.
+  scoped_refptr<DownloadImageObserver> download_image_observer =
+      new DownloadImageObserver();
+  scoped_refptr<MessageLoopRunner> loop_runner =
+      new MessageLoopRunner();
+
+  // Set up expectation and stub.
+  EXPECT_CALL(*download_image_observer,
+              OnFinishDownloadImage(_, expected_http_status, _, _, _))
+      .WillOnce(InvokeWithoutArgs([&]() {
+            loop_runner->Quit();
+          }));
+
+  // Load test URL and start download image.
+  shell->LoadURL(page_url);
+  shell->web_contents()->DownloadImage(
+      image_url, false, 1024, false,
+      base::Bind(&DownloadImageObserver::OnFinishDownloadImage,
+                 download_image_observer));
+
+  // Wait for response.
+  loop_runner->Run();
+}
+
+}  // anonymous namespace
+
+IN_PROC_BROWSER_TEST_F(WebContentsImplBrowserTest,
+                       DownloadImage_HttpPage_HttpImage) {
+  ASSERT_TRUE(embedded_test_server()->Start());
+
+  const GURL kPageUrl =
+      embedded_test_server()->GetURL("/simple_page.html");

[pkotwicz, 2016/07/04 17:17:01]

I don't think that the page URL affects whether the download is successful.

If the page URL is a file URL and image URL is a HTTP URL the download still
passes

I believe that WebContents::DownloadImage() can be used to download a URL
completely unrelated to the current page.

[Zhiqiang Zhang (Slow), 2016/07/04 19:34:27]

OK, so now I just test for cases: file/http page * file/http image. I was
thinking of testing cross-origin scenario but given this I don't need do that.

Though I'm a bit worried that does it violate security rules if we can fetch any
http url from a page? (maybe for downloadImage() is fine)

+  const GURL kImageUrl =
+      embedded_test_server()->GetURL("/image.jpg");
+
+  DownloadImageTestInternal(shell(), kPageUrl, kImageUrl, 200);
+}
+
+IN_PROC_BROWSER_TEST_F(WebContentsImplBrowserTest,
+                       DownloadImage_Deny_HttpPage_FileImage) {
+  ASSERT_TRUE(embedded_test_server()->Start());
+
+  const GURL kPageUrl =
+      embedded_test_server()->GetURL("/simple_page.html");
+  const GURL kImageUrl =
+      GetTestUrl("", "image.jpg");
+
+  DownloadImageTestInternal(shell(), kPageUrl, kImageUrl, 0);
+}
+
+IN_PROC_BROWSER_TEST_F(WebContentsImplBrowserTest,
+                       DownloadImage_Deny_FilePage_FileImage) {
+  const GURL kPageUrl = GetTestUrl("", "simple-page.html");
+  const GURL kImageUrl = GetTestUrl("", "image.jpg");
+
+  DownloadImageTestInternal(shell(), kPageUrl, kImageUrl, 0);
+}
+
 }  // namespace content