Home | History | Annotate | Download | only in drive 
      1 // Copyright (c) 2013 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 "google_apis/drive/base_requests.h"
      6 
      7 #include "base/bind.h"
      8 #include "base/files/file_util.h"
      9 #include "base/files/scoped_temp_dir.h"
     10 #include "base/memory/scoped_ptr.h"
     11 #include "base/message_loop/message_loop.h"
     12 #include "base/run_loop.h"
     13 #include "google_apis/drive/dummy_auth_service.h"
     14 #include "google_apis/drive/request_sender.h"
     15 #include "google_apis/drive/task_util.h"
     16 #include "google_apis/drive/test_util.h"
     17 #include "net/test/embedded_test_server/embedded_test_server.h"
     18 #include "net/test/embedded_test_server/http_request.h"
     19 #include "net/test/embedded_test_server/http_response.h"
     20 #include "net/url_request/url_request_test_util.h"
     21 #include "testing/gtest/include/gtest/gtest.h"
     22 
     23 namespace google_apis {
     24 
     25 namespace {
     26 
     27 const char kTestUserAgent[] = "test-user-agent";
     28 
     29 }  // namespace
     30 
     31 class BaseRequestsServerTest : public testing::Test {
     32  protected:
     33   BaseRequestsServerTest() {
     34   }
     35 
     36   virtual void SetUp() OVERRIDE {
     37     ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
     38 
     39     request_context_getter_ = new net::TestURLRequestContextGetter(
     40         message_loop_.message_loop_proxy());
     41 
     42     request_sender_.reset(new RequestSender(
     43         new DummyAuthService,
     44         request_context_getter_.get(),
     45         message_loop_.message_loop_proxy(),
     46         kTestUserAgent));
     47 
     48     ASSERT_TRUE(test_server_.InitializeAndWaitUntilReady());
     49     test_server_.RegisterRequestHandler(
     50         base::Bind(&test_util::HandleDownloadFileRequest,
     51                    test_server_.base_url(),
     52                    base::Unretained(&http_request_)));
     53   }
     54 
     55   // Returns a temporary file path suitable for storing the cache file.
     56   base::FilePath GetTestCachedFilePath(const base::FilePath& file_name) {
     57     return temp_dir_.path().Append(file_name);
     58   }
     59 
     60   base::MessageLoopForIO message_loop_;  // Test server needs IO thread.
     61   net::test_server::EmbeddedTestServer test_server_;
     62   scoped_ptr<RequestSender> request_sender_;
     63   scoped_refptr<net::TestURLRequestContextGetter> request_context_getter_;
     64   base::ScopedTempDir temp_dir_;
     65 
     66   // The incoming HTTP request is saved so tests can verify the request
     67   // parameters like HTTP method (ex. some requests should use DELETE
     68   // instead of GET).
     69   net::test_server::HttpRequest http_request_;
     70 };
     71 
     72 TEST_F(BaseRequestsServerTest, DownloadFileRequest_ValidFile) {
     73   GDataErrorCode result_code = GDATA_OTHER_ERROR;
     74   base::FilePath temp_file;
     75   {
     76     base::RunLoop run_loop;
     77     DownloadFileRequestBase* request = new DownloadFileRequestBase(
     78         request_sender_.get(),
     79         test_util::CreateQuitCallback(
     80             &run_loop,
     81             test_util::CreateCopyResultCallback(&result_code, &temp_file)),
     82         GetContentCallback(),
     83         ProgressCallback(),
     84         test_server_.GetURL("/files/drive/testfile.txt"),
     85         GetTestCachedFilePath(
     86             base::FilePath::FromUTF8Unsafe("cached_testfile.txt")));
     87     request_sender_->StartRequestWithRetry(request);
     88     run_loop.Run();
     89   }
     90 
     91   std::string contents;
     92   base::ReadFileToString(temp_file, &contents);
     93   base::DeleteFile(temp_file, false);
     94 
     95   EXPECT_EQ(HTTP_SUCCESS, result_code);
     96   EXPECT_EQ(net::test_server::METHOD_GET, http_request_.method);
     97   EXPECT_EQ("/files/drive/testfile.txt", http_request_.relative_url);
     98 
     99   const base::FilePath expected_path =
    100       test_util::GetTestFilePath("drive/testfile.txt");
    101   std::string expected_contents;
    102   base::ReadFileToString(expected_path, &expected_contents);
    103   EXPECT_EQ(expected_contents, contents);
    104 }
    105 
    106 TEST_F(BaseRequestsServerTest, DownloadFileRequest_NonExistentFile) {
    107   GDataErrorCode result_code = GDATA_OTHER_ERROR;
    108   base::FilePath temp_file;
    109   {
    110     base::RunLoop run_loop;
    111     DownloadFileRequestBase* request = new DownloadFileRequestBase(
    112         request_sender_.get(),
    113         test_util::CreateQuitCallback(
    114             &run_loop,
    115             test_util::CreateCopyResultCallback(&result_code, &temp_file)),
    116         GetContentCallback(),
    117         ProgressCallback(),
    118         test_server_.GetURL("/files/gdata/no-such-file.txt"),
    119         GetTestCachedFilePath(
    120             base::FilePath::FromUTF8Unsafe("cache_no-such-file.txt")));
    121     request_sender_->StartRequestWithRetry(request);
    122     run_loop.Run();
    123   }
    124   EXPECT_EQ(HTTP_NOT_FOUND, result_code);
    125   EXPECT_EQ(net::test_server::METHOD_GET, http_request_.method);
    126   EXPECT_EQ("/files/gdata/no-such-file.txt",
    127             http_request_.relative_url);
    128   // Do not verify the not found message.
    129 }
    130 
    131 }  // namespace google_apis
    132