chrome/browser/chromeos/gdata/gdata_wapi_service_browsertest.cc

// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "base/bind.h"
#include "base/command_line.h"
#include "base/file_path.h"
#include "base/file_util.h"
#include "base/json/json_reader.h"
#include "chrome/browser/chromeos/gdata/gdata_wapi_service.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/ui/browser.h"
#include "chrome/test/base/in_process_browser_test.h"
#include "chrome/test/base/ui_test_utils.h"
#include "net/test/test_server.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace gdata {

namespace {

class GDataTest : public InProcessBrowserTest {
 public:
  GDataTest()
      : InProcessBrowserTest(),
        gdata_test_server_(net::TestServer::TYPE_GDATA,
                           net::TestServer::kLocalhost,
                           FilePath(FILE_PATH_LITERAL("chrome/test/data"))) {
  }

  virtual void SetUpOnMainThread() OVERRIDE {
    ASSERT_TRUE(gdata_test_server_.Start());
    service_.reset(new gdata::GDataWapiService);
    service_->Initialize(browser()->profile());
    service_->auth_service_for_testing()->set_access_token_for_testing(
        net::TestServer::kGDataAuthToken);
  }

  virtual void CleanUpOnMainThread() {
    service_.reset();
  }

 protected:
  FilePath GetTestCachedFilePath(const FilePath& file_name) {
    return browser()->profile()->GetPath().Append(file_name);
  }

  net::TestServer gdata_test_server_;
  scoped_ptr<gdata::GDataWapiService> service_;
};

// The test callback for GDataWapiService::DownloadFile().
void TestDownloadCallback(gdata::GDataErrorCode* result,
                          std::string* contents,
                          gdata::GDataErrorCode error,
                          const GURL& content_url,
                          const FilePath& temp_file) {
  *result = error;
  file_util::ReadFileToString(temp_file, contents);
  file_util::Delete(temp_file, false);
  MessageLoop::current()->Quit();
}

// The test callback for GDataWapiService::GetDocuments().
void TestGetDocumentsCallback(gdata::GDataErrorCode* result_code,
                              base::Value** result_data,
                              gdata::GDataErrorCode error,
                              scoped_ptr<base::Value> feed_data) {
  *result_code = error;
  *result_data = feed_data.release();
  MessageLoop::current()->Quit();
}

}  // namespace

IN_PROC_BROWSER_TEST_F(GDataTest, Download) {
  gdata::GDataErrorCode result = gdata::GDATA_OTHER_ERROR;
  std::string contents;
  service_->DownloadFile(
      FilePath("/dummy/gdata/testfile.txt"),
      GetTestCachedFilePath(FilePath("cached_testfile.txt")),
      gdata_test_server_.GetURL("files/chromeos/gdata/testfile.txt"),
      base::Bind(&TestDownloadCallback, &result, &contents),
      gdata::GetContentCallback());
  content::RunMessageLoop();

  EXPECT_EQ(gdata::HTTP_SUCCESS, result);
  FilePath expected_filepath = gdata_test_server_.document_root().Append(
      FilePath(FILE_PATH_LITERAL("chromeos/gdata/testfile.txt")));
  std::string expected_contents;
  file_util::ReadFileToString(expected_filepath, &expected_contents);
  EXPECT_EQ(expected_contents, contents);
}

IN_PROC_BROWSER_TEST_F(GDataTest, NonExistingDownload) {
  gdata::GDataErrorCode result = gdata::GDATA_OTHER_ERROR;
  std::string dummy_contents;
  service_->DownloadFile(
      FilePath("/dummy/gdata/no-such-file.txt"),
      GetTestCachedFilePath(FilePath("cache_no-such-file.txt")),
      gdata_test_server_.GetURL("files/chromeos/gdata/no-such-file.txt"),
      base::Bind(&TestDownloadCallback, &result, &dummy_contents),
      gdata::GetContentCallback());
  content::RunMessageLoop();

  EXPECT_EQ(gdata::HTTP_NOT_FOUND, result);
  // Do not verify the not found message.
}

IN_PROC_BROWSER_TEST_F(GDataTest, GetDocuments) {
  gdata::GDataErrorCode result = gdata::GDATA_OTHER_ERROR;
  base::Value* result_data = NULL;
  service_->GetDocuments(
      gdata_test_server_.GetURL("files/chromeos/gdata/root_feed.json"),
      0,  // start_changestamp
      std::string(),  // search string
      std::string(),  // directory resource ID
      base::Bind(&TestGetDocumentsCallback, &result, &result_data));
  content::RunMessageLoop();

  EXPECT_EQ(gdata::HTTP_SUCCESS, result);
  ASSERT_TRUE(result_data);
  FilePath expected_filepath = gdata_test_server_.document_root().Append(
      FilePath(FILE_PATH_LITERAL("chromeos/gdata/root_feed.json")));
  std::string expected_contents;
  file_util::ReadFileToString(expected_filepath, &expected_contents);
  scoped_ptr<base::Value> expected_data(
      base::JSONReader::Read(expected_contents));
  EXPECT_TRUE(base::Value::Equals(expected_data.get(), result_data));
  delete result_data;
}

IN_PROC_BROWSER_TEST_F(GDataTest, GetDocumentsFailure) {
  // testfile.txt exists but the response is not JSON, so it should
  // emit a parse error instead.
  gdata::GDataErrorCode result = gdata::GDATA_OTHER_ERROR;
  base::Value* result_data = NULL;
  service_->GetDocuments(
      gdata_test_server_.GetURL("files/chromeos/gdata/testfile.txt"),
      0,  // start_changestamp
      std::string(),  // search string
      std::string(),  // directory resource ID
      base::Bind(&TestGetDocumentsCallback, &result, &result_data));
  content::RunMessageLoop();

  EXPECT_EQ(gdata::GDATA_PARSE_ERROR, result);
  EXPECT_FALSE(result_data);
}

}  // namespace gdata