Enables Activity Log to extract URLs from URL assignment API calls

chrome/browser/extensions/activity_log/activity_log_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 "base/command_line.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
 #include "base/synchronization/waitable_event.h"
 #include "chrome/browser/extensions/activity_log/activity_action_constants.h"
@@ skipping 16 matching lines @@
 #if defined(OS_CHROMEOS)
 #include "chrome/browser/chromeos/login/user_manager.h"
 #include "chrome/browser/chromeos/settings/cros_settings.h"
 #include "chrome/browser/chromeos/settings/device_settings_service.h"
 #endif
 
 namespace {
 
 const char kExtensionId[] = "abc";
 
+const char* kUrlApiCalls[] = {
+    "HTMLButtonElement.formAction", "HTMLEmbedElement.src",
+    "HTMLFormElement.action",       "HTMLFrameElement.src",
+    "HTMLHtmlElement.manifest",     "HTMLIFrameElement.src",
+    "HTMLImageElement.longDesc",    "HTMLImageElement.src",
+    "HTMLImageElement.lowsrc",      "HTMLInputElement.formAction",
+    "HTMLInputElement.src",         "HTMLLinkElement.href",
+    "HTMLMediaElement.src",         "HTMLMediaElement.currentSrc",
+    "HTMLModElement.cite",          "HTMLObjectElement.data",
+    "HTMLQuoteElement.cite",        "HTMLScriptElement.src",
+    "HTMLSourceElement.src",        "HTMLTrackElement.src",
+    "HTMLVideoElement.poster"};
+
 }  // namespace
 
 namespace extensions {
 
 class ActivityLogTest : public ChromeRenderViewHostTestHarness {
  protected:
   virtual void SetUp() OVERRIDE {
     ChromeRenderViewHostTestHarness::SetUp();
 #if defined OS_CHROMEOS
     test_user_manager_.reset(new chromeos::ScopedTestUserManager());
@@ skipping 94 matching lines @@
               ActivityLogPolicy::Util::Serialize(action->args()));
     ASSERT_FALSE(action->arg_url().is_valid());
 
     action = i->at(3);
     ASSERT_EQ("windows.create", action->api_name());
     ASSERT_EQ("[{\"url\":\"\\u003Carg_url>\"}]",
               ActivityLogPolicy::Util::Serialize(action->args()));
     ASSERT_EQ("http://www.google.co.uk/", action->arg_url().spec());
   }
 
+  static void RetrieveActions_ArgUrlApiCalls(
+      scoped_ptr<std::vector<scoped_refptr<Action> > > actions) {
+    size_t api_calls_size = arraysize(kUrlApiCalls);
+    const base::DictionaryValue* other = NULL;
+    int dom_verb = -1;
+
+    ASSERT_EQ(api_calls_size, actions->size());
+
+    for (size_t i = 0; i < actions->size(); i++) {
+      scoped_refptr<Action> action = actions->at(i);
+      ASSERT_EQ(kExtensionId, action->extension_id());
+      ASSERT_EQ(Action::ACTION_DOM_ACCESS, action->action_type());
+      ASSERT_EQ(kUrlApiCalls[i], action->api_name());
+      ASSERT_EQ("[\"\\u003Carg_url>\"]",
+                ActivityLogPolicy::Util::Serialize(action->args()));
+      ASSERT_EQ("http://www.google.co.uk/", action->arg_url().spec());
+      other = action->other();
+      ASSERT_TRUE(other);
+      ASSERT_TRUE(
+          other->GetInteger(activity_log_constants::kActionDomVerb, &dom_verb));
+      ASSERT_EQ(DomActionType::SETTER, dom_verb);
+    }
+  }
+
   ExtensionService* extension_service_;
 
 #if defined OS_CHROMEOS
   chromeos::ScopedTestDeviceSettingsService test_device_settings_service_;
   chromeos::ScopedTestCrosSettings test_cros_settings_;
   scoped_ptr<chromeos::ScopedTestUserManager> test_user_manager_;
 #endif
 };
 
 TEST_F(ActivityLogTest, Construct) {
@@ skipping 172 matching lines @@
   activity_log->GetFilteredActions(
       extension->id(),
       Action::ACTION_ANY,
       "",
       "",
       "",
       -1,
       base::Bind(ActivityLogTest::RetrieveActions_LogAndFetchActions0));
 }
 
+TEST_F(ActivityLogTest, ArgUrlApiCalls) {
+  ActivityLog* activity_log = ActivityLog::GetInstance(profile());
+  scoped_ptr<base::ListValue> args(new base::ListValue());
+  base::Time now = base::Time::Now();
+  int api_calls_size = arraysize(kUrlApiCalls);
+  scoped_refptr<Action> action;
+
+  for (int i = 0; i < api_calls_size; i++) {
+    action = new Action(kExtensionId,
+                        now - base::TimeDelta::FromSeconds(i),
+                        Action::ACTION_DOM_ACCESS,
+                        kUrlApiCalls[i]);
+    action->mutable_args()->AppendString("http://www.google.co.uk");
+    action->mutable_other()->SetInteger(activity_log_constants::kActionDomVerb,
+                                        DomActionType::SETTER);
+    activity_log->LogAction(action);
+  }
+
+  activity_log->GetFilteredActions(
+      kExtensionId,
+      Action::ACTION_ANY,
+      "",
+      "",
+      "",
+      -1,
+      base::Bind(ActivityLogTest::RetrieveActions_ArgUrlApiCalls));
+}
+
 }  // namespace extensions