content/browser/download/save_package_browsertest.cc

   1 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "base/files/scoped_temp_dir.h"
   6 #include "content/browser/download/save_package.h"
   7 #include "content/public/test/content_browser_test.h"
   8 #include "content/public/test/content_browser_test_utils.h"
   9 #include "content/shell/browser/shell.h"
  10
  11 namespace content {
  12
  13 const char kTestFile[] = "files/simple_page.html";
  14
  15 class SavePackageBrowserTest : public ContentBrowserTest {
  16  protected:
  17   void SetUp() override {
  18     ASSERT_TRUE(save_dir_.CreateUniqueTempDir());
  19     ContentBrowserTest::SetUp();
  20   }
  21
  22   // Returns full paths of destination file and directory.
  23   void GetDestinationPaths(const std::string& prefix,
  24                            base::FilePath* full_file_name,
  25                            base::FilePath* dir) {
  26     *full_file_name = save_dir_.path().AppendASCII(prefix + ".htm");
  27     *dir = save_dir_.path().AppendASCII(prefix + "_files");
  28   }
  29
  30   // Temporary directory we will save pages to.
  31   base::ScopedTempDir save_dir_;
  32 };
  33
  34 // Create a SavePackage and delete it without calling Init.
  35 // SavePackage dtor has various asserts/checks that should not fire.
  36 IN_PROC_BROWSER_TEST_F(SavePackageBrowserTest, ImplicitCancel) {
  37   ASSERT_TRUE(test_server()->Start());
  38   GURL url = test_server()->GetURL(kTestFile);
  39   NavigateToURL(shell(), url);
  40   base::FilePath full_file_name, dir;
  41   GetDestinationPaths("a", &full_file_name, &dir);
  42   scoped_refptr<SavePackage> save_package(new SavePackage(
  43       shell()->web_contents(), SAVE_PAGE_TYPE_AS_ONLY_HTML, full_file_name,
  44       dir));
  45   ASSERT_TRUE(test_server()->Stop());
  46 }
  47
  48 // Create a SavePackage, call Cancel, then delete it.
  49 // SavePackage dtor has various asserts/checks that should not fire.
  50 IN_PROC_BROWSER_TEST_F(SavePackageBrowserTest, ExplicitCancel) {
  51   ASSERT_TRUE(test_server()->Start());
  52   GURL url = test_server()->GetURL(kTestFile);
  53   NavigateToURL(shell(), url);
  54   base::FilePath full_file_name, dir;
  55   GetDestinationPaths("a", &full_file_name, &dir);
  56   scoped_refptr<SavePackage> save_package(new SavePackage(
  57       shell()->web_contents(), SAVE_PAGE_TYPE_AS_ONLY_HTML, full_file_name,
  58       dir));
  59   save_package->Cancel(true);
  60   ASSERT_TRUE(test_server()->Stop());
  61 }
  62
  63 }  // namespace content