Extension activity log database refactoring (step 2)

chrome/browser/extensions/activity_log/activity_database_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 <string>
 #include "base/command_line.h"
 #include "base/file_util.h"
 #include "base/files/file_path.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/run_loop.h"
 #include "base/test/simple_test_clock.h"
 #include "base/time/time.h"
 #include "chrome/browser/extensions/activity_log/activity_database.h"
 #include "chrome/browser/extensions/activity_log/api_actions.h"
 #include "chrome/browser/extensions/activity_log/blocked_actions.h"
 #include "chrome/browser/extensions/activity_log/dom_actions.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/dom_action_types.h"
 #include "chrome/test/base/chrome_render_view_host_test_harness.h"
 #include "chrome/test/base/testing_profile.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/test/test_browser_thread.h"
 #include "sql/statement.h"
@@ skipping 13 matching lines @@
 namespace extensions {
 
 // A dummy implementation of ActivityDatabase::Delegate, sufficient for
 // the unit tests.
 class ActivityDatabaseTestPolicy : public ActivityDatabase::Delegate {
  public:
   ActivityDatabaseTestPolicy() {};
 
  protected:
   virtual bool OnDatabaseInit(sql::Connection* db) OVERRIDE {
-    if (!DOMAction::InitializeTable(db)) return false;
-    if (!APIAction::InitializeTable(db)) return false;
-    if (!BlockedAction::InitializeTable(db)) return false;
-    return true;
+    return ActivityDatabase::InitializeTable(
+        db,
+        FullStreamUIPolicy::kTableName,
+        FullStreamUIPolicy::kTableContentFields,
+        FullStreamUIPolicy::kTableFieldTypes,
+        FullStreamUIPolicy::kTableFieldCount);
   }
 
   // Called by ActivityDatabase just before the ActivityDatabase object is
   // deleted.  The database will make no further callbacks after invoking this
   // method, so it is an appropriate time for the policy to delete itself.
   virtual void OnDatabaseClose() OVERRIDE {
     delete this;
   }
 };
 
@@ skipping 28 matching lines @@
  private:
 #if defined OS_CHROMEOS
   chromeos::ScopedStubCrosEnabler stub_cros_enabler_;
   chromeos::ScopedTestDeviceSettingsService test_device_settings_service_;
   chromeos::ScopedTestCrosSettings test_cros_settings_;
   scoped_ptr<chromeos::ScopedTestUserManager> test_user_manager_;
 #endif
 
 };
 
-
 // Check that the database is initialized properly.
 TEST_F(ActivityDatabaseTest, Init) {
   base::ScopedTempDir temp_dir;
   base::FilePath db_file;
   ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
   db_file = temp_dir.path().AppendASCII("ActivityInit.db");
   base::Delete(db_file, false);
 
   ActivityDatabase* activity_db = OpenDatabase(db_file);
   activity_db->Close();
 
   sql::Connection db;
   ASSERT_TRUE(db.Open(db_file));
-  ASSERT_TRUE(db.DoesTableExist(DOMAction::kTableName));
-  ASSERT_TRUE(db.DoesTableExist(APIAction::kTableName));
-  ASSERT_TRUE(db.DoesTableExist(BlockedAction::kTableName));
+  ASSERT_TRUE(db.DoesTableExist(FullStreamUIPolicy::kTableName));
   db.Close();
 }
 
 // Check that API actions are recorded in the db.
 TEST_F(ActivityDatabaseTest, RecordAPIAction) {
   base::ScopedTempDir temp_dir;
   base::FilePath db_file;
   ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
   db_file = temp_dir.path().AppendASCII("ActivityRecord.db");
   base::Delete(db_file, false);
 
   ActivityDatabase* activity_db = OpenDatabase(db_file);
   activity_db->SetBatchModeForTesting(false);
+  base::ListValue args_list;
+  args_list.AppendString("woof");
   scoped_refptr<APIAction> action = new APIAction(
       "punky",
       base::Time::Now(),
       APIAction::CALL,
       "brewster",
       "woof",
+      args_list,
       "extra");
   activity_db->RecordAction(action);
   activity_db->Close();
 
   sql::Connection db;
   ASSERT_TRUE(db.Open(db_file));
 
-  ASSERT_TRUE(db.DoesTableExist(APIAction::kTableName));
+  ASSERT_TRUE(db.DoesTableExist(FullStreamUIPolicy::kTableName));
   std::string sql_str = "SELECT * FROM " +
-      std::string(APIAction::kTableName);
+      std::string(FullStreamUIPolicy::kTableName);
   sql::Statement statement(db.GetUniqueStatement(sql_str.c_str()));
   ASSERT_TRUE(statement.Step());
   ASSERT_EQ("punky", statement.ColumnString(0));
-  ASSERT_EQ(0, statement.ColumnInt(2));
+  ASSERT_EQ(static_cast<int>(Action::ACTION_API_CALL), statement.ColumnInt(2));
   ASSERT_EQ("brewster", statement.ColumnString(3));
-  ASSERT_EQ("woof", statement.ColumnString(4));
+  ASSERT_EQ("[\"woof\"]", statement.ColumnString(4));
 }
 
 // Check that DOM actions are recorded in the db.
 TEST_F(ActivityDatabaseTest, RecordDOMAction) {
   base::ScopedTempDir temp_dir;
   base::FilePath db_file;
   ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
   db_file = temp_dir.path().AppendASCII("ActivityRecord.db");
   base::Delete(db_file, false);
 
   ActivityDatabase* activity_db = OpenDatabase(db_file);
   activity_db->SetBatchModeForTesting(false);
   scoped_refptr<DOMAction> action = new DOMAction(
       "punky",
       base::Time::Now(),
       DomActionType::MODIFIED,
       GURL("http://www.google.com/foo?bar"),
       string16(),
       "lets",
       "vamoose",
       "extra");
   activity_db->RecordAction(action);
   activity_db->Close();
 
   sql::Connection db;
   ASSERT_TRUE(db.Open(db_file));
 
-  ASSERT_TRUE(db.DoesTableExist(APIAction::kTableName));
+  ASSERT_TRUE(db.DoesTableExist(FullStreamUIPolicy::kTableName));
   std::string sql_str = "SELECT * FROM " +
-      std::string(DOMAction::kTableName);
+      std::string(FullStreamUIPolicy::kTableName);
   sql::Statement statement(db.GetUniqueStatement(sql_str.c_str()));
   ASSERT_TRUE(statement.Step());
   ASSERT_EQ("punky", statement.ColumnString(0));
-  ASSERT_EQ(DomActionType::MODIFIED, statement.ColumnInt(2));
-  ASSERT_EQ("http://www.google.com", statement.ColumnString(3));
+  ASSERT_EQ(static_cast<int>(Action::ACTION_DOM_ACCESS),
+            statement.ColumnInt(2));
+  // TODO(mvrable): This test doesn't work properly, due to crbug.com/260784
+  // This will be fixed when URL sanitization is moved into the activity log
+  // policies in some upcoming code refactoring.
   if (CommandLine::ForCurrentProcess()->HasSwitch(
       switches::kEnableExtensionActivityLogTesting))
-    ASSERT_EQ("/foo?bar", statement.ColumnString(4));
+    ASSERT_EQ("http://www.google.com/foo?bar", statement.ColumnString(5));
   else
-    ASSERT_EQ("/foo", statement.ColumnString(4));
-  ASSERT_EQ("lets", statement.ColumnString(6));
-  ASSERT_EQ("vamoose", statement.ColumnString(7));
-  ASSERT_EQ("extra", statement.ColumnString(8));
+    ASSERT_EQ("http://www.google.com/foo", statement.ColumnString(5));
+  ASSERT_EQ("lets", statement.ColumnString(3));
+  ASSERT_EQ("[\"vamoose\"]", statement.ColumnString(4));
+  ASSERT_EQ("{\"dom_verb\":6,\"extra\":\"extra\",\"page_title\":\"\"}",
+            statement.ColumnString(7));
 }
 
 // Check that blocked actions are recorded in the db.
 TEST_F(ActivityDatabaseTest, RecordBlockedAction) {
   base::ScopedTempDir temp_dir;
   base::FilePath db_file;
   ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
   db_file = temp_dir.path().AppendASCII("ActivityRecord.db");
   base::Delete(db_file, false);
 
   ActivityDatabase* activity_db = OpenDatabase(db_file);
   scoped_refptr<BlockedAction> action = new BlockedAction(
       "punky",
       base::Time::Now(),
       "do.evilThings",
       "1, 2",
       BlockedAction::ACCESS_DENIED,
       "extra");
   activity_db->RecordAction(action);
   activity_db->Close();
 
   sql::Connection db;
   ASSERT_TRUE(db.Open(db_file));
 
-  ASSERT_TRUE(db.DoesTableExist(BlockedAction::kTableName));
+  ASSERT_TRUE(db.DoesTableExist(FullStreamUIPolicy::kTableName));
   std::string sql_str = "SELECT * FROM " +
-      std::string(BlockedAction::kTableName);
+      std::string(FullStreamUIPolicy::kTableName);
   sql::Statement statement(db.GetUniqueStatement(sql_str.c_str()));
   ASSERT_TRUE(statement.Step());
   ASSERT_EQ("punky", statement.ColumnString(0));
-  ASSERT_EQ("do.evilThings", statement.ColumnString(2));
-  ASSERT_EQ("1, 2", statement.ColumnString(3));
-  ASSERT_EQ(1, statement.ColumnInt(4));
-  ASSERT_EQ("extra", statement.ColumnString(5));
+  ASSERT_EQ(static_cast<int>(Action::ACTION_API_BLOCKED),
+            statement.ColumnInt(2));
+  ASSERT_EQ("do.evilThings", statement.ColumnString(3));
+  ASSERT_EQ("1, 2", statement.ColumnString(4));
+  //ASSERT_EQ(1, statement.ColumnInt(4));
+  ASSERT_EQ("{\"reason\":1}", statement.ColumnString(7));
 }
 
 // Check that we can read back recent actions in the db.
 TEST_F(ActivityDatabaseTest, GetTodaysActions) {
   base::ScopedTempDir temp_dir;
   base::FilePath db_file;
   ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
   db_file = temp_dir.path().AppendASCII("ActivityRecord.db");
   base::Delete(db_file, false);
 
   // Use a mock clock to ensure that events are not recorded on the wrong day
   // when the test is run close to local midnight.
   base::SimpleTestClock mock_clock;
   mock_clock.SetNow(base::Time::Now().LocalMidnight() +
                     base::TimeDelta::FromHours(12));
 
   // Record some actions
   ActivityDatabase* activity_db = OpenDatabase(db_file);
+  base::ListValue args_list;
+  args_list.AppendString("woof");
   scoped_refptr<APIAction> api_action = new APIAction(
       "punky",
       mock_clock.Now() - base::TimeDelta::FromMinutes(40),
       APIAction::CALL,
       "brewster",
       "woof",
+      args_list,
       "extra");
   scoped_refptr<DOMAction> dom_action = new DOMAction(
       "punky",
       mock_clock.Now(),
       DomActionType::MODIFIED,
       GURL("http://www.google.com"),
       string16(),
       "lets",
       "vamoose",
       "extra");
   scoped_refptr<DOMAction> extra_dom_action = new DOMAction(
       "scoobydoo",
       mock_clock.Now(),
       DomActionType::MODIFIED,
       GURL("http://www.google.com"),
       string16(),
       "lets",
       "vamoose",
       "extra");
   activity_db->RecordAction(api_action);
   activity_db->RecordAction(dom_action);
   activity_db->RecordAction(extra_dom_action);
 
   // Read them back
-  std::string api_print = "ID: punky, CATEGORY: call, "
-      "API: brewster, ARGS: woof";
-  std::string dom_print = "DOM API CALL: lets, ARGS: vamoose, VERB: modified";
+  std::string api_print =
+      "ID=punky CATEGORY=api_call API=brewster ARGS=[\"woof\"] OTHER={}";
+  std::string dom_print =
+      "ID=punky CATEGORY=dom_access API=lets ARGS=[\"vamoose\"] "
+      "PAGE_URL=http://www.google.com/ OTHER={}";
   scoped_ptr<std::vector<scoped_refptr<Action> > > actions =
       activity_db->GetActions("punky", 0);
   ASSERT_EQ(2, static_cast<int>(actions->size()));
   ASSERT_EQ(dom_print, actions->at(0)->PrintForDebug());
   ASSERT_EQ(api_print, actions->at(1)->PrintForDebug());
 
   activity_db->Close();
 }
 
 // Check that we can read back recent actions in the db.
 TEST_F(ActivityDatabaseTest, GetOlderActions) {
   base::ScopedTempDir temp_dir;
   base::FilePath db_file;
   ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
   db_file = temp_dir.path().AppendASCII("ActivityRecord.db");
   base::Delete(db_file, false);
 
   // Use a mock clock to ensure that events are not recorded on the wrong day
   // when the test is run close to local midnight.
   base::SimpleTestClock mock_clock;
   mock_clock.SetNow(base::Time::Now().LocalMidnight() +
                     base::TimeDelta::FromHours(12));
 
   // Record some actions
   ActivityDatabase* activity_db = OpenDatabase(db_file);
+  base::ListValue args_list;
+  args_list.AppendString("woof");
   scoped_refptr<APIAction> api_action = new APIAction(
       "punky",
       mock_clock.Now() - base::TimeDelta::FromDays(3)
           - base::TimeDelta::FromMinutes(40),
       APIAction::CALL,
       "brewster",
       "woof",
+      args_list,
       "extra");
   scoped_refptr<DOMAction> dom_action = new DOMAction(
       "punky",
       mock_clock.Now() - base::TimeDelta::FromDays(3),
       DomActionType::MODIFIED,
       GURL("http://www.google.com"),
       string16(),
       "lets",
       "vamoose",
       "extra");
@@ skipping 14 matching lines @@
       string16(),
       "too old",
       "vamoose",
       "extra");
   activity_db->RecordAction(api_action);
   activity_db->RecordAction(dom_action);
   activity_db->RecordAction(toonew_dom_action);
   activity_db->RecordAction(tooold_dom_action);
 
   // Read them back
-  std::string api_print = "ID: punky, CATEGORY: call, "
-      "API: brewster, ARGS: woof";
-  std::string dom_print = "DOM API CALL: lets, ARGS: vamoose, VERB: modified";
+  std::string api_print =
+      "ID=punky CATEGORY=api_call API=brewster ARGS=[\"woof\"] OTHER={}";
+  std::string dom_print =
+      "ID=punky CATEGORY=dom_access API=lets ARGS=[\"vamoose\"] "
+      "PAGE_URL=http://www.google.com/ OTHER={}";
   scoped_ptr<std::vector<scoped_refptr<Action> > > actions =
       activity_db->GetActions("punky", 3);
   ASSERT_EQ(2, static_cast<int>(actions->size()));
   ASSERT_EQ(dom_print, actions->at(0)->PrintForDebug());
   ASSERT_EQ(api_print, actions->at(1)->PrintForDebug());
 
   activity_db->Close();
 }
 
 TEST_F(ActivityDatabaseTest, BatchModeOff) {
   base::ScopedTempDir temp_dir;
   base::FilePath db_file;
   ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
   db_file = temp_dir.path().AppendASCII("ActivityRecord.db");
   base::Delete(db_file, false);
 
   // Use a mock clock to ensure that events are not recorded on the wrong day
   // when the test is run close to local midnight.
   base::SimpleTestClock mock_clock;
   mock_clock.SetNow(base::Time::Now().LocalMidnight() +
                     base::TimeDelta::FromHours(12));
 
   // Record some actions
   ActivityDatabase* activity_db = OpenDatabase(db_file);
   activity_db->SetBatchModeForTesting(false);
   activity_db->SetClockForTesting(&mock_clock);
+  base::ListValue args_list;
+  args_list.AppendString("woof");
   scoped_refptr<APIAction> api_action = new APIAction(
       "punky",
       mock_clock.Now() - base::TimeDelta::FromMinutes(40),
       APIAction::CALL,
       "brewster",
       "woof",
+      args_list,
       "extra");
   activity_db->RecordAction(api_action);
 
   scoped_ptr<std::vector<scoped_refptr<Action> > > actions =
       activity_db->GetActions("punky", 0);
   ASSERT_EQ(1, static_cast<int>(actions->size()));
   activity_db->Close();
 }
 
 TEST_F(ActivityDatabaseTest, BatchModeOn) {
   base::ScopedTempDir temp_dir;
   base::FilePath db_file;
   ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
   db_file = temp_dir.path().AppendASCII("ActivityRecord.db");
   base::Delete(db_file, false);
 
   // Use a mock clock to set the time, and a special timer to control the
   // timing and skip ahead in time.
   base::SimpleTestClock mock_clock;
   mock_clock.SetNow(base::Time::Now().LocalMidnight() +
                     base::TimeDelta::FromHours(11));
 
   // Record some actions
   ActivityDatabase* activity_db = OpenDatabase(db_file);
   activity_db->SetBatchModeForTesting(true);
   activity_db->SetClockForTesting(&mock_clock);
+  base::ListValue args_list;
+  args_list.AppendString("woof");
   scoped_refptr<APIAction> api_action = new APIAction(
       "punky",
       mock_clock.Now() - base::TimeDelta::FromMinutes(40),
       APIAction::CALL,
       "brewster",
       "woof",
+      args_list,
       "extra");
   activity_db->RecordAction(api_action);
 
   scoped_ptr<std::vector<scoped_refptr<Action> > > actions_before =
       activity_db->GetActions("punky", 0);
   ASSERT_EQ(0, static_cast<int>(actions_before->size()));
 
   // Artificially trigger and then stop the timer.
   activity_db->SetTimerForTesting(0);
   base::MessageLoop::current()->RunUntilIdle();
 
   scoped_ptr<std::vector<scoped_refptr<Action> > > actions_after =
         activity_db->GetActions("punky", 0);
     ASSERT_EQ(1, static_cast<int>(actions_after->size()));
 
   activity_db->Close();
 }
 
 // Check that nothing explodes if the DB isn't initialized.
 TEST_F(ActivityDatabaseTest, InitFailure) {
   base::ScopedTempDir temp_dir;
   base::FilePath db_file;
   ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
   db_file = temp_dir.path().AppendASCII("ActivityRecord.db");
   base::Delete(db_file, false);
 
   ActivityDatabase* activity_db =
       new ActivityDatabase(new ActivityDatabaseTestPolicy());
+  base::ListValue args_list;
+  args_list.AppendString("woof");
   scoped_refptr<APIAction> action = new APIAction(
       "punky",
       base::Time::Now(),
       APIAction::CALL,
       "brewster",
       "woooof",
+      args_list,
       "extra");
   activity_db->RecordAction(action);
   activity_db->Close();
 }
 
 }  // namespace extensions