source: tests/pybin/test_run.py @ d895e32

import os

from pybin.tools import *

import pybin.settings

# Test class that defines what a test is
class Test:
        def __init__(self):
                self.name = ''
                self.path = ''
                self.arch = ''
                self.astv = ''

        def toString(self):
                return "{:25s} ({:5s} arch, {:s} ast: {:s})".format( self.name, self.arch if self.arch else "Any", self.astv if self.astv else "Any", self.target() )

        def prepare(self):
                mkdir( (self.output_log(), self.error_log(), self.input()            ) )
                rm   ( (self.output_log(), self.error_log(), self.target_executable()) )

        def expect(self):
                arch = '' if not self.arch else ".%s" % self.arch
                astv = '' if not self.astv else ".nast" if self.astv == "new" else ".oast"
                return os.path.normpath( os.path.join(settings.SRCDIR  , self.path, ".expect", "%s%s%s.txt" % (self.name,astv,arch)) )

        def error_log(self):
                return os.path.normpath( os.path.join(settings.BUILDDIR, self.path, ".err"   , "%s.log" % self.name) )

        def output_log(self):
                return os.path.normpath( os.path.join(settings.BUILDDIR, self.path, ".out"   , "%s.log" % self.name) )

        def input(self):
                return os.path.normpath( os.path.join(settings.SRCDIR  , self.path, ".in"    , "%s.txt" % self.name) )

        def target_output(self):
                return self.output_log() if not settings.generating else self.expect()

        def target(self):
                return os.path.normpath( os.path.join(self.path, self.name) )

        def target_executable(self):
                return os.path.normpath( os.path.join(settings.BUILDDIR, self.path, self.name) )

        def format_target(self, width):
                target = self.target()
                length = len(target)
                if length < width:
                        return '{0:{width}}'.format(target, width=width)
                elif length == width:
                        return target
                else:
                        return '...' + target[3-width:]

        @staticmethod
        def valid_name(name):
                return not name.endswith( ('.c', '.cc', '.cpp', '.cfa') )

        @staticmethod
        def new_target(target, arch, astv):
                test = Test()
                test.name = os.path.basename(target)
                test.path = os.path.relpath (os.path.dirname(target), settings.SRCDIR)
                test.arch = arch.target if arch else ''
                test.astv = astv.target if astv else ''
                return test


class TestResult:
        SUCCESS = 0
        FAILURE = 1
        TIMEOUT = 124

        @classmethod
        def toString( cls, retcode, duration ):
                if settings.generating :
                        if   retcode == TestResult.SUCCESS:     key = 'pass'; text = "Done   "
                        elif retcode == TestResult.TIMEOUT:     key = 'time'; text = "TIMEOUT"
                        else :  key = 'fail';   text = "ERROR code %d" % retcode
                else :
                        if   retcode == TestResult.SUCCESS:     key = 'pass'; text = "PASSED "
                        elif retcode == TestResult.TIMEOUT:     key = 'time'; text = "TIMEOUT"
                        else :  key = 'fail';   text = "FAILED with code %d" % retcode

                text += "    C%s - R%s" % (fmtDur(duration[0]), fmtDur(duration[1]))
                return key, text