Use FileTypePolicies for download danger classifications.

chrome/browser/download/download_item_model_unittest.cc

 // Copyright (c) 2012 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/download/download_item_model.h"
 
 #include <stddef.h>
 #include <stdint.h>
 
 #include <vector>
 
 #include "base/i18n/rtl.h"
 #include "base/logging.h"
 #include "base/macros.h"
 #include "base/message_loop/message_loop.h"
 #include "base/strings/string16.h"
 #include "base/strings/string_split.h"
 #include "base/strings/string_util.h"
 #include "base/strings/utf_string_conversions.h"
 #include "content/public/test/mock_download_item.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "ui/base/resource/resource_bundle.h"
 #include "ui/base/text/bytes_formatting.h"
 #include "ui/gfx/font_list.h"
 #include "ui/gfx/text_utils.h"
 
 using content::DownloadItem;
+using safe_browsing::DownloadFileType;
 using ::testing::Mock;
 using ::testing::NiceMock;
 using ::testing::Return;
 using ::testing::ReturnRefOfCopy;
 using ::testing::SetArgPointee;
 using ::testing::_;
 
 namespace {
 
 // Create a char array that has as many elements as there are download
@@ skipping 329 matching lines @@
   EXPECT_FALSE(model().ShouldShowInShelf());
 
   model().SetShouldShowInShelf(true);
   EXPECT_TRUE(model().ShouldShowInShelf());
 }
 
 TEST_F(DownloadItemModelTest, DangerLevel) {
   SetupDownloadItemDefaults();
 
   // Default danger level is NOT_DANGEROUS.
-  EXPECT_EQ(download_util::NOT_DANGEROUS, model().GetDangerLevel());
+  EXPECT_EQ(DownloadFileType::NOT_DANGEROUS, model().GetDangerLevel());
 
-  model().SetDangerLevel(download_util::ALLOW_ON_USER_GESTURE);
-  EXPECT_EQ(download_util::ALLOW_ON_USER_GESTURE, model().GetDangerLevel());
+  model().SetDangerLevel(DownloadFileType::ALLOW_ON_USER_GESTURE);
+  EXPECT_EQ(DownloadFileType::ALLOW_ON_USER_GESTURE, model().GetDangerLevel());
 }
 
 TEST_F(DownloadItemModelTest, ShouldRemoveFromShelfWhenComplete) {
   const struct TestCase {
     DownloadItem::DownloadState state;
     bool is_dangerous;  // Expectation for IsDangerous().
     bool is_auto_open;  // Expectation for GetOpenWhenComplete().
     bool auto_opened;   // Whether the download was successfully
                         // auto-opened. Expecation for GetAutoOpened().
     bool expected_result;
@@ skipping 36 matching lines @@
     EXPECT_CALL(item(), GetAutoOpened())
         .WillRepeatedly(Return(test_case.auto_opened));
 
     EXPECT_EQ(test_case.expected_result,
               model().ShouldRemoveFromShelfWhenComplete())
         << "Test case: " << i;
     Mock::VerifyAndClearExpectations(&item());
     Mock::VerifyAndClearExpectations(&model());
   }
 }