// 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.

// This file contains mocks for classes in drive_service_interface.h

#ifndef CHROME_BROWSER_GOOGLE_APIS_MOCK_DRIVE_SERVICE_H_
#define CHROME_BROWSER_GOOGLE_APIS_MOCK_DRIVE_SERVICE_H_

#include <string>

#include "base/memory/scoped_ptr.h"
#include "chrome/browser/google_apis/drive_service_interface.h"
#include "testing/gmock/include/gmock/gmock.h"

class FilePath;

namespace google_apis {

class MockDriveService : public DriveServiceInterface {
 public:
  // DriveService is usually owned and created by DriveFileSystem.
  MockDriveService();
  virtual ~MockDriveService();

  // DriveServiceInterface overrides.
  MOCK_METHOD1(Initialize, void(Profile* profile));
  MOCK_METHOD1(AddObserver, void(DriveServiceObserver* observer));
  MOCK_METHOD1(RemoveObserver,
      void(DriveServiceObserver* observer));
  MOCK_CONST_METHOD0(CanStartOperation, bool());
  MOCK_METHOD0(CancelAll, void(void));
  MOCK_METHOD1(CancelForFilePath, bool(const FilePath& file_path));
  MOCK_CONST_METHOD0(GetProgressStatusList,
      OperationProgressStatusList());
  MOCK_CONST_METHOD0(GetRootResourceId, std::string());
  MOCK_METHOD6(GetResourceList,
      void(const GURL& feed_url,
          int64 start_changestamp,
          const std::string& search_string,
          bool shared_with_me,
          const std::string& directory_resource_id,
          const GetResourceListCallback& callback));
  MOCK_METHOD2(GetResourceEntry,
      void(const std::string& resource_id,
          const GetResourceEntryCallback& callback));
  MOCK_METHOD1(GetAccountMetadata,
      void(const GetAccountMetadataCallback& callback));
  MOCK_METHOD1(GetAppList, void(const GetAppListCallback& callback));
  MOCK_METHOD2(DeleteResource,
      void(const std::string& resource_id,
          const EntryActionCallback& callback));
  MOCK_METHOD3(CopyHostedDocument,
      void(const std::string& resource_id,
          const std::string& new_name,
          const GetResourceEntryCallback& callback));
  MOCK_METHOD3(RenameResource,
      void(const std::string& resource_id,
          const std::string& new_name,
          const EntryActionCallback& callback));
  MOCK_METHOD3(AddResourceToDirectory,
      void(const std::string& parent_resource_id,
          const GURL& edit_url,
          const EntryActionCallback& callback));
  MOCK_METHOD3(RemoveResourceFromDirectory,
      void(const std::string& parent_resource_id,
          const std::string& resource_id,
          const EntryActionCallback& callback));
  MOCK_METHOD3(AddNewDirectory,
      void(const std::string& parent_resource_id,
          const std::string& directory_name,
          const GetResourceEntryCallback& callback));
  MOCK_METHOD5(
      DownloadFile,
      void(const FilePath& virtual_path,
          const FilePath& local_cache_path,
          const GURL& content_url,
          const DownloadActionCallback&
          donwload_action_callback,
          const GetContentCallback& get_content_callback));
  MOCK_METHOD2(InitiateUpload,
      void(const InitiateUploadParams& upload_file_info,
          const InitiateUploadCallback& callback));
  MOCK_METHOD2(ResumeUpload,
      void(const ResumeUploadParams& upload_file_info,
          const ResumeUploadCallback& callback));
  MOCK_METHOD3(AuthorizeApp,
      void(const GURL& edit_url,
          const std::string& app_id,
          const AuthorizeAppCallback& callback));
  MOCK_CONST_METHOD0(HasAccessToken, bool());
  MOCK_CONST_METHOD0(HasRefreshToken, bool());

  void set_file_data(std::string* file_data) {
    file_data_.reset(file_data);
  }

  void set_search_result(const std::string& search_result_feed);

 private:
  // Helper stub methods for functions which take callbacks, so that
  // the callbacks get called with testable results.

  // Will call |callback| with HTTP_SUCCESS and a StringValue with the current
  // value of |feed_data_|.
  void GetResourceListStub(const GURL& feed_url,
      int64 start_changestamp,
      const std::string& search_string,
      bool shared_with_me,
      const std::string& directory_resource_id,
      const GetResourceListCallback& callback);

  // Will call |callback| with HTTP_SUCCESS and a StringValue with the current
  // value of |account_metadata_|.
  void GetAccountMetadataStub(const GetAccountMetadataCallback& callback);

  // Will call |callback| with HTTP_SUCCESS.
  void DeleteResourceStub(const std::string& resource_id,
                          const EntryActionCallback& callback);

  // Will call |callback| with HTTP_SUCCESS and the current value of
  // |document_data_|.
  void CopyHostedDocumentStub(const std::string& resource_id,
                              const std::string& new_name,
                              const GetResourceEntryCallback& callback);

  // Will call |callback| with HTTP_SUCCESS.
  void RenameResourceStub(const std::string& resource_id,
                          const std::string& new_name,
                          const EntryActionCallback& callback);

  // Will call |callback| with HTTP_SUCCESS.
  void AddResourceToDirectoryStub(
      const std::string& parent_resource_id,
      const GURL& edit_url,
      const EntryActionCallback& callback);

  // Will call |callback| with HTTP_SUCCESS.
  void RemoveResourceFromDirectoryStub(
      const std::string& parent_resource_id,
      const std::string& resource_id,
      const EntryActionCallback& callback);

  // Will call |callback| with HTTP_SUCCESS and the current value of
  // |directory_data_|.
  void CreateDirectoryStub(const std::string& parent_resource_id,
                           const std::string& directory_name,
                           const GetResourceEntryCallback& callback);

  // Will call |callback| with HTTP_SUCCESS, the given URL, and the host+path
  // portion of the URL as the temporary file path. If |file_data_| is not null,
  // |file_data_| is written to the temporary file.
  void DownloadFileStub(
      const FilePath& virtual_path,
      const FilePath& local_tmp_path,
      const GURL& content_url,
      const DownloadActionCallback& download_action_callback,
      const GetContentCallback& get_content_callback);

  // Account meta data to be returned from GetAccountMetadata.
  scoped_ptr<base::Value> account_metadata_data_;

  // Feed data to be returned from GetResourceList.
  scoped_ptr<base::Value> feed_data_;

  // Feed data to be returned from CreateDirectory.
  scoped_ptr<base::Value> directory_data_;

  // Feed data to be returned from CopyHostedDocument.
  scoped_ptr<base::Value> document_data_;

  // Feed data to be returned from GetResourceList if the search path is
  // specified. The feed contains subset of the root_feed.
  scoped_ptr<base::Value> search_result_;

  // File data to be written to the local temporary file when
  // DownloadFileStub is called.
  scoped_ptr<std::string> file_data_;
};

}  // namespace google_apis

#endif  // CHROME_BROWSER_GOOGLE_APIS_MOCK_DRIVE_SERVICE_H_