Add an extractor for DOM features to be used for client side phishing detection.

chrome/renderer/safe_browsing/features_unittest.cc

 // Copyright (c) 2010 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/renderer/safe_browsing/features.h"
 
 #include "base/format_macros.h"
 #include "base/string_util.h"
+#include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace safe_browsing {
 
 TEST(PhishingFeaturesTest, TooManyFeatures) {
   FeatureMap features;
   for (size_t i = 0; i < FeatureMap::kMaxFeatureMapSize; ++i) {
     EXPECT_TRUE(features.AddBooleanFeature(StringPrintf("Feature%" PRIuS, i)));
   }
   EXPECT_EQ(FeatureMap::kMaxFeatureMapSize, features.features().size());
 
   // Attempting to add more features should fail.
   for (size_t i = 0; i < 3; ++i) {
     EXPECT_FALSE(features.AddBooleanFeature(StringPrintf("Extra%" PRIuS, i)));
   }
   EXPECT_EQ(FeatureMap::kMaxFeatureMapSize, features.features().size());
 }
 
+TEST(PhishingFeaturesTest, IllegalFeatureValue) {
+  FeatureMap features;
+  EXPECT_FALSE(features.AddRealFeature("toosmall", -0.1));
+  EXPECT_TRUE(features.AddRealFeature("zero", 0.0));
+  EXPECT_TRUE(features.AddRealFeature("pointfive", 0.5));
+  EXPECT_TRUE(features.AddRealFeature("one", 1.0));
+  EXPECT_FALSE(features.AddRealFeature("toolarge", 1.1));
+
+  FeatureMap expected_features;
+  expected_features.AddRealFeature("zero", 0.0);
+  expected_features.AddRealFeature("pointfive", 0.5);
+  expected_features.AddRealFeature("one", 1.0);
+  EXPECT_THAT(features.features(),
+              ::testing::ContainerEq(expected_features.features()));
+}
+
 }  // namespace safe_browsing