Add Pleroma.Test.Helpers.get_query_parameter/2 to retrieve specific query parameter values

1 files changed, 11 insertions, 0 deletions

diff --git a/test/support/helpers.ex b/test/support/helpers.ex
index bcdc49e4f..7fa6c31a4 100644
--- a/test/support/helpers.ex
+++ b/test/support/helpers.ex
@@ -32,6 +32,17 @@ defmodule Pleroma.Tests.Helpers do
     |> URI.to_string()
   end
 
+  @doc "Returns the value of the specified query parameter for the provided URL"
+  def get_query_parameter(url, param) do
+    url
+    |> URI.parse()
+    |> Map.get(:query)
+    |> URI.query_decoder()
+    |> Enum.to_list()
+    |> Enum.into(%{}, fn {x, y} -> {x, y} end)
+    |> Map.get(param)
+  end
+
   defmacro clear_config(config_path) do
     quote do
       clear_config(unquote(config_path)) do