Use FileTypePolicies for is_archive and is_supported classifications.

chrome/common/safe_browsing/download_protection_util_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 "chrome/common/safe_browsing/download_protection_util.h"
 
 #include "base/files/file_path.h"
+#include "chrome/common/safe_browsing/file_type_policies.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace safe_browsing {
 namespace download_protection_util {
 
-TEST(DownloadProtectionUtilTest, KnownValues) {
-  EXPECT_EQ(0, GetSBClientDownloadExtensionValueForUMA(
-                   base::FilePath(FILE_PATH_LITERAL("foo.exe"))));
-  EXPECT_EQ(0, GetSBClientDownloadExtensionValueForUMA(
-                   base::FilePath(FILE_PATH_LITERAL("foo.exe."))));
-  EXPECT_EQ(0, GetSBClientDownloadExtensionValueForUMA(
-                   base::FilePath(FILE_PATH_LITERAL("foo.exe "))));
-  EXPECT_EQ(0, GetSBClientDownloadExtensionValueForUMA(
-                   base::FilePath(FILE_PATH_LITERAL("foo.exe.. . . .   "))));
-  EXPECT_EQ(9, GetSBClientDownloadExtensionValueForUMA(
-                   base::FilePath(FILE_PATH_LITERAL("foo.dll"))));
-  EXPECT_EQ(29, GetSBClientDownloadExtensionValueForUMA(
-                    base::FilePath(FILE_PATH_LITERAL("foo.jse"))));
-  EXPECT_EQ(18, GetSBClientDownloadExtensionValueForUMA(
-                    base::FilePath(FILE_PATH_LITERAL("foo.111"))));
-  EXPECT_EQ(18, GetSBClientDownloadExtensionValueForUMA(
-                    base::FilePath(FILE_PATH_LITERAL("foo.zzz"))));
-  EXPECT_EQ(152, GetSBClientDownloadExtensionValueForUMA(
-                    base::FilePath(FILE_PATH_LITERAL("foo.tbz2"))));
-  EXPECT_TRUE(IsArchiveFile(base::FilePath(FILE_PATH_LITERAL("foo.zip"))));
-  EXPECT_TRUE(IsArchiveFile(base::FilePath(FILE_PATH_LITERAL("foo.zip."))));
-  EXPECT_TRUE(IsArchiveFile(base::FilePath(FILE_PATH_LITERAL("foo.zip "))));
-  EXPECT_TRUE(
-      IsArchiveFile(base::FilePath(FILE_PATH_LITERAL("foo.zip   ...."))));
-  EXPECT_TRUE(
-      IsSupportedBinaryFile(base::FilePath(FILE_PATH_LITERAL("foo.zip"))));
-  EXPECT_TRUE(IsArchiveFile(base::FilePath(FILE_PATH_LITERAL("foo.rar"))));
-  EXPECT_TRUE(
-      IsSupportedBinaryFile(base::FilePath(FILE_PATH_LITERAL("foo.rar"))));
-  EXPECT_FALSE(IsArchiveFile(base::FilePath(FILE_PATH_LITERAL("foo.ps1"))));
-  EXPECT_FALSE(
-      IsArchiveFile(base::FilePath(FILE_PATH_LITERAL("foo.unknownextension"))));
-  EXPECT_FALSE(
-      IsArchiveFile(base::FilePath(FILE_PATH_LITERAL("no_extension"))));
-  EXPECT_TRUE(IsSupportedBinaryFile(base::FilePath(FILE_PATH_LITERAL(".exe"))));
-  EXPECT_TRUE(
-      IsSupportedBinaryFile(base::FilePath(FILE_PATH_LITERAL("foo.msi"))));
-  EXPECT_TRUE(
-      IsSupportedBinaryFile(base::FilePath(FILE_PATH_LITERAL("foo.msi."))));
-  EXPECT_TRUE(
-      IsSupportedBinaryFile(base::FilePath(FILE_PATH_LITERAL("foo.msi "))));
-  EXPECT_TRUE(IsSupportedBinaryFile(
-      base::FilePath(FILE_PATH_LITERAL("foo.msi  .. .."))));
-  EXPECT_FALSE(
-      IsSupportedBinaryFile(base::FilePath(FILE_PATH_LITERAL("foo.txt"))));
-  EXPECT_FALSE(IsSupportedBinaryFile(
-      base::FilePath(FILE_PATH_LITERAL("foo.unknownextension"))));
-  EXPECT_FALSE(
-      IsSupportedBinaryFile(base::FilePath(FILE_PATH_LITERAL("no_extension"))));
+class DownloadProtectionUtilTest : public testing::Test {
+ public:
+  ~DownloadProtectionUtilTest() override {}
 
+ protected:
+  FileTypePoliciesManager file_type_policies_manager_;
+};
+
+TEST_F(DownloadProtectionUtilTest, KnownValues) {
   EXPECT_EQ(ClientDownloadRequest::WIN_EXECUTABLE,
             GetDownloadType(base::FilePath(FILE_PATH_LITERAL("foo.exe"))));
   EXPECT_EQ(ClientDownloadRequest::CHROME_EXTENSION,
             GetDownloadType(base::FilePath(FILE_PATH_LITERAL("foo.crx"))));
   EXPECT_EQ(ClientDownloadRequest::ZIPPED_EXECUTABLE,
             GetDownloadType(base::FilePath(FILE_PATH_LITERAL("foo.zip"))));
   EXPECT_EQ(ClientDownloadRequest::ARCHIVE,
             GetDownloadType(base::FilePath(FILE_PATH_LITERAL("foo.rar"))));
   EXPECT_EQ(ClientDownloadRequest::MAC_EXECUTABLE,
             GetDownloadType(base::FilePath(FILE_PATH_LITERAL("foo.pkg"))));
   EXPECT_EQ(ClientDownloadRequest::ANDROID_APK,
             GetDownloadType(base::FilePath(FILE_PATH_LITERAL("foo.apk"))));
 }
 
 } // namespace download_protection_util
 } // namespace safe_browsing