gm/rebaseline_server/base_unittest.py

#!/usr/bin/python

"""
Copyright 2014 Google Inc.

Use of this source code is governed by a BSD-style license that can be
found in the LICENSE file.

A wrapper around the standard Python unittest library, adding features we need
for various unittests within this directory.
"""

# System-level imports.
import os
import sys

PARENT_DIR = os.path.abspath(os.path.dirname(__file__))
TRUNK_DIR = os.path.abspath(os.path.join(PARENT_DIR, os.pardir, os.pardir))

# Import the superclass base_unittest module from the tools dir.
TOOLS_DIR = os.path.join(TRUNK_DIR, 'tools')
if TOOLS_DIR not in sys.path:
  sys.path.append(TOOLS_DIR)
import tests.base_unittest as superclass_module


class TestCase(superclass_module.TestCase):

  def __init__(self, *args, **kwargs):
    super(TestCase, self).__init__(*args, **kwargs)
    # Some of the tests within this package want their output validated,
    # so we declare where the expected and actual output will be.
    self._testdata_dir = os.path.join(PARENT_DIR, 'testdata')

def main(*args, **kwargs):
  superclass_module.main(*args, **kwargs)