Compressed activity log database storage

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"
@@ skipping 85 matching lines @@
     }
   }
 
   queue_.clear();
   return true;
 }
 
 void ActivityDatabaseTestPolicy::Record(ActivityDatabase* db,
                                         scoped_refptr<Action> action) {
   queue_.push_back(action);
-  db->NotifyAction();
+  db->AdviseFlush(queue_.size());
 }
 
 class ActivityDatabaseTest : public ChromeRenderViewHostTestHarness {
  protected:
   virtual void SetUp() OVERRIDE {
     ChromeRenderViewHostTestHarness::SetUp();
 #if defined OS_CHROMEOS
     test_user_manager_.reset(new chromeos::ScopedTestUserManager());
 #endif
     CommandLine command_line(CommandLine::NO_PROGRAM);
@@ skipping 27 matching lines @@
   void Record(ActivityDatabase* db, scoped_refptr<Action> action) {
     db_delegate_->Record(db, action);
   }
 
   int CountActions(sql::Connection* db, const std::string& api_name_pattern) {
     if (!db->DoesTableExist(ActivityDatabaseTestPolicy::kTableName))
       return -1;
     std::string sql_str = "SELECT COUNT(*) FROM " +
                           std::string(ActivityDatabaseTestPolicy::kTableName) +
                           " WHERE api_name LIKE ?";
-    sql::Statement statement(db->GetUniqueStatement(sql_str.c_str()));
+    sql::Statement statement(db->GetCachedStatement(
+        sql::StatementID(SQL_FROM_HERE), sql_str.c_str()));
     statement.BindString(0, api_name_pattern);
     if (!statement.Step())
       return -1;
     return statement.ColumnInt(0);
   }
 
  private:
 #if defined OS_CHROMEOS
   chromeos::ScopedStubNetworkLibraryEnabler stub_network_library_enabler_;
   chromeos::ScopedTestDeviceSettingsService test_device_settings_service_;
@@ skipping 74 matching lines @@
   ASSERT_EQ(0, CountActions(&activity_db->db_, "brewster"));
 
   // Artificially trigger and then stop the timer.
   activity_db->SetTimerForTesting(0);
   base::MessageLoop::current()->RunUntilIdle();
   ASSERT_EQ(1, CountActions(&activity_db->db_, "brewster"));
 
   activity_db->Close();
 }
 
+TEST_F(ActivityDatabaseTest, BatchModeFlush) {
+  base::ScopedTempDir temp_dir;
+  base::FilePath db_file;
+  ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
+  db_file = temp_dir.path().AppendASCII("ActivityFlush.db");
+  base::DeleteFile(db_file, false);
+
+  // Record some actions
+  ActivityDatabase* activity_db = OpenDatabase(db_file);
+  activity_db->SetBatchModeForTesting(true);
+  scoped_refptr<Action> action = CreateAction(base::Time::Now(), "brewster");
+  Record(activity_db, action);
+  ASSERT_EQ(0, CountActions(&activity_db->db_, "brewster"));
+
+  // Request an immediate database flush.
+  activity_db->AdviseFlush(ActivityDatabase::kFlushImmediately);
+  ASSERT_EQ(1, CountActions(&activity_db->db_, "brewster"));
+
+  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::DeleteFile(db_file, false);
 
   ActivityDatabaseTestPolicy* delegate = new ActivityDatabaseTestPolicy();
   ActivityDatabase* activity_db = new ActivityDatabase(delegate);
   scoped_refptr<Action> action = new Action(
       "punky", base::Time::Now(), Action::ACTION_API_CALL, "brewster");
   action->mutable_args()->AppendString("woof");
   delegate->Record(activity_db, action);
   activity_db->Close();
 }
 
 }  // namespace extensions