First
[anni] / test / pleroma / captcha_test.exs
1 # Pleroma: A lightweight social networking server
2 # Copyright © 2017-2022 Pleroma Authors <https://pleroma.social/>
3 # SPDX-License-Identifier: AGPL-3.0-only
4
5 defmodule Pleroma.CaptchaTest do
6   use Pleroma.DataCase
7
8   import Tesla.Mock
9
10   alias Pleroma.Captcha
11   alias Pleroma.Captcha.Kocaptcha
12   alias Pleroma.Captcha.Native
13
14   @ets_options [:ordered_set, :private, :named_table, {:read_concurrency, true}]
15   setup do: clear_config([Pleroma.Captcha, :enabled])
16
17   describe "Kocaptcha" do
18     setup do
19       ets_name = Kocaptcha.Ets
20       ^ets_name = :ets.new(ets_name, @ets_options)
21
22       mock(fn
23         %{method: :get, url: "https://captcha.kotobank.ch/new"} ->
24           json(%{
25             md5: "63615261b77f5354fb8c4e4986477555",
26             token: "afa1815e14e29355e6c8f6b143a39fa2",
27             url: "/captchas/afa1815e14e29355e6c8f6b143a39fa2.png"
28           })
29       end)
30
31       :ok
32     end
33
34     test "new and validate" do
35       new = Kocaptcha.new()
36
37       token = "afa1815e14e29355e6c8f6b143a39fa2"
38       url = "https://captcha.kotobank.ch/captchas/afa1815e14e29355e6c8f6b143a39fa2.png"
39
40       assert %{
41                answer_data: answer,
42                token: ^token,
43                url: ^url,
44                type: :kocaptcha,
45                seconds_valid: 300
46              } = new
47
48       assert Kocaptcha.validate(token, "7oEy8c", answer) == :ok
49     end
50   end
51
52   describe "Native" do
53     test "new and validate" do
54       new = Native.new()
55
56       assert %{
57                answer_data: answer,
58                token: token,
59                type: :native,
60                url: "data:image/png;base64," <> _,
61                seconds_valid: 300
62              } = new
63
64       assert is_binary(answer)
65       assert :ok = Native.validate(token, answer, answer)
66       assert {:error, :invalid} == Native.validate(token, answer, answer <> "foobar")
67     end
68   end
69
70   describe "Captcha Wrapper" do
71     test "validate" do
72       clear_config([Pleroma.Captcha, :enabled], true)
73
74       new = Captcha.new()
75
76       assert %{
77                answer_data: answer,
78                token: token
79              } = new
80
81       assert is_binary(answer)
82       assert :ok = Captcha.validate(token, "63615261b77f5354fb8c4e4986477555", answer)
83     end
84
85     test "doesn't validate invalid answer" do
86       clear_config([Pleroma.Captcha, :enabled], true)
87
88       new = Captcha.new()
89
90       assert %{
91                answer_data: answer,
92                token: token
93              } = new
94
95       assert is_binary(answer)
96
97       assert {:error, :invalid_answer_data} =
98                Captcha.validate(token, "63615261b77f5354fb8c4e4986477555", answer <> "foobar")
99     end
100
101     test "nil answer_data" do
102       clear_config([Pleroma.Captcha, :enabled], true)
103
104       new = Captcha.new()
105
106       assert %{
107                answer_data: answer,
108                token: token
109              } = new
110
111       assert is_binary(answer)
112
113       assert {:error, :invalid_answer_data} =
114                Captcha.validate(token, "63615261b77f5354fb8c4e4986477555", nil)
115     end
116   end
117 end