src/cobalt/black_box_tests/tests/persistent_cookie.py

"""Open Cobalt in preload mode and find a basic element."""

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import _env  # pylint: disable=unused-import

from cobalt.black_box_tests import black_box_tests
from cobalt.black_box_tests.threaded_web_server import ThreadedWebServer
from cobalt.tools.automated_testing import webdriver_utils

keys = webdriver_utils.import_selenium_module('webdriver.common.keys')

_MAX_RESUME_WAIT_SECONDS = 30


class PersistentCookieTest(black_box_tests.BlackBoxTestCase):

  # The same page has to be used since cookie are stored per URL.

  def test_simple(self):

    with ThreadedWebServer() as server:
      url = server.GetURL(file_name='testdata/persistent_cookie.html')

      # The webpage listens for NUMPAD1, NUMPAD2 and NUMPAD3 at opening.
      with self.CreateCobaltRunner(url=url) as runner:
        # Press NUMPAD1 to verify basic cookie functionality and set
        # a persistent cookie.
        runner.WaitForJSTestsSetup()
        runner.SendKeys(keys.Keys.NUMPAD1)
        self.assertTrue(runner.JSTestsSucceeded())

      with self.CreateCobaltRunner(url=url) as runner:
        runner.WaitForJSTestsSetup()
        # Press NUMPAD2 to indicate this is the second time we opened
        # the webpage and verify a persistent cookie is on device. Then
        # clear this persistent cookie.
        runner.SendKeys(keys.Keys.NUMPAD2)
        self.assertTrue(runner.JSTestsSucceeded())

      with self.CreateCobaltRunner(url=url) as runner:
        runner.WaitForJSTestsSetup()
        # Press NUMPAD3 to verify the persistent cookie we cleared is
        # not on the device for this URL any more.
        runner.SendKeys(keys.Keys.NUMPAD3)
        self.assertTrue(runner.JSTestsSucceeded())