class CookieTest(unittest.TestCase): """Cookie test for the json webdriver protocol""" def setUp(self): self._launcher = ChromeDriverLauncher(root_path=os.path.dirname(__file__)) self._driver = WebDriver(self._launcher.GetURL(), DesiredCapabilities.CHROME) def tearDown(self): self._driver.quit() self._launcher.Kill() def testAddCookie(self): self._driver.get(self._launcher.GetURL() + '/test_page.html') cookie_dict = None cookie_dict = self._driver.get_cookie("chromedriver_cookie_test") cookie_dict = {} cookie_dict["name"]= "chromedriver_cookie_test" cookie_dict["value"] = "this is a test" self._driver.add_cookie(cookie_dict) 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): self.testAddCookie(); self._driver.delete_cookie("chromedriver_cookie_test") cookie_dict = self._driver.get_cookie("chromedriver_cookie_test") self.assertEqual(cookie_dict, None)