Support contents upload on FakeDriveService

chrome/browser/drive/fake_drive_service.h

 // 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"
@@ skipping 188 matching lines @@
 
   // 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 pointer to the entry that matches |upload_url|, or NULL if
-  // not found.
-  base::DictionaryValue* FindEntryByUploadUrl(const GURL& upload_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 to
+  // Increments |largest_changestamp_| and adds the new changestamp and ETag to
   // |entry|.
-  void AddNewChangestamp(base::DictionaryValue* 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_