OLD | NEW |
1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "ios/web/public/test/http_server_util.h" | 5 #include "ios/web/public/test/http_server_util.h" |
6 | 6 |
7 #include "base/memory/ptr_util.h" | 7 #include "base/memory/ptr_util.h" |
8 #include "base/path_service.h" | 8 #include "base/path_service.h" |
9 #import "ios/web/public/test/http_server.h" | 9 #import "ios/web/public/test/http_server.h" |
10 #include "ios/web/public/test/response_providers/file_based_response_provider.h" | 10 #include "ios/web/public/test/response_providers/file_based_response_provider.h" |
11 #include "ios/web/public/test/response_providers/html_response_provider.h" | 11 #include "ios/web/public/test/response_providers/html_response_provider.h" |
12 | 12 |
13 namespace web { | 13 namespace web { |
14 namespace test { | 14 namespace test { |
15 | 15 |
16 void SetUpSimpleHttpServer(const std::map<GURL, std::string>& responses) { | 16 void SetUpSimpleHttpServer(const std::map<GURL, std::string>& responses) { |
17 SetUpHttpServer(base::MakeUnique<HtmlResponseProvider>(responses)); | 17 SetUpHttpServer(base::MakeUnique<HtmlResponseProvider>(responses)); |
18 } | 18 } |
19 | 19 |
| 20 void SetUpSimpleHttpServerWithSetCookie( |
| 21 const std::map<GURL, std::string>& responses, |
| 22 const std::string& cookie) { |
| 23 SetUpHttpServer(base::MakeUnique<HtmlResponseProvider>(responses, cookie)); |
| 24 } |
| 25 |
20 void SetUpFileBasedHttpServer() { | 26 void SetUpFileBasedHttpServer() { |
21 base::FilePath path; | 27 base::FilePath path; |
22 PathService::Get(base::DIR_MODULE, &path); | 28 PathService::Get(base::DIR_MODULE, &path); |
23 SetUpHttpServer(base::MakeUnique<FileBasedResponseProvider>(path)); | 29 SetUpHttpServer(base::MakeUnique<FileBasedResponseProvider>(path)); |
24 } | 30 } |
25 | 31 |
26 void SetUpHttpServer(std::unique_ptr<web::ResponseProvider> provider) { | 32 void SetUpHttpServer(std::unique_ptr<web::ResponseProvider> provider) { |
27 web::test::HttpServer& server = web::test::HttpServer::GetSharedInstance(); | 33 web::test::HttpServer& server = web::test::HttpServer::GetSharedInstance(); |
28 DCHECK(server.IsRunning()); | 34 DCHECK(server.IsRunning()); |
29 | 35 |
30 server.RemoveAllResponseProviders(); | 36 server.RemoveAllResponseProviders(); |
31 server.AddResponseProvider(std::move(provider)); | 37 server.AddResponseProvider(std::move(provider)); |
32 } | 38 } |
33 | 39 |
34 void AddResponseProvider(std::unique_ptr<web::ResponseProvider> provider) { | 40 void AddResponseProvider(std::unique_ptr<web::ResponseProvider> provider) { |
35 web::test::HttpServer& server = web::test::HttpServer::GetSharedInstance(); | 41 web::test::HttpServer& server = web::test::HttpServer::GetSharedInstance(); |
36 DCHECK(server.IsRunning()); | 42 DCHECK(server.IsRunning()); |
37 server.AddResponseProvider(std::move(provider)); | 43 server.AddResponseProvider(std::move(provider)); |
38 } | 44 } |
39 | 45 |
40 } // namespace test | 46 } // namespace test |
41 } // namespace web | 47 } // namespace web |
OLD | NEW |