git.squeep.com Git - akkoma/blob - test/plugs/rate_limit_plug_test.exs

   1 defmodule Pleroma.Plugs.RateLimitPlugTest do
   2   use ExUnit.Case, async: true
   3   use Plug.Test
   4
   5   alias Pleroma.Plugs.RateLimitPlug
   6
   7   @opts RateLimitPlug.init(%{max_requests: 5, interval: 1})
   8
   9   setup do
  10     enabled = Pleroma.Config.get([:app_account_creation, :enabled])
  11
  12     Pleroma.Config.put([:app_account_creation, :enabled], true)
  13
  14     on_exit(fn ->
  15       Pleroma.Config.put([:app_account_creation, :enabled], enabled)
  16     end)
  17
  18     :ok
  19   end
  20
  21   test "it restricts by opts" do
  22     conn = conn(:get, "/")
  23     bucket_name = conn.remote_ip |> Tuple.to_list() |> Enum.join(".")
  24     ms = 1000
  25
  26     conn = RateLimitPlug.call(conn, @opts)
  27     {1, 4, _, _, _} = ExRated.inspect_bucket(bucket_name, ms, 5)
  28     conn = RateLimitPlug.call(conn, @opts)
  29     {2, 3, _, _, _} = ExRated.inspect_bucket(bucket_name, ms, 5)
  30     conn = RateLimitPlug.call(conn, @opts)
  31     {3, 2, _, _, _} = ExRated.inspect_bucket(bucket_name, ms, 5)
  32     conn = RateLimitPlug.call(conn, @opts)
  33     {4, 1, _, _, _} = ExRated.inspect_bucket(bucket_name, ms, 5)
  34     conn = RateLimitPlug.call(conn, @opts)
  35     {5, 0, to_reset, _, _} = ExRated.inspect_bucket(bucket_name, ms, 5)
  36     conn = RateLimitPlug.call(conn, @opts)
  37     assert conn.status == 403
  38     assert conn.halted
  39     assert conn.resp_body == "{\"error\":\"Rate limit exceeded.\"}"
  40
  41     Process.sleep(to_reset)
  42
  43     conn = conn(:get, "/")
  44     conn = RateLimitPlug.call(conn, @opts)
  45     {1, 4, _, _, _} = ExRated.inspect_bucket(bucket_name, ms, 5)
  46     refute conn.status == 403
  47     refute conn.halted
  48     refute conn.resp_body
  49   end
  50 end