scripts/cbuildbot_view_config_unittest.py

#!/usr/bin/python

# Copyright (c) 2013 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.

"""Unittests for cbuildbot_view_config."""

import json
import os
import subprocess
import sys

sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.dirname(
    os.path.abspath(__file__)))))

from chromite.buildbot import constants
from chromite.lib import cros_test_lib


class JsonDumpTest(cros_test_lib.TestCase):
  """Test the json dumping functionality of cbuildbot_view_config."""

  def setUp(self):
    bin_name = os.path.basename(__file__).rstrip('_unittest.py')
    self.bin_path = os.path.join(constants.CHROMITE_BIN_DIR, bin_name)

  # TODO(build): Add test for compare functionality
  def testJSONDumpLoadable(self):
    """Make sure config export functionality works."""
    for args in (['--dump'], ['--dump', '--pretty']):
      # TODO(build): Leverage cbuildbot_view_config.main() directly and
      # capture output with cros_test_lib.OutputTestCase (or similar).
      output = subprocess.Popen([self.bin_path] + args,
                                stdout=subprocess.PIPE).communicate()[0]
      configs = json.loads(output)
      self.assertFalse(not configs)

  def testJSONBuildbotDumpHasOrder(self):
    """Make sure config export functionality works."""
    output = subprocess.Popen([self.bin_path, '--dump',
                               '--for-buildbot'],
                              stdout=subprocess.PIPE).communicate()[0]
    configs = json.loads(output)
    for cfg in configs.itervalues():
      self.assertTrue(cfg['display_position'] is not None)

    self.assertFalse(not configs)


if __name__ == '__main__':
  cros_test_lib.main()