git.squeep.com Git - akkoma/blob - test/pleroma/captcha_test.exs

   1 # Pleroma: A lightweight social networking server
   2 # Copyright © 2017-2021 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