99 lines
2.6 KiB
Elixir
99 lines
2.6 KiB
Elixir
# Pleroma: A lightweight social networking server
|
|
# Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
|
|
# SPDX-License-Identifier: AGPL-3.0-only
|
|
|
|
defmodule Pleroma.Web.ConnCase do
|
|
@moduledoc """
|
|
This module defines the test case to be used by
|
|
tests that require setting up a connection.
|
|
|
|
Such tests rely on `Phoenix.ConnTest` and also
|
|
import other functionality to make it easier
|
|
to build common datastructures and query the data layer.
|
|
|
|
Finally, if the test case interacts with the database,
|
|
it cannot be async. For this reason, every test runs
|
|
inside a transaction which is reset at the beginning
|
|
of the test unless the test case is marked as async.
|
|
"""
|
|
|
|
use ExUnit.CaseTemplate
|
|
|
|
using do
|
|
quote do
|
|
# Import conveniences for testing with connections
|
|
use Phoenix.ConnTest
|
|
use Pleroma.Tests.Helpers
|
|
import Pleroma.Web.Router.Helpers
|
|
|
|
alias Pleroma.Config
|
|
|
|
# The default endpoint for testing
|
|
@endpoint Pleroma.Web.Endpoint
|
|
|
|
# Sets up OAuth access with specified scopes
|
|
defp oauth_access(scopes, opts \\ []) do
|
|
user =
|
|
Keyword.get_lazy(opts, :user, fn ->
|
|
Pleroma.Factory.insert(:user)
|
|
end)
|
|
|
|
token =
|
|
Keyword.get_lazy(opts, :oauth_token, fn ->
|
|
Pleroma.Factory.insert(:oauth_token, user: user, scopes: scopes)
|
|
end)
|
|
|
|
conn =
|
|
build_conn()
|
|
|> assign(:user, user)
|
|
|> assign(:token, token)
|
|
|
|
%{user: user, token: token, conn: conn}
|
|
end
|
|
|
|
defp request_content_type(%{conn: conn}) do
|
|
conn = put_req_header(conn, "content-type", "multipart/form-data")
|
|
[conn: conn]
|
|
end
|
|
|
|
defp ensure_federating_or_authenticated(conn, url, user) do
|
|
initial_setting = Config.get([:instance, :federating])
|
|
on_exit(fn -> Config.put([:instance, :federating], initial_setting) end)
|
|
|
|
Config.put([:instance, :federating], false)
|
|
|
|
conn
|
|
|> get(url)
|
|
|> response(403)
|
|
|
|
conn
|
|
|> assign(:user, user)
|
|
|> get(url)
|
|
|> response(200)
|
|
|
|
Config.put([:instance, :federating], true)
|
|
|
|
conn
|
|
|> get(url)
|
|
|> response(200)
|
|
end
|
|
end
|
|
end
|
|
|
|
setup tags do
|
|
Cachex.clear(:user_cache)
|
|
Cachex.clear(:object_cache)
|
|
:ok = Ecto.Adapters.SQL.Sandbox.checkout(Pleroma.Repo)
|
|
|
|
unless tags[:async] do
|
|
Ecto.Adapters.SQL.Sandbox.mode(Pleroma.Repo, {:shared, self()})
|
|
end
|
|
|
|
if tags[:needs_streamer] do
|
|
start_supervised(Pleroma.Web.Streamer.supervisor())
|
|
end
|
|
|
|
{:ok, conn: Phoenix.ConnTest.build_conn()}
|
|
end
|
|
end
|