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

   1 # Pleroma: A lightweight social networking server
   2 # Copyright © 2017-2020 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
  16   clear_config([Pleroma.Captcha, :enabled])
  17
  18   describe "Kocaptcha" do
  19     setup do
  20       ets_name = Kocaptcha.Ets
  21       ^ets_name = :ets.new(ets_name, @ets_options)
  22
  23       mock(fn
  24         %{method: :get, url: "https://captcha.kotobank.ch/new"} ->
  25           json(%{
  26             md5: "63615261b77f5354fb8c4e4986477555",
  27             token: "afa1815e14e29355e6c8f6b143a39fa2",
  28             url: "/captchas/afa1815e14e29355e6c8f6b143a39fa2.png"
  29           })
  30       end)
  31
  32       :ok
  33     end
  34
  35     test "new and validate" do
  36       new = Kocaptcha.new()
  37
  38       token = "afa1815e14e29355e6c8f6b143a39fa2"
  39       url = "https://captcha.kotobank.ch/captchas/afa1815e14e29355e6c8f6b143a39fa2.png"
  40
  41       assert %{
  42                answer_data: answer,
  43                token: ^token,
  44                url: ^url,
  45                type: :kocaptcha
  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              } = new
  62
  63       assert is_binary(answer)
  64       assert :ok = Native.validate(token, answer, answer)
  65       assert {:error, "Invalid CAPTCHA"} == Native.validate(token, answer, answer <> "foobar")
  66     end
  67   end
  68
  69   describe "Captcha Wrapper" do
  70     test "validate" do
  71       Pleroma.Config.put([Pleroma.Captcha, :enabled], true)
  72
  73       new = Captcha.new()
  74
  75       assert %{
  76                answer_data: answer,
  77                token: token
  78              } = new
  79
  80       assert is_binary(answer)
  81       assert :ok = Captcha.validate(token, "63615261b77f5354fb8c4e4986477555", answer)
  82     end
  83
  84     test "doesn't validate invalid answer" do
  85       Pleroma.Config.put([Pleroma.Captcha, :enabled], true)
  86
  87       new = Captcha.new()
  88
  89       assert %{
  90                answer_data: answer,
  91                token: token
  92              } = new
  93
  94       assert is_binary(answer)
  95
  96       assert {:error, "Invalid answer data"} =
  97                Captcha.validate(token, "63615261b77f5354fb8c4e4986477555", answer <> "foobar")
  98     end
  99
 100     test "nil answer_data" do
 101       Pleroma.Config.put([Pleroma.Captcha, :enabled], true)
 102
 103       new = Captcha.new()
 104
 105       assert %{
 106                answer_data: answer,
 107                token: token
 108              } = new
 109
 110       assert is_binary(answer)
 111
 112       assert {:error, "Invalid answer data"} =
 113                Captcha.validate(token, "63615261b77f5354fb8c4e4986477555", nil)
 114     end
 115   end
 116 end