1 // Copyright (c) 2012 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 #ifndef CONTENT_PUBLIC_TEST_TEST_BROWSER_CONTEXT_H_ 6 #define CONTENT_PUBLIC_TEST_TEST_BROWSER_CONTEXT_H_ 7 8 #include "base/compiler_specific.h" 9 #include "base/files/file_path.h" 10 #include "base/files/scoped_temp_dir.h" 11 #include "base/gtest_prod_util.h" 12 #include "base/memory/ref_counted.h" 13 #include "base/memory/scoped_ptr.h" 14 #include "content/public/browser/browser_context.h" 15 16 namespace content { 17 class MockResourceContext; 18 19 class TestBrowserContext : public BrowserContext { 20 public: 21 TestBrowserContext(); 22 virtual ~TestBrowserContext(); 23 24 // Takes ownership of the temporary directory so that it's not deleted when 25 // this object is destructed. 26 base::FilePath TakePath(); 27 28 void SetSpecialStoragePolicy(quota::SpecialStoragePolicy* policy); 29 30 virtual base::FilePath GetPath() const OVERRIDE; 31 virtual bool IsOffTheRecord() const OVERRIDE; 32 virtual DownloadManagerDelegate* GetDownloadManagerDelegate() OVERRIDE; 33 virtual net::URLRequestContextGetter* GetRequestContext() OVERRIDE; 34 virtual net::URLRequestContextGetter* GetRequestContextForRenderProcess( 35 int renderer_child_id) OVERRIDE; 36 virtual net::URLRequestContextGetter* GetMediaRequestContext() OVERRIDE; 37 virtual net::URLRequestContextGetter* GetMediaRequestContextForRenderProcess( 38 int renderer_child_id) OVERRIDE; 39 virtual net::URLRequestContextGetter* 40 GetMediaRequestContextForStoragePartition( 41 const base::FilePath& partition_path, 42 bool in_memory) OVERRIDE; 43 virtual void RequestMIDISysExPermission( 44 int render_process_id, 45 int render_view_id, 46 int bridge_id, 47 const GURL& requesting_frame, 48 const MIDISysExPermissionCallback& callback) OVERRIDE; 49 virtual void CancelMIDISysExPermissionRequest( 50 int render_process_id, 51 int render_view_id, 52 int bridge_id, 53 const GURL& requesting_frame) OVERRIDE; 54 virtual ResourceContext* GetResourceContext() OVERRIDE; 55 virtual GeolocationPermissionContext* 56 GetGeolocationPermissionContext() OVERRIDE; 57 virtual quota::SpecialStoragePolicy* GetSpecialStoragePolicy() OVERRIDE; 58 59 private: 60 FRIEND_TEST_ALL_PREFIXES(DOMStorageTest, SessionOnly); 61 FRIEND_TEST_ALL_PREFIXES(DOMStorageTest, SaveSessionState); 62 63 scoped_refptr<net::URLRequestContextGetter> request_context_; 64 scoped_ptr<MockResourceContext> resource_context_; 65 base::ScopedTempDir browser_context_dir_; 66 scoped_refptr<quota::SpecialStoragePolicy> special_storage_policy_; 67 68 DISALLOW_COPY_AND_ASSIGN(TestBrowserContext); 69 }; 70 71 } // namespace content 72 73 #endif // CONTENT_PUBLIC_TEST_TEST_BROWSER_CONTEXT_H_ 74