src/chrome/test/chromedriver/server/http_handler.h

   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
   3 // found in the LICENSE file.
   4
   5 #ifndef CHROME_TEST_CHROMEDRIVER_SERVER_HTTP_HANDLER_H_
   6 #define CHROME_TEST_CHROMEDRIVER_SERVER_HTTP_HANDLER_H_
   7
   8 #include <string>
   9 #include <vector>
  10
  11 #include "base/callback.h"
  12 #include "base/compiler_specific.h"
  13 #include "base/gtest_prod_util.h"
  14 #include "base/memory/ref_counted.h"
  15 #include "base/memory/scoped_ptr.h"
  16 #include "base/memory/weak_ptr.h"
  17 #include "base/threading/thread_checker.h"
  18 #include "chrome/test/chromedriver/command.h"
  19 #include "chrome/test/chromedriver/commands.h"
  20 #include "chrome/test/chromedriver/element_commands.h"
  21 #include "chrome/test/chromedriver/net/sync_websocket_factory.h"
  22 #include "chrome/test/chromedriver/session_commands.h"
  23 #include "chrome/test/chromedriver/session_thread_map.h"
  24 #include "chrome/test/chromedriver/window_commands.h"
  25
  26 namespace base {
  27 class DictionaryValue;
  28 class SingleThreadTaskRunner;
  29 }
  30
  31 namespace net {
  32 class HttpServerRequestInfo;
  33 class HttpServerResponseInfo;
  34 }
  35
  36 class Adb;
  37 class DeviceManager;
  38 class PortManager;
  39 class PortServer;
  40 class URLRequestContextGetter;
  41
  42 enum HttpMethod {
  43   kGet,
  44   kPost,
  45   kDelete,
  46 };
  47
  48 struct CommandMapping {
  49   CommandMapping(HttpMethod method,
  50                  const std::string& path_pattern,
  51                  const Command& command);
  52   ~CommandMapping();
  53
  54   HttpMethod method;
  55   std::string path_pattern;
  56   Command command;
  57 };
  58
  59 typedef base::Callback<void(scoped_ptr<net::HttpServerResponseInfo>)>
  60     HttpResponseSenderFunc;
  61
  62 class HttpHandler {
  63  public:
  64   explicit HttpHandler(const std::string& url_base);
  65   HttpHandler(const base::Closure& quit_func,
  66               const scoped_refptr<base::SingleThreadTaskRunner> io_task_runner,
  67               const std::string& url_base,
  68               int adb_port,
  69               scoped_ptr<PortServer> port_server);
  70   ~HttpHandler();
  71
  72   void Handle(const net::HttpServerRequestInfo& request,
  73               const HttpResponseSenderFunc& send_response_func);
  74
  75  private:
  76   FRIEND_TEST_ALL_PREFIXES(HttpHandlerTest, HandleUnknownCommand);
  77   FRIEND_TEST_ALL_PREFIXES(HttpHandlerTest, HandleNewSession);
  78   FRIEND_TEST_ALL_PREFIXES(HttpHandlerTest, HandleInvalidPost);
  79   FRIEND_TEST_ALL_PREFIXES(HttpHandlerTest, HandleUnimplementedCommand);
  80   FRIEND_TEST_ALL_PREFIXES(HttpHandlerTest, HandleCommand);
  81   typedef std::vector<CommandMapping> CommandMap;
  82
  83   Command WrapToCommand(const char* name,
  84                         const SessionCommand& session_command);
  85   Command WrapToCommand(const char* name, const WindowCommand& window_command);
  86   Command WrapToCommand(const char* name,
  87                         const ElementCommand& element_command);
  88   void HandleCommand(const net::HttpServerRequestInfo& request,
  89                      const std::string& trimmed_path,
  90                      const HttpResponseSenderFunc& send_response_func);
  91   void PrepareResponse(const std::string& trimmed_path,
  92                        const HttpResponseSenderFunc& send_response_func,
  93                        const Status& status,
  94                        scoped_ptr<base::Value> value,
  95                        const std::string& session_id);
  96   scoped_ptr<net::HttpServerResponseInfo> PrepareResponseHelper(
  97       const std::string& trimmed_path,
  98       const Status& status,
  99       scoped_ptr<base::Value> value,
 100       const std::string& session_id);
 101
 102   base::ThreadChecker thread_checker_;
 103   base::Closure quit_func_;
 104   std::string url_base_;
 105   bool received_shutdown_;
 106   scoped_refptr<URLRequestContextGetter> context_getter_;
 107   SyncWebSocketFactory socket_factory_;
 108   SessionThreadMap session_thread_map_;
 109   scoped_ptr<CommandMap> command_map_;
 110   scoped_ptr<Adb> adb_;
 111   scoped_ptr<DeviceManager> device_manager_;
 112   scoped_ptr<PortServer> port_server_;
 113   scoped_ptr<PortManager> port_manager_;
 114
 115   base::WeakPtrFactory<HttpHandler> weak_ptr_factory_;
 116
 117   DISALLOW_COPY_AND_ASSIGN(HttpHandler);
 118 };
 119
 120 namespace internal {
 121
 122 extern const char kNewSessionPathPattern[];
 123
 124 bool MatchesCommand(const std::string& method,
 125                     const std::string& path,
 126                     const CommandMapping& command,
 127                     std::string* session_id,
 128                     base::DictionaryValue* out_params);
 129
 130 }  // namespace internal
 131
 132 #endif  // CHROME_TEST_CHROMEDRIVER_SERVER_HTTP_HANDLER_H_