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
11namespace content {
12
13const char kTestFile[] = "files/simple_page.html";
14
15class SavePackageBrowserTest : public ContentBrowserTest {
16 protected:
17  virtual 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.
36IN_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.
50IN_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
64