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 "base/auto_reset.h" | 5 #include "base/auto_reset.h" |
6 #include "base/files/scoped_temp_dir.h" | 6 #include "base/files/scoped_temp_dir.h" |
7 #include "base/json/json_reader.h" | 7 #include "base/json/json_reader.h" |
8 #include "base/prefs/pref_service.h" | 8 #include "base/prefs/pref_service.h" |
9 #include "base/values.h" | 9 #include "base/values.h" |
10 #include "chrome/browser/download/download_item_model.h" | 10 #include "chrome/browser/download/download_item_model.h" |
(...skipping 43 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
54 public: | 54 public: |
55 explicit MockDownloadsDOMHandler(content::DownloadManager* dlm) | 55 explicit MockDownloadsDOMHandler(content::DownloadManager* dlm) |
56 : DownloadsDOMHandler(dlm), | 56 : DownloadsDOMHandler(dlm), |
57 waiting_list_(false), | 57 waiting_list_(false), |
58 waiting_updated_(false), | 58 waiting_updated_(false), |
59 manager_(nullptr) { | 59 manager_(nullptr) { |
60 } | 60 } |
61 ~MockDownloadsDOMHandler() override {} | 61 ~MockDownloadsDOMHandler() override {} |
62 | 62 |
63 base::ListValue* downloads_list() { return downloads_list_.get(); } | 63 base::ListValue* downloads_list() { return downloads_list_.get(); } |
64 base::ListValue* download_updated() { return download_updated_.get(); } | 64 base::DictionaryValue* download_updated() { return download_updated_.get(); } |
65 | 65 |
66 void WaitForDownloadsList() { | 66 void WaitForDownloadsList() { |
67 if (downloads_list_.get()) | 67 if (downloads_list_) |
68 return; | 68 return; |
69 base::AutoReset<bool> reset_waiting(&waiting_list_, true); | 69 base::AutoReset<bool> reset_waiting(&waiting_list_, true); |
70 content::RunMessageLoop(); | 70 content::RunMessageLoop(); |
71 } | 71 } |
72 | 72 |
73 void WaitForDownloadUpdated() { | 73 void WaitForDownloadUpdated() { |
74 if (download_updated_.get()) | 74 if (download_updated_) |
75 return; | 75 return; |
76 base::AutoReset<bool> reset_waiting(&waiting_updated_, true); | 76 base::AutoReset<bool> reset_waiting(&waiting_updated_, true); |
77 content::RunMessageLoop(); | 77 content::RunMessageLoop(); |
78 } | 78 } |
79 | 79 |
80 void ForceSendCurrentDownloads() { | 80 void ForceSendCurrentDownloads() { |
81 ScheduleSendCurrentDownloads(); | 81 ScheduleSendCurrentDownloads(); |
82 } | 82 } |
83 | 83 |
84 void reset_downloads_list() { downloads_list_.reset(); } | 84 void reset_downloads_list() { downloads_list_.reset(); } |
85 void reset_download_updated() { download_updated_.reset(); } | 85 void reset_download_updated() { download_updated_.reset(); } |
86 | 86 |
87 void set_manager(content::DownloadManager* manager) { manager_ = manager; } | 87 void set_manager(content::DownloadManager* manager) { manager_ = manager; } |
88 | 88 |
89 using DownloadsDOMHandler::FinalizeRemovals; | 89 using DownloadsDOMHandler::FinalizeRemovals; |
90 | 90 |
91 protected: | 91 protected: |
92 content::WebContents* GetWebUIWebContents() override { return NULL; } | 92 content::WebContents* GetWebUIWebContents() override { return NULL; } |
93 | 93 |
94 void CallDownloadsList(const base::ListValue& downloads) override { | 94 void CallUpdateAll(const base::ListValue& list) override { |
95 downloads_list_.reset(downloads.DeepCopy()); | 95 downloads_list_.reset(list.DeepCopy()); |
96 if (waiting_list_) { | 96 if (waiting_list_) { |
97 content::BrowserThread::PostTask(content::BrowserThread::UI, | 97 content::BrowserThread::PostTask(content::BrowserThread::UI, |
98 FROM_HERE, | 98 FROM_HERE, |
99 base::MessageLoop::QuitClosure()); | 99 base::MessageLoop::QuitClosure()); |
100 } | 100 } |
101 } | 101 } |
102 | 102 |
103 void CallDownloadUpdated(const base::ListValue& download) override { | 103 void CallUpdateItem(const base::DictionaryValue& item) override { |
104 download_updated_.reset(download.DeepCopy()); | 104 download_updated_.reset(item.DeepCopy()); |
105 if (waiting_updated_) { | 105 if (waiting_updated_) { |
106 content::BrowserThread::PostTask(content::BrowserThread::UI, | 106 content::BrowserThread::PostTask(content::BrowserThread::UI, |
107 FROM_HERE, | 107 FROM_HERE, |
108 base::MessageLoop::QuitClosure()); | 108 base::MessageLoop::QuitClosure()); |
109 } | 109 } |
110 } | 110 } |
111 | 111 |
112 content::DownloadManager* GetMainNotifierManager() override { | 112 content::DownloadManager* GetMainNotifierManager() override { |
113 return manager_ ? manager_ : DownloadsDOMHandler::GetMainNotifierManager(); | 113 return manager_ ? manager_ : DownloadsDOMHandler::GetMainNotifierManager(); |
114 } | 114 } |
115 | 115 |
116 private: | 116 private: |
117 scoped_ptr<base::ListValue> downloads_list_; | 117 scoped_ptr<base::ListValue> downloads_list_; |
118 scoped_ptr<base::ListValue> download_updated_; | 118 scoped_ptr<base::DictionaryValue> download_updated_; |
119 bool waiting_list_; | 119 bool waiting_list_; |
120 bool waiting_updated_; | 120 bool waiting_updated_; |
121 content::DownloadManager* manager_; // weak. | 121 content::DownloadManager* manager_; // weak. |
122 | 122 |
123 DISALLOW_COPY_AND_ASSIGN(MockDownloadsDOMHandler); | 123 DISALLOW_COPY_AND_ASSIGN(MockDownloadsDOMHandler); |
124 }; | 124 }; |
125 | 125 |
126 } // namespace | 126 } // namespace |
127 | 127 |
128 class DownloadsDOMHandlerTest : public InProcessBrowserTest { | 128 class DownloadsDOMHandlerTest : public InProcessBrowserTest { |
(...skipping 32 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
161 current, | 161 current, |
162 std::string(), | 162 std::string(), |
163 std::string(), | 163 std::string(), |
164 128, | 164 128, |
165 128, | 165 128, |
166 content::DownloadItem::COMPLETE, | 166 content::DownloadItem::COMPLETE, |
167 content::DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS, | 167 content::DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS, |
168 content::DOWNLOAD_INTERRUPT_REASON_NONE, | 168 content::DOWNLOAD_INTERRUPT_REASON_NONE, |
169 false); | 169 false); |
170 | 170 |
| 171 mock_handler_->ForceSendCurrentDownloads(); |
171 mock_handler_->WaitForDownloadsList(); | 172 mock_handler_->WaitForDownloadsList(); |
172 ASSERT_EQ(1, static_cast<int>(mock_handler_->downloads_list()->GetSize())); | 173 ASSERT_EQ(1, static_cast<int>(mock_handler_->downloads_list()->GetSize())); |
173 EXPECT_TRUE(ListMatches( | 174 EXPECT_TRUE(ListMatches( |
174 mock_handler_->downloads_list(), | 175 mock_handler_->downloads_list(), |
175 "[{\"file_externally_removed\": false," | 176 "[{\"file_externally_removed\": false," |
176 " \"file_name\": \"file\"," | 177 " \"file_name\": \"file\"," |
177 " \"id\": \"1\"," | 178 " \"id\": \"1\"," |
178 " \"otr\": false," | 179 " \"otr\": false," |
179 " \"since_string\": \"Today\"," | 180 " \"since_string\": \"Today\"," |
180 " \"state\": \"COMPLETE\"," | 181 " \"state\": \"COMPLETE\"," |
(...skipping 72 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
253 EXPECT_TRUE(DownloadItemModel(&item).ShouldShowInShelf()); | 254 EXPECT_TRUE(DownloadItemModel(&item).ShouldShowInShelf()); |
254 } | 255 } |
255 | 256 |
256 // Tests that DownloadsDOMHandler detects new downloads and relays them to the | 257 // Tests that DownloadsDOMHandler detects new downloads and relays them to the |
257 // renderer. | 258 // renderer. |
258 // crbug.com/159390: This test fails when daylight savings time ends. | 259 // crbug.com/159390: This test fails when daylight savings time ends. |
259 IN_PROC_BROWSER_TEST_F(DownloadsDOMHandlerTest, DownloadsRelayed) { | 260 IN_PROC_BROWSER_TEST_F(DownloadsDOMHandlerTest, DownloadsRelayed) { |
260 DownloadAnItem(); | 261 DownloadAnItem(); |
261 | 262 |
262 mock_handler_->WaitForDownloadUpdated(); | 263 mock_handler_->WaitForDownloadUpdated(); |
263 ASSERT_EQ(1, static_cast<int>(mock_handler_->download_updated()->GetSize())); | 264 const base::DictionaryValue* update = mock_handler_->download_updated(); |
264 EXPECT_TRUE(ListMatches( | 265 ASSERT_TRUE(update); |
265 mock_handler_->download_updated(), | 266 |
266 "[{\"file_externally_removed\": true," | 267 bool removed; |
267 " \"id\": \"1\"}]")); | 268 ASSERT_TRUE(update->GetBoolean("file_externally_removed", &removed)); |
| 269 EXPECT_TRUE(removed); |
| 270 |
| 271 std::string id; |
| 272 ASSERT_TRUE(update->GetString("id", &id)); |
| 273 EXPECT_EQ("1", id); |
268 | 274 |
269 mock_handler_->reset_downloads_list(); | 275 mock_handler_->reset_downloads_list(); |
270 browser()->profile()->GetPrefs()->SetBoolean( | 276 browser()->profile()->GetPrefs()->SetBoolean( |
271 prefs::kAllowDeletingBrowserHistory, true); | 277 prefs::kAllowDeletingBrowserHistory, true); |
272 mock_handler_->HandleClearAll(NULL); | 278 mock_handler_->HandleClearAll(NULL); |
273 mock_handler_->WaitForDownloadsList(); | 279 mock_handler_->WaitForDownloadsList(); |
274 EXPECT_EQ(0, static_cast<int>(mock_handler_->downloads_list()->GetSize())); | 280 EXPECT_EQ(0, static_cast<int>(mock_handler_->downloads_list()->GetSize())); |
275 } | 281 } |
276 | 282 |
277 // Tests that DownloadsDOMHandler actually calls DownloadItem::Remove() when | 283 // Tests that DownloadsDOMHandler actually calls DownloadItem::Remove() when |
(...skipping 31 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
309 // TODO(benjhayden): Test incognito, both downloads_list() and that on-record | 315 // TODO(benjhayden): Test incognito, both downloads_list() and that on-record |
310 // calls can't access off-record items. | 316 // calls can't access off-record items. |
311 | 317 |
312 // TODO(benjhayden): Test that bad download ids incoming from the javascript are | 318 // TODO(benjhayden): Test that bad download ids incoming from the javascript are |
313 // dropped on the floor. | 319 // dropped on the floor. |
314 | 320 |
315 // TODO(benjhayden): Test that IsTemporary() downloads are not shown. | 321 // TODO(benjhayden): Test that IsTemporary() downloads are not shown. |
316 | 322 |
317 // TODO(benjhayden): Test that RemoveObserver is called on all download items, | 323 // TODO(benjhayden): Test that RemoveObserver is called on all download items, |
318 // including items that crossed IsTemporary() and back. | 324 // including items that crossed IsTemporary() and back. |
OLD | NEW |