This source file includes following definitions.
- TEST_F
- TEST_F
- TEST_F
- TEST_F
- TEST_F
#include "base/time/time.h"
#include "chrome/browser/extensions/activity_log/activity_action_constants.h"
#include "chrome/browser/extensions/activity_log/uma_policy.h"
#include "chrome/common/extensions/dom_action_types.h"
#include "chrome/test/base/testing_profile.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace extensions {
class UmaPolicyTest : public testing::Test {
public:
UmaPolicyTest() {
profile_.reset(new TestingProfile());
}
protected:
scoped_ptr<TestingProfile> profile_;
};
TEST_F(UmaPolicyTest, Construct) {
ActivityLogPolicy* policy = new UmaPolicy(profile_.get());
policy->Close();
}
TEST_F(UmaPolicyTest, MatchActionToStatusTest) {
UmaPolicy* policy = new UmaPolicy(profile_.get());
scoped_refptr<Action> action = new Action(
"id", base::Time::Now(), Action::ACTION_API_CALL, "extension.connect");
ASSERT_EQ(UmaPolicy::NONE, policy->MatchActionToStatus(action));
action = new Action(
"id", base::Time::Now(), Action::ACTION_API_CALL, "tabs.executeScript");
ASSERT_EQ(
(1 << UmaPolicy::CONTENT_SCRIPT), policy->MatchActionToStatus(action));
action = new Action(
"id", base::Time::Now(), Action::ACTION_CONTENT_SCRIPT, "");
ASSERT_TRUE(
(1 << UmaPolicy::CONTENT_SCRIPT) & policy->MatchActionToStatus(action));
ASSERT_EQ(
(1 << UmaPolicy::CONTENT_SCRIPT), policy->MatchActionToStatus(action));
action = new Action(
"id", base::Time::Now(), Action::ACTION_DOM_ACCESS, "Document.location");
action->mutable_other()->SetInteger(activity_log_constants::kActionDomVerb,
DomActionType::GETTER);
ASSERT_TRUE((1 << UmaPolicy::READ_DOM) & policy->MatchActionToStatus(action));
ASSERT_EQ((1 << UmaPolicy::READ_DOM), policy->MatchActionToStatus(action));
action->mutable_other()->SetInteger(activity_log_constants::kActionDomVerb,
DomActionType::SETTER);
ASSERT_TRUE(
(1 << UmaPolicy::MODIFIED_DOM) & policy->MatchActionToStatus(action));
ASSERT_EQ(
(1 << UmaPolicy::MODIFIED_DOM), policy->MatchActionToStatus(action));
action = new Action(
"id", base::Time::Now(), Action::ACTION_DOM_ACCESS, "HTMLDocument.write");
action->mutable_other()->SetInteger(activity_log_constants::kActionDomVerb,
DomActionType::METHOD);
ASSERT_TRUE(
(1 << UmaPolicy::DOCUMENT_WRITE) & policy->MatchActionToStatus(action));
ASSERT_TRUE(
(1 << UmaPolicy::DOM_METHOD) & policy->MatchActionToStatus(action));
action = new Action("id",
base::Time::Now(),
Action::ACTION_DOM_ACCESS,
"Document.createElement");
scoped_ptr<base::ListValue> args(new base::ListValue());
args->Set(0, new base::StringValue("script"));
action->set_args(args.Pass());
ASSERT_EQ(UmaPolicy::NONE, policy->MatchActionToStatus(action));
action->mutable_other()->SetInteger(activity_log_constants::kActionDomVerb,
DomActionType::METHOD);
ASSERT_TRUE(
(1 << UmaPolicy::CREATED_SCRIPT) & policy->MatchActionToStatus(action));
ASSERT_TRUE(
(1 << UmaPolicy::DOM_METHOD) & policy->MatchActionToStatus(action));
policy->Close();
}
TEST_F(UmaPolicyTest, SiteUrlTest) {
UmaPolicy* policy = new UmaPolicy(profile_.get());
const std::string site0 = "http://www.zzz.com";
const std::string site1 = "http://www.foo.com";
const std::string site2 = "http://www.google.com#a";
const std::string site3 = "http://www.google.com#bb";
policy->SetupOpenedPage(site1);
policy->SetupOpenedPage(site2);
policy->SetupOpenedPage(site2);
policy->SetupOpenedPage(site3);
policy->SetupOpenedPage(site3);
policy->SetupOpenedPage(site3);
ASSERT_EQ(3u, policy->url_status().size());
ASSERT_EQ(1, policy->url_status()[site1][UmaPolicy::kNumberOfTabs]);
ASSERT_EQ(2, policy->url_status()[site2][UmaPolicy::kNumberOfTabs]);
ASSERT_EQ(3, policy->url_status()[site3][UmaPolicy::kNumberOfTabs]);
policy->CleanupClosedPage(site0);
policy->CleanupClosedPage(site2);
policy->CleanupClosedPage(site2);
policy->CleanupClosedPage(site3);
ASSERT_EQ(2u, policy->url_status().size());
ASSERT_EQ(1, policy->url_status()[site1][UmaPolicy::kNumberOfTabs]);
ASSERT_EQ(2, policy->url_status()[site3][UmaPolicy::kNumberOfTabs]);
policy->Close();
}
TEST_F(UmaPolicyTest, ProcessActionTest) {
const std::string site_a = "http://www.zzz.com/";
const std::string site_b = "http://www.foo.com/";
const std::string ext_a = "a";
const std::string ext_b = "b";
UmaPolicy* policy = new UmaPolicy(profile_.get());
policy->SetupOpenedPage(site_a);
policy->SetupOpenedPage(site_b);
scoped_refptr<Action> action1 = new Action(
ext_a, base::Time::Now(), Action::ACTION_CONTENT_SCRIPT, "");
action1->set_page_url(GURL(site_a));
policy->ProcessAction(action1);
scoped_refptr<Action> action2 = new Action(
ext_a, base::Time::Now(), Action::ACTION_CONTENT_SCRIPT, "");
action2->set_page_url(GURL(site_a));
policy->ProcessAction(action2);
scoped_refptr<Action> action3 = new Action(
ext_b, base::Time::Now(), Action::ACTION_DOM_ACCESS, "Document.location");
action3->mutable_other()->SetInteger(activity_log_constants::kActionDomVerb,
DomActionType::GETTER);
action3->set_page_url(GURL(site_a));
policy->ProcessAction(action3);
scoped_refptr<Action> action4 = new Action(
ext_a, base::Time::Now(), Action::ACTION_DOM_ACCESS, "Document.location");
action4->mutable_other()->SetInteger(activity_log_constants::kActionDomVerb,
DomActionType::SETTER);
action4->set_page_url(GURL(site_b));
policy->ProcessAction(action4);
scoped_refptr<Action> action5 = new Action(
ext_b, base::Time::Now(), Action::ACTION_DOM_ACCESS, "Document.location");
action5->mutable_other()->SetInteger(activity_log_constants::kActionDomVerb,
DomActionType::SETTER);
action5->set_page_url(GURL(site_b));
policy->ProcessAction(action5);
scoped_refptr<Action> action6 = new Action(
ext_b, base::Time::Now(), Action::ACTION_API_CALL, "tabs.executeScript");
action6->set_arg_url(GURL(site_b));
policy->ProcessAction(action6);
ASSERT_EQ(2u, policy->url_status().size());
ASSERT_EQ(3u, policy->url_status()[site_a].size());
ASSERT_TRUE(
(1 << UmaPolicy::CONTENT_SCRIPT) & policy->url_status()[site_a][ext_a]);
ASSERT_FALSE(
(1 << UmaPolicy::CONTENT_SCRIPT) & policy->url_status()[site_a][ext_b]);
ASSERT_TRUE((1 << UmaPolicy::READ_DOM) & policy->url_status()[site_a][ext_b]);
ASSERT_FALSE(
(1 << UmaPolicy::READ_DOM) & policy->url_status()[site_a][ext_a]);
ASSERT_EQ(3u, policy->url_status()[site_b].size());
ASSERT_TRUE(
(1 << UmaPolicy::MODIFIED_DOM) & policy->url_status()[site_b][ext_a]);
ASSERT_TRUE(
(1 << UmaPolicy::MODIFIED_DOM) & policy->url_status()[site_b][ext_b]);
ASSERT_TRUE(
(1 << UmaPolicy::CONTENT_SCRIPT) & policy->url_status()[site_b][ext_b]);
policy->Close();
}
TEST_F(UmaPolicyTest, CleanURLTest) {
ASSERT_EQ("http://www.google.com/",
UmaPolicy::CleanURL(GURL("http://www.google.com/")));
ASSERT_EQ("http://www.google.com/",
UmaPolicy::CleanURL(GURL("http://www.google.com")));
ASSERT_EQ("http://www.google.com:8080/a.html",
UmaPolicy::CleanURL(GURL("http://www.google.com:8080/a.html")));
ASSERT_EQ("http://www.google.com/",
UmaPolicy::CleanURL(GURL("http://www.google.com/#a")));
ASSERT_EQ("http://www.google.com/",
UmaPolicy::CleanURL(GURL("http://www.google.com/#aaaa")));
ASSERT_EQ("http://www.google.com/?q=a",
UmaPolicy::CleanURL(GURL("http://www.google.com/?q=a")));
ASSERT_EQ("http://www.cnn.com/",
UmaPolicy::CleanURL(GURL("http://www.cnn.com/")));
ASSERT_EQ("http://www.cnn.com:8080/a.html",
UmaPolicy::CleanURL(GURL("http://www.cnn.com:8080/a.html")));
ASSERT_EQ("http://www.cnn.com/",
UmaPolicy::CleanURL(GURL("http://www.cnn.com")));
ASSERT_EQ("http://www.cnn.com/",
UmaPolicy::CleanURL(GURL("http://www.cnn.com/#a")));
ASSERT_EQ("http://www.cnn.com/",
UmaPolicy::CleanURL(GURL("http://www.cnn.com/#aaaa")));
ASSERT_EQ("http://www.cnn.com/?q=a",
UmaPolicy::CleanURL(GURL("http://www.cnn.com/?q=a")));
}
}