| OLD | NEW |
| 1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
| 2 # Copyright (c) 2012 The Chromium Authors. All rights reserved. | 2 # Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 3 # Use of this source code is governed by a BSD-style license that can be | 3 # Use of this source code is governed by a BSD-style license that can be |
| 4 # found in the LICENSE file. | 4 # found in the LICENSE file. |
| 5 | 5 |
| 6 import os | 6 import os |
| 7 import logging | 7 import logging |
| 8 | 8 |
| 9 import pyauto_functional # Must be imported before pyauto | 9 import pyauto_functional # Must be imported before pyauto |
| 10 import pyauto | 10 import pyauto |
| (...skipping 125 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 136 self.assertEquals({u'cookies': 2}, | 136 self.assertEquals({u'cookies': 2}, |
| 137 self.GetPrefsInfo().Prefs(pyauto.kDefaultContentSettings), | 137 self.GetPrefsInfo().Prefs(pyauto.kDefaultContentSettings), |
| 138 msg='Cookie setting did not persist after restarting session.') | 138 msg='Cookie setting did not persist after restarting session.') |
| 139 self.assertFalse(self.GetCookie(pyauto.GURL(file_url)), | 139 self.assertFalse(self.GetCookie(pyauto.GURL(file_url)), |
| 140 msg='Cookies are not blocked.') | 140 msg='Cookies are not blocked.') |
| 141 self.assertFalse(self.GetCookie(pyauto.GURL(http_url)), | 141 self.assertFalse(self.GetCookie(pyauto.GURL(http_url)), |
| 142 msg='Cookies are not blocked.') | 142 msg='Cookies are not blocked.') |
| 143 self.assertFalse(self.GetCookie(pyauto.GURL(https_url)), | 143 self.assertFalse(self.GetCookie(pyauto.GURL(https_url)), |
| 144 msg='Cookies are not blocked.') | 144 msg='Cookies are not blocked.') |
| 145 | 145 |
| 146 def testClearCookiesOnEndingSession(self): | |
| 147 """Verify that cookies are cleared when the browsing session is closed.""" | |
| 148 | |
| 149 # This test fails on ChromeOS because kRestoreOnStartup is ignored and | |
| 150 # the startup preference is always "continue where I left off." | |
| 151 if self.IsChromeOS(): | |
| 152 logging.info('Skipping testClearCookiesOnEndingSession on ChromeOS') | |
| 153 return | |
| 154 | |
| 155 file_url = self.GetFileURLForDataPath('setcookie.html') | |
| 156 self.assertFalse(self.GetCookie(pyauto.GURL(file_url)), | |
| 157 msg='There should be no cookies on %s' % file_url) | |
| 158 | |
| 159 # Set the option to clear cookies when the browser session is closed. | |
| 160 self.SetPrefs(pyauto.kClearSiteDataOnExit, True) | |
| 161 | |
| 162 self.NavigateToURL(file_url) | |
| 163 self.assertEqual('name=Good', self.GetCookie(pyauto.GURL(file_url)), | |
| 164 msg='Unable to retrieve the cookie name=Good') | |
| 165 | |
| 166 # Restart and verify that cookie does not persist | |
| 167 self.RestartBrowser(clear_profile=False) | |
| 168 self.assertFalse(self.GetCookie(pyauto.GURL(file_url)), | |
| 169 msg='Cookie persisted after restarting session.') | |
| 170 | |
| 171 def testAllowCookiesUsingExceptions(self): | 146 def testAllowCookiesUsingExceptions(self): |
| 172 """Verify that cookies can be allowed and set using exceptions for | 147 """Verify that cookies can be allowed and set using exceptions for |
| 173 particular website(s) when all others are blocked.""" | 148 particular website(s) when all others are blocked.""" |
| 174 http_url = 'http://%s' % self.test_host | 149 http_url = 'http://%s' % self.test_host |
| 175 self.assertFalse(self.GetCookie(pyauto.GURL(http_url)), | 150 self.assertFalse(self.GetCookie(pyauto.GURL(http_url)), |
| 176 msg='There should be no cookies on %s' % http_url) | 151 msg='There should be no cookies on %s' % http_url) |
| 177 | 152 |
| 178 # Set the preference to block all cookies. | 153 # Set the preference to block all cookies. |
| 179 self.SetPrefs(pyauto.kDefaultContentSettings, {u'cookies': 2}) | 154 self.SetPrefs(pyauto.kDefaultContentSettings, {u'cookies': 2}) |
| 180 | 155 |
| (...skipping 60 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 241 # Restart the browser to check that the cookie doesn't persist. | 216 # Restart the browser to check that the cookie doesn't persist. |
| 242 # (This fails on ChromeOS because kRestoreOnStartup is ignored and | 217 # (This fails on ChromeOS because kRestoreOnStartup is ignored and |
| 243 # the startup preference is always "continue where I left off.") | 218 # the startup preference is always "continue where I left off.") |
| 244 if not self.IsChromeOS(): | 219 if not self.IsChromeOS(): |
| 245 self.RestartBrowser(clear_profile=False) | 220 self.RestartBrowser(clear_profile=False) |
| 246 self.assertFalse(self.GetCookie(pyauto.GURL(http_url)), | 221 self.assertFalse(self.GetCookie(pyauto.GURL(http_url)), |
| 247 msg='Cookie persisted after restarting session.') | 222 msg='Cookie persisted after restarting session.') |
| 248 | 223 |
| 249 if __name__ == '__main__': | 224 if __name__ == '__main__': |
| 250 pyauto_functional.Main() | 225 pyauto_functional.Main() |
| OLD | NEW |