Move PageTransition from //content/public/common to //ui/base

chrome/browser/safe_browsing/browser_feature_extractor_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/safe_browsing/browser_feature_extractor.h"
 
 #include <map>
 #include <string>
 #include <vector>
 
 #include "base/memory/scoped_ptr.h"
 #include "base/message_loop/message_loop.h"
 #include "base/strings/stringprintf.h"
 #include "base/time/time.h"
 #include "chrome/browser/history/history_backend.h"
 #include "chrome/browser/history/history_service.h"
 #include "chrome/browser/history/history_service_factory.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/safe_browsing/browser_features.h"
 #include "chrome/browser/safe_browsing/client_side_detection_host.h"
 #include "chrome/browser/safe_browsing/database_manager.h"
 #include "chrome/browser/safe_browsing/safe_browsing_service.h"
 #include "chrome/browser/safe_browsing/ui_manager.h"
 #include "chrome/common/safe_browsing/csd.pb.h"
 #include "chrome/test/base/chrome_render_view_host_test_harness.h"
 #include "chrome/test/base/testing_profile.h"
 #include "content/public/browser/navigation_controller.h"
 #include "content/public/browser/web_contents.h"
-#include "content/public/common/page_transition_types.h"
 #include "content/public/common/referrer.h"
 #include "content/public/test/test_browser_thread.h"
 #include "content/public/test/web_contents_tester.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
+#include "ui/base/page_transition_types.h"
 #include "url/gurl.h"
 
 using content::BrowserThread;
 using content::ResourceType;
 using content::WebContentsTester;
 
 using testing::DoAll;
 using testing::Return;
 using testing::StrictMock;
 
@@ skipping 69 matching lines @@
       browse_info_->host_redirects = redirect_chain;
     }
   }
 
   // Wrapper around NavigateAndCommit that also sets the redirect chain to
   // a sane value.
   void SimpleNavigateAndCommit(const GURL& url) {
     std::vector<GURL> redirect_chain;
     redirect_chain.push_back(url);
     SetRedirectChain(redirect_chain, true);
-    NavigateAndCommit(url, GURL(), content::PAGE_TRANSITION_LINK);
+    NavigateAndCommit(url, GURL(), ui::PAGE_TRANSITION_LINK);
   }
 
   // This is similar to NavigateAndCommit that is in WebContentsTester, but
   // allows us to specify the referrer and page_transition_type.
   void NavigateAndCommit(const GURL& url,
                          const GURL& referrer,
-                         content::PageTransition type) {
+                         ui::PageTransition type) {
     web_contents()->GetController().LoadURL(
         url, content::Referrer(referrer, blink::WebReferrerPolicyDefault),
         type, std::string());
 
     static int page_id = 0;
     content::RenderFrameHost* rfh =
         WebContentsTester::For(web_contents())->GetPendingMainFrame();
     if (!rfh) {
       rfh = web_contents()->GetMainFrame();
     }
@@ skipping 120 matching lines @@
                              history::SOURCE_BROWSED);  // same host HTTPS.
   history_service()->AddPage(GURL("http://www.foo.com/gaa.html"),
                              base::Time::Now(),
                              history::SOURCE_BROWSED);  // same host HTTP.
   history_service()->AddPage(GURL("http://bar.foo.com/gaa.html"),
                              base::Time::Now(),
                              history::SOURCE_BROWSED);  // different host.
   history_service()->AddPage(GURL("http://www.foo.com/bar.html?a=b"),
                              base::Time::Now() - base::TimeDelta::FromHours(23),
                              NULL, 0, GURL(), history::RedirectList(),
-                             content::PAGE_TRANSITION_LINK,
+                             ui::PAGE_TRANSITION_LINK,
                              history::SOURCE_BROWSED, false);
   history_service()->AddPage(GURL("http://www.foo.com/bar.html"),
                              base::Time::Now() - base::TimeDelta::FromHours(25),
                              NULL, 0, GURL(), history::RedirectList(),
-                             content::PAGE_TRANSITION_TYPED,
+                             ui::PAGE_TRANSITION_TYPED,
                              history::SOURCE_BROWSED, false);
   history_service()->AddPage(GURL("https://www.foo.com/goo.html"),
                              base::Time::Now() - base::TimeDelta::FromDays(5),
                              NULL, 0, GURL(), history::RedirectList(),
-                             content::PAGE_TRANSITION_TYPED,
+                             ui::PAGE_TRANSITION_TYPED,
                              history::SOURCE_BROWSED, false);
 
   SimpleNavigateAndCommit(GURL("http://www.foo.com/bar.html"));
 
   ClientPhishingRequest request;
   request.set_url("http://www.foo.com/bar.html");
   request.set_client_score(0.5);
   EXPECT_TRUE(ExtractFeatures(&request));
   std::map<std::string, double> features;
   GetFeatureMap(request, &features);
@@ skipping 72 matching lines @@
   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));
+                    ui::PageTransitionFromInt(
+                        ui::PAGE_TRANSITION_AUTO_BOOKMARK |
+                        ui::PAGE_TRANSITION_FORWARD_BACK));
 
   EXPECT_TRUE(ExtractFeatures(&request));
   std::map<std::string, double> features;
   GetFeatureMap(request, &features);
 
   EXPECT_EQ(1.0,
             features[base::StringPrintf("%s=%s",
                                         features::kReferrer,
                                         "http://google.com/")]);
   EXPECT_EQ(1.0,
@@ skipping 15 matching lines @@
   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));
+                    ui::PageTransitionFromInt(
+                        ui::PAGE_TRANSITION_TYPED |
+                        ui::PAGE_TRANSITION_CHAIN_START |
+                        ui::PAGE_TRANSITION_CLIENT_REDIRECT));
 
   EXPECT_TRUE(ExtractFeatures(&request));
   features.clear();
   GetFeatureMap(request, &features);
 
   EXPECT_EQ(1,
             features[base::StringPrintf("%s=%s",
                                         features::kReferrer,
                                         "http://www.foo.com/")]);
   EXPECT_EQ(1.0,
@@ skipping 29 matching lines @@
 
   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"),
-                    content::PageTransitionFromInt(
-                        content::PAGE_TRANSITION_LINK |
-                        content::PAGE_TRANSITION_CHAIN_END |
-                        content::PAGE_TRANSITION_CLIENT_REDIRECT));
+                    ui::PageTransitionFromInt(
+                        ui::PAGE_TRANSITION_LINK |
+                        ui::PAGE_TRANSITION_CHAIN_END |
+                        ui::PAGE_TRANSITION_CLIENT_REDIRECT));
 
   EXPECT_TRUE(ExtractFeatures(&request));
   features.clear();
   GetFeatureMap(request, &features);
 
   EXPECT_EQ(1.0,
             features[base::StringPrintf("%s=%s",
                                         features::kReferrer,
                                         "http://www.foo.com/page.html")]);
   EXPECT_EQ(1.0,
@@ skipping 15 matching lines @@
                                               features::kIsFirstNavigation)));
 
   request.Clear();
   request.set_url("http://www.bar.com/other_page.html");
   request.set_client_score(0.5);
   redirect_chain.clear();
   redirect_chain.push_back(GURL("http://www.bar.com/other_page.html"));
   SetRedirectChain(redirect_chain, false);
   NavigateAndCommit(GURL("http://www.bar.com/other_page.html"),
                     GURL("http://www.bar.com/"),
-                    content::PAGE_TRANSITION_LINK);
+                    ui::PAGE_TRANSITION_LINK);
 
   EXPECT_TRUE(ExtractFeatures(&request));
   features.clear();
   GetFeatureMap(request, &features);
 
   EXPECT_EQ(1.0,
             features[base::StringPrintf("%s=%s",
                                         features::kReferrer,
                                         "http://www.bar.com/")]);
   EXPECT_EQ(0.0, features[features::kHasSSLReferrer]);
@@ skipping 19 matching lines @@
                                         features::kIsFirstNavigation)]);
   request.Clear();
   request.set_url("http://www.baz.com/");
   request.set_client_score(0.5);
   redirect_chain.clear();
   redirect_chain.push_back(GURL("https://bankofamerica.com"));
   redirect_chain.push_back(GURL("http://www.baz.com/"));
   SetRedirectChain(redirect_chain, true);
   NavigateAndCommit(GURL("http://www.baz.com"),
                     GURL("https://bankofamerica.com"),
-                    content::PAGE_TRANSITION_GENERATED);
+                    ui::PAGE_TRANSITION_GENERATED);
 
   EXPECT_TRUE(ExtractFeatures(&request));
   features.clear();
   GetFeatureMap(request, &features);
 
   EXPECT_EQ(1.0,
             features[base::StringPrintf("%s[0]=%s",
                                         features::kRedirect,
                                         features::kSecureRedirectValue)]);
   EXPECT_EQ(1.0, features[features::kHasSSLReferrer]);
@@ skipping 94 matching lines @@
     // First ip is good but all the others are bad.
     EXPECT_CALL(*db_manager_, MatchMalwareIP(ip)).WillOnce(Return(i > 0));
   }
 
   ExtractMalwareFeatures(&request);
   // The number of IP matched url we store is capped at 5 IPs per request.
   EXPECT_EQ(5, request.bad_ip_url_info_size());
 }
 
 }  // namespace safe_browsing