| OLD | NEW |
| 1 // Copyright (c) 2006-2008 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2006-2008 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 "base/logging.h" | 5 #include "base/logging.h" |
| 6 #include "chrome/browser/download/save_package.h" | 6 #include "chrome/browser/download/save_package.h" |
| 7 #include "testing/gtest/include/gtest/gtest.h" | 7 #include "testing/gtest/include/gtest/gtest.h" |
| 8 | 8 |
| 9 // 0 1 2 3 4 5 6 | 9 // 0 1 2 3 4 5 6 |
| 10 // 0123456789012345678901234567890123456789012345678901234567890123456789 | 10 // 0123456789012345678901234567890123456789012345678901234567890123456789 |
| (...skipping 20 matching lines...) Expand all Loading... |
| 31 bool GetGeneratedFilename(bool need_success_generate_filename, | 31 bool GetGeneratedFilename(bool need_success_generate_filename, |
| 32 const std::string& disposition, | 32 const std::string& disposition, |
| 33 const std::wstring& url, | 33 const std::wstring& url, |
| 34 bool need_htm_ext, | 34 bool need_htm_ext, |
| 35 std::wstring* generated_name) { | 35 std::wstring* generated_name) { |
| 36 SavePackage* save_package; | 36 SavePackage* save_package; |
| 37 if (need_success_generate_filename) | 37 if (need_success_generate_filename) |
| 38 save_package = save_package_success_.get(); | 38 save_package = save_package_success_.get(); |
| 39 else | 39 else |
| 40 save_package = save_package_fail_.get(); | 40 save_package = save_package_fail_.get(); |
| 41 return save_package->GenerateFilename(disposition, url, need_htm_ext, | 41 return save_package->GenerateFilename(disposition, GURL(url), need_htm_ext, |
| 42 generated_name); | 42 generated_name); |
| 43 } | 43 } |
| 44 | 44 |
| 45 protected: | 45 protected: |
| 46 DISALLOW_EVIL_CONSTRUCTORS(SavePackageTest); | 46 DISALLOW_EVIL_CONSTRUCTORS(SavePackageTest); |
| 47 | 47 |
| 48 private: | 48 private: |
| 49 // SavePackage for successfully generating file name. | 49 // SavePackage for successfully generating file name. |
| 50 scoped_refptr<SavePackage> save_package_success_; | 50 scoped_refptr<SavePackage> save_package_success_; |
| 51 // SavePackage for failed generating file name. | 51 // SavePackage for failed generating file name. |
| (...skipping 93 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 145 | 145 |
| 146 TEST_F(SavePackageTest, TestSuccessfullyGenerateSavePackageFilename) { | 146 TEST_F(SavePackageTest, TestSuccessfullyGenerateSavePackageFilename) { |
| 147 for (int i = 0; i < arraysize(kGeneratedFiles); ++i) { | 147 for (int i = 0; i < arraysize(kGeneratedFiles); ++i) { |
| 148 std::wstring file_name; | 148 std::wstring file_name; |
| 149 bool ok = GetGeneratedFilename(true, | 149 bool ok = GetGeneratedFilename(true, |
| 150 kGeneratedFiles[i].disposition, | 150 kGeneratedFiles[i].disposition, |
| 151 kGeneratedFiles[i].url, | 151 kGeneratedFiles[i].url, |
| 152 kGeneratedFiles[i].need_htm_ext, | 152 kGeneratedFiles[i].need_htm_ext, |
| 153 &file_name); | 153 &file_name); |
| 154 ASSERT_TRUE(ok); | 154 ASSERT_TRUE(ok); |
| 155 EXPECT_EQ(file_name, kGeneratedFiles[i].expected_name); | 155 EXPECT_EQ(kGeneratedFiles[i].expected_name, file_name); |
| 156 } | 156 } |
| 157 } | 157 } |
| 158 | 158 |
| 159 TEST_F(SavePackageTest, TestUnSuccessfullyGenerateSavePackageFilename) { | 159 TEST_F(SavePackageTest, TestUnSuccessfullyGenerateSavePackageFilename) { |
| 160 for (int i = 0; i < arraysize(kGeneratedFiles); ++i) { | 160 for (int i = 0; i < arraysize(kGeneratedFiles); ++i) { |
| 161 std::wstring file_name; | 161 std::wstring file_name; |
| 162 bool ok = GetGeneratedFilename(false, | 162 bool ok = GetGeneratedFilename(false, |
| 163 kGeneratedFiles[i].disposition, | 163 kGeneratedFiles[i].disposition, |
| 164 kGeneratedFiles[i].url, | 164 kGeneratedFiles[i].url, |
| 165 kGeneratedFiles[i].need_htm_ext, | 165 kGeneratedFiles[i].need_htm_ext, |
| 166 &file_name); | 166 &file_name); |
| 167 ASSERT_FALSE(ok); | 167 ASSERT_FALSE(ok); |
| 168 } | 168 } |
| 169 } | 169 } |
| 170 | 170 |
| OLD | NEW |