pleroma/test/support/conn_case.ex

138 lines
3.9 KiB
Elixir
Raw Normal View History

2018-12-23 12:11:29 -08:00
# Pleroma: A lightweight social networking server
# Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
2018-12-23 12:11:29 -08:00
# SPDX-License-Identifier: AGPL-3.0-only
2017-03-17 09:09:58 -07:00
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
2020-10-07 08:31:08 -07:00
import Plug.Conn
import Phoenix.ConnTest
2018-12-11 04:31:52 -08:00
use Pleroma.Tests.Helpers
2017-03-17 09:09:58 -07:00
import Pleroma.Web.Router.Helpers
alias Pleroma.Config
2017-03-17 09:09:58 -07:00
# 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 empty_json_response(conn) do
body = response(conn, 204)
2020-08-11 02:08:27 -07:00
response_content_type(conn, :json)
body
end
defp json_response_and_validate_schema(
%{
private: %{
open_api_spex: %{operation_id: op_id, operation_lookup: lookup, spec: spec}
}
} = conn,
status
) do
content_type =
conn
|> Plug.Conn.get_resp_header("content-type")
|> List.first()
|> String.split(";")
|> List.first()
status = Plug.Conn.Status.code(status)
unless lookup[op_id].responses[status] do
2020-05-05 06:05:34 -07:00
err = "Response schema not found for #{status} #{conn.method} #{conn.request_path}"
flunk(err)
end
schema = lookup[op_id].responses[status].content[content_type].schema
json = if status == 204, do: empty_json_response(conn), else: json_response(conn, status)
case OpenApiSpex.cast_value(json, schema, spec) do
{:ok, _data} ->
json
{:error, errors} ->
errors =
Enum.map(errors, fn error ->
message = OpenApiSpex.Cast.Error.message(error)
path = OpenApiSpex.Cast.Error.path_to_string(error)
"#{message} at #{path}"
end)
flunk(
"Response does not conform to schema of #{op_id} operation: #{
Enum.join(errors, "\n")
}\n#{inspect(json)}"
)
end
end
defp json_response_and_validate_schema(conn, _status) do
flunk("Response schema not found for #{conn.method} #{conn.request_path} #{conn.status}")
end
2017-03-17 09:09:58 -07:00
end
end
setup tags do
2018-03-22 04:44:32 -07:00
Cachex.clear(:user_cache)
2019-02-03 09:54:39 -08:00
Cachex.clear(:object_cache)
2017-03-17 09:09:58 -07:00
:ok = Ecto.Adapters.SQL.Sandbox.checkout(Pleroma.Repo)
2018-03-30 06:01:53 -07:00
2017-03-17 09:09:58 -07:00
unless tags[:async] do
Ecto.Adapters.SQL.Sandbox.mode(Pleroma.Repo, {:shared, self()})
end
2018-03-30 06:01:53 -07:00
if tags[:needs_streamer] do
2020-05-07 02:13:32 -07:00
start_supervised(%{
id: Pleroma.Web.Streamer.registry(),
start:
{Registry, :start_link, [[keys: :duplicate, name: Pleroma.Web.Streamer.registry()]]}
})
end
2017-03-17 09:09:58 -07:00
{:ok, conn: Phoenix.ConnTest.build_conn()}
end
end