Prerender: omnibox and instant field trial parameters.

chrome/browser/prerender/prerender_test_utils.h

 // Copyright (c) 2016 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.
 
 #ifndef CHROME_BROWSER_PRERENDER_PRERENDER_TEST_UTILS_H_
 #define CHROME_BROWSER_PRERENDER_PRERENDER_TEST_UTILS_H_
 
 #include <deque>
 #include <functional>
 #include <memory>
@@ skipping 368 matching lines @@
       safe_browsing_factory_;
   TestPrerenderContentsFactory* prerender_contents_factory_;
   Browser* explicitly_set_browser_;
   bool autostart_test_server_;
   base::HistogramTester histogram_tester_;
   std::unique_ptr<net::EmbeddedTestServer> https_src_server_;
 
   DISALLOW_COPY_AND_ASSIGN(PrerenderInProcessBrowserTest);
 };
 
+// RAII class to save and restore the prerender mode.
+class RestorePrerenderMode {
+ public:
+  RestorePrerenderMode()
+      : prev_mode_(PrerenderManager::GetMode(ORIGIN_NONE)),
+        prev_omnibox_mode_(PrerenderManager::GetMode(ORIGIN_OMNIBOX)),
+        prev_instant_mode_(PrerenderManager::GetMode(ORIGIN_INSTANT)) {}
+
+  ~RestorePrerenderMode() {
+    PrerenderManager::SetMode(prev_mode_);
+    PrerenderManager::SetOmniboxMode(prev_omnibox_mode_);
+    PrerenderManager::SetInstantMode(prev_instant_mode_);
+  }
+
+ private:
+  PrerenderManager::PrerenderManagerMode prev_mode_;
+  PrerenderManager::PrerenderManagerMode prev_omnibox_mode_;
+  PrerenderManager::PrerenderManagerMode prev_instant_mode_;
+
+  DISALLOW_COPY_AND_ASSIGN(RestorePrerenderMode);
+};
+
 // Makes |url| respond to requests with the contents of |file|, counting the
 // number that start in |counter|.
 void CreateCountingInterceptorOnIO(
     const GURL& url,
     const base::FilePath& file,
     const base::WeakPtr<RequestCounter>& counter);
 
 // When the |url| hits the net::URLRequestFilter (on the IO thread), executes
 // the |callback_io| providing the request to it, also pings the |counter| on UI
 // thread. Does not modify the behavior or the request job.
 void InterceptRequestAndCount(
     const GURL& url,
     RequestCounter* counter,
     base::Callback<void(net::URLRequest*)> callback_io);
 
 // Makes |url| respond to requests with the contents of |file|.
 void CreateMockInterceptorOnIO(const GURL& url, const base::FilePath& file);
 
 // Makes |url| never respond on the first load, and then with the contents of
 // |file| afterwards. When the first load has been scheduled, runs |callback| on
 // the UI thread.
 void CreateHangingFirstRequestInterceptorOnIO(
     const GURL& url, const base::FilePath& file, base::Closure callback);
 
 }  // namespace test_utils
 
 }  // namespace prerender
 
 #endif  // CHROME_BROWSER_PRERENDER_PRERENDER_TEST_UTILS_H_