Port perf.py's _LoginToGoogleAccount to chrome_remote_control

tools/chrome_remote_control/chrome_remote_control/credentials_backend_unittest.py

Index: tools/chrome_remote_control/chrome_remote_control/credentials_backend_unittest.py
diff --git a/tools/chrome_remote_control/chrome_remote_control/credentials_backend_unittest.py b/tools/chrome_remote_control/chrome_remote_control/credentials_backend_unittest.py
new file mode 100644
index 0000000000000000000000000000000000000000..b975d097498b1b9f735dff756fda91505d11ddc1
--- /dev/null
+++ b/tools/chrome_remote_control/chrome_remote_control/credentials_backend_unittest.py
@@ -0,0 +1,159 @@
+# Copyright (c) 2012 The Chromium Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+import unittest
+import tempfile
+import os
+
+from chrome_remote_control import credentials_backend
+from chrome_remote_control import browser_finder
+from chrome_remote_control import browser_options
+from chrome_remote_control import util
+
+simple_config_string = """
+{
+  "google_username": "example",
+  "google_password": "asdf",
+
+  "google_account_url": null,
+
+
+  "othersite_username": "test",
+  "othersite_password": "1234"
+}
+"""
+simple_config_dict = {'google_username':'example', 'google_password':'asdf',
+                      'google_account_url': None, 'othersite_username':'test',
+                      'othersite_password':'1234'}
+
+desired_config = {'google_username':None, 'othersite_password':None}
+result_config = {'google_username':'example', 'othersite_password':'1234'}
+
+google_credentials_file = os.path.join(os.path.dirname(__file__),
+                                       'credentials.json')
+
+google_credentials = {'google_username':None, 'google_password':None,
+                      'google_account_url': None}
+
+def getPossibleBrowser():
+  assert browser_options.options_for_unittests
+  options = (
+    browser_options.options_for_unittests.Copy())
+  temp_parser = options.CreateParser()
+  defaults = temp_parser.get_default_values()
+  for k, v in defaults.__dict__.items():
+    if hasattr(options, k):
+      continue
+    setattr(options, k, v)
+
+  possible_browser = browser_finder.FindBrowser(options)
+
+  return possible_browser
+
+def waitForExist(element_id, tab):
+  def exists():
+    return tab.runtime.Evaluate(
+        'document.querySelector("#%s")!== null' % element_id)
+
+  # Wait until the form is submitted and the page completes loading.
+  util.WaitFor(lambda: exists(), 60) # pylint: disable=W0108
+
+class TestCredentialsBackend(unittest.TestCase):
+  def testSetCredentialsConfigFile(self):
+    with tempfile.NamedTemporaryFile() as f:
+      f.write(simple_config_string)
+      f.flush()
+
+      credentials_backend.CredentialsBackend.ResetConfig()
+      credentials_backend.CredentialsBackend.SetCredentialsConfigFile(f.name)
+      config = credentials_backend.CredentialsBackend.GetConfig(desired_config)
+
+      self.assertEqual(config, result_config)
+
+  def testGetConfig(self):
+    with tempfile.NamedTemporaryFile() as f:
+      f.write(simple_config_string)
+      f.flush()
+
+      credentials_backend.CredentialsBackend.SetCredentialsConfigFile(f.name)
+      credentials_backend.CredentialsBackend.ResetConfig()
+      config = credentials_backend.CredentialsBackend.GetConfig(desired_config)
+
+      self.assertEqual(config, result_config)
+
+  def testSubmitForm(self):
+    possible_browser = getPossibleBrowser()
+    self.assertIsNotNone(possible_browser)
+
+    credentials_backend.CredentialsBackend.SetCredentialsConfigFile(
+        google_credentials_file)
+    credentials_backend.CredentialsBackend.ResetConfig()
+    try:
+      config = credentials_backend.CredentialsBackend.GetConfig(
+          google_credentials)
+    except IOError:
+      self.skipTest('Google Credentials File Not Found: "%s"' %
+                    google_credentials_file)
+
+    with possible_browser.Create() as browser:
+      with browser.ConnectToNthTab(0) as tab:
+        tab.page.Navigate('http://accounts.google.com')
+        form_id = 'gaia_loginform'
+        waitForExist(form_id, tab)
+        tab.WaitForDocumentReadyStateToBeInteractiveOrBetter()
+
+        username = config['google_username']
+        password = config['google_password']
+
+        email_id = 'document.getElementById("Email").value = "%s"; ' % username
+        password = 'document.getElementById("Passwd").value = "%s"; ' % password
+        tab.runtime.Execute(email_id)
+        tab.runtime.Execute(password)
+
+        credentials_backend.CredentialsBackend.SubmitForm(form_id, tab)
+
+        waitForExist('nav-privacy', tab)
+
+  def testWaitForFormToLoad(self):
+    possible_browser = getPossibleBrowser()
+    self.assertIsNotNone(possible_browser)
+
+    with possible_browser.Create() as browser:
+      with browser.ConnectToNthTab(0) as tab:
+        tab.page.Navigate('http://accounts.google.com')
+        form_id = 'gaia_loginform'
+        credentials_backend.CredentialsBackend.WaitForFormToLoad(form_id, tab)
+        waitForExist(form_id, tab)
+
+  def testSubmitFormAndWait(self):
+    possible_browser = getPossibleBrowser()
+    self.assertIsNotNone(possible_browser)
+
+    credentials_backend.CredentialsBackend.SetCredentialsConfigFile(
+        google_credentials_file)
+    credentials_backend.CredentialsBackend.ResetConfig()
+    try:
+      config = credentials_backend.CredentialsBackend.GetConfig(
+          google_credentials)
+    except IOError:
+      self.skipTest('Google Credentials File Not Found: "%s"' %
+                    google_credentials_file)
+
+    with possible_browser.Create() as browser:
+      with browser.ConnectToNthTab(0) as tab:
+        tab.page.Navigate('http://accounts.google.com')
+        form_id = 'gaia_loginform'
+        waitForExist(form_id, tab)
+        tab.WaitForDocumentReadyStateToBeInteractiveOrBetter()
+
+        username = config['google_username']
+        password = config['google_password']
+
+        email_id = 'document.getElementById("Email").value = "%s"; ' % username
+        password = 'document.getElementById("Passwd").value = "%s"; ' % password
+        tab.runtime.Execute(email_id)
+        tab.runtime.Execute(password)
+
+        credentials_backend.CredentialsBackend.SubmitFormAndWait(form_id, tab)
+
+        waitForExist('nav-privacy', tab)