py/test/factory_unittest.py

#!/usr/bin/python -u
#
# -*- coding: utf-8 -*-
#
# Copyright (c) 2012 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.


"""Unittest for factory.py."""


import factory_common # pylint: disable=W0611

import glob
import logging
import os
import traceback
import unittest

from cros.factory.test import factory


SRCROOT = os.environ.get('CROS_WORKON_SRCROOT')


class FactoryModuleTest(unittest.TestCase):
  """Unittest for Factory module."""
  def test_parse_test_lists(self):
    '''Checks that all known test lists are parseable.'''
    # This test is located in a full source checkout (e.g.,
    # src/third_party/autotest/files/client/cros/factory/
    # factory_unittest.py). Construct the paths to any test lists in
    # private overlays.
    test_lists = []

    test_lists.extend(os.path.realpath(x) for x in glob.glob(
        os.path.join(SRCROOT, 'src/private-overlays/*/'
               'chromeos-base/autotest-private-board/'
               'files/test_list*')))

    failures = []
    for test_list in test_lists:
      logging.info('Parsing test list %s', test_list)
      try:
        factory.read_test_list(test_list)
      except:  # pylint: disable=W0702
        failures.append(test_list)
        traceback.print_exc()

    if failures:
      self.fail('Errors in test lists: %r' % failures)

    self.assertEqual([], failures)

  def test_options(self):
    base_test_list = 'TEST_LIST = []\n'

    # This is a valid option.
    factory.read_test_list(
      text=base_test_list +
      'options.auto_run_on_start = True')

    try:
      factory.read_test_list(
        text=base_test_list + 'options.auto_run_on_start = 3')
      self.fail('Expected exception')
    except factory.TestListError as e:
      self.assertTrue(
        'Option auto_run_on_start has unexpected type' in e[0], e)

    try:
      factory.read_test_list(
        text=base_test_list + 'options.fly_me_to_the_moon = 3')
      self.fail('Expected exception')
    except factory.TestListError as e:
      # Sorry, swinging among the stars is currently unsupported.
      self.assertTrue(
        'Unknown option fly_me_to_the_moon' in e[0], e)

  def test_py_test_name_to_id(self):
    for name, test_id in (('a', 'A'),
                          ('_', '_'),
                          ('ab', 'Ab'),
                          ('a_b', 'AB'),
                          ('foo_bar', 'FooBar')):
      self.assertEqual(test_id, factory.FactoryTest.pytest_name_to_id(name))

if __name__ == "__main__":
  factory.init_logging('factory_unittest')
  unittest.main()