# vim: set ts=2 sw=4 tw=99 ft=python et:

 import os

 import os.path

 class Result:

   def __init__(self, ok, out, err, code, timeout, message):

     self.ok = ok

     self.out = out

     self.err = err

     self.code = code

     self.timeout = timeout

     self.message = message

 class Test:

   def __init__(self, path, name, type):

     self.path = path

     self.name = name

     self.type = type

   def cmd(self):

     return os.path.join(self.path, self.name + '.sp')

   def compare(self, text):

     ok = True

     message = None

     with open(os.path.join(self.path, self.name + '.' + self.type), 'rt') as f:

       text = str.replace(text, '\r', '')

       lines = text.split('\n')

       i = 0

       for line in f:

         line = line.rstrip('\r\n')

         if "Displaying compilation errors for" in lines[i]:

           i += 1

           continue

         if line != lines[i]:

           message = 'Got "' + lines[i] + '", expected "' + line + '"'

           ok = False

           break

         i += 1

     return (ok, message)

   def check(self, out, err, code, timeout):

     if not out:

       out = ''

     if not err:

       err = ''

     if self.type == 'err':

       if code == 0:

         message = 'Expected failure, got success'

         ok = False

       else:

         (ok, message) = self.compare(err)

     elif self.type == 'out':

       if code != 0:

         message = 'Expected success, got failure (code: ' + str(code) + ')'

         ok = False

       else:

         (ok, message) = self.compare(out)

     if timeout:

       ok = False

       message = 'timeout'

     return Result(ok, out, err, code, timeout, message)