Class Runner

test/java_impl_tests/run_tests.py:37–115

class Runner:
    def __init__(self, server: str, client: str, port: int = 12345, mvn: str = "mvn", host="localhost"):
        self.port = port
        self.host = host
        self.mvn = mvn

        if server == "java":
            self.server = self.java_server
        elif server == "py":
            self.server = self.python_server
        else:
            raise AttributeError

        if client == "java":
            self.client = self.java_client
        elif client == "py":
            self.client = self.python_client
        else:
            raise AttributeError

    def execute_test(self, test_name: str):
        print("=================================")
        print(f"Running test case: {test_case}")
        print("=================================")

        server = self.server(test_name)
        sleep(1)
        client = self.client(test_name)

        try:
            print("============ SERVER =============")
            stdout, stderr = server.communicate(timeout=10)
        except TimeoutExpired:
            print("ERROR: TIMEOUT")
            server.kill()
            stdout, stderr = server.communicate()

        print(stdout.decode(errors="ignore"))
        print(stderr.decode())
        print("=================================")

        try:
            print("============ CLIENT =============")
            stdout, stderr = client.communicate(timeout=10)
        except TimeoutExpired:
            print("ERROR: TIMEOUT")
            client.kill()
            stdout, stderr = client.communicate()

        print(stdout.decode(errors="ignore"))
        print(stderr.decode())

    def java_client(self, test_name: str):
        return Popen(
            f"{self.mvn} test -Dtest={to_pascal_case(test_name)} -Dclient "
            f"-Dport={self.port} -Dhost={self.host}",
            stdout=PIPE, stderr=PIPE
        )