Adding pyauto tests for the NaCl SDK.

functional/nacl_sdk.py

+#!/usr/bin/python
+# Copyright (c) 2011 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 copy
+import ctypes
+from distutils import version
+import fnmatch
+import glob
+import hashlib
+import logging
+import os
+import platform
+import shutil
+import subprocess
+import sys
+import tarfile
+import time
+import urllib2
+
+import pyauto_functional  # Must be imported before pyauto.
+import pyauto
+
+class NaClSDKTest(pyauto.PyUITest):
+  """Tests for the NaCl SDK."""
+  _download_dir = os.path.join(pyauto.PyUITest.DataDir(), 'downloads')
+  _extracted_sdk_path = os.path.join(_download_dir, 'extracted_nacl_sdk')
+  
+  def setUp(self):
+    pyauto.PyUITest.setUp(self)
+    self._RemoveDownloadedTestFile()
+    
+  def testNaClSDK(self):
+    """Verify that NaCl SDK is working properly."""
+    if self._HasAllSystemRequirements() == False:

[dennis_jeffrey, 2011/08/09 17:14:21]

if not self._HasAllSystemRequirements():

[chrisphan, 2011/08/09 17:42:57]

Done.

+      logging.info('System does not meet the requirements.')
+      return
+    
+    self._OpenExamplesAndStartTest(
+        self._GetTestSetting()['gallery_examples'])
+    self._VerifyDownloadLinks()
+    self._VerifyNaClSDKInstaller()
+    self._VerifyBuildStubProject()
+    self._LaunchServerAndVerifyExamples()
+    self._VerifyRebuildExamples()
+    self._VerifySelldrAndNcval()
+    
+  def testVerifyNaClSDKChecksum(self):
+    """Verify NaCl SDK Checksum."""
+    if self._HasAllSystemRequirements() == False:
+      logging.info('System does not meet the requirements.')
+      return
+    
+    settings = self._GetTestSetting()
+    
+    self._DownloadNaClSDK()
+    
+    if pyauto.PyUITest.IsWin():
+      expected_shasum = settings['release_win_expected_shasum']
+      file_path = os.path.join(self._download_dir, 'naclsdk_win.exe')
+    elif pyauto.PyUITest.IsMac():
+      expected_shasum = settings['release_mac_expected_shasum']
+      file_path = os.path.join(self._download_dir, 'naclsdk_mac.tgz')
+    elif pyauto.PyUITest.IsLinux():
+      expected_shasum = settings['release_lin_expected_shasum']
+      file_path = os.path.join(self._download_dir, 'naclsdk_linux.tgz')
+    else:
+      self.fail(msg='NaCl SDK does not support this OS.')
+      
+    sha = hashlib.sha1()
+    try:
+      f = open(file_path, 'rb')
+      sha.update(f.read())
+      shasum = sha.hexdigest()
+      self.assertEqual(expected_shasum, shasum,
+                       msg='Unexpected checksum. Expected: %s, got: %s'
+                       % (expected_shasum, shasum))
+    except IOError:
+      self.fail(msg='Cannot open %s.' % file_path)
+    finally:
+      f.close()
+  
+  def testVerifyPrereleaseGallery(self):
+    """Verify Pre-release gallery examples."""
+    if self._HasAllSystemRequirements() == False:
+      logging.info('System does not meet the requirements.')
+      return
+    self._OpenExamplesAndStartTest(
+        self._GetTestSetting()['prerelease_gallery'])
+    
+  def _VerifyDownloadLinks(self):
+    """Verify the download links."""
+    settings = self._GetTestSetting()
+    self.NavigateToURL(settings['post_sdk_download_url'])
+    html = self.GetTabContents()
+
+    # Make sure the correct URL is under the correct label.
+    if pyauto.PyUITest.IsWin():
+      win_sdk_url = settings['post_win_sdk_url']
+      win_url_index = html.find(win_sdk_url)
+      self.assertTrue(win_url_index > -1,
+                      msg='Missing SDK download URL: %s' % win_sdk_url)
+      win_keyword_index = html.rfind('Windows', 0, win_url_index)
+      self.assertTrue(win_keyword_index > -1,
+                      msg='Misplaced download link: %s' % win_sdk_url)
+    elif pyauto.PyUITest.IsMac():
+      mac_sdk_url = settings['post_mac_sdk_url']
+      mac_url_index = html.find(mac_sdk_url)
+      self.assertTrue(mac_url_index > -1,
+                      msg='Missing SDK download URL: %s' % mac_sdk_url)
+      mac_keyword_index = html.rfind('Macintosh', 0, mac_url_index)
+      self.assertTrue(mac_keyword_index > -1,
+                      msg='Misplaced download link: %s' % mac_sdk_url)
+    elif pyauto.PyUITest.IsLinux():
+      lin_sdk_url = settings['post_lin_sdk_url']
+      lin_url_index = html.find(lin_sdk_url)
+      self.assertTrue(lin_url_index > -1,
+                      msg='Missing SDK download URL: %s' % lin_sdk_url)
+      lin_keyword_index = html.rfind('Linux', 0, lin_url_index)
+      self.assertTrue(lin_keyword_index > -1,
+                      msg='Misplaced download link: %s' % lin_sdk_url)
+    else:
+      self.fail(msg='NaCl SDK does not support this OS.')
+    
+  def _VerifyNaClSDKInstaller(self):
+    """Verify NaCl SDK installer."""
+    search_list = [
+        'build.scons',
+        'favicon.ico',
+        'geturl/',
+        'hello_world/',
+        'hello_world_c/',
+        'httpd.py',
+        'index.html',
+        'nacl_sdk_scons/',
+        'pi_generator/',
+        'scons',
+        'sine_synth/'
+    ]
+    
+    mac_lin_additional_search_items = [
+      'sel_ldr_x86_32',
+      'sel_ldr_x86_64',
+      'ncval_x86_32',
+      'ncval_x86_64'
+    ]
+
+    win_additional_search_items = [
+        'httpd.cmd',
+        'sel_ldr_x86_32.exe',
+        'sel_ldr_x86_64.exe',
+        'ncval_x86_32.exe',
+        'ncval_x86_64.exe'
+    ]
+    
+    self._DownloadNaClSDK()
+    
+    if pyauto.PyUITest.IsWin():
+      source_file = os.path.join(self._download_dir, 'naclsdk_win.exe')
+      self._SearchNaClSDKFileWindows(
+          search_list + win_additional_search_items, source_file)
+    elif pyauto.PyUITest.IsMac():
+      source_file = os.path.join(self._download_dir, 'naclsdk_mac.tgz')
+      self._SearchNaClSDKTarFile(search_list + mac_lin_additional_search_items,
+                                 source_file)
+    elif pyauto.PyUITest.IsLinux():
+      source_file = os.path.join(self._download_dir, 'naclsdk_linux.tgz')
+      self._SearchNaClSDKTarFile(search_list + mac_lin_additional_search_items,
+                                 source_file)
+    else:
+      self.fail(msg='NaCl SDK does not support this OS.')
+    
+    self._ExtractNaClSDK()
+      
+  def _VerifyBuildStubProject(self):
+    """Build stub project."""
+    stub_project_files = [
+        'build.scons',
+        'scons'
+    ]
+    project_template_path = self._GetDirectoryPath('project_templates', 
+                                                   self._extracted_sdk_path)
+    examples_path = self._GetDirectoryPath('examples', 
+                                           self._extracted_sdk_path)
+    init_project_path = os.path.join(project_template_path, 'init_project.py')
+    
+    # Build a C project.
+    self._BuildStubProject(init_project_path, 'hello_c', examples_path,
+                           stub_project_files)
+    
+    # Build a C++ project.
+    self._BuildStubProject(init_project_path, 'hello_cc', examples_path,
+                           stub_project_files)
+    
+  def _BuildStubProject(self, init_project_path, name, examples_path,
+                        stub_project_files):
+    """Build stub project."""
+    proc = subprocess.Popen(
+        ['python', init_project_path, '-n', name, '-c', '-d',
+        examples_path], stdout=subprocess.PIPE)
+    proc.communicate()
+    
+    hello_c_path = os.path.join(examples_path, name)
+    for file in stub_project_files:
+        self.assertTrue(self._HasFile(file, hello_c_path),
+                        msg='Cannot build %s stub project.' % name)
+    
+  def _LaunchServerAndVerifyExamples(self):
+    """Start local HTTP server and verify examples."""
+    # Make sure server is not open.
+    if self._IsURLAlive('http://localhost:5103'):
+      self._CloseHTTPServer()
+    
+    # Start HTTP server.
+    examples_path = self._GetDirectoryPath('examples',
+                                           self._extracted_sdk_path)
+    if pyauto.PyUITest.IsWin():
+      http_path = os.path.join(examples_path, 'httpd.cmd')
+      proc = subprocess.Popen([http_path], cwd=examples_path)
+    else:
+      http_path = os.path.join(examples_path, 'httpd.py')
+      proc = subprocess.Popen(['python', http_path], cwd=examples_path)
+    
+    sucess = self.WaitUntil(

[dennis_jeffrey, 2011/08/09 17:14:21]

nit: 'sucess' --> 'success'

[chrisphan, 2011/08/09 17:42:57]

Done.

+        lambda: self._IsURLAlive('http://localhost:5103'),
+        timeout=5, retry_sleep=1, expect_retval=True)

[dennis_jeffrey, 2011/08/09 17:14:21]

Do you think a timeout of 5 seconds is enough?  If the local machine is slow,
perhaps it could take longer.  Would there be any problem if we used a timeout
of, say, 30 seconds?  It seems a little safer to me.

[chrisphan, 2011/08/09 17:42:57]

Done.

+    self.assertTrue(sucess,
+                    msg='Cannot open HTTP server. %s' %
+                    self.GetActiveTabTitle())
+    
+    examples = {
+        'hello_world_c': 'http://localhost:5103/hello_world_c/'
+        'hello_world.html',
+        'hello_world': 'http://localhost:5103/hello_world/hello_world.html',
+        'geturl': 'http://localhost:5103/geturl/geturl.html',
+        'pi_generator': 'http://localhost:5103/pi_generator/pi_generator.html',
+        'sine_synth': 'http://localhost:5103/sine_synth/sine_synth.html',
+    }
+    try:
+      self._OpenExamplesAndStartTest(examples)
+    finally:
+      self._CloseHTTPServer(proc)
+      
+  def _VerifyRebuildExamples(self):
+    """Re-build the examples and verify they are as expected."""
+    examples_path = self._GetDirectoryPath('examples',
+                                           self._extracted_sdk_path)
+    scons_path = os.path.join(examples_path, 'scons -c')
+
+    example_dirs = [
+        'geturl',
+        'hello_world',
+        'hello_world_c',
+        'pi_generator',
+        'sine_synth'
+    ]
+
+    proc = subprocess.Popen([scons_path], cwd=examples_path, shell=True)
+    proc.communicate()
+    for x in example_dirs:
+      ex_path = os.path.join(examples_path, x)
+      if self._HasFile('*.nmf', ex_path):
+        self.fail(msg='Failed running scons -c.')
+    
+    scons_path = os.path.join(examples_path, 'scons')
+    proc = subprocess.Popen([scons_path], cwd=examples_path,
+                            stdout=subprocess.PIPE, shell=True)
+    proc.communicate()
+    
+    # Verify each example directory contains .nmf file.
+    for dir in example_dirs:
+      dir = os.path.join(examples_path, dir)
+      if not self._HasFile('*.nmf', dir):
+        self.fail(msg='Failed running scons.')
+    
+    self._LaunchServerAndVerifyExamples()
+    
+  def _VerifySelldrAndNcval(self):
+    """Verify sel_ldr and ncval."""
+    architecture = self._GetPlatformArchitecture()
+    scons_arg = None
+    if pyauto.PyUITest.IsWin():
+      if architecture == '64bit':
+        scons_arg = 'test64'
+      else:
+        scons_arg = 'test32'
+    elif pyauto.PyUITest.IsMac():
+      scons_arg = 'test64'
+    elif pyauto.PyUITest.IsLinux():
+      scons_arg = 'test64'
+      
+    examples_path = self._GetDirectoryPath('examples',
+                                           self._extracted_sdk_path)
+    scons_path = os.path.join(examples_path, 'scons ' + scons_arg)
+
+    # Build and run the unit test.
+    proc = subprocess.Popen([scons_path], stdout=subprocess.PIPE,
+                            shell=True, cwd=examples_path)
+    stdout = proc.communicate()[0]
+    lines = stdout.splitlines()
+    test_ran = False
+    for line in lines:
+      if 'Check:' in line:
+        self.assertTrue('passed' in line,
+                        msg='Nacl-sel_ldr unit test failed.')
+        test_ran = True
+    self.assertTrue(test_ran,
+                    msg='Failed to build and run nacl-sel_ldr unit test.')
+
+    if architecture == '64bit':
+      if not self._HasPathInTree('hello_world_x86_64.nexe',
+                                 True, root=examples_path):
+        self.fail(msg='Missing file: hello_world_x86_64.nexe.')
+    else:
+      if not self._HasPathInTree('hello_world_x86_32.nexe',
+                                 True, root=examples_path):
+        self.fail(msg='Missing file: hello_world_x86_32.nexe.')
+    
+    # Verify that a mismatch of sel_ldr and .nexe produces an error.
+    toolchain_path = self._GetDirectoryPath('toolchain',
+                                            self._extracted_sdk_path)
+    bin_path = self._GetDirectoryPath('bin', toolchain_path)
+    hello_world_path = self._GetDirectoryPath('hello_world', examples_path)
+    sel_32_path = os.path.join(bin_path, 'sel_ldr_x86_32')
+    sel_64_path = os.path.join(bin_path, 'sel_ldr_x86_64')
+    nexe_32_path = os.path.join(hello_world_path, 'hello_world_x86_32.nexe')
+    nexe_64_path = os.path.join(hello_world_path, 'hello_world_x86_64.nexe')
+    
+    if architecture == '64bit':
+      success = self._RunProcessAndCheckOutput(
+          [sel_64_path, nexe_32_path], 'Error while loading')
+    else:
+      success = self._RunProcessAndCheckOutput(
+          [sel_32_path, nexe_64_path], 'Error while loading')
+    self.assertTrue(success, 
+                    msg='Failed to verify sel_ldr and .nexe mismatch.')
+    
+    # Run the appropriate ncval for the platform on the matching .nexe.
+    ncval_32_path = os.path.join(bin_path, 'ncval_x86_32')
+    ncval_64_path = os.path.join(bin_path, 'ncval_x86_64')
+
+    if architecture == '64bit':
+      success = self._RunProcessAndCheckOutput(
+          [ncval_64_path, nexe_64_path], 'is safe')
+    else:
+      success = self._RunProcessAndCheckOutput(
+          [ncval_32_path, nexe_32_path], 'is safe')
+    self.assertTrue(success, msg='Failed to verify ncval.')
+    
+    # Verify that a mismatch of ncval and .nexe produces an error.
+    if architecture == '64bit':
+      success = self._RunProcessAndCheckOutput(
+          [ncval_64_path, nexe_32_path], 'is safe', is_in=False)
+    else:
+      success = self._RunProcessAndCheckOutput(
+          [ncval_32_path, nexe_64_path], 'is safe', is_in=False)
+    self.assertTrue(success, msg='Failed to verify ncval and .nexe mismatch.')
+
+  def _RemoveDownloadedTestFile(self):
+    """Delete downloaded files and dirs from downloads directory."""
+    if os.path.exists(self._extracted_sdk_path):
+      try:
+        shutil.rmtree(self._extracted_sdk_path)
+      except:
+        self.fail(msg='Cannot remove %s' % self._extracted_sdk_path)
+
+    for sdk in ['naclsdk_win.exe', 'naclsdk_mac.tgz',
+                     'naclsdk_linux.tgz']:

[dennis_jeffrey, 2011/08/09 17:14:21]

nit: indent this line by 5 fewer spaces.

[chrisphan, 2011/08/09 17:42:57]

Done.

+      sdk_path = os.path.join(self._download_dir, sdk)
+      if os.path.exists(sdk_path):
+        try:
+          os.remove(sdk_path)
+        except:
+          self.fail(msg='Cannot remove %s' % sdk_path)
+                          
+  def _RunProcessAndCheckOutput(self, args, look_for, is_in=True):
+    """Run process and look for string in output.
+    
+    Args:
+      args: Argument strings to pass to subprocess.
+      look_for: The string to search in output.
+      is_in: True if checking if param look_for is in output.
+             False if checking if param look_for is not in output.
+      
+    Returns:
+      True, if output contains parameter |look_for| and |is_in| is True, or
+      False otherwise.
+    """
+    proc = subprocess.Popen(args, stdout=subprocess.PIPE,
+                            stderr=subprocess.PIPE)
+    (stdout, stderr) = proc.communicate()
+    lines = stdout.splitlines()
+    for line in lines:
+      if look_for in line:
+        return is_in
+    
+    lines = stderr.splitlines()
+    for line in lines:
+      if look_for in line:
+        return is_in
+    return not is_in
+
+  def _OpenExamplesAndStartTest(self, examples):
+    """Open each example and verify that it's working.
+    
+    Args:
+      examples: A dict of name to url of examples.
+    """
+    self._EnableNaClPlugin()
+
+    # Open all examples.
+    for name, url in examples.items():
+      self.AppendTab(pyauto.GURL(url))
+      self._CheckForCrashes()
+
+    # Verify all examples are working.
+    for name, url in examples.items():
+      self._VerifyAnExample(name, url)
+    self._CheckForCrashes()
+
+    # Reload all examples.
+    for _ in range(2):
+      for tab_index in range(self.GetTabCount()):
+        self.GetBrowserWindow(0).GetTab(tab_index).Reload()
+    self._CheckForCrashes()
+
+    # Verify all examples are working.
+    for name, url in examples.items():
+      self._VerifyAnExample(name, url)
+    self._CheckForCrashes()
+    
+    # Close each tab, check for crashes and verify all open
+    # examples operate correctly.
+    tab_count = self.GetTabCount()
+    for index in xrange(tab_count - 1, 0, -1):
+      self.GetBrowserWindow(0).GetTab(index).Close(True)
+      self._CheckForCrashes()
+      
+      tabs = self.GetBrowserInfo()['windows'][0]['tabs']
+      for tab in tabs:
+        if tab['index'] > 0:
+          for name, url in examples.items():
+            if url == tab['url']:
+              self._VerifyAnExample(name, url)
+              break
+      
+  def _VerifyAnExample(self, name, url):
+    """Verify NaCl example is working.
+    
+    Args:
+      name: A string name of the example.
+      url: A string url of the example.
+    """
+    available_example_tests = {
+      'hello_world_c': self._VerifyHelloWorldExample,
+      'hello_world': self._VerifyHelloWorldExample,
+      'pi_generator': self._VerifyPiGeneratorExample,
+      'sine_synth': self._VerifySineSynthExample,
+      'geturl': self._VerifyGetURLExample,
+      'life': self._VerifyConwaysLifeExample
+    }
+    
+    if not name in available_example_tests:
+      self.fail(msg='No test available for %s.' % name)
+    
+    info = self.GetBrowserInfo()
+    tabs = info['windows'][0]['tabs']
+    tab_index = None
+    for tab in tabs:
+      if url == tab['url']:
+        self.ActivateTab(tab['index'])
+        tab_index = tab['index']
+        break
+    
+    if tab_index != None:

[dennis_jeffrey, 2011/08/09 17:14:21]

I think this should work too:

if tab_index:

[chrisphan, 2011/08/09 17:42:57]

Done.

+      available_example_tests[name](tab_index, name, url)
+
+  def _VerifyHelloWorldExample(self, tab_index, name, url):
+    """Verify Hello World Example.
+
+    Args:
+      tab_index: Tab index integer that the example is on.
+      name: A string name of the example.
+      url: A string url of the example.
+    """
+    success = self.WaitUntil(
+        lambda: self.GetDOMValue(
+                    'document.getElementById("statusField").innerHTML',
+                    0, tab_index),
+        timeout=60, expect_retval='SUCCESS')
+    self.assertTrue(success, msg='Example %s failed. URL: %s' % (name, url))
+    
+    js_code = """
+      window.alert = function(e) {
+        window.domAutomationController.send(String(e));
+      }
+      window.domAutomationController.send("done");
+    """
+    self.ExecuteJavascript(js_code, 0, tab_index)
+
+    result = self.ExecuteJavascript('document.helloForm.elements[1].click();',
+                                    0, tab_index)
+    self.assertEqual(result, '42',
+                     msg='Example %s failed. URL: %s' % (name, url))
+    
+    result = self.ExecuteJavascript('document.helloForm.elements[2].click();',
+                                    0, tab_index)
+    self.assertEqual(result, 'dlrow olleH',
+                     msg='Example %s failed. URL: %s' % (name, url))
+    
+  def _VerifyPiGeneratorExample(self, tab_index, name, url):
+    """Verify Pi Generator Example.
+
+    Args:
+      tab_index: Tab index integer that the example is on.
+      name: A string name of the example.
+      url: A string url of the example.
+    """
+    success = self.WaitUntil(
+        lambda: self.GetDOMValue('document.form.pi.value', 0, tab_index)[0:3],
+        timeout=120, expect_retval='3.1')
+    self.assertTrue(success, msg='Example %s failed. URL: %s' % (name, url))
+    
+    # Get top corner of Pi image.
+    js_code = """
+      var obj = document.getElementById('piGenerator');
+      var curleft = curtop = 0;
+      do {
+          curleft += obj.offsetLeft;
+          curtop += obj.offsetTop;
+      } while (obj = obj.offsetParent);
+      window.domAutomationController.send(curleft + "," + curtop);
+    """
+    result = self.ExecuteJavascript(js_code, 0, tab_index)
+    result_split = result.split(",")
+    x = int(result_split[0])
+    y = int(result_split[1])
+    window = self.GetBrowserInfo()['windows'][0]
+    window_to_content_x = 2
+    window_to_content_y = 80
+    pi_image_x = x + window['x'] + window_to_content_x
+    pi_image_y = y + window['y'] + window_to_content_y
+    
+    if self._IsGetPixelSupported():
+      is_animating = self._IsColorChanging(pi_image_x, pi_image_y, 50, 50)
+      self.assertTrue(is_animating,
+                      msg='Example %s failed. URL: %s' % (name, url))
+    
+  def _VerifySineSynthExample(self, tab_index, name, url):
+    """Verify Sine Wave Synthesizer Example.
+
+    Args:
+      tab_index: Tab index integer that the example is on.
+      name: A string name of the example.
+      url: A string url of the example.
+    """
+    success = self.WaitUntil(
+        lambda: self.GetDOMValue(
+                    'document.getElementById("frequency_field").value',
+                    0, tab_index),
+        timeout=30, expect_retval='440')
+    self.assertTrue(success, msg='Example %s failed. URL: %s' % (name, url))
+    
+    self.ExecuteJavascript(
+        'document.body.getElementsByTagName("button")[0].click();'
+        'window.domAutomationController.send("done")',
+        0, tab_index)
+    
+    # TODO(chrisphan): Verify sound.
+    
+  def _VerifyGetURLExample(self, tab_index, name, url):
+    """Verify GetURL Example.
+
+    Args:
+      tab_index: Tab index integer that the example is on.
+      name: A string name of the example.
+      url: A string url of the example.
+    """
+    success = self.WaitUntil(
+        lambda: self.GetDOMValue(
+                    'document.getElementById("status_field").innerHTML',
+                    0, tab_index),
+        timeout=60, expect_retval='SUCCESS')
+    self.assertTrue(success, msg='Example %s failed. URL: %s' % (name, url))
+    
+    self.ExecuteJavascript(
+        'document.geturl_form.elements[0].click();'
+        'window.domAutomationController.send("done")',
+        0, tab_index)
+
+    js_code = """
+      var output = document.getElementById("general_output").innerHTML;
+      var result;
+      if (output.indexOf("test passed") != -1)
+        result = "pass";
+      else
+        result = "fail";
+      window.domAutomationController.send(result);
+    """
+    success = self.WaitUntil(
+        lambda: self.ExecuteJavascript(js_code, 0, tab_index),
+        timeout=30, expect_retval='pass')
+    self.assertTrue(success, msg='Example %s failed. URL: %s' % (name, url))
+    
+  def _VerifyConwaysLifeExample(self, tab_index, name, url):
+    """Verify Conway's Life Example.
+
+    Args:
+      tab_index: Tab index integer that the example is on.
+      name: A string name of the example.
+      url: A string url of the example.
+    """
+    window = self.GetBrowserInfo()['windows'][0]
+    window_to_content_x = 2
+    window_to_content_y = 80
+    x = window['x'] + window_to_content_x
+    y = window['y'] + window_to_content_y
+    offset_pixel = 100
+    if self._IsGetPixelSupported():
+      wait_for = 30

[dennis_jeffrey, 2011/08/09 17:14:21]

Maybe no need to define this as a separate variable.  Just inline the value
where needed at line 624 below.

[chrisphan, 2011/08/09 17:42:57]

Done.

+      success = self.WaitUntil(
+          lambda: self._GetPixel(x + offset_pixel, y + offset_pixel), 
+          wait_for, expect_retval=16777215)

[dennis_jeffrey, 2011/08/09 17:14:21]

'wait_for' --> 'timeout=30'

[chrisphan, 2011/08/09 17:42:57]

Done.

+      self.assertTrue(success, msg='Example %s failed. URL: %s' % (name, url))
+
+  def _IsColorChanging(self, x, y, width, height, tries=3, retry_sleep=2):
+    """Check screen for anything that is moving.
+
+    Args:
+      x: X coordinate on the screen.
+      y: Y coordinate on the screen.
+      width: Width of the area to scan.
+      height: Height of the area to scan.
+      tries: Number of tries.
+      retry_sleep: Sleep time in-between each try.
+      
+    Returns:
+      True, if pixel color in area is changing, or
+      False otherwise.
+    """
+    color_a = self._GetAreaPixelColor(x, y, width, height)
+    for _ in xrange(tries):
+      time.sleep(retry_sleep)
+      color_b = self._GetAreaPixelColor(x, y, width, height)
+      if color_a != color_b:
+        return True
+    return False
+  
+  def _IsGetPixelSupported(self):
+    """Check if get pixel is supported.
+      
+    Returns:
+      True, if get pixel is supported, or
+      False otherwise.
+    """
+    return pyauto.PyUITest.IsWin()
+ 
+  def _GetAreaPixelColor(self, x, y, width, height):
+    """Get an area of pixel color and return a list of color code values.
+
+    Args:
+      x: X coordinate on the screen.
+      y: Y coordinate on the screen.
+      width: Width of the area to scan.
+      height: Height of the area to scan.
+      
+    Returns:
+      A list containing color codes.
+    """
+    if pyauto.PyUITest.IsMac():
+      pass  # TODO(chrisphan): Do Mac.
+    elif pyauto.PyUITest.IsWin():
+      return self._GetAreaPixelColorWin(x, y, width, height)
+    elif pyauto.PyUITest.IsLinux():
+      pass  # TODO(chrisphan): Do Linux.
+    return None
+  
+  def _GetAreaPixelColorWin(self, x, y, width, height):
+    """Get an area of pixel color for Windows and return a list.
+
+    Args:
+      x: X coordinate on the screen.
+      y: Y coordinate on the screen.
+      width: Width of the area to scan.
+      height: Height of the area to scan.
+      
+    Returns:
+      A list containing color codes.
+    """
+    colors = []
+    hdc = ctypes.windll.user32.GetDC(0)
+    for x_pos in xrange(x, x + width, 1):
+      for y_pos in xrange(y, y + height, 1):
+        color = ctypes.windll.gdi32.GetPixel(hdc, x_pos, y_pos)
+        colors.append(color)
+    return colors
+   
+  def _GetPixel(self, x, y):
+    """Get pixel color at coordinate x and y.
+
+    Args:
+      x: X coordinate on the screen.
+      y: Y coordinate on the screen.
+      
+    Returns:
+      An integer color code.
+    """
+    if pyauto.PyUITest.IsMac():
+      pass  # TODO(chrisphan): Do Mac.
+    elif pyauto.PyUITest.IsWin():
+      return self._GetPixelWin(x, y)
+    elif pyauto.PyUITest.IsLinux():
+      pass  # TODO(chrisphan): Do Linux.
+    return None
+  
+  def _GetPixelWin(self, x, y):
+    """Get pixel color at coordinate x and y for Windows
+
+    Args:
+      x: X coordinate on the screen.
+      y: Y coordinate on the screen.
+      
+    Returns:
+      An integer color code.
+    """
+    hdc = ctypes.windll.user32.GetDC(0)
+    color = ctypes.windll.gdi32.GetPixel(hdc, x, y)
+    return color
+  
+  def _CheckForCrashes(self, last_action=None, last_action_param=None):
+    """Check for any browser/tab crashes and hangs.
+
+    Args:
+      last_action: Specify action taken before checking for crashes.
+      last_action_param: Parameter for last action.
+    """
+    self.assertTrue(self.GetBrowserWindowCount(),
+                    msg='Browser crashed, no window is open.')
+    
+    info = self.GetBrowserInfo()
+    breakpad_folder = info['properties']['DIR_CRASH_DUMPS']
+    old_dmp_files = glob.glob(os.path.join(breakpad_folder, '*.dmp'))
+    
+    # Verify there're no crash dump files.s

[dennis_jeffrey, 2011/08/09 17:14:21]

remove the 's' at the end of this line.

[chrisphan, 2011/08/09 17:42:57]

Done.

+    for dmp_file in glob.glob(os.path.join(breakpad_folder, '*.dmp')):
+      self.assertTrue(dmp_file in old_dmp_files,
+                      msg='Crash dump %s found' % dmp_file)
+    
+    # Check for any crashed tabs.
+    tabs = info['windows'][0]['tabs']
+    for tab in tabs:
+      if tab['url'] != 'about:blank':
+        if self.GetDOMValue('document.body.innerHTML', 0, tab['index']) == '':

[dennis_jeffrey, 2011/08/09 17:14:21]

if not self.GetDOMValue('document.body.innerHTML', 0, tab['index']):

[chrisphan, 2011/08/09 17:42:57]

Done.

+          self.fail(msg='Tab crashed on %s' % tab['url'])
+          
+    # TODO(chrisphan): Check for tab hangs and browser hangs.
+    # TODO(chrisphan): Handle specific action: close browser, close tab.
+    if last_action == 'close tab':
+      pass
+    elif last_action == 'close browser':
+      pass
+    else:
+      pass
+    
+  def _GetPlatformArchitecture(self):
+    """Get platform architecture.
+
+    Args:
+      last_action: Last action taken before checking for crashes.
+      last_action_param: Parameter for last action.
+      
+    Returns:
+      A string representing the platform architecture.
+    """
+    if pyauto.PyUITest.IsWin():
+      if os.environ['PROGRAMFILES'] == 'C:\\Program Files (x86)':
+        return '64bit'
+      else:
+        return '32bit'
+    elif pyauto.PyUITest.IsMac() or pyauto.PyUITest.IsLinux():
+      if platform.machine() == 'x86_64':
+        return '64bit'
+      else:
+        return '32bit'
+    return '32bit'
+
+  def _HasFile(self, pattern, root=os.curdir):
+    """Check if a file matching the specified pattern exists in a directory.
+
+    Args:
+      pattern: Pattern of file name.
+      root: Directory to start looking.
+      
+    Returns:
+      True, if root contains the file name pattern, or
+      False otherwise.
+    """
+    return len(glob.glob(os.path.join(root, pattern)))
+  
+  def _HasPathInTree(self, pattern, is_file, root=os.curdir):
+    """Check if a file or directory matching the specified pattern exists
+       in a directory and subdirectories.

[dennis_jeffrey, 2011/08/09 17:14:21]

This first line in the docstring should be a 1-line summary.  How about this:

"""Recursively checks if a file/directory matching a pattern exists."""

[chrisphan, 2011/08/09 17:42:57]

Done.

+
+    Args:
+      pattern: Pattern of file or directory name.
+      is_file: True if looking for file, or False if looking for directory.
+      root: Directory to start looking.
+      
+    Returns:
+      True, if root contains the directory name pattern, or
+      False otherwise.
+    """
+    for path, dirs, files in os.walk(os.path.abspath(root)):
+      if is_file:
+        if len(fnmatch.filter(files, pattern)) > 0:

[dennis_jeffrey, 2011/08/09 17:14:21]

I think this would be equivalent if we remove the '> 0'.

[chrisphan, 2011/08/09 17:42:57]

Done.

+          return True
+      else:
+        if len(fnmatch.filter(dirs, pattern)) > 0:

[dennis_jeffrey, 2011/08/09 17:14:21]

I think this would be equivalent if we remove the '> 0'.

[chrisphan, 2011/08/09 17:42:57]

Done.

+          return True
+    return False
+    
+  def _GetDirectoryPath(self, pattern, root=os.curdir):
+    """Get the path of a directory in another directory.
+
+    Args:
+      pattern: Pattern of directory name.
+      root: Directory to start looking.
+    
+    Returns:
+      A string of the path.
+    """
+    for path, dirs, files in os.walk(os.path.abspath(root)):
+        result = fnmatch.filter(dirs, pattern)
+        if len(result) > 0:
+            return os.path.join(path, result[0])
+    return None
+    
+  def _HasAllSystemRequirements(self):
+    """Verify NaCl SDK installation system requirements.
+    
+    Returns:
+        True, if system passed requirements, or
+        False otherwise.
+    """
+    # Check python version.
+    if sys.version_info[0:2] < (2, 5):
+      return False
+    
+    # Check OS requirements.
+    if pyauto.PyUITest.IsMac():
+      mac_min_version = version.StrictVersion('10.6')
+      mac_version = version.StrictVersion(platform.mac_ver()[0])
+      if mac_version < mac_min_version:
+        return False
+    elif pyauto.PyUITest.IsWin():
+      if not (self.IsWin7() or self.IsWinVista() or self.IsWinXP()):
+        return False
+    elif pyauto.PyUITest.IsLinux():
+      pass  # TODO(chrisphan): Check Lin requirements.
+    else:
+      return False
+    
+    # Check for Chrome version compatibility. 
+    # NaCl supports Chrome 10 and higher builds.
+    settings = self._GetTestSetting()
+    min_required_chrome_build = settings['min_required_chrome_build']
+    browser_info = self.GetBrowserInfo()
+    chrome_version =  browser_info['properties']['ChromeVersion']
+    chrome_build = int(chrome_version.split('.')[0])
+    if chrome_build < min_required_chrome_build:
+      return False
+
+    return True

[dennis_jeffrey, 2011/08/09 17:14:21]

Change lines 871-874 to this:

return chrome_build >= min_required_chrome_build

[chrisphan, 2011/08/09 17:42:57]

Done.

+    
+  def _DownloadNaClSDK(self):
+    """Download NaCl SDK."""
+    settings = self._GetTestSetting()
+    
+    if pyauto.PyUITest.IsWin():
+      dl_file = urllib2.urlopen(settings['release_win_sdk_url'])
+      file_path = os.path.join(self._download_dir, 'naclsdk_win.exe')
+    elif pyauto.PyUITest.IsMac():
+      dl_file = urllib2.urlopen(settings['release_mac_sdk_url'])
+      file_path = os.path.join(self._download_dir, 'naclsdk_mac.tgz')
+    elif pyauto.PyUITest.IsLinux():
+      dl_file = urllib2.urlopen(settings['release_lin_sdk_url'])
+      file_path = os.path.join(self._download_dir, 'naclsdk_linux.tgz')
+    else:
+      self.fail(msg='NaCl SDK does not support this OS.')
+    
+    try:
+      f = open(file_path, 'wb')
+      f.write(dl_file.read())
+    except IOError:
+      self.fail(msg='Cannot open %s.' % file_path)
+    finally:
+      f.close()
+    
+  def _ExtractNaClSDK(self):
+    """Extract NaCl SDK."""
+    os.makedirs(self._extracted_sdk_path)
+    
+    if pyauto.PyUITest.IsWin():
+      # Requires 7-Zip to extract self-install archive.
+      seven_z_file_path = self._GetWin7ZipPath()
+      self.assertNotEqual(seven_z_file_path, None,
+                          '7-Zip is required but could not be found.')
+      
+      source_file = os.path.join(self._download_dir, 'naclsdk_win.exe')
+      proc = subprocess.Popen(
+          [seven_z_file_path, 'x', source_file, '-o' + self._extracted_sdk_path],
+          stdout=subprocess.PIPE)
+      proc.communicate()
+    elif pyauto.PyUITest.IsMac():
+      source_file = os.path.join(self._download_dir, 'naclsdk_mac.tgz')
+      tar = tarfile.open(source_file, 'r')
+      tar.extractall(self._extracted_sdk_path)
+    elif pyauto.PyUITest.IsLinux():
+      source_file = os.path.join(self._download_dir, 'naclsdk_linux.tgz')
+      tar = tarfile.open(source_file, 'r')
+      tar.extractall(self._extracted_sdk_path)
+    else:
+      self.fail(msg='NaCl SDK does not support this OS.')
+      
+  def _GetWin7ZipPath(self):
+    """Check if 7-Zip is installed on Windows.
+    
+    Returns:
+      String path to the 7-Zip executable, or None if it cannot be found.
+    """
+    current_dir = os.path.dirname(__file__)
+    seven_zip_path = os.path.join(
+        current_dir, os.pardir, os.pardir,
+        os.pardir, 'third_party', '7-Zip', '7z.exe')
+    
+    if os.path.isfile(seven_zip_path):
+      return seven_zip_path
+    
+    # Attempt to find 7-Zip in Program Files.
+    program_files_path = os.getenv('ProgramFiles')
+    if program_files_path != None:
+      seven_zip_path = os.path.join(program_files_path, '7-Zip', '7z.exe')
+      if os.path.isfile(seven_zip_path):
+        return seven_zip_path
+    program_files_path = os.getenv('ProgramW6432')
+    if program_files_path != None:
+      seven_zip_path = os.path.join(program_files_path, '7-Zip', '7z.exe')
+      if os.path.isfile(seven_zip_path):
+        return seven_zip_path
+    
+    return None
+
+  def _IsURLAlive(self, url):
+    """Test if URL is alive."""
+    try:
+      urllib2.urlopen(url)
+    except:
+      return False
+    return True
+  
+  def _CloseHTTPServer(self, proc=None):
+    """Close HTTP server.
+
+    Args:
+      proc: Process that opened the HTTP server.
+    """
+    if not self._IsURLAlive('http://localhost:5103'):
+      return
+    response = urllib2.urlopen('http://localhost:5103')
+    html = response.read()
+    if not 'Native Client' in html:
+      self.fail(msg='Port 5103 is in use.')
+    
+    urllib2.urlopen('http://localhost:5103?quit=1')
+    success = self.WaitUntil(
+        lambda: self._IsURLAlive('http://localhost:5103'),
+        timeout=30, expect_retval=False)
+    if not success:
+      if proc == None:
+        self.fail(msg='Fail to close HTTP server.')
+      else:
+        if proc.poll() == None:
+          try:
+            proc.kill()
+          except:
+            self.fail(msg='Fail to close HTTP server')

[dennis_jeffrey, 2011/08/09 17:14:21]

nit: 'fail' --> 'failed'

[chrisphan, 2011/08/09 17:42:57]

Done.

+        
+  def _SearchNaClSDKTarFile(self, search_list, source_file):
+    """Search NaCl SDK tar file for example files and directories.
+    
+    Args:
+      search_list: A list of strings, representing file and
+                   directory names for which to search.
+      source_file: The string name of an NaCl SDK tar file.
+    """
+    tar = tarfile.open(source_file, 'r')
+
+    # Look for files and directories in examples.
+    files = copy.deepcopy(search_list)
+    for tar_info in tar:
+      file_name = tar_info.name
+      if tar_info.isdir() and not file_name.endswith('/'):
+        file_name = file_name + '/'
+      
+      for name in files:
+        if file_name.find('examples/' + name):
+          files.remove(name)
+      if len(files) == 0:
+        break
+    
+    tar.close()
+    
+    self.assertEqual(len(files), 0,
+                     msg='Missing files or directories: %s' %
+                         ', '.join(map(str, files)))
+    
+  def _SearchNaClSDKFileWindows(self, search_list, source_file):
+    """Search NaCl SDK file for example files and directories in Windows.
+    
+    Args:
+      search_list: A list of strings, representing file and
+                   directory names for which to search.
+      source_file: The string name of an NaCl SDK Windows
+                   self-install archive file.
+    """
+    files = []
+    for i in xrange(len(search_list)):
+      files.append(search_list[i].replace(r'/', '\\'))
+
+    # Requires 7-Zip to look at self-install archive.
+    seven_z_file_path = self._GetWin7ZipPath()
+    self.assertNotEqual(seven_z_file_path, None,
+                        msg='7-Zip is required but could not be found.')
+    
+    proc = subprocess.Popen([seven_z_file_path, 'l', source_file],
+                            stdout=subprocess.PIPE)
+    stdout = proc.communicate()[0]
+    lines = stdout.splitlines()
+    for x in lines:
+      item_name = x.split(' ')[-1]
+      for name in files:
+        if item_name.find(name) > 0:
+          files.remove(name)
+      if len(files) == 0:
+        break
+    
+    self.assertEqual(len(files), 0,
+                     msg='Missing files or directories: %s' %
+                         ', '.join(map(str, files)))
+    
+  def _EnableNaClPlugin(self):
+    """"Enable NaCl Plugin."""

[dennis_jeffrey, 2011/08/09 17:14:21]

nit: 'Plugin' --> 'plugin'

[chrisphan, 2011/08/09 17:42:57]

Done.

+    nacl_plugin = self.GetPluginsInfo().PluginForName('Chrome NaCl')
+    if len(nacl_plugin) == 0:

[dennis_jeffrey, 2011/08/09 17:14:21]

if not nacl_plugin:

[chrisphan, 2011/08/09 17:42:57]

Done.

+      nacl_plugin = self.GetPluginsInfo().PluginForName('Native Client')
+    if len(nacl_plugin) == 0:

[dennis_jeffrey, 2011/08/09 17:14:21]

if not nacl_plugin:

[chrisphan, 2011/08/09 17:42:57]

Done.

+      self.fail(msg='No NaCl plugin found.')
+    self.EnablePlugin(nacl_plugin[0]['path'])
+    
+    self.NavigateToURL('about:flags')
+    
+    js_code = """
+      chrome.send('enableFlagsExperiment', ['enable-nacl', 'true']);
+      requestFlagsExperimentsData();
+      window.domAutomationController.send('done');
+    """
+    self.ExecuteJavascript(js_code)
+    self.RestartBrowser(False)
+    
+  def _GetTestSetting(self):
+    """Read the given data file and return a dictionary of items.
+    
+    Returns:
+      A dict mapping of keys/values in the NaCl SDK setting file.
+    """
+    data_file = os.path.join(self.DataDir(), 'nacl_sdk', 'nacl_sdk_setting')
+    
+    try:
+      f = open(data_file, 'r')
+    except IOError:
+      self.fail(msg='Cannot open %s.' % data_file)
+    
+    try:
+      dictionary = eval(f.read(), {'__builtins__': None}, None)
+    except SyntaxError:
+      self.fail(msg='%s is an invalid setting file.' % data_file)
+    finally:
+      f.close()
+    
+    return dictionary
+
+
+if __name__ == '__main__':
+  pyauto_functional.Main()