Materialize/misc/python/materialize/mzcompose/services/sql_logic_test.py

# Copyright Materialize, Inc. and contributors. All rights reserved.
#
# Use of this software is governed by the Business Source License
# included in the LICENSE file at the root of this repository.
#
# As of the Change Date specified in that file, in accordance with
# the Business Source License, use of this software will be governed
# by the Apache License, Version 2.0.


from materialize.mzcompose import (
    get_default_system_parameters,
)
from materialize.mzcompose.service import (
    Service,
)


class SqlLogicTest(Service):
    def __init__(
        self,
        name: str = "sqllogictest",
        mzbuild: str = "sqllogictest",
        environment: list[str] | None = None,
        volumes: list[str] = ["../..:/workdir"],
    ) -> None:
        if environment is None:
            environment = [
                "MZ_SOFT_ASSERTIONS=1",
                "LD_PRELOAD=libeatmydata.so",
            ]
        environment += [
            "MZ_SYSTEM_PARAMETER_DEFAULT="
            + ";".join(
                [
                    f"{key}={value}"
                    for key, value in get_default_system_parameters().items()
                ]
                + ["enable_lgalloc=false"]
            )
        ]

        super().__init__(
            name=name,
            config={
                "mzbuild": mzbuild,
                "environment": environment,
                "volumes": volumes,
                "tmpfs": ["/tmp"],
                "propagate_uid_gid": True,
                "init": True,
            },
        )