Chromium Code Reviews| 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 { |
| 50 CommandMapping(); | |
|
chrisgao (Use stgao instead)
2013/07/02 01:31:40
remove
chrisgao (Use stgao instead)
2013/07/02 18:13:04
Done.
| |
| 39 CommandMapping(HttpMethod method, | 51 CommandMapping(HttpMethod method, |
| 40 const std::string& path_pattern, | 52 const std::string& path_pattern, |
| 41 const std::string& name); | 53 const Command& command); |
| 42 ~CommandMapping(); | 54 ~CommandMapping(); |
| 43 | 55 |
| 44 HttpMethod method; | 56 HttpMethod method; |
| 45 std::string path_pattern; | 57 std::string path_pattern; |
| 46 std::string name; | 58 Command command; |
| 47 }; | 59 }; |
| 48 | 60 |
| 49 class HttpHandler { | 61 class HttpHandler { |
| 50 public: | 62 public: |
| 51 typedef std::vector<CommandMapping> CommandMap; | 63 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(); | 64 ~HttpHandler(); |
| 59 | 65 |
| 60 void Handle(const HttpRequest& request, | 66 void Init(); |
|
kkania
2013/07/02 01:57:08
can we remove the init and just do in the construc
chrisgao (Use stgao instead)
2013/07/02 18:13:04
Done.
| |
| 61 HttpResponse* response); | 67 void Handle(const HttpRequest& request, HttpResponse* response); |
| 62 | |
| 63 bool ShouldShutdown(const HttpRequest& request); | 68 bool ShouldShutdown(const HttpRequest& request); |
| 64 | 69 |
| 65 private: | 70 private: |
| 66 void HandleInternal(const HttpRequest& request, | 71 FRIEND_TEST_ALL_PREFIXES(HttpHandlerTest, HandleUnknownCommand); |
| 67 HttpResponse* response); | 72 FRIEND_TEST_ALL_PREFIXES(HttpHandlerTest, HandleNewSession); |
| 73 FRIEND_TEST_ALL_PREFIXES(HttpHandlerTest, HandleInvalidPost); | |
| 74 FRIEND_TEST_ALL_PREFIXES(HttpHandlerTest, HandleUnimplementedCommand); | |
| 75 FRIEND_TEST_ALL_PREFIXES(HttpHandlerTest, HandleCommand); | |
| 76 typedef std::vector<CommandMapping> CommandMap; | |
| 77 | |
| 78 Command WrapToNonSessionCommand(const SessionCommand& session_command); | |
| 79 Command WrapToNonSessionCommand(const WindowCommand& window_command); | |
| 80 Command WrapToNonSessionCommand(const ElementCommand& element_command); | |
| 81 void HandleInternal(const HttpRequest& request, HttpResponse* response); | |
| 68 bool HandleWebDriverCommand( | 82 bool HandleWebDriverCommand( |
| 69 const HttpRequest& request, | 83 const HttpRequest& request, |
| 70 const std::string& trimmed_path, | 84 const std::string& trimmed_path, |
| 71 HttpResponse* response); | 85 HttpResponse* response); |
| 72 | 86 |
| 73 Log* log_; | 87 Log* log_; |
| 74 scoped_ptr<CommandExecutor> executor_; | 88 base::Thread io_thread_; |
| 89 std::string url_base_; | |
| 90 scoped_refptr<URLRequestContextGetter> context_getter_; | |
| 91 SyncWebSocketFactory socket_factory_; | |
| 92 SessionMap session_map_; | |
| 75 scoped_ptr<CommandMap> command_map_; | 93 scoped_ptr<CommandMap> command_map_; |
| 76 std::string url_base_; | 94 scoped_ptr<Adb> adb_; |
| 95 scoped_ptr<DeviceManager> device_manager_; | |
| 96 | |
| 97 DISALLOW_COPY_AND_ASSIGN(HttpHandler); | |
| 77 }; | 98 }; |
| 78 | 99 |
| 79 namespace internal { | 100 namespace internal { |
| 80 extern const char kNewSessionIdCommand[]; | 101 |
| 102 extern const char kNewSessionPathPattern[]; | |
| 103 | |
| 81 bool MatchesCommand(HttpMethod method, | 104 bool MatchesCommand(HttpMethod method, |
| 82 const std::string& path, | 105 const std::string& path, |
| 83 const CommandMapping& command, | 106 const CommandMapping& command, |
| 84 std::string* session_id, | 107 std::string* session_id, |
| 85 base::DictionaryValue* out_params); | 108 base::DictionaryValue* out_params); |
| 109 | |
| 86 } // namespace internal | 110 } // namespace internal |
| 87 | 111 |
| 88 #endif // CHROME_TEST_CHROMEDRIVER_SERVER_HTTP_HANDLER_H_ | 112 #endif // CHROME_TEST_CHROMEDRIVER_SERVER_HTTP_HANDLER_H_ |
| OLD | NEW |