Home | History | Annotate | Download | only in download 
      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.
      4 
      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/download_test_observer.h"
     18 #include "content/public/test/test_utils.h"
     19 #include "content/shell/browser/shell.h"
     20 #include "content/shell/browser/shell_browser_context.h"
     21 #include "content/shell/browser/shell_download_manager_delegate.h"
     22 #include "content/test/content_browser_test.h"
     23 #include "content/test/content_browser_test_utils.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"
     28 #include "url/gurl.h"
     29 
     30 using testing::_;
     31 using testing::InvokeWithoutArgs;
     32 
     33 namespace content {
     34 
     35 class MockDownloadFileObserver : public ui::DownloadFileObserver {
     36  public:
     37   MockDownloadFileObserver() {}
     38 
     39   MOCK_METHOD1(OnDownloadCompleted, void(const base::FilePath& file_path));
     40   MOCK_METHOD0(OnDownloadAborted, void());
     41 
     42  private:
     43   virtual ~MockDownloadFileObserver() {}
     44 
     45   DISALLOW_COPY_AND_ASSIGN(MockDownloadFileObserver);
     46 };
     47 
     48 class DragDownloadFileTest : public ContentBrowserTest {
     49  public:
     50   DragDownloadFileTest() {}
     51   virtual ~DragDownloadFileTest() {}
     52 
     53   void Succeed() {
     54     BrowserThread::PostTask(BrowserThread::UI,
     55                             FROM_HERE,
     56                             base::MessageLoopForUI::current()->QuitClosure());
     57   }
     58 
     59   void FailFast() {
     60     CHECK(false);
     61   }
     62 
     63  protected:
     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());
     71   }
     72 
     73   void SetUpServer() {
     74     base::FilePath mock_base(GetTestFilePath("download", ""));
     75     BrowserThread::PostTask(
     76         BrowserThread::IO, FROM_HERE,
     77         base::Bind(&URLRequestMockHTTPJob::AddUrlHandler, mock_base));
     78   }
     79 
     80   const base::FilePath& downloads_directory() const {
     81     return downloads_directory_.path();
     82   }
     83 
     84  private:
     85   base::ScopedTempDir downloads_directory_;
     86 
     87   DISALLOW_COPY_AND_ASSIGN(DragDownloadFileTest);
     88 };
     89 
     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"))));
     95   Referrer referrer;
     96   std::string referrer_encoding;
     97   scoped_refptr<DragDownloadFile> file(
     98       new DragDownloadFile(name, scoped_ptr<net::FileStream>(), 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());
    107   RunMessageLoop();
    108 }
    109 
    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"))));
    115   Referrer referrer;
    116   std::string referrer_encoding;
    117   net::FileStream* stream = NULL;
    118   SetUpServer();
    119   scoped_refptr<DragDownloadFile> file(new DragDownloadFile(
    120       name, scoped_ptr<net::FileStream>(stream), url, referrer,
    121       referrer_encoding, shell()->web_contents()));
    122   scoped_refptr<MockDownloadFileObserver> observer(
    123       new MockDownloadFileObserver());
    124   EXPECT_CALL(*observer.get(), OnDownloadCompleted(_))
    125       .WillOnce(InvokeWithoutArgs(this, &DragDownloadFileTest::Succeed));
    126   ON_CALL(*observer.get(), OnDownloadAborted())
    127       .WillByDefault(InvokeWithoutArgs(this, &DragDownloadFileTest::FailFast));
    128   file->Start(observer.get());
    129   RunMessageLoop();
    130 }
    131 
    132 // TODO(benjhayden): Test Stop().
    133 
    134 }  // namespace content
    135