mirror of
https://github.com/plausible/analytics.git
synced 2024-12-25 18:48:06 +03:00
33 lines
719 B
Elixir
33 lines
719 B
Elixir
defmodule PlausibleWeb.FirewallTest do
|
|
use Plausible.DataCase
|
|
use Plug.Test
|
|
alias PlausibleWeb.Firewall
|
|
|
|
@allowed_ip "127.0.0.1"
|
|
@blocked_ip "127.0.0.2"
|
|
@opts [blocklist: [@blocked_ip]]
|
|
|
|
setup do
|
|
Application.put_env(:plausible, PlausibleWeb.Firewall, blocklist: [@blocked_ip])
|
|
:ok
|
|
end
|
|
|
|
test "ignores request if IP is allowed" do
|
|
conn =
|
|
conn(:get, "/")
|
|
|> put_req_header("x-forwarded-for", @allowed_ip)
|
|
|> Firewall.call(@opts)
|
|
|
|
assert conn.status == nil
|
|
end
|
|
|
|
test "responds with 404 if IP is blocked" do
|
|
conn =
|
|
conn(:get, "/")
|
|
|> put_req_header("x-forwarded-for", @blocked_ip)
|
|
|> Firewall.call(@opts)
|
|
|
|
assert conn.status == 404
|
|
end
|
|
end
|