| 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 // Disable everything on windows only. http://crbug.com/306144 | 5 // Disable everything on windows only. http://crbug.com/306144 |
| 6 #ifndef OS_WIN | 6 #ifndef OS_WIN |
| 7 | 7 |
| 8 #include <algorithm> | 8 #include <algorithm> |
| 9 | 9 |
| 10 #include "base/file_util.h" | 10 #include "base/file_util.h" |
| (...skipping 45 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 56 #include "webkit/browser/fileapi/file_system_url.h" | 56 #include "webkit/browser/fileapi/file_system_url.h" |
| 57 | 57 |
| 58 using content::BrowserContext; | 58 using content::BrowserContext; |
| 59 using content::BrowserThread; | 59 using content::BrowserThread; |
| 60 using content::DownloadItem; | 60 using content::DownloadItem; |
| 61 using content::DownloadManager; | 61 using content::DownloadManager; |
| 62 using content::URLRequestSlowDownloadJob; | 62 using content::URLRequestSlowDownloadJob; |
| 63 | 63 |
| 64 namespace errors = download_extension_errors; | 64 namespace errors = download_extension_errors; |
| 65 | 65 |
| 66 namespace downloads = extensions::api::downloads; | |
| 67 | |
| 68 namespace extensions { | 66 namespace extensions { |
| 67 namespace downloads = api::downloads; |
| 69 | 68 |
| 70 namespace { | 69 namespace { |
| 71 | 70 |
| 72 // Comparator that orders download items by their ID. Can be used with | 71 // Comparator that orders download items by their ID. Can be used with |
| 73 // std::sort. | 72 // std::sort. |
| 74 struct DownloadIdComparator { | 73 struct DownloadIdComparator { |
| 75 bool operator() (DownloadItem* first, DownloadItem* second) { | 74 bool operator() (DownloadItem* first, DownloadItem* second) { |
| 76 return first->GetId() < second->GetId(); | 75 return first->GetId() < second->GetId(); |
| 77 } | 76 } |
| 78 }; | 77 }; |
| (...skipping 190 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 269 void LoadExtension(const char* name) { | 268 void LoadExtension(const char* name) { |
| 270 // Store the created Extension object so that we can attach it to | 269 // Store the created Extension object so that we can attach it to |
| 271 // ExtensionFunctions. Also load the extension in incognito profiles for | 270 // ExtensionFunctions. Also load the extension in incognito profiles for |
| 272 // testing incognito. | 271 // testing incognito. |
| 273 extension_ = LoadExtensionIncognito(test_data_dir_.AppendASCII(name)); | 272 extension_ = LoadExtensionIncognito(test_data_dir_.AppendASCII(name)); |
| 274 CHECK(extension_); | 273 CHECK(extension_); |
| 275 content::WebContents* tab = chrome::AddSelectedTabWithURL( | 274 content::WebContents* tab = chrome::AddSelectedTabWithURL( |
| 276 current_browser(), | 275 current_browser(), |
| 277 extension_->GetResourceURL("empty.html"), | 276 extension_->GetResourceURL("empty.html"), |
| 278 content::PAGE_TRANSITION_LINK); | 277 content::PAGE_TRANSITION_LINK); |
| 279 extensions::EventRouter::Get(current_browser()->profile()) | 278 EventRouter::Get(current_browser()->profile()) |
| 280 ->AddEventListener(downloads::OnCreated::kEventName, | 279 ->AddEventListener(downloads::OnCreated::kEventName, |
| 281 tab->GetRenderProcessHost(), | 280 tab->GetRenderProcessHost(), |
| 282 GetExtensionId()); | 281 GetExtensionId()); |
| 283 extensions::EventRouter::Get(current_browser()->profile()) | 282 EventRouter::Get(current_browser()->profile()) |
| 284 ->AddEventListener(downloads::OnChanged::kEventName, | 283 ->AddEventListener(downloads::OnChanged::kEventName, |
| 285 tab->GetRenderProcessHost(), | 284 tab->GetRenderProcessHost(), |
| 286 GetExtensionId()); | 285 GetExtensionId()); |
| 287 extensions::EventRouter::Get(current_browser()->profile()) | 286 EventRouter::Get(current_browser()->profile()) |
| 288 ->AddEventListener(downloads::OnErased::kEventName, | 287 ->AddEventListener(downloads::OnErased::kEventName, |
| 289 tab->GetRenderProcessHost(), | 288 tab->GetRenderProcessHost(), |
| 290 GetExtensionId()); | 289 GetExtensionId()); |
| 291 } | 290 } |
| 292 | 291 |
| 293 content::RenderProcessHost* AddFilenameDeterminer() { | 292 content::RenderProcessHost* AddFilenameDeterminer() { |
| 294 ExtensionDownloadsEventRouter::SetDetermineFilenameTimeoutSecondsForTesting( | 293 ExtensionDownloadsEventRouter::SetDetermineFilenameTimeoutSecondsForTesting( |
| 295 2); | 294 2); |
| 296 content::WebContents* tab = chrome::AddSelectedTabWithURL( | 295 content::WebContents* tab = chrome::AddSelectedTabWithURL( |
| 297 current_browser(), | 296 current_browser(), |
| 298 extension_->GetResourceURL("empty.html"), | 297 extension_->GetResourceURL("empty.html"), |
| 299 content::PAGE_TRANSITION_LINK); | 298 content::PAGE_TRANSITION_LINK); |
| 300 extensions::ExtensionSystem::Get(current_browser()->profile()) | 299 EventRouter::Get(current_browser()->profile()) |
| 301 ->event_router() | |
| 302 ->AddEventListener(downloads::OnDeterminingFilename::kEventName, | 300 ->AddEventListener(downloads::OnDeterminingFilename::kEventName, |
| 303 tab->GetRenderProcessHost(), | 301 tab->GetRenderProcessHost(), |
| 304 GetExtensionId()); | 302 GetExtensionId()); |
| 305 return tab->GetRenderProcessHost(); | 303 return tab->GetRenderProcessHost(); |
| 306 } | 304 } |
| 307 | 305 |
| 308 void RemoveFilenameDeterminer(content::RenderProcessHost* host) { | 306 void RemoveFilenameDeterminer(content::RenderProcessHost* host) { |
| 309 extensions::ExtensionSystem::Get(current_browser()->profile()) | 307 EventRouter::Get(current_browser()->profile())->RemoveEventListener( |
| 310 ->event_router() | 308 downloads::OnDeterminingFilename::kEventName, host, GetExtensionId()); |
| 311 ->RemoveEventListener(downloads::OnDeterminingFilename::kEventName, | |
| 312 host, | |
| 313 GetExtensionId()); | |
| 314 } | 309 } |
| 315 | 310 |
| 316 Browser* current_browser() { return current_browser_; } | 311 Browser* current_browser() { return current_browser_; } |
| 317 | 312 |
| 318 // InProcessBrowserTest | 313 // InProcessBrowserTest |
| 319 virtual void SetUpOnMainThread() OVERRIDE { | 314 virtual void SetUpOnMainThread() OVERRIDE { |
| 320 ExtensionApiTest::SetUpOnMainThread(); | 315 ExtensionApiTest::SetUpOnMainThread(); |
| 321 BrowserThread::PostTask( | 316 BrowserThread::PostTask( |
| 322 BrowserThread::IO, FROM_HERE, | 317 BrowserThread::IO, FROM_HERE, |
| 323 base::Bind(&chrome_browser_net::SetUrlRequestMocksEnabled, true)); | 318 base::Bind(&chrome_browser_net::SetUrlRequestMocksEnabled, true)); |
| (...skipping 272 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 596 } | 591 } |
| 597 | 592 |
| 598 void CreateAndSetDownloadsDirectory() { | 593 void CreateAndSetDownloadsDirectory() { |
| 599 ASSERT_TRUE(downloads_directory_.CreateUniqueTempDir()); | 594 ASSERT_TRUE(downloads_directory_.CreateUniqueTempDir()); |
| 600 current_browser()->profile()->GetPrefs()->SetFilePath( | 595 current_browser()->profile()->GetPrefs()->SetFilePath( |
| 601 prefs::kDownloadDefaultDirectory, | 596 prefs::kDownloadDefaultDirectory, |
| 602 downloads_directory_.path()); | 597 downloads_directory_.path()); |
| 603 } | 598 } |
| 604 | 599 |
| 605 base::ScopedTempDir downloads_directory_; | 600 base::ScopedTempDir downloads_directory_; |
| 606 const extensions::Extension* extension_; | 601 const Extension* extension_; |
| 607 Browser* incognito_browser_; | 602 Browser* incognito_browser_; |
| 608 Browser* current_browser_; | 603 Browser* current_browser_; |
| 609 scoped_ptr<DownloadsEventsListener> events_listener_; | 604 scoped_ptr<DownloadsEventsListener> events_listener_; |
| 610 | 605 |
| 611 DISALLOW_COPY_AND_ASSIGN(DownloadExtensionTest); | 606 DISALLOW_COPY_AND_ASSIGN(DownloadExtensionTest); |
| 612 }; | 607 }; |
| 613 | 608 |
| 614 class MockIconExtractorImpl : public DownloadFileIconExtractor { | 609 class MockIconExtractorImpl : public DownloadFileIconExtractor { |
| 615 public: | 610 public: |
| 616 MockIconExtractorImpl(const base::FilePath& path, | 611 MockIconExtractorImpl(const base::FilePath& path, |
| (...skipping 3414 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 4031 #include "content/public/browser/download_interrupt_reason_values.h" | 4026 #include "content/public/browser/download_interrupt_reason_values.h" |
| 4032 #undef INTERRUPT_REASON | 4027 #undef INTERRUPT_REASON |
| 4033 } | 4028 } |
| 4034 | 4029 |
| 4035 TEST(ExtensionDetermineDownloadFilenameInternal, | 4030 TEST(ExtensionDetermineDownloadFilenameInternal, |
| 4036 ExtensionDetermineDownloadFilenameInternal) { | 4031 ExtensionDetermineDownloadFilenameInternal) { |
| 4037 std::string winner_id; | 4032 std::string winner_id; |
| 4038 base::FilePath filename; | 4033 base::FilePath filename; |
| 4039 downloads::FilenameConflictAction conflict_action = | 4034 downloads::FilenameConflictAction conflict_action = |
| 4040 downloads::FILENAME_CONFLICT_ACTION_UNIQUIFY; | 4035 downloads::FILENAME_CONFLICT_ACTION_UNIQUIFY; |
| 4041 extensions::ExtensionWarningSet warnings; | 4036 ExtensionWarningSet warnings; |
| 4042 | 4037 |
| 4043 // Empty incumbent determiner | 4038 // Empty incumbent determiner |
| 4044 warnings.clear(); | 4039 warnings.clear(); |
| 4045 ExtensionDownloadsEventRouter::DetermineFilenameInternal( | 4040 ExtensionDownloadsEventRouter::DetermineFilenameInternal( |
| 4046 base::FilePath(FILE_PATH_LITERAL("a")), | 4041 base::FilePath(FILE_PATH_LITERAL("a")), |
| 4047 downloads::FILENAME_CONFLICT_ACTION_OVERWRITE, | 4042 downloads::FILENAME_CONFLICT_ACTION_OVERWRITE, |
| 4048 "suggester", | 4043 "suggester", |
| 4049 base::Time::Now(), | 4044 base::Time::Now(), |
| 4050 "", | 4045 "", |
| 4051 base::Time(), | 4046 base::Time(), |
| (...skipping 16 matching lines...) Expand all Loading... |
| 4068 "incumbent", | 4063 "incumbent", |
| 4069 base::Time::Now(), | 4064 base::Time::Now(), |
| 4070 &winner_id, | 4065 &winner_id, |
| 4071 &filename, | 4066 &filename, |
| 4072 &conflict_action, | 4067 &conflict_action, |
| 4073 &warnings); | 4068 &warnings); |
| 4074 EXPECT_EQ("incumbent", winner_id); | 4069 EXPECT_EQ("incumbent", winner_id); |
| 4075 EXPECT_EQ(FILE_PATH_LITERAL("a"), filename.value()); | 4070 EXPECT_EQ(FILE_PATH_LITERAL("a"), filename.value()); |
| 4076 EXPECT_EQ(downloads::FILENAME_CONFLICT_ACTION_OVERWRITE, conflict_action); | 4071 EXPECT_EQ(downloads::FILENAME_CONFLICT_ACTION_OVERWRITE, conflict_action); |
| 4077 EXPECT_FALSE(warnings.empty()); | 4072 EXPECT_FALSE(warnings.empty()); |
| 4078 EXPECT_EQ(extensions::ExtensionWarning::kDownloadFilenameConflict, | 4073 EXPECT_EQ(ExtensionWarning::kDownloadFilenameConflict, |
| 4079 warnings.begin()->warning_type()); | 4074 warnings.begin()->warning_type()); |
| 4080 EXPECT_EQ("suggester", warnings.begin()->extension_id()); | 4075 EXPECT_EQ("suggester", warnings.begin()->extension_id()); |
| 4081 | 4076 |
| 4082 // Suggester wins | 4077 // Suggester wins |
| 4083 warnings.clear(); | 4078 warnings.clear(); |
| 4084 ExtensionDownloadsEventRouter::DetermineFilenameInternal( | 4079 ExtensionDownloadsEventRouter::DetermineFilenameInternal( |
| 4085 base::FilePath(FILE_PATH_LITERAL("b")), | 4080 base::FilePath(FILE_PATH_LITERAL("b")), |
| 4086 downloads::FILENAME_CONFLICT_ACTION_PROMPT, | 4081 downloads::FILENAME_CONFLICT_ACTION_PROMPT, |
| 4087 "suggester", | 4082 "suggester", |
| 4088 base::Time::Now(), | 4083 base::Time::Now(), |
| 4089 "incumbent", | 4084 "incumbent", |
| 4090 base::Time::Now() - base::TimeDelta::FromDays(1), | 4085 base::Time::Now() - base::TimeDelta::FromDays(1), |
| 4091 &winner_id, | 4086 &winner_id, |
| 4092 &filename, | 4087 &filename, |
| 4093 &conflict_action, | 4088 &conflict_action, |
| 4094 &warnings); | 4089 &warnings); |
| 4095 EXPECT_EQ("suggester", winner_id); | 4090 EXPECT_EQ("suggester", winner_id); |
| 4096 EXPECT_EQ(FILE_PATH_LITERAL("b"), filename.value()); | 4091 EXPECT_EQ(FILE_PATH_LITERAL("b"), filename.value()); |
| 4097 EXPECT_EQ(downloads::FILENAME_CONFLICT_ACTION_PROMPT, conflict_action); | 4092 EXPECT_EQ(downloads::FILENAME_CONFLICT_ACTION_PROMPT, conflict_action); |
| 4098 EXPECT_FALSE(warnings.empty()); | 4093 EXPECT_FALSE(warnings.empty()); |
| 4099 EXPECT_EQ(extensions::ExtensionWarning::kDownloadFilenameConflict, | 4094 EXPECT_EQ(ExtensionWarning::kDownloadFilenameConflict, |
| 4100 warnings.begin()->warning_type()); | 4095 warnings.begin()->warning_type()); |
| 4101 EXPECT_EQ("incumbent", warnings.begin()->extension_id()); | 4096 EXPECT_EQ("incumbent", warnings.begin()->extension_id()); |
| 4102 } | 4097 } |
| 4103 | 4098 |
| 4104 } // namespace extensions | 4099 } // namespace extensions |
| 4105 | 4100 |
| 4106 #endif // http://crbug.com/3061144 | 4101 #endif // http://crbug.com/3061144 |
| OLD | NEW |