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$/,
   8              ~r/^(.*\.)*unsafe.tld$/
   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$/, ~r/^unsafe2.tld$/]
  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$/]
  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$/]
  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   end
  46 end