From 67bd3526c231536974cb9d85e90443b52e11af63 Mon Sep 17 00:00:00 2001 From: steve-chavez Date: Fri, 20 Jun 2025 10:55:14 -0500 Subject: [PATCH] nix: use json file for generated jwk on loadtest Having the JWK as .http file was confusing. Also split genTargets into two to avoid confusion. --- .gitignore | 2 +- nix/tools/loadtest.nix | 19 +++++++++++++------ 2 files changed, 14 insertions(+), 7 deletions(-) diff --git a/.gitignore b/.gitignore index 7794fdd3c..af81dde65 100644 --- a/.gitignore +++ b/.gitignore @@ -25,4 +25,4 @@ loadtest .history .docs-build gen_targets.http -gen_jwk.http +gen_jwk.json diff --git a/nix/tools/loadtest.nix b/nix/tools/loadtest.nix index 6f3c7462a..7ab7a97c6 100644 --- a/nix/tools/loadtest.nix +++ b/nix/tools/loadtest.nix @@ -69,7 +69,7 @@ let case "$_arg_kind" in jwt-hs) - ${genTargets ./generate_targets.py} "$_arg_testdir"/gen_targets.http + ${genTargetsHS} "$_arg_testdir"/gen_targets.http if [ "$_arg_jwtcache" = "off" ]; then export PGRST_JWT_CACHE_MAX_LIFETIME="0" @@ -84,9 +84,9 @@ let jwt-rsa) - ${genTargets ./generate_targets_rsa.py} "$_arg_testdir"/gen_targets.http "$_arg_testdir"/gen_jwk.http + ${genTargetsRSA} "$_arg_testdir"/gen_targets.http "$_arg_testdir"/gen_jwk.json - export PGRST_JWT_SECRET="@$_arg_testdir/gen_jwk.http" + export PGRST_JWT_SECRET="@$_arg_testdir/gen_jwk.json" if [ "$_arg_jwtcache" = "off" ]; then export PGRST_JWT_CACHE_MAX_LIFETIME="0" @@ -245,12 +245,19 @@ let | ${mergeMonitorResults} ''; - genTargets = genTargetsScript: - writers.writePython3 "postgrest-gen-loadtest-targets" + genTargetsHS = + writers.writePython3 "postgrest-gen-loadtest-targets-hs" { libraries = [ python3Packages.pyjwt python3Packages.jwcrypto ]; } - (builtins.readFile genTargetsScript); + (builtins.readFile ./generate_targets.py); + + genTargetsRSA = + writers.writePython3 "postgrest-gen-loadtest-targets-rsa" + { + libraries = [ python3Packages.pyjwt python3Packages.jwcrypto ]; + } + (builtins.readFile ./generate_targets_rsa.py); mergeMonitorResults = writers.writePython3 "postgrest-merge-monitor-results"