Activity Log: allow searching by day

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/cancelable_callback.h"
 #include "base/command_line.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/run_loop.h"
 #include "base/synchronization/waitable_event.h"
 #include "base/test/simple_test_clock.h"
@@ skipping 42 matching lines @@
 #if defined OS_CHROMEOS
     test_user_manager_.reset();
 #endif
     base::RunLoop().RunUntilIdle();
     profile_.reset(NULL);
     base::RunLoop().RunUntilIdle();
     // Restore the original command line and undo the affects of SetUp().
     *CommandLine::ForCurrentProcess() = saved_cmdline_;
   }
 
-  // A helper function to call ReadData on a policy object and wait for the
-  // results to be processed.
+  // A wrapper function for CheckReadFilteredData, so that we don't need to
+  // enter empty string values for parameters we don't care about.
   void CheckReadData(
       ActivityLogPolicy* policy,
       const std::string& extension_id,
-      const int day,
+      int day,
       const base::Callback<void(scoped_ptr<Action::ActionVector>)>& checker) {
-    // Submit a request to the policy to read back some data, and call the
-    // checker function when results are available.  This will happen on the
-    // database thread.
-    policy->ReadData(
-        extension_id,
-        day,
-        base::Bind(&FullStreamUIPolicyTest::CheckWrapper,
-                   checker,
-                   base::MessageLoop::current()->QuitClosure()));
-
-    // Set up a timeout that will trigger after 5 seconds; if we haven't
-    // received any results by then assume that the test is broken.
-    base::CancelableClosure timeout(
-        base::Bind(&FullStreamUIPolicyTest::TimeoutCallback));
-    base::MessageLoop::current()->PostDelayedTask(
-        FROM_HERE, timeout.callback(), base::TimeDelta::FromSeconds(5));
-
-    // Wait for results; either the checker or the timeout callbacks should
-    // cause the main loop to exit.
-    base::MessageLoop::current()->Run();
-
-    timeout.Cancel();
+    CheckReadFilteredData(
+        policy, extension_id, Action::ACTION_ANY, "", "", "", day, checker);
   }
 
   // A helper function to call ReadFilteredData on a policy object and wait for
   // the results to be processed.
   void CheckReadFilteredData(
       ActivityLogPolicy* policy,
       const std::string& extension_id,
       const Action::ActionType type,
       const std::string& api_name,
       const std::string& page_url,
       const std::string& arg_url,
+      const int days_ago,
       const base::Callback<void(scoped_ptr<Action::ActionVector>)>& checker) {
     // Submit a request to the policy to read back some data, and call the
     // checker function when results are available.  This will happen on the
     // database thread.
     policy->ReadFilteredData(
         extension_id,
         type,
         api_name,
         page_url,
         arg_url,
+        days_ago,
         base::Bind(&FullStreamUIPolicyTest::CheckWrapper,
                    checker,
                    base::MessageLoop::current()->QuitClosure()));
 
     // Set up a timeout that will trigger after 5 seconds; if we haven't
     // received any results by then assume that the test is broken.
     base::CancelableClosure timeout(
         base::Bind(&FullStreamUIPolicyTest::TimeoutCallback));
     base::MessageLoop::current()->PostDelayedTask(
         FROM_HERE, timeout.callback(), base::TimeDelta::FromSeconds(5));
@@ skipping 211 matching lines @@
   action_dom->set_page_url(gurl);
   policy->ProcessAction(action_dom);
 
   CheckReadFilteredData(
       policy,
       extension->id(),
       Action::ACTION_API_CALL,
       "tabs.testMethod",
       "",
       "",
+      -1,
       base::Bind(
           &FullStreamUIPolicyTest::RetrieveActions_FetchFilteredActions1));
 
   CheckReadFilteredData(
       policy,
       "",
       Action::ACTION_DOM_ACCESS,
       "",
       "",
       "",
+      -1,
       base::Bind(
           &FullStreamUIPolicyTest::RetrieveActions_FetchFilteredActions1));
 
   CheckReadFilteredData(
       policy,
       "",
       Action::ACTION_DOM_ACCESS,
       "",
       "http://www.google.com/",
       "",
+      -1,
       base::Bind(
           &FullStreamUIPolicyTest::RetrieveActions_FetchFilteredActions1));
 
   CheckReadFilteredData(
       policy,
       "",
       Action::ACTION_DOM_ACCESS,
       "",
       "http://www.google.com",
       "",
+      -1,
       base::Bind(
           &FullStreamUIPolicyTest::RetrieveActions_FetchFilteredActions1));
 
   CheckReadFilteredData(
       policy,
       "",
       Action::ACTION_DOM_ACCESS,
       "",
       "http://www.goo",
       "",
+      -1,
       base::Bind(
           &FullStreamUIPolicyTest::RetrieveActions_FetchFilteredActions1));
 
   CheckReadFilteredData(
       policy,
       extension->id(),
       Action::ACTION_ANY,
       "",
       "",
       "",
+      -1,
       base::Bind(
           &FullStreamUIPolicyTest::RetrieveActions_FetchFilteredActions2));
 
   policy->Close();
 }
 
 TEST_F(FullStreamUIPolicyTest, LogWithArguments) {
   ActivityLogPolicy* policy = new FullStreamUIPolicy(profile_.get());
   scoped_refptr<const Extension> extension =
       ExtensionBuilder()
@@ skipping 232 matching lines @@
 
   CheckReadData(
       policy,
       "punky",
       0,
       base::Bind(&FullStreamUIPolicyTest::SomeURLsRemoved));
   policy->Close();
 }
 
 }  // namespace extensions