src/chrome/browser/extensions/activity_log/uma_policy_unittest.cc

   1 // Copyright 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "base/time/time.h"
   6 #include "chrome/browser/extensions/activity_log/activity_action_constants.h"
   7 #include "chrome/browser/extensions/activity_log/uma_policy.h"
   8 #include "chrome/common/extensions/dom_action_types.h"
   9 #include "chrome/test/base/testing_profile.h"
  10 #include "testing/gtest/include/gtest/gtest.h"
  11
  12 namespace extensions {
  13
  14 class UmaPolicyTest : public testing::Test {
  15  public:
  16   UmaPolicyTest() {
  17     profile_.reset(new TestingProfile());
  18   }
  19
  20  protected:
  21   scoped_ptr<TestingProfile> profile_;
  22 };
  23
  24 TEST_F(UmaPolicyTest, Construct) {
  25   ActivityLogPolicy* policy = new UmaPolicy(profile_.get());
  26   policy->Close();
  27 }
  28
  29 TEST_F(UmaPolicyTest, MatchActionToStatusTest) {
  30   UmaPolicy* policy = new UmaPolicy(profile_.get());
  31
  32   scoped_refptr<Action> action = new Action(
  33       "id", base::Time::Now(), Action::ACTION_API_CALL, "extension.connect");
  34   ASSERT_EQ(UmaPolicy::NONE, policy->MatchActionToStatus(action));
  35
  36   action = new Action(
  37       "id", base::Time::Now(), Action::ACTION_API_CALL, "tabs.executeScript");
  38   ASSERT_EQ(
  39       (1 << UmaPolicy::CONTENT_SCRIPT), policy->MatchActionToStatus(action));
  40
  41   action = new Action(
  42       "id", base::Time::Now(), Action::ACTION_CONTENT_SCRIPT, "");
  43   ASSERT_TRUE(
  44       (1 << UmaPolicy::CONTENT_SCRIPT) & policy->MatchActionToStatus(action));
  45   ASSERT_EQ(
  46       (1 << UmaPolicy::CONTENT_SCRIPT), policy->MatchActionToStatus(action));
  47
  48   action = new Action(
  49       "id", base::Time::Now(), Action::ACTION_DOM_ACCESS, "Document.location");
  50   action->mutable_other()->SetInteger(activity_log_constants::kActionDomVerb,
  51                                       DomActionType::GETTER);
  52   ASSERT_TRUE((1 << UmaPolicy::READ_DOM) & policy->MatchActionToStatus(action));
  53   ASSERT_EQ((1 << UmaPolicy::READ_DOM), policy->MatchActionToStatus(action));
  54
  55   action->mutable_other()->SetInteger(activity_log_constants::kActionDomVerb,
  56                                       DomActionType::SETTER);
  57   ASSERT_TRUE(
  58       (1 << UmaPolicy::MODIFIED_DOM) & policy->MatchActionToStatus(action));
  59   ASSERT_EQ(
  60       (1 << UmaPolicy::MODIFIED_DOM), policy->MatchActionToStatus(action));
  61
  62   action = new Action(
  63       "id", base::Time::Now(), Action::ACTION_DOM_ACCESS, "HTMLDocument.write");
  64   action->mutable_other()->SetInteger(activity_log_constants::kActionDomVerb,
  65                                       DomActionType::METHOD);
  66   ASSERT_TRUE(
  67       (1 << UmaPolicy::DOCUMENT_WRITE) & policy->MatchActionToStatus(action));
  68   ASSERT_TRUE(
  69       (1 << UmaPolicy::DOM_METHOD) & policy->MatchActionToStatus(action));
  70
  71   action = new Action("id",
  72                       base::Time::Now(),
  73                       Action::ACTION_DOM_ACCESS,
  74                       "Document.createElement");
  75   scoped_ptr<base::ListValue> args(new base::ListValue());
  76   args->Set(0, new base::StringValue("script"));
  77   action->set_args(args.Pass());
  78   ASSERT_EQ(UmaPolicy::NONE, policy->MatchActionToStatus(action));
  79   action->mutable_other()->SetInteger(activity_log_constants::kActionDomVerb,
  80                                       DomActionType::METHOD);
  81   ASSERT_TRUE(
  82       (1 << UmaPolicy::CREATED_SCRIPT) & policy->MatchActionToStatus(action));
  83   ASSERT_TRUE(
  84       (1 << UmaPolicy::DOM_METHOD) & policy->MatchActionToStatus(action));
  85
  86   policy->Close();
  87 }
  88
  89 TEST_F(UmaPolicyTest, SiteUrlTest) {
  90   UmaPolicy* policy = new UmaPolicy(profile_.get());
  91
  92   const std::string site0 = "http://www.zzz.com";
  93   const std::string site1 = "http://www.foo.com";
  94   const std::string site2 = "http://www.google.com#a";
  95   const std::string site3 = "http://www.google.com#bb";
  96
  97   // Record some opened sites.
  98   policy->SetupOpenedPage(site1);
  99   policy->SetupOpenedPage(site2);
 100   policy->SetupOpenedPage(site2);
 101   policy->SetupOpenedPage(site3);
 102   policy->SetupOpenedPage(site3);
 103   policy->SetupOpenedPage(site3);
 104
 105   // Check that site1, site2, and site3 were recorded.
 106   ASSERT_EQ(3u, policy->url_status().size());
 107   ASSERT_EQ(1, policy->url_status()[site1][UmaPolicy::kNumberOfTabs]);
 108   ASSERT_EQ(2, policy->url_status()[site2][UmaPolicy::kNumberOfTabs]);
 109   ASSERT_EQ(3, policy->url_status()[site3][UmaPolicy::kNumberOfTabs]);
 110
 111   // Remove some sites.
 112   policy->CleanupClosedPage(site0);
 113   policy->CleanupClosedPage(site2);
 114   policy->CleanupClosedPage(site2);
 115   policy->CleanupClosedPage(site3);
 116
 117   // Check that the removal worked.
 118   ASSERT_EQ(2u, policy->url_status().size());
 119   ASSERT_EQ(1, policy->url_status()[site1][UmaPolicy::kNumberOfTabs]);
 120   ASSERT_EQ(2, policy->url_status()[site3][UmaPolicy::kNumberOfTabs]);
 121
 122   policy->Close();
 123 }
 124
 125 TEST_F(UmaPolicyTest, ProcessActionTest) {
 126   const std::string site_a = "http://www.zzz.com/";
 127   const std::string site_b = "http://www.foo.com/";
 128   const std::string ext_a = "a";
 129   const std::string ext_b = "b";
 130   UmaPolicy* policy = new UmaPolicy(profile_.get());
 131
 132   // Populate with a few different pages.
 133   policy->SetupOpenedPage(site_a);
 134   policy->SetupOpenedPage(site_b);
 135
 136   // Process a few actions for site_a.
 137   scoped_refptr<Action> action1 = new Action(
 138       ext_a, base::Time::Now(), Action::ACTION_CONTENT_SCRIPT, "");
 139   action1->set_page_url(GURL(site_a));
 140   policy->ProcessAction(action1);
 141
 142   scoped_refptr<Action> action2 = new Action(
 143       ext_a, base::Time::Now(), Action::ACTION_CONTENT_SCRIPT, "");
 144   action2->set_page_url(GURL(site_a));
 145   policy->ProcessAction(action2);
 146
 147   scoped_refptr<Action> action3 = new Action(
 148       ext_b, base::Time::Now(), Action::ACTION_DOM_ACCESS, "Document.location");
 149   action3->mutable_other()->SetInteger(activity_log_constants::kActionDomVerb,
 150                                        DomActionType::GETTER);
 151   action3->set_page_url(GURL(site_a));
 152   policy->ProcessAction(action3);
 153
 154   // Process an action for site_b.
 155   scoped_refptr<Action> action4 = new Action(
 156       ext_a, base::Time::Now(), Action::ACTION_DOM_ACCESS, "Document.location");
 157   action4->mutable_other()->SetInteger(activity_log_constants::kActionDomVerb,
 158                                        DomActionType::SETTER);
 159   action4->set_page_url(GURL(site_b));
 160   policy->ProcessAction(action4);
 161
 162   scoped_refptr<Action> action5 = new Action(
 163       ext_b, base::Time::Now(), Action::ACTION_DOM_ACCESS, "Document.location");
 164   action5->mutable_other()->SetInteger(activity_log_constants::kActionDomVerb,
 165                                        DomActionType::SETTER);
 166   action5->set_page_url(GURL(site_b));
 167   policy->ProcessAction(action5);
 168
 169   scoped_refptr<Action> action6 = new Action(
 170       ext_b, base::Time::Now(), Action::ACTION_API_CALL, "tabs.executeScript");
 171   action6->set_arg_url(GURL(site_b));
 172   policy->ProcessAction(action6);
 173
 174   // Now check what's been recorded.
 175   ASSERT_EQ(2u, policy->url_status().size());
 176
 177   ASSERT_EQ(3u, policy->url_status()[site_a].size());
 178   ASSERT_TRUE(
 179       (1 << UmaPolicy::CONTENT_SCRIPT) & policy->url_status()[site_a][ext_a]);
 180   ASSERT_FALSE(
 181       (1 << UmaPolicy::CONTENT_SCRIPT) & policy->url_status()[site_a][ext_b]);
 182   ASSERT_TRUE((1 << UmaPolicy::READ_DOM) & policy->url_status()[site_a][ext_b]);
 183   ASSERT_FALSE(
 184       (1 << UmaPolicy::READ_DOM) & policy->url_status()[site_a][ext_a]);
 185
 186   ASSERT_EQ(3u, policy->url_status()[site_b].size());
 187   ASSERT_TRUE(
 188       (1 << UmaPolicy::MODIFIED_DOM) & policy->url_status()[site_b][ext_a]);
 189   ASSERT_TRUE(
 190       (1 << UmaPolicy::MODIFIED_DOM) & policy->url_status()[site_b][ext_b]);
 191   ASSERT_TRUE(
 192       (1 << UmaPolicy::CONTENT_SCRIPT) & policy->url_status()[site_b][ext_b]);
 193
 194   policy->Close();
 195 }
 196
 197 TEST_F(UmaPolicyTest, CleanURLTest) {
 198   ASSERT_EQ("http://www.google.com/",
 199             UmaPolicy::CleanURL(GURL("http://www.google.com/")));
 200   ASSERT_EQ("http://www.google.com/",
 201             UmaPolicy::CleanURL(GURL("http://www.google.com")));
 202   ASSERT_EQ("http://www.google.com:8080/a.html",
 203             UmaPolicy::CleanURL(GURL("http://www.google.com:8080/a.html")));
 204   ASSERT_EQ("http://www.google.com/",
 205             UmaPolicy::CleanURL(GURL("http://www.google.com/#a")));
 206   ASSERT_EQ("http://www.google.com/",
 207             UmaPolicy::CleanURL(GURL("http://www.google.com/#aaaa")));
 208   ASSERT_EQ("http://www.google.com/?q=a",
 209             UmaPolicy::CleanURL(GURL("http://www.google.com/?q=a")));
 210
 211   ASSERT_EQ("http://www.cnn.com/",
 212             UmaPolicy::CleanURL(GURL("http://www.cnn.com/")));
 213   ASSERT_EQ("http://www.cnn.com:8080/a.html",
 214             UmaPolicy::CleanURL(GURL("http://www.cnn.com:8080/a.html")));
 215   ASSERT_EQ("http://www.cnn.com/",
 216             UmaPolicy::CleanURL(GURL("http://www.cnn.com")));
 217   ASSERT_EQ("http://www.cnn.com/",
 218             UmaPolicy::CleanURL(GURL("http://www.cnn.com/#a")));
 219   ASSERT_EQ("http://www.cnn.com/",
 220             UmaPolicy::CleanURL(GURL("http://www.cnn.com/#aaaa")));
 221   ASSERT_EQ("http://www.cnn.com/?q=a",
 222             UmaPolicy::CleanURL(GURL("http://www.cnn.com/?q=a")));
 223 }
 224
 225 }  // namespace extensions