Fetch file metadata of files under Team Drives.

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"
 #include "google_apis/drive/test_util.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "url/gurl.h"
 #include "url/url_util.h"
 
 namespace google_apis {
 namespace {
 // The URLs used for production may be different for Chromium OS and Chrome
 // OS, so use testing base urls.
 const char kBaseUrlForTesting[] = "https://www.example.com";
 const char kBaseThumbnailUrlForTesting[] = "https://thumbnail.example.com";
 }  // namespace
 
 class DriveApiUrlGeneratorTest : public testing::Test {
  public:
   DriveApiUrlGeneratorTest()
       : url_generator_(GURL(kBaseUrlForTesting),
-                       GURL(kBaseThumbnailUrlForTesting)) {}
+                       GURL(kBaseThumbnailUrlForTesting),
+                       TEAM_DRIVES_INTEGRATION_DISABLED),
+        team_drives_url_generator_(GURL(kBaseUrlForTesting),
+                                   GURL(kBaseThumbnailUrlForTesting),
+                                   TEAM_DRIVES_INTEGRATION_ENABLED) {}
 
  protected:
   DriveApiUrlGenerator url_generator_;
+  DriveApiUrlGenerator team_drives_url_generator_;
 };
 
 // Make sure the hard-coded urls are returned.
 TEST_F(DriveApiUrlGeneratorTest, GetAboutGetUrl) {
   EXPECT_EQ("https://www.example.com/drive/v2/about",
             url_generator_.GetAboutGetUrl().spec());
 }
 
 TEST_F(DriveApiUrlGeneratorTest, GetAppsListUrl) {
   const bool use_internal_url = true;
@@ skipping 10 matching lines @@
 
 TEST_F(DriveApiUrlGeneratorTest, GetFilesGetUrl) {
   // |file_id| should be embedded into the url.
   EXPECT_EQ("https://www.example.com/drive/v2/files/0ADK06pfg",
             url_generator_.GetFilesGetUrl("0ADK06pfg", false, GURL()).spec());
   EXPECT_EQ("https://www.example.com/drive/v2/files/0Bz0bd074",
             url_generator_.GetFilesGetUrl("0Bz0bd074", false, GURL()).spec());
   EXPECT_EQ(
       "https://www.example.com/drive/v2/files/file%3Afile_id",
       url_generator_.GetFilesGetUrl("file:file_id", false, GURL()).spec());
+  EXPECT_EQ("https://www.example.com/drive/v2beta/files/0Bz0bd074"
+                "?supportsTeamDrives=true",
+            team_drives_url_generator_.GetFilesGetUrl(
+                "0Bz0bd074", false, GURL()).spec());
 
   // If |use_internal_endpoint| is true, the generated url should point to the
   // v2internal.
   EXPECT_EQ("https://www.example.com/drive/v2internal/files/0ADK06pfg",
             url_generator_.GetFilesGetUrl("0ADK06pfg", true, GURL()).spec());
 
   // If |embed_origin| is not empty, it should be added as a query parameter.
   url::AddStandardScheme("chrome-extension", url::SCHEME_WITHOUT_PORT);
   EXPECT_EQ(
       "https://www.example.com/drive/v2/files/0ADK06pfg"
@@ skipping 77 matching lines @@
 
 TEST_F(DriveApiUrlGeneratorTest, GetFilesListUrl) {
   struct TestPattern {
     int max_results;
     const std::string page_token;
     const std::string q;
     const std::string expected_query;
   };
   const TestPattern kTestPatterns[] = {
     { 100, "", "", "" },
-    { 150, "", "", "?maxResults=150" },
-    { 10, "", "", "?maxResults=10" },
-    { 100, "token", "", "?pageToken=token" },
-    { 150, "token", "", "?maxResults=150&pageToken=token" },
-    { 10, "token", "", "?maxResults=10&pageToken=token" },
-    { 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" },
+    { 150, "", "", "maxResults=150" },
+    { 10, "", "", "maxResults=10" },
+    { 100, "token", "", "pageToken=token" },
+    { 150, "token", "", "maxResults=150&pageToken=token" },
+    { 10, "token", "", "maxResults=10&pageToken=token" },
+    { 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 kV2BetaFilesUrlPrefix =
+      "https://www.example.com/drive/v2beta/files?"
+      "supportsTeamDrives=true&includeTeamDriveItems=true";
 
   for (size_t i = 0; i < arraysize(kTestPatterns); ++i) {
-    EXPECT_EQ("https://www.example.com/drive/v2/files" +
+    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());
+    EXPECT_EQ(kV2BetaFilesUrlPrefix +
+                  (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());
   }
 }
 
 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",
@@ skipping 13 matching lines @@
 TEST_F(DriveApiUrlGeneratorTest, GetChangesListUrl) {
   struct TestPattern {
     bool include_deleted;
     int max_results;
     const std::string page_token;
     int64_t start_change_id;
     const std::string expected_query;
   };
   const TestPattern kTestPatterns[] = {
     { true, 100, "", 0, "" },
-    { false, 100, "", 0, "?includeDeleted=false" },
-    { true, 150, "", 0, "?maxResults=150" },
-    { false, 150, "", 0, "?includeDeleted=false&maxResults=150" },
-    { true, 10, "", 0, "?maxResults=10" },
-    { false, 10, "", 0, "?includeDeleted=false&maxResults=10" },
+    { false, 100, "", 0, "includeDeleted=false" },
+    { true, 150, "", 0, "maxResults=150" },
+    { false, 150, "", 0, "includeDeleted=false&maxResults=150" },
+    { true, 10, "", 0, "maxResults=10" },
+    { false, 10, "", 0, "includeDeleted=false&maxResults=10" },
 
-    { true, 100, "token", 0, "?pageToken=token" },
-    { false, 100, "token", 0, "?includeDeleted=false&pageToken=token" },
-    { true, 150, "token", 0, "?maxResults=150&pageToken=token" },
+    { true, 100, "token", 0, "pageToken=token" },
+    { false, 100, "token", 0, "includeDeleted=false&pageToken=token" },
+    { true, 150, "token", 0, "maxResults=150&pageToken=token" },
     { false, 150, "token", 0,
-      "?includeDeleted=false&maxResults=150&pageToken=token" },
-    { true, 10, "token", 0, "?maxResults=10&pageToken=token" },
+      "includeDeleted=false&maxResults=150&pageToken=token" },
+    { true, 10, "token", 0, "maxResults=10&pageToken=token" },
     { false, 10, "token", 0,
-      "?includeDeleted=false&maxResults=10&pageToken=token" },
+      "includeDeleted=false&maxResults=10&pageToken=token" },
 
-    { true, 100, "", 12345, "?startChangeId=12345" },
-    { false, 100, "", 12345, "?includeDeleted=false&startChangeId=12345" },
-    { true, 150, "", 12345, "?maxResults=150&startChangeId=12345" },
+    { true, 100, "", 12345, "startChangeId=12345" },
+    { false, 100, "", 12345, "includeDeleted=false&startChangeId=12345" },
+    { true, 150, "", 12345, "maxResults=150&startChangeId=12345" },
     { false, 150, "", 12345,
-      "?includeDeleted=false&maxResults=150&startChangeId=12345" },
-    { true, 10, "", 12345, "?maxResults=10&startChangeId=12345" },
+      "includeDeleted=false&maxResults=150&startChangeId=12345" },
+    { true, 10, "", 12345, "maxResults=10&startChangeId=12345" },
     { false, 10, "", 12345,
-      "?includeDeleted=false&maxResults=10&startChangeId=12345" },
+      "includeDeleted=false&maxResults=10&startChangeId=12345" },
 
-    { true, 100, "token", 12345, "?pageToken=token&startChangeId=12345" },
+    { true, 100, "token", 12345, "pageToken=token&startChangeId=12345" },
     { false, 100, "token", 12345,
-      "?includeDeleted=false&pageToken=token&startChangeId=12345" },
+      "includeDeleted=false&pageToken=token&startChangeId=12345" },
     { true, 150, "token", 12345,
-      "?maxResults=150&pageToken=token&startChangeId=12345" },
+      "maxResults=150&pageToken=token&startChangeId=12345" },
     { false, 150, "token", 12345,
-      "?includeDeleted=false&maxResults=150&pageToken=token"
+      "includeDeleted=false&maxResults=150&pageToken=token"
       "&startChangeId=12345" },
     { true, 10, "token", 12345,
-      "?maxResults=10&pageToken=token&startChangeId=12345" },
+      "maxResults=10&pageToken=token&startChangeId=12345" },
     { false, 10, "token", 12345,
-      "?includeDeleted=false&maxResults=10&pageToken=token"
+      "includeDeleted=false&maxResults=10&pageToken=token"
       "&startChangeId=12345" },
   };
 
+  const std::string kV2ChangesUrlPrefix =
+      "https://www.example.com/drive/v2/changes";
+  const std::string kV2BetaChangesUrlPrefix =
+      "https://www.example.com/drive/v2beta/changes?"
+      "supportsTeamDrives=true&includeTeamDriveItems=true";
   for (size_t i = 0; i < arraysize(kTestPatterns); ++i) {
-    EXPECT_EQ("https://www.example.com/drive/v2/changes" +
+    EXPECT_EQ(kV2ChangesUrlPrefix +
+                  (kTestPatterns[i].expected_query.empty() ? "" : "?") +
                   kTestPatterns[i].expected_query,
               url_generator_.GetChangesListUrl(kTestPatterns[i].include_deleted,
                                                kTestPatterns[i].max_results,
                                                kTestPatterns[i].page_token,
                                                kTestPatterns[i].start_change_id)
                   .spec());
+    EXPECT_EQ(kV2BetaChangesUrlPrefix +
+                  (kTestPatterns[i].expected_query.empty() ? "" : "&") +
+                  kTestPatterns[i].expected_query,
+              team_drives_url_generator_.GetChangesListUrl(
+                  kTestPatterns[i].include_deleted,
+                  kTestPatterns[i].max_results,
+                  kTestPatterns[i].page_token,
+                  kTestPatterns[i].start_change_id)
+                  .spec());
   }
 }
 
 TEST_F(DriveApiUrlGeneratorTest, GetChildrenInsertUrl) {
   // |file_id| should be embedded into the url.
   EXPECT_EQ("https://www.example.com/drive/v2/files/0ADK06pfg/children",
             url_generator_.GetChildrenInsertUrl("0ADK06pfg").spec());
   EXPECT_EQ("https://www.example.com/drive/v2/files/0Bz0bd074/children",
             url_generator_.GetChildrenInsertUrl("0Bz0bd074").spec());
   EXPECT_EQ("https://www.example.com/drive/v2/files/file%3Afolder_id/children",
@@ skipping 114 matching lines @@
       "https://thumbnail.example.com/d/0ADK06pfg=w360-h380-c",
       url_generator_.GetThumbnailUrl("0ADK06pfg", 360, 380, true).spec());
 }
 
 TEST_F(DriveApiUrlGeneratorTest, BatchUploadUrl) {
   EXPECT_EQ("https://www.example.com/upload/drive",
             url_generator_.GetBatchUploadUrl().spec());
 }
 
 }  // namespace google_apis