Adding WebDriver cookie tests

Signed-off-by: AutomatedTester <dburns@mozilla.com>
web-platform-tests · Jan 22, 2014 · 9fe5000 · 9fe5000
1 parent 66f434f
commit 9fe5000
Show file tree

Hide file tree

Showing 2 changed files with 76 additions and 0 deletions.
diff --git a/webdriver/cookie/cookie_container.html b/webdriver/cookie/cookie_container.html
@@ -0,0 +1,5 @@
+<!DOCTYPE html>
+<meta charset="utf-8">
+<title>Cookie Container</title>
+<body>
+</body>
diff --git a/webdriver/cookie/cookie_test.py b/webdriver/cookie/cookie_test.py
@@ -0,0 +1,71 @@
+# -*- mode: python; fill-column: 100; comment-column: 100; -*-
+
+import os
+import sys
+import unittest
+from selenium.common.exceptions import UnableToSetCookieException
+from selenium.common.exceptions import InvalidCookieDomainException
+
+sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), os.path.pardir)))
+
+import base_test
+
+class CookieTest( base_test.WebDriverBaseTest ):
+
+    def test_can_create_a_well_formed_cookie( self ):
+        name = 'foo'
+        value = 'bar'
+
+        self.driver.add_cookie({ 'name': name, 'value': value })
+
+    def test_cookies_should_allow_secure_to_be_set( self ):
+        name = 'foo'
+        value = 'bar'
+        secure = True
+
+        self.driver.add_cookie({ 'name': name,
+                                 'value': value,
+                                 'path': '/',
+                                 'secure': secure})
+        self.assertTrue(self.driver.get_cookie(name)['secure'])
+
+    def test_secure_defaults_to_false( self ):
+        name = 'foo'
+        value = 'bar'
+
+        self.driver.add_cookie({ 'name': name,
+                                 'value': value})
+
+        self.assertFalse( self.driver.get_cookie( name )[ 'secure' ] )
+
+    def test_should_throw_an_exception_when_semicolon_exists_in_the_cookie_attribute(self):
+        invalid_name = 'foo;bar'
+        value = 'foobar'
+
+        self.driver.get( self.webserver.where_is( "cookie/cookie_container.html" ))
+
+        try:
+            self.driver.add_cookie({ 'name': invalid_name, 'value': value })
+            self.fail( 'should have thrown exceptions.' )
+
+        except UnableToSetCookieException:
+            pass
+        except InvalidCookieDomainException:
+            pass
+
+    def test_should_throw_an_exception_the_name_is_null(self):
+        VAL = 'foobar'
+
+        self.driver.get( self.webserver.where_is( "cookie/cookie_container.html" ))
+
+        try:
+            self.driver.add_cookie({ 'name': None, 'value': VAL })
+            self.fail( 'should have thrown exceptions.' )
+
+        except UnableToSetCookieException:
+            pass
+        except InvalidCookieDomainException:
+            pass
+
+if __name__ == '__main__':
+    unittest.main()