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