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/files/file_path.h" | 5 #include "base/files/file_path.h" |
6 #include "base/files/scoped_temp_dir.h" | 6 #include "base/files/scoped_temp_dir.h" |
| 7 #include "base/memory/ref_counted.h" |
7 #include "content/browser/download/download_file_factory.h" | 8 #include "content/browser/download/download_file_factory.h" |
8 #include "content/browser/download/download_file_impl.h" | 9 #include "content/browser/download/download_file_impl.h" |
9 #include "content/browser/download/download_item_impl.h" | 10 #include "content/browser/download/download_item_impl.h" |
10 #include "content/browser/download/download_manager_impl.h" | 11 #include "content/browser/download/download_manager_impl.h" |
11 #include "content/browser/download/drag_download_file.h" | 12 #include "content/browser/download/drag_download_file.h" |
12 #include "content/browser/download/drag_download_util.h" | 13 #include "content/browser/download/drag_download_util.h" |
13 #include "content/browser/web_contents/web_contents_impl.h" | 14 #include "content/browser/web_contents/web_contents_impl.h" |
14 #include "content/public/browser/content_browser_client.h" | 15 #include "content/public/browser/content_browser_client.h" |
15 #include "content/public/browser/power_save_blocker.h" | 16 #include "content/public/browser/power_save_blocker.h" |
16 #include "content/public/common/content_client.h" | 17 #include "content/public/common/content_client.h" |
17 #include "content/public/test/content_browser_test.h" | 18 #include "content/public/test/content_browser_test.h" |
18 #include "content/public/test/content_browser_test_utils.h" | 19 #include "content/public/test/content_browser_test_utils.h" |
19 #include "content/public/test/download_test_observer.h" | 20 #include "content/public/test/download_test_observer.h" |
20 #include "content/public/test/test_utils.h" | 21 #include "content/public/test/test_utils.h" |
21 #include "content/shell/browser/shell.h" | 22 #include "content/shell/browser/shell.h" |
22 #include "content/shell/browser/shell_browser_context.h" | 23 #include "content/shell/browser/shell_browser_context.h" |
23 #include "content/shell/browser/shell_download_manager_delegate.h" | 24 #include "content/shell/browser/shell_download_manager_delegate.h" |
24 #include "content/test/net/url_request_mock_http_job.h" | |
25 #include "content/test/net/url_request_slow_download_job.h" | 25 #include "content/test/net/url_request_slow_download_job.h" |
| 26 #include "net/test/url_request/url_request_mock_http_job.h" |
26 #include "testing/gmock/include/gmock/gmock.h" | 27 #include "testing/gmock/include/gmock/gmock.h" |
27 #include "testing/gtest/include/gtest/gtest.h" | 28 #include "testing/gtest/include/gtest/gtest.h" |
28 #include "url/gurl.h" | 29 #include "url/gurl.h" |
29 | 30 |
30 using testing::_; | 31 using testing::_; |
31 using testing::InvokeWithoutArgs; | 32 using testing::InvokeWithoutArgs; |
32 | 33 |
33 namespace content { | 34 namespace content { |
34 | 35 |
35 class MockDownloadFileObserver : public ui::DownloadFileObserver { | 36 class MockDownloadFileObserver : public ui::DownloadFileObserver { |
(...skipping 30 matching lines...) Expand all Loading... |
66 ShellDownloadManagerDelegate* delegate = | 67 ShellDownloadManagerDelegate* delegate = |
67 static_cast<ShellDownloadManagerDelegate*>( | 68 static_cast<ShellDownloadManagerDelegate*>( |
68 shell()->web_contents()->GetBrowserContext() | 69 shell()->web_contents()->GetBrowserContext() |
69 ->GetDownloadManagerDelegate()); | 70 ->GetDownloadManagerDelegate()); |
70 delegate->SetDownloadBehaviorForTesting(downloads_directory()); | 71 delegate->SetDownloadBehaviorForTesting(downloads_directory()); |
71 } | 72 } |
72 | 73 |
73 void SetUpServer() { | 74 void SetUpServer() { |
74 base::FilePath mock_base(GetTestFilePath("download", "")); | 75 base::FilePath mock_base(GetTestFilePath("download", "")); |
75 BrowserThread::PostTask( | 76 BrowserThread::PostTask( |
76 BrowserThread::IO, FROM_HERE, | 77 BrowserThread::IO, |
77 base::Bind(&URLRequestMockHTTPJob::AddUrlHandler, mock_base)); | 78 FROM_HERE, |
| 79 base::Bind( |
| 80 &net::URLRequestMockHTTPJob::AddUrlHandler, |
| 81 mock_base, |
| 82 make_scoped_refptr(content::BrowserThread::GetBlockingPool()))); |
78 } | 83 } |
79 | 84 |
80 const base::FilePath& downloads_directory() const { | 85 const base::FilePath& downloads_directory() const { |
81 return downloads_directory_.path(); | 86 return downloads_directory_.path(); |
82 } | 87 } |
83 | 88 |
84 private: | 89 private: |
85 base::ScopedTempDir downloads_directory_; | 90 base::ScopedTempDir downloads_directory_; |
86 | 91 |
87 DISALLOW_COPY_AND_ASSIGN(DragDownloadFileTest); | 92 DISALLOW_COPY_AND_ASSIGN(DragDownloadFileTest); |
88 }; | 93 }; |
89 | 94 |
90 IN_PROC_BROWSER_TEST_F(DragDownloadFileTest, DragDownloadFileTest_NetError) { | 95 IN_PROC_BROWSER_TEST_F(DragDownloadFileTest, DragDownloadFileTest_NetError) { |
91 base::FilePath name(downloads_directory().AppendASCII( | 96 base::FilePath name(downloads_directory().AppendASCII( |
92 "DragDownloadFileTest_NetError.txt")); | 97 "DragDownloadFileTest_NetError.txt")); |
93 GURL url(URLRequestMockHTTPJob::GetMockUrl(base::FilePath(FILE_PATH_LITERAL( | 98 GURL url(net::URLRequestMockHTTPJob::GetMockUrl( |
94 "download-test.lib")))); | 99 base::FilePath(FILE_PATH_LITERAL("download-test.lib")))); |
95 Referrer referrer; | 100 Referrer referrer; |
96 std::string referrer_encoding; | 101 std::string referrer_encoding; |
97 scoped_refptr<DragDownloadFile> file( | 102 scoped_refptr<DragDownloadFile> file( |
98 new DragDownloadFile(name, base::File(), url, referrer, | 103 new DragDownloadFile(name, base::File(), url, referrer, |
99 referrer_encoding, shell()->web_contents())); | 104 referrer_encoding, shell()->web_contents())); |
100 scoped_refptr<MockDownloadFileObserver> observer( | 105 scoped_refptr<MockDownloadFileObserver> observer( |
101 new MockDownloadFileObserver()); | 106 new MockDownloadFileObserver()); |
102 EXPECT_CALL(*observer.get(), OnDownloadAborted()) | 107 EXPECT_CALL(*observer.get(), OnDownloadAborted()) |
103 .WillOnce(InvokeWithoutArgs(this, &DragDownloadFileTest::Succeed)); | 108 .WillOnce(InvokeWithoutArgs(this, &DragDownloadFileTest::Succeed)); |
104 ON_CALL(*observer.get(), OnDownloadCompleted(_)) | 109 ON_CALL(*observer.get(), OnDownloadCompleted(_)) |
105 .WillByDefault(InvokeWithoutArgs(this, &DragDownloadFileTest::FailFast)); | 110 .WillByDefault(InvokeWithoutArgs(this, &DragDownloadFileTest::FailFast)); |
106 file->Start(observer.get()); | 111 file->Start(observer.get()); |
107 RunMessageLoop(); | 112 RunMessageLoop(); |
108 } | 113 } |
109 | 114 |
110 IN_PROC_BROWSER_TEST_F(DragDownloadFileTest, DragDownloadFileTest_Complete) { | 115 IN_PROC_BROWSER_TEST_F(DragDownloadFileTest, DragDownloadFileTest_Complete) { |
111 base::FilePath name(downloads_directory().AppendASCII( | 116 base::FilePath name(downloads_directory().AppendASCII( |
112 "DragDownloadFileTest_Complete.txt")); | 117 "DragDownloadFileTest_Complete.txt")); |
113 GURL url(URLRequestMockHTTPJob::GetMockUrl(base::FilePath(FILE_PATH_LITERAL( | 118 GURL url(net::URLRequestMockHTTPJob::GetMockUrl( |
114 "download-test.lib")))); | 119 base::FilePath(FILE_PATH_LITERAL("download-test.lib")))); |
115 Referrer referrer; | 120 Referrer referrer; |
116 std::string referrer_encoding; | 121 std::string referrer_encoding; |
117 SetUpServer(); | 122 SetUpServer(); |
118 scoped_refptr<DragDownloadFile> file(new DragDownloadFile( | 123 scoped_refptr<DragDownloadFile> file(new DragDownloadFile( |
119 name, base::File(), url, referrer, | 124 name, base::File(), url, referrer, |
120 referrer_encoding, shell()->web_contents())); | 125 referrer_encoding, shell()->web_contents())); |
121 scoped_refptr<MockDownloadFileObserver> observer( | 126 scoped_refptr<MockDownloadFileObserver> observer( |
122 new MockDownloadFileObserver()); | 127 new MockDownloadFileObserver()); |
123 EXPECT_CALL(*observer.get(), OnDownloadCompleted(_)) | 128 EXPECT_CALL(*observer.get(), OnDownloadCompleted(_)) |
124 .WillOnce(InvokeWithoutArgs(this, &DragDownloadFileTest::Succeed)); | 129 .WillOnce(InvokeWithoutArgs(this, &DragDownloadFileTest::Succeed)); |
125 ON_CALL(*observer.get(), OnDownloadAborted()) | 130 ON_CALL(*observer.get(), OnDownloadAborted()) |
126 .WillByDefault(InvokeWithoutArgs(this, &DragDownloadFileTest::FailFast)); | 131 .WillByDefault(InvokeWithoutArgs(this, &DragDownloadFileTest::FailFast)); |
127 file->Start(observer.get()); | 132 file->Start(observer.get()); |
128 RunMessageLoop(); | 133 RunMessageLoop(); |
129 } | 134 } |
130 | 135 |
131 // TODO(benjhayden): Test Stop(). | 136 // TODO(benjhayden): Test Stop(). |
132 | 137 |
133 } // namespace content | 138 } // namespace content |
OLD | NEW |