| // 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. |
| |
| #ifndef CHROME_BROWSER_DRIVE_FAKE_DRIVE_SERVICE_H_ |
| #define CHROME_BROWSER_DRIVE_FAKE_DRIVE_SERVICE_H_ |
| |
| #include "base/files/file_path.h" |
| #include "base/values.h" |
| #include "chrome/browser/drive/drive_service_interface.h" |
| #include "chrome/browser/google_apis/auth_service_interface.h" |
| |
| namespace drive { |
| |
| // This class implements a fake DriveService which acts like a real Drive |
| // service. The fake service works as follows: |
| // |
| // 1) Load JSON files and construct the in-memory resource list. |
| // 2) Return valid responses based on the the in-memory resource list. |
| // 3) Update the in-memory resource list by requests like DeleteResource(). |
| class FakeDriveService : public DriveServiceInterface { |
| public: |
| FakeDriveService(); |
| virtual ~FakeDriveService(); |
| |
| // Loads the resource list for WAPI. Returns true on success. |
| bool LoadResourceListForWapi(const std::string& relative_path); |
| |
| // Loads the account metadata for WAPI. Returns true on success. Also adds |
| // the largest changestamp in the account metadata to the existing |
| // entries. The changestamp information will be used to generate change |
| // lists in GetResourceList() when non-zero |start_changestamp| is |
| // specified. |
| bool LoadAccountMetadataForWapi(const std::string& relative_path); |
| |
| // Loads the app list for Drive API. Returns true on success. |
| bool LoadAppListForDriveApi(const std::string& relative_path); |
| |
| // Changes the offline state. All functions fail with GDATA_NO_CONNECTION |
| // when offline. By default the offline state is false. |
| void set_offline(bool offline) { offline_ = offline; } |
| |
| // Changes the default max results returned from GetResourceList(). |
| // By default, it's set to 0, which is unlimited. |
| void set_default_max_results(int default_max_results) { |
| default_max_results_ = default_max_results; |
| } |
| |
| // Returns the largest changestamp, which starts from 0 by default. See |
| // also comments at LoadAccountMetadataForWapi(). |
| int64 largest_changestamp() const { return largest_changestamp_; } |
| |
| // Returns the number of times the resource list is successfully loaded by |
| // GetAllResourceList(). |
| int resource_list_load_count() const { return resource_list_load_count_; } |
| |
| // Returns the number of times the resource list is successfully loaded by |
| // GetChangeList(). |
| int change_list_load_count() const { return change_list_load_count_; } |
| |
| // Returns the number of times the resource list is successfully loaded by |
| // GetResourceListInDirectory(). |
| int directory_load_count() const { return directory_load_count_; } |
| |
| // Returns the number of times the about resource is successfully loaded |
| // by GetAboutResource(). |
| int about_resource_load_count() const { |
| return about_resource_load_count_; |
| } |
| |
| // Returns the file path whose request is cancelled just before this method |
| // invocation. |
| const base::FilePath& last_cancelled_file() const { |
| return last_cancelled_file_; |
| } |
| |
| // Returns the (fake) URL for the link. |
| static GURL GetFakeLinkUrl(const std::string& resource_id); |
| |
| // DriveServiceInterface Overrides |
| virtual void Initialize() OVERRIDE; |
| virtual void AddObserver(DriveServiceObserver* observer) OVERRIDE; |
| virtual void RemoveObserver(DriveServiceObserver* observer) OVERRIDE; |
| virtual bool CanSendRequest() const OVERRIDE; |
| virtual std::string CanonicalizeResourceId( |
| const std::string& resource_id) const OVERRIDE; |
| virtual std::string GetRootResourceId() const OVERRIDE; |
| virtual bool HasAccessToken() const OVERRIDE; |
| virtual void RequestAccessToken( |
| const google_apis::AuthStatusCallback& callback) OVERRIDE; |
| virtual bool HasRefreshToken() const OVERRIDE; |
| virtual void ClearAccessToken() OVERRIDE; |
| virtual void ClearRefreshToken() OVERRIDE; |
| virtual google_apis::CancelCallback GetAllResourceList( |
| const google_apis::GetResourceListCallback& callback) OVERRIDE; |
| virtual google_apis::CancelCallback GetResourceListInDirectory( |
| const std::string& directory_resource_id, |
| const google_apis::GetResourceListCallback& callback) OVERRIDE; |
| // See the comment for EntryMatchWidthQuery() in .cc file for details about |
| // the supported search query types. |
| virtual google_apis::CancelCallback Search( |
| const std::string& search_query, |
| const google_apis::GetResourceListCallback& callback) OVERRIDE; |
| virtual google_apis::CancelCallback SearchByTitle( |
| const std::string& title, |
| const std::string& directory_resource_id, |
| const google_apis::GetResourceListCallback& callback) OVERRIDE; |
| virtual google_apis::CancelCallback GetChangeList( |
| int64 start_changestamp, |
| const google_apis::GetResourceListCallback& callback) OVERRIDE; |
| virtual google_apis::CancelCallback ContinueGetResourceList( |
| const GURL& override_url, |
| const google_apis::GetResourceListCallback& callback) OVERRIDE; |
| virtual google_apis::CancelCallback GetResourceEntry( |
| const std::string& resource_id, |
| const google_apis::GetResourceEntryCallback& callback) OVERRIDE; |
| virtual google_apis::CancelCallback GetShareUrl( |
| const std::string& resource_id, |
| const GURL& embed_origin, |
| const google_apis::GetShareUrlCallback& callback) OVERRIDE; |
| virtual google_apis::CancelCallback GetAboutResource( |
| const google_apis::GetAboutResourceCallback& callback) OVERRIDE; |
| virtual google_apis::CancelCallback GetAppList( |
| const google_apis::GetAppListCallback& callback) OVERRIDE; |
| virtual google_apis::CancelCallback DeleteResource( |
| const std::string& resource_id, |
| const std::string& etag, |
| const google_apis::EntryActionCallback& callback) OVERRIDE; |
| virtual google_apis::CancelCallback DownloadFile( |
| const base::FilePath& local_cache_path, |
| const std::string& resource_id, |
| const google_apis::DownloadActionCallback& download_action_callback, |
| const google_apis::GetContentCallback& get_content_callback, |
| const google_apis::ProgressCallback& progress_callback) OVERRIDE; |
| virtual google_apis::CancelCallback CopyResource( |
| const std::string& resource_id, |
| const std::string& parent_resource_id, |
| const std::string& new_title, |
| const google_apis::GetResourceEntryCallback& callback) OVERRIDE; |
| // The new resource ID for the copied document will look like |
| // |resource_id| + "_copied". |
| virtual google_apis::CancelCallback CopyHostedDocument( |
| const std::string& resource_id, |
| const std::string& new_title, |
| const google_apis::GetResourceEntryCallback& callback) OVERRIDE; |
| virtual google_apis::CancelCallback RenameResource( |
| const std::string& resource_id, |
| const std::string& new_title, |
| const google_apis::EntryActionCallback& callback) OVERRIDE; |
| virtual google_apis::CancelCallback TouchResource( |
| const std::string& resource_id, |
| const base::Time& modified_date, |
| const base::Time& last_viewed_by_me_date, |
| const google_apis::GetResourceEntryCallback& callback) OVERRIDE; |
| virtual google_apis::CancelCallback AddResourceToDirectory( |
| const std::string& parent_resource_id, |
| const std::string& resource_id, |
| const google_apis::EntryActionCallback& callback) OVERRIDE; |
| virtual google_apis::CancelCallback RemoveResourceFromDirectory( |
| const std::string& parent_resource_id, |
| const std::string& resource_id, |
| const google_apis::EntryActionCallback& callback) OVERRIDE; |
| virtual google_apis::CancelCallback AddNewDirectory( |
| const std::string& parent_resource_id, |
| const std::string& directory_title, |
| const google_apis::GetResourceEntryCallback& callback) OVERRIDE; |
| virtual google_apis::CancelCallback InitiateUploadNewFile( |
| const std::string& content_type, |
| int64 content_length, |
| const std::string& parent_resource_id, |
| const std::string& title, |
| const google_apis::InitiateUploadCallback& callback) OVERRIDE; |
| virtual google_apis::CancelCallback InitiateUploadExistingFile( |
| const std::string& content_type, |
| int64 content_length, |
| const std::string& resource_id, |
| const std::string& etag, |
| const google_apis::InitiateUploadCallback& callback) OVERRIDE; |
| virtual google_apis::CancelCallback ResumeUpload( |
| const GURL& upload_url, |
| int64 start_position, |
| int64 end_position, |
| int64 content_length, |
| const std::string& content_type, |
| const base::FilePath& local_file_path, |
| const google_apis::UploadRangeCallback& callback, |
| const google_apis::ProgressCallback& progress_callback) OVERRIDE; |
| virtual google_apis::CancelCallback GetUploadStatus( |
| const GURL& upload_url, |
| int64 content_length, |
| const google_apis::UploadRangeCallback& callback) OVERRIDE; |
| virtual google_apis::CancelCallback AuthorizeApp( |
| const std::string& resource_id, |
| const std::string& app_id, |
| const google_apis::AuthorizeAppCallback& callback) OVERRIDE; |
| |
| // Adds a new file with the given parameters. On success, returns |
| // HTTP_CREATED with the parsed entry. |
| // |callback| must not be null. |
| void AddNewFile(const std::string& content_type, |
| const std::string& content_data, |
| const std::string& parent_resource_id, |
| const std::string& title, |
| bool shared_with_me, |
| const google_apis::GetResourceEntryCallback& callback); |
| |
| // Sets the last modified time for an entry specified by |resource_id|. |
| // On success, returns HTTP_SUCCESS with the parsed entry. |
| // |callback| must not be null. |
| void SetLastModifiedTime( |
| const std::string& resource_id, |
| const base::Time& last_modified_time, |
| const google_apis::GetResourceEntryCallback& callback); |
| |
| private: |
| struct UploadSession; |
| |
| // Returns a pointer to the entry that matches |resource_id|, or NULL if |
| // not found. |
| base::DictionaryValue* FindEntryByResourceId(const std::string& resource_id); |
| |
| // Returns a pointer to the entry that matches |content_url|, or NULL if |
| // not found. |
| base::DictionaryValue* FindEntryByContentUrl(const GURL& content_url); |
| |
| // Returns a new resource ID, which looks like "resource_id_<num>" where |
| // <num> is a monotonically increasing number starting from 1. |
| std::string GetNewResourceId(); |
| |
| // Increments |largest_changestamp_| and adds the new changestamp and ETag to |
| // |entry|. |
| void AddNewChangestampAndETag(base::DictionaryValue* entry); |
| |
| // Adds a new entry based on the given parameters. |entry_kind| should be |
| // "file" or "folder". Returns a pointer to the newly added entry, or NULL |
| // if failed. |
| const base::DictionaryValue* AddNewEntry( |
| const std::string& content_type, |
| const std::string& content_data, |
| const std::string& parent_resource_id, |
| const std::string& title, |
| bool shared_with_me, |
| const std::string& entry_kind); |
| |
| // Core implementation of GetResourceList. |
| // This method returns the slice of the all matched entries, and its range |
| // is between |start_offset| (inclusive) and |start_offset| + |max_results| |
| // (exclusive). |
| // Increments *load_counter by 1 before it returns successfully. |
| void GetResourceListInternal( |
| int64 start_changestamp, |
| const std::string& search_query, |
| const std::string& directory_resource_id, |
| int start_offset, |
| int max_results, |
| int* load_counter, |
| const google_apis::GetResourceListCallback& callback); |
| |
| // Returns new upload session URL. |
| GURL GetNewUploadSessionUrl(); |
| |
| scoped_ptr<base::DictionaryValue> resource_list_value_; |
| scoped_ptr<base::Value> account_metadata_value_; |
| scoped_ptr<base::Value> app_info_value_; |
| std::map<GURL, UploadSession> upload_sessions_; |
| int64 largest_changestamp_; |
| int64 published_date_seq_; |
| int64 next_upload_sequence_number_; |
| int default_max_results_; |
| int resource_id_count_; |
| int resource_list_load_count_; |
| int change_list_load_count_; |
| int directory_load_count_; |
| int about_resource_load_count_; |
| bool offline_; |
| base::FilePath last_cancelled_file_; |
| |
| DISALLOW_COPY_AND_ASSIGN(FakeDriveService); |
| }; |
| |
| } // namespace drive |
| |
| #endif // CHROME_BROWSER_DRIVE_FAKE_DRIVE_SERVICE_H_ |