| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include <string> | 5 #include <string> |
| 6 | 6 |
| 7 #include "base/files/file_path.h" | 7 #include "base/files/file_path.h" |
| 8 #include "base/files/scoped_temp_dir.h" | 8 #include "base/files/scoped_temp_dir.h" |
| 9 #include "base/path_service.h" | 9 #include "base/path_service.h" |
| 10 #include "base/string_util.h" | 10 #include "base/string_util.h" |
| 11 #include "base/utf_string_conversions.h" | 11 #include "base/utf_string_conversions.h" |
| 12 #include "content/browser/browser_thread_impl.h" | |
| 13 #include "content/browser/download/save_package.h" | 12 #include "content/browser/download/save_package.h" |
| 14 #include "content/browser/renderer_host/test_render_view_host.h" | 13 #include "content/browser/renderer_host/test_render_view_host.h" |
| 15 #include "content/test/net/url_request_mock_http_job.h" | 14 #include "content/test/net/url_request_mock_http_job.h" |
| 16 #include "content/test/test_web_contents.h" | 15 #include "content/test/test_web_contents.h" |
| 17 #include "googleurl/src/gurl.h" | 16 #include "googleurl/src/gurl.h" |
| 18 #include "testing/gtest/include/gtest/gtest.h" | 17 #include "testing/gtest/include/gtest/gtest.h" |
| 19 | 18 |
| 20 namespace content { | 19 namespace content { |
| 21 | 20 |
| 22 #define FPL FILE_PATH_LITERAL | 21 #define FPL FILE_PATH_LITERAL |
| (...skipping 38 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 61 return false; | 60 return false; |
| 62 } | 61 } |
| 63 | 62 |
| 64 return true; | 63 return true; |
| 65 } | 64 } |
| 66 | 65 |
| 67 } // namespace | 66 } // namespace |
| 68 | 67 |
| 69 class SavePackageTest : public RenderViewHostImplTestHarness { | 68 class SavePackageTest : public RenderViewHostImplTestHarness { |
| 70 public: | 69 public: |
| 71 SavePackageTest() : browser_thread_(BrowserThread::UI, &message_loop_) { | |
| 72 } | |
| 73 | |
| 74 bool GetGeneratedFilename(bool need_success_generate_filename, | 70 bool GetGeneratedFilename(bool need_success_generate_filename, |
| 75 const std::string& disposition, | 71 const std::string& disposition, |
| 76 const std::string& url, | 72 const std::string& url, |
| 77 bool need_htm_ext, | 73 bool need_htm_ext, |
| 78 base::FilePath::StringType* generated_name) { | 74 base::FilePath::StringType* generated_name) { |
| 79 SavePackage* save_package; | 75 SavePackage* save_package; |
| 80 if (need_success_generate_filename) | 76 if (need_success_generate_filename) |
| 81 save_package = save_package_success_.get(); | 77 save_package = save_package_success_.get(); |
| 82 else | 78 else |
| 83 save_package = save_package_fail_.get(); | 79 save_package = save_package_fail_.get(); |
| (...skipping 32 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 116 long_file_name += long_file_name; | 112 long_file_name += long_file_name; |
| 117 long_file_name.resize( | 113 long_file_name.resize( |
| 118 kMaxFilePathLength - 9 - temp_dir_.path().value().length()); | 114 kMaxFilePathLength - 9 - temp_dir_.path().value().length()); |
| 119 | 115 |
| 120 save_package_fail_ = new SavePackage(contents(), | 116 save_package_fail_ = new SavePackage(contents(), |
| 121 temp_dir_.path().AppendASCII(long_file_name + HTML_EXTENSION), | 117 temp_dir_.path().AppendASCII(long_file_name + HTML_EXTENSION), |
| 122 temp_dir_.path().AppendASCII(long_file_name + "_files")); | 118 temp_dir_.path().AppendASCII(long_file_name + "_files")); |
| 123 } | 119 } |
| 124 | 120 |
| 125 private: | 121 private: |
| 126 BrowserThreadImpl browser_thread_; | |
| 127 | |
| 128 // SavePackage for successfully generating file name. | 122 // SavePackage for successfully generating file name. |
| 129 scoped_refptr<SavePackage> save_package_success_; | 123 scoped_refptr<SavePackage> save_package_success_; |
| 130 // SavePackage for failed generating file name. | 124 // SavePackage for failed generating file name. |
| 131 scoped_refptr<SavePackage> save_package_fail_; | 125 scoped_refptr<SavePackage> save_package_fail_; |
| 132 | 126 |
| 133 base::ScopedTempDir temp_dir_; | 127 base::ScopedTempDir temp_dir_; |
| 134 | |
| 135 DISALLOW_COPY_AND_ASSIGN(SavePackageTest); | |
| 136 }; | 128 }; |
| 137 | 129 |
| 138 static const struct { | 130 static const struct { |
| 139 const char* disposition; | 131 const char* disposition; |
| 140 const char* url; | 132 const char* url; |
| 141 const base::FilePath::CharType* expected_name; | 133 const base::FilePath::CharType* expected_name; |
| 142 bool need_htm_ext; | 134 bool need_htm_ext; |
| 143 } kGeneratedFiles[] = { | 135 } kGeneratedFiles[] = { |
| 144 // We mainly focus on testing duplicated names here, since retrieving file | 136 // We mainly focus on testing duplicated names here, since retrieving file |
| 145 // name from disposition and url has been tested in DownloadManagerTest. | 137 // name from disposition and url has been tested in DownloadManagerTest. |
| (...skipping 293 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 439 GURL view_source_url = URLRequestMockHTTPJob::GetMockViewSourceUrl( | 431 GURL view_source_url = URLRequestMockHTTPJob::GetMockViewSourceUrl( |
| 440 base::FilePath(kTestDir).Append(file_name)); | 432 base::FilePath(kTestDir).Append(file_name)); |
| 441 GURL actual_url = URLRequestMockHTTPJob::GetMockUrl( | 433 GURL actual_url = URLRequestMockHTTPJob::GetMockUrl( |
| 442 base::FilePath(kTestDir).Append(file_name)); | 434 base::FilePath(kTestDir).Append(file_name)); |
| 443 NavigateAndCommit(view_source_url); | 435 NavigateAndCommit(view_source_url); |
| 444 EXPECT_EQ(actual_url, GetUrlToBeSaved()); | 436 EXPECT_EQ(actual_url, GetUrlToBeSaved()); |
| 445 EXPECT_EQ(view_source_url, contents()->GetURL()); | 437 EXPECT_EQ(view_source_url, contents()->GetURL()); |
| 446 } | 438 } |
| 447 | 439 |
| 448 } // namespace content | 440 } // namespace content |
| OLD | NEW |