blob: 100016c62842461e987b0c04f82efa71fe6a45a2 (
plain)
| 1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
 | # Pleroma: A lightweight social networking server
# Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
# SPDX-License-Identifier: AGPL-3.0-only
defmodule Pleroma.Plugs.AdminSecretAuthenticationPlugTest do
  use Pleroma.Web.ConnCase, async: true
  import Pleroma.Factory
  alias Pleroma.Plugs.AdminSecretAuthenticationPlug
  test "does nothing if a user is assigned", %{conn: conn} do
    user = insert(:user)
    conn =
      conn
      |> assign(:user, user)
    ret_conn =
      conn
      |> AdminSecretAuthenticationPlug.call(%{})
    assert conn == ret_conn
  end
  describe "when secret set it assigns an admin user" do
    setup do: clear_config([:admin_token])
    test "with `admin_token` query parameter", %{conn: conn} do
      Pleroma.Config.put(:admin_token, "password123")
      conn =
        %{conn | params: %{"admin_token" => "wrong_password"}}
        |> AdminSecretAuthenticationPlug.call(%{})
      refute conn.assigns[:user]
      conn =
        %{conn | params: %{"admin_token" => "password123"}}
        |> AdminSecretAuthenticationPlug.call(%{})
      assert conn.assigns[:user].is_admin
    end
    test "with `x-admin-token` HTTP header", %{conn: conn} do
      Pleroma.Config.put(:admin_token, "☕️")
      conn =
        conn
        |> put_req_header("x-admin-token", "🥛")
        |> AdminSecretAuthenticationPlug.call(%{})
      refute conn.assigns[:user]
      conn =
        conn
        |> put_req_header("x-admin-token", "☕️")
        |> AdminSecretAuthenticationPlug.call(%{})
      assert conn.assigns[:user].is_admin
    end
  end
end
 |