Unify HasGDocFileExtension and IsHostedDocumentByExtentoin.

chrome/browser/drive/drive_api_util_unittest.cc

 // Copyright 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 "chrome/browser/drive/drive_api_util.h"
 
 #include "base/files/scoped_temp_dir.h"
 #include "base/md5.h"
 #include "google_apis/drive/drive_api_parser.h"
 #include "google_apis/drive/gdata_wapi_parser.h"
@@ skipping 146 matching lines @@
   base::ScopedTempDir temp_dir;
   ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
 
   base::FilePath path = temp_dir.path().AppendASCII("test.txt");
   const char kTestData[] = "abcdefghijklmnopqrstuvwxyz0123456789";
   ASSERT_TRUE(google_apis::test_util::WriteStringToFile(path, kTestData));
 
   EXPECT_EQ(base::MD5String(kTestData), GetMd5Digest(path));
 }
 
+TEST(DriveAPIUtilTest, HasHostedDocumentExtension) {
+  EXPECT_TRUE(
+      HasHostedDocumentExtension(base::FilePath::FromUTF8Unsafe("xx.gdoc")));
+  EXPECT_TRUE(
+      HasHostedDocumentExtension(base::FilePath::FromUTF8Unsafe("xx.gsheet")));
+  EXPECT_TRUE(
+      HasHostedDocumentExtension(base::FilePath::FromUTF8Unsafe("xx.gslides")));
+  EXPECT_TRUE(
+      HasHostedDocumentExtension(base::FilePath::FromUTF8Unsafe("xx.gdraw")));
+  EXPECT_TRUE(
+      HasHostedDocumentExtension(base::FilePath::FromUTF8Unsafe("xx.gtable")));
+  EXPECT_TRUE(
+      HasHostedDocumentExtension(base::FilePath::FromUTF8Unsafe("xx.gform")));
+
+  EXPECT_FALSE(
+      HasHostedDocumentExtension(base::FilePath::FromUTF8Unsafe("xx.gdocs")));
+  EXPECT_FALSE(
+      HasHostedDocumentExtension(base::FilePath::FromUTF8Unsafe("xx.docx")));
+  EXPECT_FALSE(
+      HasHostedDocumentExtension(base::FilePath::FromUTF8Unsafe("xx.jpg")));
+}
+
 }  // namespace util
 }  // namespace drive