Cookie commands for the webdriver protocol

chrome/test/webdriver/webdriver_remote_tests.py

 #!/usr/bin/python
 # 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.
 
 try:
   import json
 except ImportError:  # < 2.6
   import simplejson as json
 
@@ skipping 80 matching lines @@
         WEBDRIVER_PROCESS = None
 
   """Preforms a string search ignoring case"""
   def assertFind(self, text, search):
     text = string.lower(text)
     search = string.lower(search)
     self.assertNotEqual(-1, string.find(text, search))
 
 class TestFindElement(RemoteWebDriverTest):
   def testFindByName(self):
-    navigate(SEARCH)
+    self.navigate(SEARCH)
     # Find the Google search button.
     q = self.driver.find_element_by_name("q")
     self.assertTrue(isinstance(q, WebElement))
     # Trying looking for an element not on the page.
     self.assertRaises(NoSuchElementException,
                       self.driver.find_elment_by_name, "q2")
     # Try to find the Google search button using the multiple find method.
     q = self.driver.find_elements_by_name("q")
     self.assertTrue(isinstance(q, list))
     self.assertTrue(len(q), 1)
     self.assertTrue(isinstance(q[0], WebElement))
     # Try finding something not on page, with multiple find an empty array
     # should return and no exception thrown.
     q = self.driver.find_elements_by_name("q2")
     assertTrue(q == [])
     # Find a hidden element on the page
     q = self.driver.find_element_by_name("oq")
     self.assertTrue(isinstance(q, WebElement))
 
  def testFindElementById(self):
-    navigate(SEARCH)
+    self.navigate(SEARCH)
     # Find the padding for the logo near the search bar.
     elem = self.driver.find_element_by_id("logocont")
     self.assertTrue(isinstance(elem, WebElement))
     # Look for an ID not there.
     self.assertRaises(NoSuchElementException,
                       self.driver.find_element_by_id, "logocont")
 
   def testFindElementById0WithTimeout(self):
     self.set_implicit_wait(0)
-    navigate(SEARCH)
+    self.navigate(SEARCH)
     # Find the padding for the logo near the search bar.
     elem = self.driver.find_element_by_id("logocont")
     self.assertTrue(isinstance(elem, WebElement))
     self.set_implicit_wait(5000)
-    navigate(SEARCH)
+    self.navigate(SEARCH)
     # Look for an ID not there.
     self.assertRaises(NoSuchElementException,
                       self.driver.find_element_by_id, "logocont")
+
+
+class testCookies(RemoteWebDriverTest):
+  def testAddCookie(self):
+    self.navigate(SEARCH)
+    cookie_dict = {}
+    cookie_dict["name"]= "chromedriver_cookie_test";
+    cookie_dict["value"] = "this is a test";
+    self.driver.add_cookie(cookie_dict)
+    cookie_dict = None
+    cookie_dict = self.driver.get_cookie("chromedriver_cookie_test")
+    self.assertNotEqual(cookie_dict, None)
+    self.assertEqual(cookie_dict["value"], "this is a test";);
+
+  def testDeleteCookie(self):
+    testAddCookie();
+    self.driver.delete_cookie("chromedriver_cookie_test")
+    cookie_dict = self.driver.get_cookie("chromedriver_cookie_test")
+    self.assertEqual(cookie_dict, None)
+
+class TestFindElement(RemoteWebDriverTest):
+  def testFindByName(self):
+    self.navigate(SEARCH)
+    # Find the Google search button.
+    q = self.driver.find_element_by_name("q")
+    self.assertTrue(isinstance(q, WebElement))
+    # Trying looking for an element not on the page.
+    self.assertRaises(NoSuchElementException,
+                      self.driver.find_elment_by_name, "q2")
+    # Try to find the Google search button using the multiple find method.
+    q = self.driver.find_elements_by_name("q")
+    self.assertTrue(isinstance(q, list))
+    self.assertTrue(len(q), 1)
+    self.assertTrue(isinstance(q[0], WebElement))
+    # Try finding something not on page, with multiple find an empty array
+    # should return and no exception thrown.
+    q = self.driver.find_elements_by_name("q2")
+    assertTrue(q == [])
+    # Find a hidden element on the page
+    q = self.driver.find_element_by_name("oq")
+    self.assertTrue(isinstance(q, WebElement))
+
+ def testFindElementById(self):
+    self.navigate(SEARCH)
+    # Find the padding for the logo near the search bar.
+    elem = self.driver.find_element_by_id("logocont")
+    self.assertTrue(isinstance(elem, WebElement))
+    # Look for an ID not there.
+    self.assertRaises(NoSuchElementException,
+                      self.driver.find_element_by_id, "logocont")
 
 
 class TestJavaScriptExecution(RemoteWebDriverTest):
   """ Test the execute javascript ability of the remote driver"""
   def testNoModification(self):
     self.driver.get("http://www.google.com")
     title = self.driver.execute_script("return document.title")
     self.assertEqual(title, self.driver.get_title())
 
   def testModification(self):
@@ skipping 108 matching lines @@
     if options.port:
       WEBDRIVER_PORT = options.port
     if options.exe:
       WEBDRIVER_EXE = options.exe
     if not os.path.exists(WEBDRIVER_EXE):
       parser.error('WebDriver server executable not found:\n\t%s\n'
                    'Please specify a valid path with the --exe flag.'
                    % WEBDRIVER_EXE)
 
   unittest.main()