Fetch files shared in Team Drives by specifying allTeamDrives corpora.

google_apis/drive/files_list_request_runner_unittest.cc

 // Copyright 2015 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 "google_apis/drive/files_list_request_runner.h"
 
 #include <memory>
 #include <utility>
 
 #include "base/message_loop/message_loop.h"
@@ skipping 121 matching lines @@
 
   // A requests and a response stored for verification in test cases.
   std::unique_ptr<net::test_server::HttpRequest> http_request_;
   std::unique_ptr<DriveApiErrorCode> response_error_;
   std::unique_ptr<FileList> response_entry_;
 };
 
 TEST_F(FilesListRequestRunnerTest, Success_NoBackoff) {
   SetFakeServerResponse(net::HTTP_OK, kSuccessResource);
   runner_->CreateAndStartWithSizeBackoff(
-      kMaxResults, kQuery, kFields,
+      kMaxResults, FilesListCorpora::DEFAULT, std::string(), kQuery, kFields,
       base::Bind(&FilesListRequestRunnerTest::OnCompleted,
                  base::Unretained(this)));
 
   base::RunLoop run_loop;
   on_completed_callback_ = run_loop.QuitClosure();
   run_loop.Run();
 
   ASSERT_TRUE(http_request_.get());
   EXPECT_EQ(
       "/drive/v2/files?maxResults=4&q=testing-query&fields=testing-fields",
       http_request_->relative_url);
 
   ASSERT_TRUE(response_error_.get());
   EXPECT_EQ(HTTP_SUCCESS, *response_error_);
   EXPECT_TRUE(response_entry_.get());
 }
 
 TEST_F(FilesListRequestRunnerTest, Success_Backoff) {
   SetFakeServerResponse(net::HTTP_INTERNAL_SERVER_ERROR,
                         kResponseTooLargeErrorResource);
   runner_->CreateAndStartWithSizeBackoff(
-      kMaxResults, kQuery, kFields,
+      kMaxResults, FilesListCorpora::DEFAULT, std::string(), kQuery, kFields,
       base::Bind(&FilesListRequestRunnerTest::OnCompleted,
                  base::Unretained(this)));
   {
     base::RunLoop run_loop;
     runner_->SetRequestCompletedCallbackForTesting(run_loop.QuitClosure());
     run_loop.Run();
 
     ASSERT_TRUE(http_request_.get());
     EXPECT_EQ(
         "/drive/v2/files?maxResults=4&q=testing-query&fields=testing-fields",
@@ skipping 17 matching lines @@
     ASSERT_TRUE(response_error_.get());
     EXPECT_EQ(HTTP_SUCCESS, *response_error_);
     EXPECT_TRUE(response_entry_.get());
   }
 }
 
 TEST_F(FilesListRequestRunnerTest, Failure_TooManyBackoffs) {
   SetFakeServerResponse(net::HTTP_INTERNAL_SERVER_ERROR,
                         kResponseTooLargeErrorResource);
   runner_->CreateAndStartWithSizeBackoff(
-      kMaxResults, kQuery, kFields,
+      kMaxResults, FilesListCorpora::DEFAULT, std::string(), kQuery, kFields,
       base::Bind(&FilesListRequestRunnerTest::OnCompleted,
                  base::Unretained(this)));
   {
     base::RunLoop run_loop;
     runner_->SetRequestCompletedCallbackForTesting(run_loop.QuitClosure());
     run_loop.Run();
 
     ASSERT_TRUE(http_request_.get());
     EXPECT_EQ(
         "/drive/v2/files?maxResults=4&q=testing-query&fields=testing-fields",
@@ skipping 35 matching lines @@
     ASSERT_TRUE(response_error_.get());
     EXPECT_EQ(DRIVE_RESPONSE_TOO_LARGE, *response_error_);
     EXPECT_FALSE(response_entry_.get());
   }
 }
 
 TEST_F(FilesListRequestRunnerTest, Failure_AnotherError) {
   SetFakeServerResponse(net::HTTP_INTERNAL_SERVER_ERROR,
                         kQuotaExceededErrorResource);
   runner_->CreateAndStartWithSizeBackoff(
-      kMaxResults, kQuery, kFields,
+      kMaxResults, FilesListCorpora::DEFAULT, std::string(), kQuery, kFields,
       base::Bind(&FilesListRequestRunnerTest::OnCompleted,
                  base::Unretained(this)));
 
   base::RunLoop run_loop;
   on_completed_callback_ = run_loop.QuitClosure();
   run_loop.Run();
 
   ASSERT_TRUE(http_request_.get());
   EXPECT_EQ(
       "/drive/v2/files?maxResults=4&q=testing-query&fields=testing-fields",
       http_request_->relative_url);
 
   // There must be no backoff in case of an error different than
   // DRIVE_RESPONSE_TOO_LARGE.
   ASSERT_TRUE(response_error_.get());
   EXPECT_EQ(DRIVE_NO_SPACE, *response_error_);
   EXPECT_FALSE(response_entry_.get());
 }
 
 }  // namespace google_apis