Cookie commands for the webdriver protocol

chrome/test/webdriver/session.cc

 // Copyright (c) 2011 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.
 
 #include "chrome/test/webdriver/session_manager.h"
 
 #include <vector>
 
 #include "base/command_line.h"
 #include "base/file_util.h"
 #include "base/logging.h"
 #include "base/message_loop_proxy.h"
 #include "base/process.h"
 #include "base/process_util.h"
 #include "base/string_util.h"
 #include "base/json/json_reader.h"
 #include "base/json/json_writer.h"
 #include "base/test/test_timeouts.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/app/chrome_command_ids.h"
 #include "chrome/common/chrome_constants.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/test/test_launcher_utils.h"
 #include "chrome/test/webdriver/utility_functions.h"
+#include "googleurl/src/gurl.h"
 #include "third_party/webdriver/atoms.h"
 
 namespace webdriver {
 
 Session::Session(const std::string& id)
     : thread_(id.c_str()),
       id_(id),
       window_num_(0),
       implicit_wait_(0),
       current_frame_xpath_("") {
@@ skipping 132 matching lines @@
 bool Session::GetURL(std::string* url) {
   bool success = false;
   RunSessionTask(NewRunnableMethod(
       automation_.get(),
       &Automation::GetURL,
       url,
       &success));
   return success;
 }
 
+bool Session::GetURL(GURL* gurl) {
+  bool success = false;
+  RunSessionTask(NewRunnableMethod(
+      automation_.get(),
+      &Automation::GetGURL,
+      gurl,
+      &success));
+  return success;
+}
+
 bool Session::GetTabTitle(std::string* tab_title) {
   bool success = false;
   RunSessionTask(NewRunnableMethod(
       automation_.get(),
       &Automation::GetTabTitle,
       tab_title,
       &success));
   return success;
 }
 
+bool Session::GetCookies(const GURL& url, std::string* cookies) {
+  bool success = false;
+  RunSessionTask(NewRunnableMethod(
+      automation_.get(),
+      &Automation::GetCookies,
+      url,
+      cookies,
+      &success));
+  return success;
+}
+
+bool Session::GetCookieByName(const GURL& url,
+                              const std::string& cookie_name,
+                              std::string* cookie) {
+  bool success = false;
+  RunSessionTask(NewRunnableMethod(
+      automation_.get(),
+      &Automation::GetCookieByName,
+      url,
+      cookie_name,
+      cookie,
+      &success));
+  return success;
+}
+
+bool Session::DeleteCookie(const GURL& url, const std::string& cookie_name) {
+  bool success = false;
+  RunSessionTask(NewRunnableMethod(
+      automation_.get(),
+      &Automation::DeleteCookie,
+      url,
+      cookie_name,
+      &success));
+  return success;
+}
+
+bool Session::SetCookie(const GURL& url, const std::string& cookie) {
+  bool success = false;
+  RunSessionTask(NewRunnableMethod(
+      automation_.get(),
+      &Automation::SetCookie,
+      url,
+      cookie,
+      &success));
+  return success;
+}
+
 void Session::RunSessionTask(Task* task) {
   base::WaitableEvent done_event(false, false);
   thread_.message_loop_proxy()->PostTask(FROM_HERE, NewRunnableMethod(
       this,
       &Session::RunSessionTaskOnSessionThread,
       task,
       &done_event));
   done_event.Wait();
 }
 
 void Session::RunSessionTaskOnSessionThread(Task* task,
                                             base::WaitableEvent* done_event) {
   task->Run();
   delete task;
   done_event->Signal();
 }
 
 void Session::InitOnSessionThread(bool* success) {
   automation_.reset(new Automation());
   automation_->Init(success);
 }
 
 void Session::TerminateOnSessionThread() {
   automation_->Terminate();
   automation_.reset();
 }
 
 }  // namespace webdriver