Cookie commands for the webdriver protocol

chrome/test/webdriver/commands/cookie_commands.h

Index: chrome/test/webdriver/commands/cookie_commands.h
diff --git a/chrome/test/webdriver/commands/cookie_commands.h b/chrome/test/webdriver/commands/cookie_commands.h
new file mode 100644
index 0000000000000000000000000000000000000000..d1024dec822924db61e10720cbb3613d447f40ef
--- /dev/null
+++ b/chrome/test/webdriver/commands/cookie_commands.h
@@ -0,0 +1,72 @@
+// Copyright (c) 2010 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#ifndef CHROME_TEST_WEBDRIVER_COMMANDS_COOKIE_COMMANDS_H_
+#define CHROME_TEST_WEBDRIVER_COMMANDS_COOKIE_COMMANDS_H_
+

[kkania, 2011/01/28 23:25:59]

add '#pragma once' here

[Joe, 2011/02/03 10:02:05]

Done.

+#include "chrome/test/webdriver/commands/webdriver_command.h"
+
+namespace webdriver {
+
+// Retrieve all cookies visible to the current page. Each cookie will be
+// returned as a JSON object with the following properties:
+// name, value, path, domain, secure, and expiry. See:
+// http://code.google.com/p/selenium/wiki/JsonWireProtocol#/session/:sessionId/cookie
+class CookieCommand: public WebDriverCommand {
+ public:
+   CookieCommand(const std::vector<std::string> path_segments,
+                 const DictionaryValue* const parameters)
+      : WebDriverCommand(path_segments, parameters) {}
+  virtual ~CookieCommand(){}
+
+  virtual bool Init(Response* const response);
+
+  virtual bool DoesDelete() { return true; }
+  virtual bool DoesGet() { return true; }
+  virtual bool DoesPost() { return true; }
+
+  virtual void ExecuteDelete(Response* const response);
+  virtual void ExecuteGet(Response* const response);
+  virtual void ExecutePost(Response* const response);
+
+ private:
+  GURL current_url_;
+
+  virtual bool RequiresValidTab() { return true; }
+
+  DISALLOW_COPY_AND_ASSIGN(CookieCommand);
+};
+
+// Set a cookie. The cookie should be specified as a JSON object with the
+// following properties: name, value, path, domain, secure, and expiry. See:
+// http://code.google.com/p/selenium/wiki/JsonWireProtocol#/session/:sessionId/cookie/:name
+class NamedCookieCommand : public WebDriverCommand {
+ public:
+   NamedCookieCommand(const std::vector<std::string> path_segments,
+                      const DictionaryValue* const parameters)
+      : WebDriverCommand(path_segments, parameters) {}
+  virtual ~NamedCookieCommand(){}
+
+  virtual bool Init(Response* const response);
+
+ protected:
+  virtual bool DoesDelete() { return true; }
+  virtual bool DoesGet() { return true; }
+
+  virtual void ExecuteDelete(Response* const response);
+  virtual void ExecuteGet(Response* const response);
+
+ private:
+  GURL current_url_;
+  std::string cookie_name_;
+
+  virtual bool RequiresValidTab() { return true; }
+
+  DISALLOW_COPY_AND_ASSIGN(NamedCookieCommand);
+};
+
+}  // namespace webdriver
+
+#endif  // CHROME_TEST_WEBDRIVER_COMMANDS_COOKIE_COMMANDS_H_
+