New architecture of the activity logging: Policies for summarization (and compression)

chrome/browser/extensions/activity_log/fullstream_ui_policy_unittest.cc

+// Copyright 2013 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.h"
+#include "base/synchronization/waitable_event.h"
+#include "chrome/browser/extensions/activity_log/activity_log.h"
+#include "chrome/browser/extensions/activity_log/fullstream_ui_policy.h"
+#include "chrome/browser/extensions/extension_service.h"
+#include "chrome/browser/extensions/test_extension_system.h"
+#include "chrome/common/chrome_constants.h"
+#include "chrome/common/chrome_switches.h"
+#include "chrome/common/extensions/extension_builder.h"
+#include "chrome/test/base/chrome_render_view_host_test_harness.h"
+#include "chrome/test/base/testing_profile.h"
+#include "content/public/browser/browser_thread.h"
+#include "content/public/test/test_browser_thread.h"
+#include "sql/statement.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+#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 extensions {
+
+class FullStreamUIPolicyTest : public ChromeRenderViewHostTestHarness {
+ public:
+  FullStreamUIPolicyTest()
+      : ui_thread_(BrowserThread::UI, base::MessageLoop::current()),
+        db_thread_(BrowserThread::DB, base::MessageLoop::current()),
+        file_thread_(BrowserThread::FILE, base::MessageLoop::current()) {}

[felt, 2013/06/13 20:24:16]

Hey, there were a bunch of changes made to the activity_log_unittest.cc. Can you
please check out the new version of those unittests and also update this one
accordingly? For example we now have a thread bundle, reset the command line,
etc. Same for the other unittest.

[dbabic, 2013/06/13 23:10:08]

Done.

+
+
+  virtual void SetUp() OVERRIDE {
+    ChromeRenderViewHostTestHarness::SetUp();
+    CommandLine command_line(CommandLine::NO_PROGRAM);
+    profile_ =
+        Profile::FromBrowserContext(web_contents()->GetBrowserContext());
+    extension_service_ = static_cast<TestExtensionSystem*>(
+        ExtensionSystem::Get(profile_))->CreateExtensionService(
+            &command_line, base::FilePath(), false);
+    CommandLine::ForCurrentProcess()->AppendSwitch(
+        switches::kEnableExtensionActivityLogTesting);
+  }
+
+  virtual ~FullStreamUIPolicyTest() {
+    base::MessageLoop::current()->PostTask(FROM_HERE,
+                                           base::MessageLoop::QuitClosure());
+    base::MessageLoop::current()->Run();
+  }
+
+  static void RetrieveActions_LogAndFetchActions(
+      scoped_ptr<std::vector<scoped_refptr<Action> > > i) {
+    ASSERT_EQ(2, static_cast<int>(i->size()));
+  }
+
+  static void Arguments_Present(
+      scoped_ptr<std::vector<scoped_refptr<Action> > > i) {
+    scoped_refptr<Action> last = i->front();
+    std::string args = "ID: odlameecjipmbmbejkplpemijjgpljce, CATEGORY: "
+      "CALL, API: extension.connect, ARGS: \"hello\", \"world\"";
+    ASSERT_EQ(args, last->PrintForDebug());
+  }
+
+ protected:
+  ExtensionService* extension_service_;
+  Profile* profile_;
+
+ private:
+  content::TestBrowserThread ui_thread_;
+  content::TestBrowserThread db_thread_;
+  content::TestBrowserThread file_thread_;
+
+#if defined OS_CHROMEOS
+  chromeos::ScopedTestDeviceSettingsService test_device_settings_service_;
+  chromeos::ScopedTestCrosSettings test_cros_settings_;
+  chromeos::ScopedTestUserManager test_user_manager_;
+#endif
+};
+
+TEST_F(FullStreamUIPolicyTest, Construct) {
+  ActivityLogPolicy* policy = new FullStreamUIPolicy(profile_,
+                                                     BrowserThread::UI);
+  scoped_refptr<const Extension> extension =
+      ExtensionBuilder()
+          .SetManifest(DictionaryBuilder()
+                       .Set("name", "Test extension")
+                       .Set("version", "1.0.0")
+                       .Set("manifest_version", 2))
+          .Build();
+  extension_service_->AddExtension(extension);
+  scoped_ptr<ListValue> args(new ListValue());
+  policy->ProcessAction(ActivityLogPolicy::ACTION_API, extension->id(),
+      std::string("tabs.testMethod"), NULL, args.get(), NULL);
+  delete policy;
+}
+
+TEST_F(FullStreamUIPolicyTest, LogAndFetchActions) {
+  ActivityLogPolicy* policy = new FullStreamUIPolicy(profile_,
+                                                     BrowserThread::UI);
+  scoped_refptr<const Extension> extension =
+      ExtensionBuilder()
+          .SetManifest(DictionaryBuilder()
+                       .Set("name", "Test extension")
+                       .Set("version", "1.0.0")
+                       .Set("manifest_version", 2))
+          .Build();
+  extension_service_->AddExtension(extension);
+  scoped_ptr<ListValue> args(new ListValue());
+  GURL* gurl = new GURL("http://www.google.com");
+
+  // Write some API calls
+  policy->ProcessAction(ActivityLogPolicy::ACTION_API, extension->id(),
+      std::string("tabs.testMethod"), NULL, args.get(), NULL);
+  policy->ProcessAction(ActivityLogPolicy::ACTION_DOM,
+      extension->id(), std::string("document.write"), gurl, args.get(), NULL);
+  policy->ReadData(extension->id(), 0,
+      base::Bind(FullStreamUIPolicyTest::RetrieveActions_LogAndFetchActions));
+  delete policy;
+  delete gurl;
+}
+
+TEST_F(FullStreamUIPolicyTest, LogWithArguments) {
+  ActivityLogPolicy* policy = new FullStreamUIPolicy(profile_,
+                                                     BrowserThread::UI);
+  scoped_refptr<const Extension> extension =
+      ExtensionBuilder()
+          .SetManifest(DictionaryBuilder()
+                       .Set("name", "Test extension")
+                       .Set("version", "1.0.0")
+                       .Set("manifest_version", 2))
+          .Build();
+  extension_service_->AddExtension(extension);
+  scoped_ptr<ListValue> args(new ListValue());
+  args->Set(0, new base::StringValue("hello"));
+  args->Set(1, new base::StringValue("world"));
+  policy->ProcessAction(ActivityLogPolicy::ACTION_API, extension->id(),
+      std::string("extension.connect"), NULL, args.get(), NULL);
+  policy->ReadData(extension->id(), 0,
+      base::Bind(FullStreamUIPolicyTest::Arguments_Present));
+}
+
+}  // namespace extensions