| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2014 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "content/test/browser_side_navigation_test_utils.h" | |
| 6 | |
| 7 #include "base/command_line.h" | |
| 8 #include "base/guid.h" | |
| 9 #include "base/lazy_instance.h" | |
| 10 #include "base/macros.h" | |
| 11 #include "content/browser/streams/stream.h" | |
| 12 #include "content/browser/streams/stream_registry.h" | |
| 13 #include "content/public/browser/browser_thread.h" | |
| 14 #include "content/public/browser/stream_handle.h" | |
| 15 #include "content/public/common/content_switches.h" | |
| 16 #include "content/test/test_navigation_url_loader_factory.h" | |
| 17 | |
| 18 namespace content { | |
| 19 | |
| 20 namespace { | |
| 21 | |
| 22 // PlzNavigate | |
| 23 // A UI thread singleton helper class for browser side navigations. When browser | |
| 24 // side navigations are enabled, initialize this class before doing any | |
| 25 // operation that may start a navigation request on the UI thread. Use TearDown | |
| 26 // at the end of the test. | |
| 27 class BrowserSideNavigationTestUtils { | |
| 28 public: | |
| 29 BrowserSideNavigationTestUtils() | |
| 30 : stream_registry_(new StreamRegistry), | |
| 31 loader_factory_(new TestNavigationURLLoaderFactory) { | |
| 32 } | |
| 33 | |
| 34 ~BrowserSideNavigationTestUtils() {} | |
| 35 StreamRegistry* stream_registry() { return stream_registry_.get();} | |
| 36 | |
| 37 private: | |
| 38 std::unique_ptr<StreamRegistry> stream_registry_; | |
| 39 std::unique_ptr<TestNavigationURLLoaderFactory> loader_factory_; | |
| 40 | |
| 41 DISALLOW_COPY_AND_ASSIGN(BrowserSideNavigationTestUtils); | |
| 42 }; | |
| 43 | |
| 44 base::LazyInstance<std::unique_ptr<BrowserSideNavigationTestUtils>> | |
| 45 browser_side_navigation_test_utils; | |
| 46 | |
| 47 } // namespace | |
| 48 | |
| 49 void BrowserSideNavigationSetUp() { | |
| 50 DCHECK_CURRENTLY_ON(BrowserThread::UI); | |
| 51 browser_side_navigation_test_utils.Get().reset( | |
| 52 new BrowserSideNavigationTestUtils); | |
| 53 } | |
| 54 | |
| 55 void BrowserSideNavigationTearDown() { | |
| 56 DCHECK_CURRENTLY_ON(BrowserThread::UI); | |
| 57 browser_side_navigation_test_utils.Get().reset(nullptr); | |
| 58 } | |
| 59 | |
| 60 std::unique_ptr<StreamHandle> MakeEmptyStream() { | |
| 61 GURL url(std::string(url::kBlobScheme) + "://" + base::GenerateGUID()); | |
| 62 StreamRegistry* stream_registry = | |
| 63 browser_side_navigation_test_utils.Get()->stream_registry(); | |
| 64 scoped_refptr<Stream> stream(new Stream(stream_registry, NULL, url)); | |
| 65 stream->Finalize(); | |
| 66 return stream->CreateHandle(); | |
| 67 } | |
| 68 | |
| 69 void EnableBrowserSideNavigation() { | |
| 70 base::CommandLine::ForCurrentProcess()->AppendSwitch( | |
| 71 switches::kEnableBrowserSideNavigation); | |
| 72 } | |
| 73 | |
| 74 } // namespace content | |
| OLD | NEW |