Pass calls to GetDocuments through the scheduler

chrome/browser/chromeos/drive/drive_scheduler_unittest.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 "chrome/browser/chromeos/drive/drive_scheduler.h"
 
 #include "base/bind.h"
+#include "base/file_util.h"
+#include "base/json/json_reader.h"
 #include "base/threading/sequenced_worker_pool.h"
 #include "chrome/browser/chromeos/drive/drive_test_util.h"
 #include "chrome/browser/chromeos/drive/file_system/drive_operations.h"
 #include "chrome/browser/chromeos/drive/file_system/copy_operation.h"
 #include "chrome/browser/chromeos/drive/file_system/move_operation.h"
 #include "chrome/browser/chromeos/drive/file_system/remove_operation.h"
 #include "chrome/browser/prefs/pref_service.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/test/base/testing_profile.h"
 #include "content/public/test/test_browser_thread.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using ::testing::AnyNumber;
 using ::testing::DoAll;
+using ::testing::Eq;
 using ::testing::Return;
 using ::testing::StrictMock;
 using ::testing::_;
 
+namespace google_apis {
+
+class FakeDriveService : public DriveServiceInterface {
+  virtual void Initialize(Profile* profile) {
+  }
+
+  virtual void AddObserver(DriveServiceObserver* observer) {
+  }
+
+  virtual void RemoveObserver(DriveServiceObserver* observer) {
+  }
+
+  virtual bool CanStartOperation() const {
+    return true;
+  }
+
+  virtual void CancelAll() {
+  }
+
+  virtual bool CancelForFilePath(const FilePath& file_path) {
+    return true;
+  }
+
+  virtual OperationProgressStatusList GetProgressStatusList() const {
+    return OperationProgressStatusList();
+  }
+
+  virtual bool HasAccessToken() const {
+    return true;
+  }
+
+  virtual bool HasRefreshToken() const {
+    return true;
+  }
+
+  virtual void GetDocuments(const GURL& feed_url,
+                            int64 start_changestamp,
+                            const std::string& search_query,
+                            bool shared_with_me,
+                            const std::string& directory_resource_id,
+                            const GetDataCallback& callback) {
+    // TODO: Make this more flexible.
+    if (feed_url == GURL("http://example.com/gdata/root_feed.json")) {
+      // Make some sample data.
+      const FilePath feed_path =
+          test_util::GetTestFilePath("gdata/root_feed.json");
+      std::string feed_contents;
+      file_util::ReadFileToString(feed_path, &feed_contents);
+      scoped_ptr<base::Value> feed_data(
+          base::JSONReader::Read(feed_contents));
+
+      base::MessageLoopProxy::current()->PostTask(FROM_HERE,
+          base::Bind(callback,
+                     HTTP_SUCCESS,
+                     base::Passed(&feed_data)));
+    } else {
+      scoped_ptr<base::Value> feed_data;
+      base::MessageLoopProxy::current()->PostTask(FROM_HERE,
+          base::Bind(callback,
+                     GDATA_PARSE_ERROR,
+                     base::Passed(&feed_data)));
+    }
+  }
+
+  virtual void GetDocumentEntry(const std::string& resource_id,
+                                const GetDataCallback& callback) {
+  }
+
+  virtual void GetAccountMetadata(const GetDataCallback& callback) {
+  }
+
+  virtual void GetApplicationInfo(const GetDataCallback& callback) {
+  }
+
+  virtual void DeleteDocument(const GURL& document_url,
+                              const EntryActionCallback& callback) {
+  }
+
+  virtual void DownloadDocument(const FilePath& virtual_path,
+                                const FilePath& local_cache_path,
+                                const GURL& content_url,
+                                DocumentExportFormat format,
+                                const DownloadActionCallback& callback) {
+  }
+
+  virtual void CopyDocument(const std::string& resource_id,
+                            const FilePath::StringType& new_name,
+                            const GetDataCallback& callback) {
+  }
+
+  virtual void RenameResource(const GURL& resource_url,
+                              const FilePath::StringType& new_name,
+                              const EntryActionCallback& callback) {
+  }
+
+  virtual void AddResourceToDirectory(const GURL& parent_content_url,
+                                      const GURL& resource_url,
+                                      const EntryActionCallback& callback) {
+  }
+
+  virtual void RemoveResourceFromDirectory(
+      const GURL& parent_content_url,
+      const std::string& resource_id,
+      const EntryActionCallback& callback) {
+  }
+
+  virtual void AddNewDirectory(const GURL& parent_content_url,
+                               const FilePath::StringType& directory_name,
+                               const GetDataCallback& callback) {
+  }
+
+  virtual void DownloadFile(
+      const FilePath& virtual_path,
+      const FilePath& local_cache_path,
+      const GURL& content_url,
+      const DownloadActionCallback& download_action_callback,
+      const GetContentCallback& get_content_callback) {
+  }
+
+  virtual void InitiateUpload(const InitiateUploadParams& params,
+                              const InitiateUploadCallback& callback) {
+  }
+
+  virtual void ResumeUpload(const ResumeUploadParams& params,
+                            const ResumeUploadCallback& callback) {
+  }
+
+  virtual void AuthorizeApp(const GURL& resource_url,
+                            const std::string& app_id,
+                            const GetDataCallback& callback) {
+  }
+
+};
+
+} // namespace google_apis
+
 namespace drive {
 
 namespace {
 
 class MockNetworkChangeNotifier : public net::NetworkChangeNotifier {
  public:
   MOCK_CONST_METHOD0(GetCurrentConnectionType,
                      net::NetworkChangeNotifier::ConnectionType());
 };
 
@@ skipping 56 matching lines @@
 class DriveSchedulerTest : public testing::Test {
  public:
   DriveSchedulerTest()
       : ui_thread_(content::BrowserThread::UI, &message_loop_),
         profile_(new TestingProfile) {
   }
 
   virtual void SetUp() OVERRIDE {
     mock_network_change_notifier_.reset(new MockNetworkChangeNotifier);
 
+    fake_drive_service_.reset(new google_apis::FakeDriveService());
     mock_copy_operation_ = new StrictMock<MockCopyOperation>();
     mock_move_operation_ = new StrictMock<MockMoveOperation>();
     mock_remove_operation_ = new StrictMock<MockRemoveOperation>();
     drive_operations_.InitForTesting(mock_copy_operation_,
                                      mock_move_operation_,
                                      mock_remove_operation_,
                                      NULL);
     scheduler_.reset(new DriveScheduler(profile_.get(),
+                                        fake_drive_service_.get(),
                                         &drive_operations_));
 
     scheduler_->Initialize();
     scheduler_->SetDisableThrottling(true);
   }
 
   virtual void TearDown() OVERRIDE {
     // The scheduler should be deleted before NetworkLibrary, as it
     // registers itself as observer of NetworkLibrary.
     scheduler_.reset();
     google_apis::test_util::RunBlockingPoolTask();
+    fake_drive_service_.reset();
     mock_network_change_notifier_.reset();
   }
 
   // Sets up MockNetworkChangeNotifier as if it's connected to a network with
   // the specified connection type.
   void ChangeConnectionType(net::NetworkChangeNotifier::ConnectionType type) {
     EXPECT_CALL(*mock_network_change_notifier_, GetCurrentConnectionType())
         .WillRepeatedly(Return(type));
     // Notify the sync client that the network is changed. This is done via
     // NetworkChangeNotifier in production, but here, we simulate the behavior
@@ skipping 20 matching lines @@
   void ConnectToNone() {
     ChangeConnectionType(net::NetworkChangeNotifier::CONNECTION_NONE);
   }
 
  protected:
   MessageLoopForUI message_loop_;
   content::TestBrowserThread ui_thread_;
   scoped_ptr<TestingProfile> profile_;
   scoped_ptr<DriveScheduler> scheduler_;
   scoped_ptr<MockNetworkChangeNotifier> mock_network_change_notifier_;
+  scoped_ptr<google_apis::FakeDriveService> fake_drive_service_;
 
   file_system::DriveOperations drive_operations_;
   StrictMock<MockCopyOperation>* mock_copy_operation_;
   StrictMock<MockMoveOperation>* mock_move_operation_;
   StrictMock<MockRemoveOperation>* mock_remove_operation_;
 };
 
 TEST_F(DriveSchedulerTest, CopyFile) {
   ConnectToWifi();
 
@@ skipping 58 matching lines @@
 
   DriveFileError error(DRIVE_FILE_ERROR_FAILED);
   scheduler_->TransferRegularFile(
       file_in_root, dest_file,
       base::Bind(&test_util::CopyErrorCodeFromFileOperationCallback, &error));
   google_apis::test_util::RunBlockingPoolTask();
 
   ASSERT_EQ(DRIVE_FILE_OK, error);
 }
 
+TEST_F(DriveSchedulerTest, GetDocuments) {
+  ConnectToWifi();
+
+  google_apis::GDataErrorCode error = google_apis::GDATA_OTHER_ERROR;
+  scoped_ptr<base::Value> value;
+
+  scheduler_->GetDocuments(
+      GURL("http://example.com/gdata/root_feed.json"),
+      0,
+      std::string(),
+      true,
+      std::string(),
+      base::Bind(&google_apis::test_util::CopyResultsFromGetDataCallback,
+                 &error,
+                 &value));
+  google_apis::test_util::RunBlockingPoolTask();
+
+  ASSERT_EQ(google_apis::HTTP_SUCCESS, error);
+  ASSERT_TRUE(value);
+}
+
 TEST_F(DriveSchedulerTest, MoveFile) {
   ConnectToWifi();
 
   FilePath file_in_root(FILE_PATH_LITERAL("drive/File 1.txt"));
   FilePath dest_file(FILE_PATH_LITERAL("drive/File 1.txt"));
   EXPECT_CALL(*mock_move_operation_, Move(file_in_root, dest_file, _))
       .WillOnce(MockOperation(DRIVE_FILE_OK));
 
   DriveFileError error(DRIVE_FILE_ERROR_FAILED);
   scheduler_->Move(
@@ skipping 149 matching lines @@
 
   FilePath file_in_root(FILE_PATH_LITERAL("drive/File 1.txt"));
   DriveFileError error;
   scheduler_->Remove(
       file_in_root, false,
       base::Bind(&test_util::CopyErrorCodeFromFileOperationCallback, &error));
   google_apis::test_util::RunBlockingPoolTask();
 }
 
 }  // namespace drive