Allow to configure audience

doctolib · Jul 20, 2024 · 686fbfb · 686fbfb
1 parent 9fb54d3
commit 686fbfb
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 3 deletions.
diff --git a/load_test/config/runtime.exs b/load_test/config/runtime.exs
@@ -15,6 +15,9 @@ config :load_test,
   sse_jwt_secret:
     System.get_env("SSE_JWT_SECRET") || "966KljJz--KyzyBnMOrFXfAkq9XMqWwPgdBV3cKTxsc"
 
+config :load_test,
+  sse_jwt_audience: System.get_env("SSE_JWT_AUDIENCE") || "public_api"
+
 config :load_test,
   publish_url: System.get_env("PUBLISH_URL") || "http://localhost:3000/v1/publish"
 
@@ -29,6 +32,9 @@ config :load_test,
   publish_jwt_secret:
     System.get_env("PUBLISH_JWT_SECRET") || "nLjJdNLlpdv3W4Xk7MyVCAZKD-hvza6FQ4yhUUFnjmg"
 
+config :load_test,
+  publish_jwt_audience: System.get_env("PUBLISH_JWT_AUDIENCE") || "internal_api"
+
 config :load_test,
   delay_between_messages_min:
     String.to_integer(System.get_env("DELAY_BETWEEN_MESSAGES_MIN") || "500")

diff --git a/load_test/lib/load_test/main.ex b/load_test/lib/load_test/main.ex
@@ -6,12 +6,14 @@ defmodule LoadTest.Main do
     defstruct [
       :sse_timeout,
       :sse_url,
-      :sse_jwt_secret,
       :sse_jwt_issuer,
+      :sse_jwt_secret,
+      :sse_jwt_audience,
       :publish_url,
       :publish_timeout,
       :publish_jwt_issuer,
       :publish_jwt_secret,
+      :publish_jwt_audience,
       :delay_between_messages_min,
       :delay_between_messages_max,
       :number_of_messages_min,
@@ -31,11 +33,13 @@ defmodule LoadTest.Main do
     {:ok, sse_url} = Application.fetch_env(:load_test, :sse_url)
     {:ok, sse_jwt_issuer} = Application.fetch_env(:load_test, :sse_jwt_issuer)
     {:ok, sse_jwt_secret} = Application.fetch_env(:load_test, :sse_jwt_secret)
+    {:ok, sse_jwt_audience} = Application.fetch_env(:load_test, :sse_jwt_audience)
 
     {:ok, publish_url} = Application.fetch_env(:load_test, :publish_url)
     {:ok, publish_timeout} = Application.fetch_env(:load_test, :publish_timeout)
     {:ok, publish_jwt_issuer} = Application.fetch_env(:load_test, :publish_jwt_issuer)
     {:ok, publish_jwt_secret} = Application.fetch_env(:load_test, :publish_jwt_secret)
+    {:ok, publish_jwt_audience} = Application.fetch_env(:load_test, :publish_jwt_audience)
 
     {:ok, delay_between_messages_min} =
       Application.fetch_env(:load_test, :delay_between_messages_min)
@@ -53,10 +57,12 @@ defmodule LoadTest.Main do
       sse_url: sse_url,
       sse_jwt_issuer: sse_jwt_issuer,
       sse_jwt_secret: JOSE.JWK.from_oct(sse_jwt_secret),
+      sse_jwt_audience: sse_jwt_audience,
       publish_url: publish_url,
       publish_timeout: publish_timeout,
       publish_jwt_issuer: publish_jwt_issuer,
       publish_jwt_secret: JOSE.JWK.from_oct(publish_jwt_secret),
+      publish_jwt_audience: publish_jwt_audience,
       delay_between_messages_min: delay_between_messages_min,
       delay_between_messages_max: delay_between_messages_max,
       number_of_messages_min: number_of_messages_min,

diff --git a/load_test/lib/load_test/user/publisher.ex b/load_test/lib/load_test/user/publisher.ex
@@ -49,7 +49,7 @@ defmodule LoadTest.User.Publisher do
       "iss" => context.publish_jwt_issuer,
       "exp" => exp,
       "iat" => iat,
-      "aud" => "internal_api"
+      "aud" => context.publish_jwt_audience
     }
 
     jws = %{

diff --git a/load_test/lib/load_test/user/sse.ex b/load_test/lib/load_test/user/sse.ex
@@ -21,7 +21,7 @@ defmodule SseUser do
       "iss" => context.sse_jwt_issuer,
       "exp" => exp,
       "iat" => iat,
-      "aud" => "public_api",
+      "aud" => context.sse_jwt_audience,
       "sub" => topic
     }