Save extension activity log to a file.

chrome/browser/extensions/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 "chrome/browser/extensions/activity_log.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/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"
+
+namespace extensions {
+namespace {

[Eric Dingle, 2012/12/13 20:37:34]

No need for the anonymous namespace.

[felt, 2012/12/15 02:51:52]

Done.

+
+class ActivityLogTest : public ChromeRenderViewHostTestHarness {
+ public:
+  ActivityLogTest()
+      : ui_thread_(BrowserThread::UI, MessageLoop::current()),
+        file_thread_(BrowserThread::FILE, MessageLoop::current()) {}
+
+  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, FilePath(), false);
+  }
+
+ protected:
+  ExtensionService* extension_service_;
+  Profile* profile_;
+
+ private:
+  content::TestBrowserThread ui_thread_;

[Eric Dingle, 2012/12/13 20:37:34]

I haven't written tests with the TestBrowserThread before, but I assume that you
need them for this test to run, even though you don't access them in any way?

[felt, 2012/12/15 02:51:52]

Yes, it seems like they're needed by ChromeRenderViewHostTestHarness in some
way.  Everything explodes if I remove them.

+  content::TestBrowserThread file_thread_;
+};
+
+TEST_F(ActivityLogTest, ConstructAndLog) {
+  ActivityLog* activity_log = ActivityLog::GetInstance(profile_);
+  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);
+  ListValue args;
+  for (int i = 0; i < 30; i++) {
+    // Run this a bunch of times and hope that if something goes wrong with
+    // threading, 30 times is enough to cause it to fail.
+    activity_log->LogManagerAction(extension,
+                                   std::string("tabs.testMethod"),
+                                   args);
+  }
+  // Need to ensure the writes were completed.
+  base::PlatformThread::Sleep(base::TimeDelta::FromSeconds(2));
+  FilePath db_file = profile_->GetPath().Append(chrome::kActivityLogFilename);
+  sql::Connection db;
+  ASSERT_TRUE(db.Open(db_file));
+  std::string sql_str = "SELECT * FROM " + std::string(manager_table_name_);
+  sql::Statement statement(db.GetUniqueStatement(sql_str.c_str()));
+  ASSERT_TRUE(statement.Step());
+  ASSERT_EQ("UNKNOWN_ACTION", statement.ColumnString(2));
+  ASSERT_EQ("TABS", statement.ColumnString(3));
+  ASSERT_EQ("tabs.testMethod()", statement.ColumnString(4));
+}
+
+}  // namespace
+}  // namespace extensions
+