chrome/browser/extensions/activity_log/ad_injection_unittest.cc

// Copyright 2014 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/prefs/testing_pref_service.h"
#include "base/time/time.h"
#include "chrome/browser/extensions/activity_log/activity_actions.h"
#include "components/rappor/test_rappor_service.h"
#include "extensions/common/value_builder.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace extensions {

namespace {

scoped_refptr<Action> CreateAction(const std::string& api_name,
                                   const std::string& element,
                                   const std::string& attr) {
  scoped_refptr<Action> action = new Action("id",
                                            base::Time::Now(),
                                            Action::ACTION_DOM_ACCESS,
                                            api_name);
  action->set_args(ListBuilder()
                   .Append(element)
                   .Append(attr)
                   .Append("http://www.google.co.uk")
                   .Append("http://www.google.co.uk")
                   .Build());
  return action;
}

}  // namespace

// Test that the actions properly upload the URLs to the RAPPOR service if
// the action may have injected the ad.
TEST(AdInjectionUnittest, CheckActionForAdInjectionTest) {
  rappor::TestRapporService rappor_service;
  ASSERT_EQ(0, rappor_service.GetReportsCount());

  scoped_refptr<Action> modify_iframe_src =
      CreateAction("blinkSetAttribute", "iframe", "src");
  modify_iframe_src->DidInjectAd(&rappor_service);
  EXPECT_EQ(1, rappor_service.GetReportsCount());

  scoped_refptr<Action> modify_anchor_href =
      CreateAction("blinkSetAttribute", "a", "href");
  modify_anchor_href->DidInjectAd(&rappor_service);
  EXPECT_EQ(1, rappor_service.GetReportsCount());

  scoped_refptr<Action> harmless_action =
      CreateAction("Location.replace", "", "");
  harmless_action->DidInjectAd(&rappor_service);
  EXPECT_EQ(0, rappor_service.GetReportsCount());
}

}  // namespace extensions