Include the HTTP status code as a feature in the ClientPhishingRequest.

chrome/browser/safe_browsing/browser_feature_extractor_unittest.cc

 // Copyright (c) 2011 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/safe_browsing/browser_feature_extractor.h"
 
 #include <map>
 #include <string>
 #include <vector>
 
@@ skipping 254 matching lines @@
   history_service()->AddPage(GURL("http://www.baz.com/"),
                              history::SOURCE_BROWSED);
 
   ClientPhishingRequest request;
   request.set_url("http://www.foo.com/");
   request.set_client_score(0.5);
   std::vector<GURL> redirect_chain;
   redirect_chain.push_back(GURL("http://somerandomwebsite.com/"));
   redirect_chain.push_back(GURL("http://www.foo.com/"));
   SetRedirectChain(redirect_chain, true);
+  browse_info_->http_status_code = 200;
   NavigateAndCommit(GURL("http://www.foo.com/"),
                     GURL("http://google.com/"),
                     content::PageTransitionFromInt(
                         content::PAGE_TRANSITION_AUTO_BOOKMARK |
                         content::PAGE_TRANSITION_FORWARD_BACK));
 
   EXPECT_TRUE(ExtractFeatures(&request));
   std::map<std::string, double> features;
   GetFeatureMap(request, &features);
 
   EXPECT_EQ(1.0,
             features[StringPrintf("%s=%s",
                                   features::kReferrer,
                                   "http://google.com/")]);
   EXPECT_EQ(1.0,
             features[StringPrintf("%s[0]=%s",
                                   features::kRedirect,
                                   "http://somerandomwebsite.com/")]);
   // We shouldn't have a feature for the last redirect in the chain, since it
   // should always be the URL that we navigated to.
   EXPECT_EQ(0.0,
             features[StringPrintf("%s[1]=%s",
                                   features::kRedirect,
                                   "http://foo.com/")]);
   EXPECT_EQ(0.0, features[features::kHasSSLReferrer]);
   EXPECT_EQ(2.0, features[features::kPageTransitionType]);
   EXPECT_EQ(1.0, features[features::kIsFirstNavigation]);
+  EXPECT_EQ(200.0, features[features::kHttpStatusCode]);
 
   request.Clear();
   request.set_url("http://www.foo.com/page.html");
   request.set_client_score(0.5);
   redirect_chain.clear();
   redirect_chain.push_back(GURL("http://www.foo.com/redirect"));
   redirect_chain.push_back(GURL("http://www.foo.com/second_redirect"));
   redirect_chain.push_back(GURL("http://www.foo.com/page.html"));
   SetRedirectChain(redirect_chain, false);
+  browse_info_->http_status_code = 404;
   NavigateAndCommit(GURL("http://www.foo.com/page.html"),
                     GURL("http://www.foo.com"),
                     content::PageTransitionFromInt(
                         content::PAGE_TRANSITION_TYPED |
                         content::PAGE_TRANSITION_CHAIN_START |
                         content::PAGE_TRANSITION_CLIENT_REDIRECT));
 
   EXPECT_TRUE(ExtractFeatures(&request));
   features.clear();
   GetFeatureMap(request, &features);
@@ skipping 24 matching lines @@
                                   features::kRedirect,
                                   "http://somerandomwebsite.com/")]);
   EXPECT_EQ(2.0,
             features[StringPrintf("%s%s",
                                   features::kHostPrefix,
                                   features::kPageTransitionType)]);
   EXPECT_EQ(1.0,
             features[StringPrintf("%s%s",
                                   features::kHostPrefix,
                                   features::kIsFirstNavigation)]);
+  EXPECT_EQ(404.0, features[features::kHttpStatusCode]);
 
   request.Clear();
   request.set_url("http://www.bar.com/");
   request.set_client_score(0.5);
   redirect_chain.clear();
   redirect_chain.push_back(GURL("http://www.foo.com/page.html"));
   redirect_chain.push_back(GURL("http://www.bar.com/"));
   SetRedirectChain(redirect_chain, true);
   NavigateAndCommit(GURL("http://www.bar.com/"),
                     GURL("http://www.foo.com/page.html"),
@@ skipping 126 matching lines @@
   EXPECT_TRUE(features.count(StringPrintf("%s%s",
                                           features::kSafeBrowsingMaliciousUrl,
                                           "http://www.malware.com/")));
   EXPECT_TRUE(features.count(StringPrintf("%s%s",
                                           features::kSafeBrowsingOriginalUrl,
                                           "http://www.good.com/")));
   EXPECT_DOUBLE_EQ(1.0, features[features::kSafeBrowsingIsSubresource]);
   EXPECT_DOUBLE_EQ(2.0, features[features::kSafeBrowsingThreatType]);
 }
 }  // namespace safe_browsing