#!/usr/bin/env python
"""Test Selenium tests JS errors detection logic."""

from absl import app

from grr_response_server.gui import gui_test_lib
from grr.test_lib import test_lib


class JavascriptErrorTest(gui_test_lib.GRRSeleniumTest):
  """Tests that Javascript errors are caught in Selenium tests."""

  def testJavascriptErrorTriggersPythonException(self):
    self.Open("/legacy")

    # Erase global Angular object.
    # Things are guaranteed to stop working correctly after this.
    self.GetJavaScriptValue("window.angular = undefined;")

    with self.assertRaisesRegex(self.failureException,
                                "Javascript error encountered"):
      # Note that self.Click(), can, on rare occasions, trigger a
      # Javascript error, because of tickers that are running on the page.
      self.Click("client_query_submit")
      self.WaitUntil(self.IsElementPresent, "css=grr-clients-list")

    # The page has some tickers running that also use Angular so there
    # is a race that they can cause more js errors after the test has
    # already finished. By navigating to the main page, we make sure
    # the Angular object is valid again which means no more errors and
    # also clear the list of recorded errors in case there have been
    # any in the meantime.
    self.Open("/legacy")


if __name__ == "__main__":
  app.run(test_lib.main)
