source: src/tests/pybin/test_run.py @ 9997fee

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 = ''

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

        def prepare(self):
                sh("mkdir -p %s" % os.path.join(self.path, '.err'))
                sh("mkdir -p %s" % os.path.join(self.path, '.out'))
                sh("mkdir -p %s" % os.path.join(self.path, '.in' ))

        def expect(self):
                return ("%s/.expect/%s%s.txt" % (self.path, self.name, '' if not self.arch else ".%s" % self.arch))

        def error_log(self):
                return ("%s/.err/%s.log"    % (self.path, self.name))

        def output_log(self):
                return ("%s/.out/%s.log"    % (self.path, self.name))

        def input(self):
                return ("%s/.in/%s.txt"     % (self.path, self.name))

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

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

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

        @classmethod
        def from_target(_, target):
                test = Test()
                test.name = os.path.basename(target)
                test.path = os.path.dirname (target)
                test.arch = settings.arch.toString() if settings.arch.cross_compile else ''
                return test


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

        @classmethod
        def toString( cls, retcode ):
                if settings.generating :
                        if   retcode == TestResult.SUCCESS:     return "Done"
                        elif retcode == TestResult.TIMEOUT:     return "TIMEOUT"
                        else :                                          return "ERROR code %d" % retcode
                else :
                        if   retcode == TestResult.SUCCESS:     return "PASSED"
                        elif retcode == TestResult.TIMEOUT:     return "TIMEOUT"
                        else :                                          return "FAILED with code %d" % retcode