OLD | NEW |
1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2013 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 #ifndef CHROME_TEST_CHROMEDRIVER_SERVER_HTTP_HANDLER_H_ | 5 #ifndef CHROME_TEST_CHROMEDRIVER_SERVER_HTTP_HANDLER_H_ |
6 #define CHROME_TEST_CHROMEDRIVER_SERVER_HTTP_HANDLER_H_ | 6 #define CHROME_TEST_CHROMEDRIVER_SERVER_HTTP_HANDLER_H_ |
7 | 7 |
8 #include <string> | 8 #include <string> |
9 #include <vector> | 9 #include <vector> |
10 | 10 |
| 11 #include "base/compiler_specific.h" |
| 12 #include "base/gtest_prod_util.h" |
11 #include "base/memory/scoped_ptr.h" | 13 #include "base/memory/scoped_ptr.h" |
| 14 #include "base/threading/thread.h" |
| 15 #include "chrome/test/chromedriver/command.h" |
| 16 #include "chrome/test/chromedriver/element_commands.h" |
| 17 #include "chrome/test/chromedriver/net/sync_websocket_factory.h" |
| 18 #include "chrome/test/chromedriver/session_commands.h" |
| 19 #include "chrome/test/chromedriver/session_map.h" |
| 20 #include "chrome/test/chromedriver/window_commands.h" |
12 | 21 |
13 namespace base { | 22 namespace base { |
14 class DictionaryValue; | 23 class DictionaryValue; |
15 } | 24 } |
16 | 25 |
17 class CommandExecutor; | 26 class Adb; |
| 27 class DeviceManager; |
| 28 class Log; |
18 class HttpResponse; | 29 class HttpResponse; |
19 class Log; | 30 class URLRequestContextGetter; |
20 | 31 |
21 enum HttpMethod { | 32 enum HttpMethod { |
22 kGet = 0, | 33 kGet = 0, |
23 kPost, | 34 kPost, |
24 kDelete, | 35 kDelete, |
25 }; | 36 }; |
26 | 37 |
27 struct HttpRequest { | 38 struct HttpRequest { |
28 HttpRequest(HttpMethod method, | 39 HttpRequest(HttpMethod method, |
29 const std::string& path, | 40 const std::string& path, |
30 const std::string& body); | 41 const std::string& body); |
31 ~HttpRequest(); | 42 ~HttpRequest(); |
32 | 43 |
33 HttpMethod method; | 44 HttpMethod method; |
34 std::string path; | 45 std::string path; |
35 std::string body; | 46 std::string body; |
36 }; | 47 }; |
37 | 48 |
38 struct CommandMapping { | 49 struct CommandMapping { |
39 CommandMapping(HttpMethod method, | 50 CommandMapping(HttpMethod method, |
40 const std::string& path_pattern, | 51 const std::string& path_pattern, |
41 const std::string& name); | 52 const Command& command); |
42 ~CommandMapping(); | 53 ~CommandMapping(); |
43 | 54 |
44 HttpMethod method; | 55 HttpMethod method; |
45 std::string path_pattern; | 56 std::string path_pattern; |
46 std::string name; | 57 Command command; |
47 }; | 58 }; |
48 | 59 |
49 class HttpHandler { | 60 class HttpHandler { |
50 public: | 61 public: |
51 typedef std::vector<CommandMapping> CommandMap; | 62 HttpHandler(Log* log, const std::string& url_base); |
52 static scoped_ptr<CommandMap> CreateCommandMap(); | |
53 | |
54 HttpHandler(Log* log, | |
55 scoped_ptr<CommandExecutor> executor, | |
56 scoped_ptr<std::vector<CommandMapping> > commands, | |
57 const std::string& url_base); | |
58 ~HttpHandler(); | 63 ~HttpHandler(); |
59 | 64 |
60 void Handle(const HttpRequest& request, | 65 void Handle(const HttpRequest& request, HttpResponse* response); |
61 HttpResponse* response); | |
62 | |
63 bool ShouldShutdown(const HttpRequest& request); | 66 bool ShouldShutdown(const HttpRequest& request); |
64 | 67 |
65 private: | 68 private: |
66 void HandleInternal(const HttpRequest& request, | 69 FRIEND_TEST_ALL_PREFIXES(HttpHandlerTest, HandleUnknownCommand); |
67 HttpResponse* response); | 70 FRIEND_TEST_ALL_PREFIXES(HttpHandlerTest, HandleNewSession); |
| 71 FRIEND_TEST_ALL_PREFIXES(HttpHandlerTest, HandleInvalidPost); |
| 72 FRIEND_TEST_ALL_PREFIXES(HttpHandlerTest, HandleUnimplementedCommand); |
| 73 FRIEND_TEST_ALL_PREFIXES(HttpHandlerTest, HandleCommand); |
| 74 typedef std::vector<CommandMapping> CommandMap; |
| 75 |
| 76 Command WrapToCommand(const SessionCommand& session_command); |
| 77 Command WrapToCommand(const WindowCommand& window_command); |
| 78 Command WrapToCommand(const ElementCommand& element_command); |
| 79 void HandleInternal(const HttpRequest& request, HttpResponse* response); |
68 bool HandleWebDriverCommand( | 80 bool HandleWebDriverCommand( |
69 const HttpRequest& request, | 81 const HttpRequest& request, |
70 const std::string& trimmed_path, | 82 const std::string& trimmed_path, |
71 HttpResponse* response); | 83 HttpResponse* response); |
72 | 84 |
73 Log* log_; | 85 Log* log_; |
74 scoped_ptr<CommandExecutor> executor_; | 86 base::Thread io_thread_; |
| 87 std::string url_base_; |
| 88 scoped_refptr<URLRequestContextGetter> context_getter_; |
| 89 SyncWebSocketFactory socket_factory_; |
| 90 SessionMap session_map_; |
75 scoped_ptr<CommandMap> command_map_; | 91 scoped_ptr<CommandMap> command_map_; |
76 std::string url_base_; | 92 scoped_ptr<Adb> adb_; |
| 93 scoped_ptr<DeviceManager> device_manager_; |
| 94 |
| 95 DISALLOW_COPY_AND_ASSIGN(HttpHandler); |
77 }; | 96 }; |
78 | 97 |
79 namespace internal { | 98 namespace internal { |
80 extern const char kNewSessionIdCommand[]; | 99 |
| 100 extern const char kNewSessionPathPattern[]; |
| 101 |
81 bool MatchesCommand(HttpMethod method, | 102 bool MatchesCommand(HttpMethod method, |
82 const std::string& path, | 103 const std::string& path, |
83 const CommandMapping& command, | 104 const CommandMapping& command, |
84 std::string* session_id, | 105 std::string* session_id, |
85 base::DictionaryValue* out_params); | 106 base::DictionaryValue* out_params); |
| 107 |
86 } // namespace internal | 108 } // namespace internal |
87 | 109 |
88 #endif // CHROME_TEST_CHROMEDRIVER_SERVER_HTTP_HANDLER_H_ | 110 #endif // CHROME_TEST_CHROMEDRIVER_SERVER_HTTP_HANDLER_H_ |
OLD | NEW |