powerline/tests/modules/__init__.py

# vim:fileencoding=utf-8:noet
from __future__ import (unicode_literals, division, absolute_import, print_function)

import sys

if sys.version_info < (2, 7):
	from unittest2 import TestCase as _TestCase  # NOQA
	from unittest2 import main as _main  # NOQA
	from unittest2.case import SkipTest  # NOQA
else:
	from unittest import TestCase as _TestCase  # NOQA
	from unittest import main as _main  # NOQA
	from unittest.case import SkipTest  # NOQA


class PowerlineDummyTest(object):
	def __enter__(self):
		return self

	def __exit__(self, *args):
		pass

	def fail(self, *args, **kwargs):
		pass

	def exception(self, *args, **kwargs):
		pass


class PowerlineTestSuite(object):
	def __init__(self, name):
		self.name = name

	def __enter__(self):
		self.saved_current_suite = os.environ['POWERLINE_CURRENT_SUITE']
		os.environ['POWERLINE_CURRENT_SUITE'] = (
			self.saved_current_suite + '/' + self.name)
		self.suite = self.saved_current_suite + '/' + self.name
		return self

	def __exit__(self, exc_type, exc_value, traceback):
		if exc_type is not None:
			self.exception(
				'suite_noexcept',
				'Exception while running test suite: {0!r}'.format(exc_value),
			)
		os.environ['POWERLINE_CURRENT_SUITE'] = self.saved_current_suite

	def record_test_failure(self, fail_char, test_name, message, allow_failure=False):
		if allow_failure:
			fail_char = 'A' + fail_char
		full_msg = '{fail_char} {suite}|{test_name} :: {message}'.format(
			fail_char=fail_char,
			suite=self.suite,
			test_name=test_name,
			message=message,
		)
		with open(os.environ['FAILURES_FILE'], 'a') as ffd:
			ffd.write(full_msg + '\n')
		return False

	def exception(self, test_name, message, allow_failure=False):
		return self.record_test_failure('E', test_name, message, allow_failure)

	def fail(self, test_name, message, allow_failure=False):
		return self.record_test_failure('F', test_name, message, allow_failure)

	def test(self, name, attempts_left=0):
		if not attempts_left:
			return PowerlineSingleTest(self, name)
		else:
			return PowerlineDummyTest()

	def subsuite(self, name):
		return PowerlineTestSuite(name)


suite = None


def main(*args, **kwargs):
	global suite
	suite = PowerlineTestSuite(sys.argv[0])
	_main(*args, **kwargs)


class TestCase(_TestCase):
	def fail(self, *args, **kwargs):
		super(TestCase, self).fail(*args, **kwargs)
		suite.fail(self.__class__.__name__, 'Failed test')