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
3 // found in the LICENSE file.
5 #include "base/files/file_path.h"
6 #include "base/files/scoped_temp_dir.h"
7 #include "content/browser/download/download_file_factory.h"
8 #include "content/browser/download/download_file_impl.h"
9 #include "content/browser/download/download_item_impl.h"
10 #include "content/browser/download/download_manager_impl.h"
11 #include "content/browser/download/drag_download_file.h"
12 #include "content/browser/download/drag_download_util.h"
13 #include "content/browser/web_contents/web_contents_impl.h"
14 #include "content/public/browser/content_browser_client.h"
15 #include "content/public/browser/power_save_blocker.h"
16 #include "content/public/common/content_client.h"
17 #include "content/public/test/content_browser_test.h"
18 #include "content/public/test/content_browser_test_utils.h"
19 #include "content/public/test/download_test_observer.h"
20 #include "content/public/test/test_utils.h"
21 #include "content/shell/browser/shell.h"
22 #include "content/shell/browser/shell_browser_context.h"
23 #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"
26 #include "testing/gmock/include/gmock/gmock.h"
27 #include "testing/gtest/include/gtest/gtest.h"
31 using testing::InvokeWithoutArgs
;
35 class MockDownloadFileObserver
: public ui::DownloadFileObserver
{
37 MockDownloadFileObserver() {}
39 MOCK_METHOD1(OnDownloadCompleted
, void(const base::FilePath
& file_path
));
40 MOCK_METHOD0(OnDownloadAborted
, void());
43 virtual ~MockDownloadFileObserver() {}
45 DISALLOW_COPY_AND_ASSIGN(MockDownloadFileObserver
);
48 class DragDownloadFileTest
: public ContentBrowserTest
{
50 DragDownloadFileTest() {}
51 virtual ~DragDownloadFileTest() {}
54 BrowserThread::PostTask(BrowserThread::UI
,
56 base::MessageLoopForUI::current()->QuitClosure());
64 virtual void SetUpOnMainThread() OVERRIDE
{
65 ASSERT_TRUE(downloads_directory_
.CreateUniqueTempDir());
66 ShellDownloadManagerDelegate
* delegate
=
67 static_cast<ShellDownloadManagerDelegate
*>(
68 shell()->web_contents()->GetBrowserContext()
69 ->GetDownloadManagerDelegate());
70 delegate
->SetDownloadBehaviorForTesting(downloads_directory());
74 base::FilePath
mock_base(GetTestFilePath("download", ""));
75 BrowserThread::PostTask(
76 BrowserThread::IO
, FROM_HERE
,
77 base::Bind(&URLRequestMockHTTPJob::AddUrlHandler
, mock_base
));
80 const base::FilePath
& downloads_directory() const {
81 return downloads_directory_
.path();
85 base::ScopedTempDir downloads_directory_
;
87 DISALLOW_COPY_AND_ASSIGN(DragDownloadFileTest
);
90 IN_PROC_BROWSER_TEST_F(DragDownloadFileTest
, DragDownloadFileTest_NetError
) {
91 base::FilePath
name(downloads_directory().AppendASCII(
92 "DragDownloadFileTest_NetError.txt"));
93 GURL
url(URLRequestMockHTTPJob::GetMockUrl(base::FilePath(FILE_PATH_LITERAL(
94 "download-test.lib"))));
96 std::string referrer_encoding
;
97 scoped_refptr
<DragDownloadFile
> file(
98 new DragDownloadFile(name
, base::File(), url
, referrer
,
99 referrer_encoding
, shell()->web_contents()));
100 scoped_refptr
<MockDownloadFileObserver
> observer(
101 new MockDownloadFileObserver());
102 EXPECT_CALL(*observer
.get(), OnDownloadAborted())
103 .WillOnce(InvokeWithoutArgs(this, &DragDownloadFileTest::Succeed
));
104 ON_CALL(*observer
.get(), OnDownloadCompleted(_
))
105 .WillByDefault(InvokeWithoutArgs(this, &DragDownloadFileTest::FailFast
));
106 file
->Start(observer
.get());
110 IN_PROC_BROWSER_TEST_F(DragDownloadFileTest
, DragDownloadFileTest_Complete
) {
111 base::FilePath
name(downloads_directory().AppendASCII(
112 "DragDownloadFileTest_Complete.txt"));
113 GURL
url(URLRequestMockHTTPJob::GetMockUrl(base::FilePath(FILE_PATH_LITERAL(
114 "download-test.lib"))));
116 std::string referrer_encoding
;
118 scoped_refptr
<DragDownloadFile
> file(new DragDownloadFile(
119 name
, base::File(), url
, referrer
,
120 referrer_encoding
, shell()->web_contents()));
121 scoped_refptr
<MockDownloadFileObserver
> observer(
122 new MockDownloadFileObserver());
123 EXPECT_CALL(*observer
.get(), OnDownloadCompleted(_
))
124 .WillOnce(InvokeWithoutArgs(this, &DragDownloadFileTest::Succeed
));
125 ON_CALL(*observer
.get(), OnDownloadAborted())
126 .WillByDefault(InvokeWithoutArgs(this, &DragDownloadFileTest::FailFast
));
127 file
->Start(observer
.get());
131 // TODO(benjhayden): Test Stop().
133 } // namespace content