mirror of
https://github.com/launchbadge/sqlx
synced 2024-09-21 06:41:56 +00:00
203 lines
5.7 KiB
Python
Executable file
203 lines
5.7 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
import subprocess
|
|
import os
|
|
import sys
|
|
import time
|
|
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()
|
|
|
|
# base dir of sqlx workspace
|
|
dir_workspace = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
|
|
|
|
# dir of tests
|
|
dir_tests = os.path.join(dir_workspace, "tests")
|
|
|
|
|
|
def run(command, comment=None, env=None, service=None, tag=None, args=None, database_url_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_url = start_database(service, database="sqlite/sqlite.db" if service == "sqlite" else "sqlx", cwd=dir_tests)
|
|
|
|
if database_url_args:
|
|
database_url += "?" + database_url_args
|
|
|
|
environ["DATABASE_URL"] = database_url
|
|
|
|
# show the database url
|
|
print(f"\x1b[94m @ {database_url}\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.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)
|
|
|
|
|
|
# before we start, we clean previous profile data
|
|
# keeping these around can cause weird errors
|
|
for path in glob(os.path.join(os.path.dirname(__file__), "target/**/*.gc*"), recursive=True):
|
|
os.remove(path)
|
|
|
|
#
|
|
# check
|
|
#
|
|
|
|
run("cargo c", comment="check with a default set of features", tag="check")
|
|
|
|
run(
|
|
"cargo c --no-default-features --features runtime-async-std-native-tls,all-databases,all-types,offline,macros",
|
|
comment="check with async-std",
|
|
tag="check_async_std"
|
|
)
|
|
|
|
run(
|
|
"cargo c --no-default-features --features runtime-tokio-native-tls,all-databases,all-types,offline,macros",
|
|
comment="check with tokio",
|
|
tag="check_tokio"
|
|
)
|
|
|
|
run(
|
|
"cargo c --no-default-features --features runtime-actix-native-tls,all-databases,all-types,offline,macros",
|
|
comment="check with actix",
|
|
tag="check_actix"
|
|
)
|
|
|
|
#
|
|
# unit test
|
|
#
|
|
|
|
run(
|
|
"cargo test --manifest-path sqlx-core/Cargo.toml --features all-databases,all-types",
|
|
comment="unit test core",
|
|
tag="unit"
|
|
)
|
|
|
|
run(
|
|
"cargo test --no-default-features --manifest-path sqlx-core/Cargo.toml --features all-databases,all-types,runtime-tokio-native-tls",
|
|
comment="unit test core",
|
|
tag="unit_tokio"
|
|
)
|
|
|
|
#
|
|
# integration tests
|
|
#
|
|
|
|
for runtime in ["async-std", "tokio", "actix"]:
|
|
|
|
#
|
|
# sqlite
|
|
#
|
|
|
|
run(
|
|
f"cargo test --no-default-features --features macros,offline,any,all-types,sqlite,runtime-{runtime}-native-tls",
|
|
comment=f"test sqlite",
|
|
service="sqlite",
|
|
tag=f"sqlite" if runtime == "async-std" else f"sqlite_{runtime}",
|
|
)
|
|
|
|
#
|
|
# postgres
|
|
#
|
|
|
|
for version in ["12", "10", "9_6", "9_5"]:
|
|
run(
|
|
f"cargo test --no-default-features --features macros,offline,any,all-types,postgres,runtime-{runtime}-native-tls",
|
|
comment=f"test postgres {version}",
|
|
service=f"postgres_{version}",
|
|
tag=f"postgres_{version}" if runtime == "async-std" else f"postgres_{version}_{runtime}",
|
|
)
|
|
|
|
# +ssl
|
|
for version in ["12", "10", "9_6", "9_5"]:
|
|
run(
|
|
f"cargo test --no-default-features --features macros,offline,any,all-types,postgres,runtime-{runtime}-native-tls",
|
|
comment=f"test postgres {version} ssl",
|
|
database_url_args="sslmode=verify-ca&sslrootcert=.%2Ftests%2Fcerts%2Fca.crt",
|
|
service=f"postgres_{version}",
|
|
tag=f"postgres_{version}_ssl" if runtime == "async-std" else f"postgres_{version}_ssl_{runtime}",
|
|
)
|
|
|
|
#
|
|
# mysql
|
|
#
|
|
|
|
for version in ["8", "5_7", "5_6"]:
|
|
run(
|
|
f"cargo test --no-default-features --features macros,offline,any,all-types,mysql,runtime-{runtime}-native-tls",
|
|
comment=f"test mysql {version}",
|
|
service=f"mysql_{version}",
|
|
tag=f"mysql_{version}" if runtime == "async-std" else f"mysql_{version}_{runtime}",
|
|
)
|
|
|
|
#
|
|
# mariadb
|
|
#
|
|
|
|
for version in ["10_5", "10_4", "10_3", "10_2", "10_1"]:
|
|
run(
|
|
f"cargo test --no-default-features --features macros,offline,any,all-types,mysql,runtime-{runtime}-native-tls",
|
|
comment=f"test mariadb {version}",
|
|
service=f"mariadb_{version}",
|
|
tag=f"mariadb_{version}" if runtime == "async-std" else f"mariadb_{version}_{runtime}",
|
|
)
|
|
|
|
#
|
|
# mssql
|
|
#
|
|
|
|
for version in ["2019"]:
|
|
run(
|
|
f"cargo test --no-default-features --features macros,offline,any,all-types,mssql,runtime-{runtime}-native-tls",
|
|
comment=f"test mssql {version}",
|
|
service=f"mssql_{version}",
|
|
tag=f"mssql_{version}" if runtime == "async-std" else f"mssql_{version}_{runtime}",
|
|
)
|
|
|
|
# TODO: Use [grcov] if available
|
|
# ~/.cargo/bin/grcov tests/.cache/target/debug -s sqlx-core/ -t html --llvm --branch -o ./target/debug/coverage
|