jimmy/postgrest
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

nix(loadtest): refactor genTargets argument

Browse Source 
Instead of passing a path to the output target file, just pass the
directory in which to create the file. This allows the script to create
multiple files later.
This commit is contained in:
Wolfgang Walther
2026-06-26 19:33:51 +00:00
parent ef72787907
commit 1446f11222
2 changed files with 11 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nix/tools/generate_targets.py
+8 -5
View File
@@ -58,9 +58,10 @@ def main():
description="Generate Vegeta targets with unique JWTs" description="Generate Vegeta targets with unique JWTs"
) )
parser.add_argument( parser.add_argument(
"targets_path", "generated_path",
metavar="TARGETS_PATH", metavar="GENERATED_PATH",
help="Path to write the generated targets file", help="Path to write the generated files",
type=Path,
) )
parser.add_argument( parser.add_argument(
"--private-key", "--private-key",
@@ -79,6 +80,8 @@ def main():
args = parser.parse_args() args = parser.parse_args()
targets_path = args.generated_path / "gen_targets.http"
rsa_private_key: Optional[jwt.algorithms.RSAAlgorithm] = None rsa_private_key: Optional[jwt.algorithms.RSAAlgorithm] = None
nsamples = 500 # per algorithm nsamples = 500 # per algorithm
@@ -134,10 +137,10 @@ def main():
lines.extend(target) lines.extend(target)
try: try:
with open(args.targets_path, "w") as f: with open(targets_path, "w") as f:
f.write("\n".join(lines)) f.write("\n".join(lines))
except IOError as e: except IOError as e:
print(f"Error writing to {args.targets_path}: {e}", file=sys.stderr) print(f"Error writing to {targets_path}: {e}", file=sys.stderr)
sys.exit(1) sys.exit(1)
print(f"Created {ntargets} targets", end=" ") print(f"Created {ntargets} targets", end=" ")
nix/tools/loadtest.nix
+3 -3
View File
@@ -70,7 +70,7 @@ let
${genKeyMaterials} --jwks="$_arg_testdir"/gen_jwks.json --private-key="$_arg_testdir"/gen_private.json ${genKeyMaterials} --jwks="$_arg_testdir"/gen_jwks.json --private-key="$_arg_testdir"/gen_private.json
export PGRST_JWT_SECRET="@$_arg_testdir/gen_jwks.json" export PGRST_JWT_SECRET="@$_arg_testdir/gen_jwks.json"
${genTargets} --private-key="$_arg_testdir"/gen_private.json "$_arg_testdir"/gen_targets.http ${genTargets} --private-key="$_arg_testdir"/gen_private.json "$_arg_testdir"
# shellcheck disable=SC2145 # shellcheck disable=SC2145
${withTools.withPg} -f "$_arg_testdir"/fixtures.sql \ ${withTools.withPg} -f "$_arg_testdir"/fixtures.sql \
@@ -83,7 +83,7 @@ let
${genKeyMaterials} --jwks="$_arg_testdir"/gen_jwks.json --private-key="$_arg_testdir"/gen_private.json ${genKeyMaterials} --jwks="$_arg_testdir"/gen_jwks.json --private-key="$_arg_testdir"/gen_private.json
export PGRST_JWT_SECRET="@$_arg_testdir/gen_jwks.json" export PGRST_JWT_SECRET="@$_arg_testdir/gen_jwks.json"
${genTargets} --private-key="$_arg_testdir"/gen_private.json "$_arg_testdir"/gen_targets.http ${genTargets} --private-key="$_arg_testdir"/gen_private.json "$_arg_testdir"
# shellcheck disable=SC2145 # shellcheck disable=SC2145
${withTools.withPg} -f "$_arg_testdir"/fixtures.sql \ ${withTools.withPg} -f "$_arg_testdir"/fixtures.sql \
@@ -96,7 +96,7 @@ let
${genKeyMaterials} --jwks="$_arg_testdir"/gen_jwks.json --private-key="$_arg_testdir"/gen_private.json ${genKeyMaterials} --jwks="$_arg_testdir"/gen_jwks.json --private-key="$_arg_testdir"/gen_private.json
export PGRST_JWT_SECRET="@$_arg_testdir/gen_jwks.json" export PGRST_JWT_SECRET="@$_arg_testdir/gen_jwks.json"
${libfaketime}/bin/faketime '2000-01-01 00:00:00' ${genTargets} --worst --private-key="$_arg_testdir"/gen_private.json "$_arg_testdir"/gen_targets.http ${libfaketime}/bin/faketime '2000-01-01 00:00:00' ${genTargets} --worst --private-key="$_arg_testdir"/gen_private.json "$_arg_testdir"
# shellcheck disable=SC2145 # shellcheck disable=SC2145
${withTools.withPg} -f "$_arg_testdir"/fixtures.sql \ ${withTools.withPg} -f "$_arg_testdir"/fixtures.sql \