git.squeep.com Git - akkoma/blob - test/web/activity_pub/mrf/mrf_test.exs

   1 defmodule Pleroma.Web.ActivityPub.MRFTest do
   2   use ExUnit.Case, async: true
   3   alias Pleroma.Web.ActivityPub.MRF
   4
   5   test "subdomains_regex/1" do
   6     assert MRF.subdomains_regex(["unsafe.tld", "*.unsafe.tld"]) == [
   7              ~r/^unsafe.tld$/i,
   8              ~r/^(.*\.)*unsafe.tld$/i
   9            ]
  10   end
  11
  12   describe "subdomain_match/2" do
  13     test "common domains" do
  14       regexes = MRF.subdomains_regex(["unsafe.tld", "unsafe2.tld"])
  15
  16       assert regexes == [~r/^unsafe.tld$/i, ~r/^unsafe2.tld$/i]
  17
  18       assert MRF.subdomain_match?(regexes, "unsafe.tld")
  19       assert MRF.subdomain_match?(regexes, "unsafe2.tld")
  20
  21       refute MRF.subdomain_match?(regexes, "example.com")
  22     end
  23
  24     test "wildcard domains with one subdomain" do
  25       regexes = MRF.subdomains_regex(["*.unsafe.tld"])
  26
  27       assert regexes == [~r/^(.*\.)*unsafe.tld$/i]
  28
  29       assert MRF.subdomain_match?(regexes, "unsafe.tld")
  30       assert MRF.subdomain_match?(regexes, "sub.unsafe.tld")
  31       refute MRF.subdomain_match?(regexes, "anotherunsafe.tld")
  32       refute MRF.subdomain_match?(regexes, "unsafe.tldanother")
  33     end
  34
  35     test "wildcard domains with two subdomains" do
  36       regexes = MRF.subdomains_regex(["*.unsafe.tld"])
  37
  38       assert regexes == [~r/^(.*\.)*unsafe.tld$/i]
  39
  40       assert MRF.subdomain_match?(regexes, "unsafe.tld")
  41       assert MRF.subdomain_match?(regexes, "sub.sub.unsafe.tld")
  42       refute MRF.subdomain_match?(regexes, "sub.anotherunsafe.tld")
  43       refute MRF.subdomain_match?(regexes, "sub.unsafe.tldanother")
  44     end
  45
  46     test "matches are case-insensitive" do
  47       regexes = MRF.subdomains_regex(["UnSafe.TLD", "UnSAFE2.Tld"])
  48
  49       assert regexes == [~r/^UnSafe.TLD$/i, ~r/^UnSAFE2.Tld$/i]
  50
  51       assert MRF.subdomain_match?(regexes, "UNSAFE.TLD")
  52       assert MRF.subdomain_match?(regexes, "UNSAFE2.TLD")
  53       assert MRF.subdomain_match?(regexes, "unsafe.tld")
  54       assert MRF.subdomain_match?(regexes, "unsafe2.tld")
  55
  56       refute MRF.subdomain_match?(regexes, "EXAMPLE.COM")
  57       refute MRF.subdomain_match?(regexes, "example.com")
  58     end
  59   end
  60 end