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( | 20 void SetUpSimpleHttpServerWithSetCookies( |
21 const std::map<GURL, std::string>& responses, | 21 const std::map<GURL, std::pair<std::string, std::string>>& responses) { |
22 const std::string& cookie) { | 22 SetUpHttpServer(base::MakeUnique<HtmlResponseProvider>(responses)); |
23 SetUpHttpServer(base::MakeUnique<HtmlResponseProvider>(responses, cookie)); | |
24 } | 23 } |
25 | 24 |
26 void SetUpFileBasedHttpServer() { | 25 void SetUpFileBasedHttpServer() { |
27 base::FilePath path; | 26 base::FilePath path; |
28 PathService::Get(base::DIR_MODULE, &path); | 27 PathService::Get(base::DIR_MODULE, &path); |
29 SetUpHttpServer(base::MakeUnique<FileBasedResponseProvider>(path)); | 28 SetUpHttpServer(base::MakeUnique<FileBasedResponseProvider>(path)); |
30 } | 29 } |
31 | 30 |
32 void SetUpHttpServer(std::unique_ptr<web::ResponseProvider> provider) { | 31 void SetUpHttpServer(std::unique_ptr<web::ResponseProvider> provider) { |
33 web::test::HttpServer& server = web::test::HttpServer::GetSharedInstance(); | 32 web::test::HttpServer& server = web::test::HttpServer::GetSharedInstance(); |
34 DCHECK(server.IsRunning()); | 33 DCHECK(server.IsRunning()); |
35 | 34 |
36 server.RemoveAllResponseProviders(); | 35 server.RemoveAllResponseProviders(); |
37 server.AddResponseProvider(std::move(provider)); | 36 server.AddResponseProvider(std::move(provider)); |
38 } | 37 } |
39 | 38 |
40 void AddResponseProvider(std::unique_ptr<web::ResponseProvider> provider) { | 39 void AddResponseProvider(std::unique_ptr<web::ResponseProvider> provider) { |
41 web::test::HttpServer& server = web::test::HttpServer::GetSharedInstance(); | 40 web::test::HttpServer& server = web::test::HttpServer::GetSharedInstance(); |
42 DCHECK(server.IsRunning()); | 41 DCHECK(server.IsRunning()); |
43 server.AddResponseProvider(std::move(provider)); | 42 server.AddResponseProvider(std::move(provider)); |
44 } | 43 } |
45 | 44 |
46 } // namespace test | 45 } // namespace test |
47 } // namespace web | 46 } // namespace web |
OLD | NEW |