Expose API to store and load data usage stats from UI task runner.

chrome/browser/renderer_context_menu/render_view_context_menu_unittest.cc

 // Copyright 2014 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 "chrome/browser/renderer_context_menu/render_view_context_menu.h"
 
 #include "base/prefs/pref_registry_simple.h"
 #include "base/prefs/pref_service.h"
 #include "base/run_loop.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/thread_task_runner_handle.h"
 #include "chrome/app/chrome_command_ids.h"
 #include "chrome/browser/custom_handlers/protocol_handler_registry.h"
 #include "chrome/browser/extensions/menu_manager.h"
 #include "chrome/browser/extensions/menu_manager_factory.h"
 #include "chrome/browser/extensions/test_extension_environment.h"
 #include "chrome/browser/extensions/test_extension_prefs.h"
 #include "chrome/browser/net/spdyproxy/data_reduction_proxy_chrome_settings.h"
 #include "chrome/browser/net/spdyproxy/data_reduction_proxy_chrome_settings_factory.h"
 #include "chrome/browser/prefs/incognito_mode_prefs.h"
 #include "chrome/browser/renderer_context_menu/render_view_context_menu_test_util.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/test/base/chrome_render_view_host_test_harness.h"
 #include "chrome/test/base/testing_profile.h"
 #include "components/data_reduction_proxy/core/browser/data_reduction_proxy_test_utils.h"
+#include "components/data_reduction_proxy/core/browser/data_store.h"
 #include "components/data_reduction_proxy/core/common/data_reduction_proxy_headers.h"
 #include "components/data_reduction_proxy/core/common/data_reduction_proxy_params.h"
 #include "components/data_reduction_proxy/core/common/data_reduction_proxy_pref_names.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/test/web_contents_tester.h"
 
 #include "extensions/browser/extension_prefs.h"
 #include "extensions/common/url_pattern.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "third_party/WebKit/public/web/WebContextMenuData.h"
@@ skipping 347 matching lines @@
     registry->RegisterDictionaryPref(prefs::kProxy);
 
     drp_test_context_->pref_service()->SetBoolean(
         data_reduction_proxy::prefs::kDataReductionProxyEnabled,
         enable_data_reduction_proxy);
     drp_test_context_->InitSettings();
 
     settings->InitDataReductionProxySettings(
         drp_test_context_->io_data(), drp_test_context_->pref_service(),
         drp_test_context_->request_context_getter(),
+        make_scoped_ptr(new data_reduction_proxy::DataStore()),
+        base::ThreadTaskRunnerHandle::Get(),
         base::ThreadTaskRunnerHandle::Get());
   }
 
+  // Force destruction of |DataReductionProxySettings| so that objects on DB
+  // task runner can be destroyed before test threads are destroyed. This method
+  // must be called by tests that call |SetupDataReductionProxy|. We cannot
+  // destroy |drp_test_context_| until browser context keyed services are
+  // destroyed since |DataReductionProxyChromeSettings| holds a pointer to the
+  // |PrefService|, which is owned by |drp_test_context_|.
+  void DestroyDataReductionProxySettings() {
+    drp_test_context_->DestroySettings();
+  }
+
  protected:
   scoped_ptr<data_reduction_proxy::DataReductionProxyTestContext>
       drp_test_context_;
 
  private:
   scoped_ptr<ProtocolHandlerRegistry> registry_;
 };
 
 // Verifies when Incognito Mode is not available (disabled by policy),
 // Open Link in Incognito Window link in the context menu is disabled.
@@ skipping 36 matching lines @@
   content::WebContents* wc = web_contents();
   scoped_ptr<TestRenderViewContextMenu> menu(
       new TestRenderViewContextMenu(wc->GetMainFrame(), params));
 
   menu->ExecuteCommand(IDC_CONTENT_CONTEXT_SAVEIMAGEAS, 0);
 
   const std::string& headers =
       content::WebContentsTester::For(web_contents())->GetSaveFrameHeaders();
   EXPECT_TRUE(headers.find("Chrome-Proxy: pass-through") != std::string::npos);
   EXPECT_TRUE(headers.find("Cache-Control: no-cache") != std::string::npos);
+
+  DestroyDataReductionProxySettings();
 }
 
 // Verify that request headers do not specify pass through when "Save Image
 // As..." is used with Data Saver disabled.
 TEST_F(RenderViewContextMenuPrefsTest, DataSaverDisabledSaveImageAs) {
   SetupDataReductionProxy(false);
 
   content::ContextMenuParams params = CreateParams(MenuItem::IMAGE);
   params.unfiltered_link_url = params.link_url;
   content::WebContents* wc = web_contents();
   scoped_ptr<TestRenderViewContextMenu> menu(
       new TestRenderViewContextMenu(wc->GetMainFrame(), params));
 
   menu->ExecuteCommand(IDC_CONTENT_CONTEXT_SAVEIMAGEAS, 0);
 
   const std::string& headers =
       content::WebContentsTester::For(web_contents())->GetSaveFrameHeaders();
   EXPECT_TRUE(headers.find("Chrome-Proxy: pass-through") == std::string::npos);
   EXPECT_TRUE(headers.find("Cache-Control: no-cache") == std::string::npos);
+
+  DestroyDataReductionProxySettings();
 }
 
 // Verify that the Chrome-Proxy Lo-Fi directive causes the context menu to
 // display the "Load Image" menu item.
 TEST_F(RenderViewContextMenuPrefsTest, DataSaverLoadImage) {
   SetupDataReductionProxy(true);
   content::ContextMenuParams params = CreateParams(MenuItem::IMAGE);
   params.properties[data_reduction_proxy::chrome_proxy_header()] =
       data_reduction_proxy::chrome_proxy_lo_fi_directive();
   params.unfiltered_link_url = params.link_url;
   content::WebContents* wc = web_contents();
   scoped_ptr<TestRenderViewContextMenu> menu(
       new TestRenderViewContextMenu(wc->GetMainFrame(), params));
   AppendImageItems(menu.get());
 
   ASSERT_TRUE(menu->IsItemPresent(IDC_CONTENT_CONTEXT_LOAD_ORIGINAL_IMAGE));
+
+  DestroyDataReductionProxySettings();
 }