Fetch files shared in Team Drives by specifying allTeamDrives corpora.

google_apis/drive/drive_api_url_generator_unittest.cc

 // Copyright (c) 2013 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/drive_api_url_generator.h"
 
 #include <stddef.h>
 #include <stdint.h>
 
 #include "base/macros.h"
@@ skipping 186 matching lines @@
     { 100, "", "query", "q=query" },
     { 150, "", "query", "maxResults=150&q=query" },
     { 10, "", "query", "maxResults=10&q=query" },
     { 100, "token", "query", "pageToken=token&q=query" },
     { 150, "token", "query", "maxResults=150&pageToken=token&q=query" },
     { 10, "token", "query", "maxResults=10&pageToken=token&q=query" },
   };
   const std::string kV2FilesUrlPrefix =
       "https://www.example.com/drive/v2/files";
   const std::string kV2FilesUrlPrefixWithTeamDrives =
-      "https://www.example.com/drive/v2/files?"
-      "supportsTeamDrives=true&includeTeamDriveItems=true";
+      "https://www.example.com/drive/v2/files?supportsTeamDrives=true&"
+      "includeTeamDriveItems=true&corpora=default%2CallTeamDrives";
 
   for (size_t i = 0; i < arraysize(kTestPatterns); ++i) {
     EXPECT_EQ(kV2FilesUrlPrefix +
                   (kTestPatterns[i].expected_query.empty() ? "" : "?") +
                   kTestPatterns[i].expected_query,
-              url_generator_.GetFilesListUrl(kTestPatterns[i].max_results,
-                                             kTestPatterns[i].page_token,
-                                             kTestPatterns[i].q).spec());
+              url_generator_
+                  .GetFilesListUrl(kTestPatterns[i].max_results,
+                                   kTestPatterns[i].page_token,
+                                   FilesListCorpora::DEFAULT, std::string(),
+                                   kTestPatterns[i].q)
+                  .spec());
     EXPECT_EQ(kV2FilesUrlPrefixWithTeamDrives +
                   (kTestPatterns[i].expected_query.empty() ? "" : "&") +
                   kTestPatterns[i].expected_query,
-              team_drives_url_generator_.GetFilesListUrl(
-                  kTestPatterns[i].max_results,
-                  kTestPatterns[i].page_token,
-                  kTestPatterns[i].q).spec());
+              team_drives_url_generator_
+                  .GetFilesListUrl(kTestPatterns[i].max_results,
+                                   kTestPatterns[i].page_token,
+                                   FilesListCorpora::ALL_TEAM_DRIVES,
+                                   std::string(), kTestPatterns[i].q)
+                  .spec());
   }
+
+  EXPECT_EQ(
+      "https://www.example.com/drive/v2/files?supportsTeamDrives=true&"
+      "includeTeamDriveItems=true&corpora=teamDrive&"
+      "teamDriveId=TheTeamDriveId&q=query",
+      team_drives_url_generator_
+          .GetFilesListUrl(100, std::string() /* page_token */,
+                           FilesListCorpora::TEAM_DRIVE, "TheTeamDriveId",
+                           "query")
+          .spec());
 }
 
 TEST_F(DriveApiUrlGeneratorTest, GetFilesDeleteUrl) {
   // |file_id| should be embedded into the url.
   EXPECT_EQ("https://www.example.com/drive/v2/files/0ADK06pfg",
             url_generator_.GetFilesDeleteUrl("0ADK06pfg").spec());
   EXPECT_EQ("https://www.example.com/drive/v2/files/0Bz0bd074",
             url_generator_.GetFilesDeleteUrl("0Bz0bd074").spec());
   EXPECT_EQ("https://www.example.com/drive/v2/files/file%3Afile_id",
             url_generator_.GetFilesDeleteUrl("file:file_id").spec());
@@ skipping 277 matching lines @@
             team_drives_url_generator_.GetTeamDriveListUrl(10, "").spec());
   EXPECT_EQ("https://www.example.com/drive/v2/teamdrives?maxResults=100",
             team_drives_url_generator_.GetTeamDriveListUrl(100, "").spec());
   EXPECT_EQ(
       "https://www.example.com/drive/v2/"
       "teamdrives?maxResults=100&pageToken=theToken",
       team_drives_url_generator_.GetTeamDriveListUrl(100, "theToken").spec());
 }
 
 }  // namespace google_apis