import subprocess
import os
import sys
import argparse
from glob import glob
from docker import start_database
parser = argparse.ArgumentParser()
parser.add_argument("-t", "--target")
parser.add_argument("-e", "--target-exact")
parser.add_argument("-l", "--list-targets", action="store_true")
parser.add_argument("--test")
argv, unknown = parser.parse_known_args()
dir_workspace = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
def run(command, comment=None, env=None, service=None, tag=None, args=None, database_dsn_args=None):
if argv.list_targets:
if tag:
print(f"{tag}")
return
if argv.target and not tag.startswith(argv.target):
return
if argv.target_exact and tag != argv.target_exact:
return
if comment is not None:
print(f"\x1b[2m # {comment}\x1b[0m")
environ = env or {}
if service is not None:
database_dsn = start_database(service, database=":memory:" if service == "sqlite" else "dbal")
if database_dsn_args:
database_dsn += "?" + database_dsn_args
environ["DATABASE_DSN"] = database_dsn
print(f"\x1b[94m @ {database_dsn}\x1b[0m")
command_args = []
if argv.test:
command_args.extend(["--test", argv.test])
if unknown:
command_args.extend(["--", *unknown])
if args is not None:
command_args.extend(args)
print(f"\x1b[93m $ {command} {' '.join(command_args)}\x1b[0m")
cwd = os.path.dirname(os.path.realpath(__file__))
res = subprocess.run(
[
*command.split(" "),
*command_args
],
env=dict(**os.environ, **environ),
cwd=cwd,
)
if res.returncode != 0:
sys.exit(res.returncode)
for path in glob(os.path.join(os.path.dirname(__file__), "target/**/*.gc*"), recursive=True):
os.remove(path)
for runtime in ["tokio"]:
for tls in ["native-tls", "rustls"]:
run(
f"cargo check --no-default-features --features sqlite,postgres,mysql,runtime-{runtime}-{tls}",
comment=f"check with {runtime}",
tag=f"check_{runtime}_{tls}"
)
for runtime in ["tokio"]:
for tls in ["native-tls", "rustls"]:
run(
f"cargo test --no-default-features --features sqlite,runtime-{runtime}-{tls}",
comment=f"test sqlite",
service="sqlite",
tag=f"sqlite" if runtime == "async-std" else f"sqlite_{runtime}",
)
for version in ["14", "13", "12", "11", "10", "9_6"]:
run(
f"cargo test --no-default-features --features postgres,runtime-{runtime}-{tls}",
comment=f"test postgres {version}",
service=f"postgres_{version}",
tag=f"postgres_{version}" if runtime == "async-std" else f"postgres_{version}_{runtime}",
)
for version in ["8", "5_7", "5_6"]:
run(
f"cargo test --no-default-features --features mysql,runtime-{runtime}-{tls}",
comment=f"test mysql {version}",
service=f"mysql_{version}",
tag=f"mysql_{version}" if runtime == "async-std" else f"mysql_{version}_{runtime}",
)
for version in ["10_6", "10_5", "10_4", "10_3", "10_2"]:
run(
f"cargo test --no-default-features --features mysql,runtime-{runtime}-{tls}",
comment=f"test mariadb {version}",
service=f"mariadb_{version}",
tag=f"mariadb_{version}" if runtime == "async-std" else f"mariadb_{version}_{runtime}",
)