syntax_tests.py

import shutil
from mjtest.environment import Environment, TestMode
from mjtest.test.tests import TestCase, BasicTestResult
from os import path


class BasicSyntaxTest(TestCase):

    FILE_ENDINGS = [".invalid.mj", ".valid.mj", ".mj"]
    MODE = TestMode.syntax

    def __init__(self, env: Environment, type: str, file: str):
        super().__init__(env, type, file)
        self._should_succeed = not file.endswith(".invalid.mj")

    def should_succeed(self) -> bool:
        return self._should_succeed

    def short_name(self) -> str:
        return path.basename(self.file)[:-3]

    def run(self) -> BasicTestResult:
        out, err, rtcode = self.env.run_mj_command(self.MODE, self.file)
        return BasicTestResult(self, rtcode, out.decode(), err.decode())

TestCase.TEST_CASE_CLASSES[TestMode.syntax].append(BasicSyntaxTest)


class JavaCompileTest(BasicSyntaxTest):
    """
    The MiniJava compiler should behave the same as javac
    """

    FILE_ENDINGS = [".java"]

    def __init__(self, env: Environment, type: str, file: str):
        super().__init__(env, type, file)
        tmp_dir = self.env.create_tmpdir()
        _, self.javac_err, self.javac_rtcode = \
            self.env.run_command("javac", path.relpath(file), "-d", tmp_dir)
        shutil.rmtree(tmp_dir, ignore_errors=True)
        self._should_succeed = self.javac_rtcode == 0

    def short_name(self) -> str:
        return path.basename(self.file)

    def run(self) -> BasicTestResult:
        ret = super().run()
        ret.add_additional_text("javac error output", self.javac_err.decode())
        ret.add_additional_text_line("javac return code", str(self.javac_rtcode))
        return ret

TestCase.TEST_CASE_CLASSES[TestMode.syntax].append(JavaCompileTest)