forked from mirrors/akkoma
0ef0aed205
This will 'time travel', i.e. change the inserted_at and update_at fields of the object in question. This is used to backdate things were we used sleeping before to ensure time differences.
128 lines
3.2 KiB
Elixir
128 lines
3.2 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.Tests.Helpers do
|
|
@moduledoc """
|
|
Helpers for use in tests.
|
|
"""
|
|
alias Pleroma.Config
|
|
|
|
defmacro clear_config(config_path) do
|
|
quote do
|
|
clear_config(unquote(config_path)) do
|
|
end
|
|
end
|
|
end
|
|
|
|
defmacro clear_config(config_path, do: yield) do
|
|
quote do
|
|
initial_setting = Config.fetch(unquote(config_path))
|
|
unquote(yield)
|
|
|
|
on_exit(fn ->
|
|
case initial_setting do
|
|
:error ->
|
|
Config.delete(unquote(config_path))
|
|
|
|
{:ok, value} ->
|
|
Config.put(unquote(config_path), value)
|
|
end
|
|
end)
|
|
|
|
:ok
|
|
end
|
|
end
|
|
|
|
defmacro clear_config(config_path, temp_setting) do
|
|
quote do
|
|
clear_config(unquote(config_path)) do
|
|
Config.put(unquote(config_path), unquote(temp_setting))
|
|
end
|
|
end
|
|
end
|
|
|
|
def require_migration(migration_name) do
|
|
[{module, _}] = Code.require_file("#{migration_name}.exs", "priv/repo/migrations")
|
|
{:ok, %{migration: module}}
|
|
end
|
|
|
|
defmacro __using__(_opts) do
|
|
quote do
|
|
import Pleroma.Tests.Helpers,
|
|
only: [
|
|
clear_config: 1,
|
|
clear_config: 2
|
|
]
|
|
|
|
def time_travel(entity, seconds) do
|
|
new_time = NaiveDateTime.add(entity.inserted_at, seconds)
|
|
|
|
entity
|
|
|> Ecto.Changeset.change(%{inserted_at: new_time, updated_at: new_time})
|
|
|> Pleroma.Repo.update()
|
|
end
|
|
|
|
def to_datetime(%NaiveDateTime{} = naive_datetime) do
|
|
naive_datetime
|
|
|> DateTime.from_naive!("Etc/UTC")
|
|
|> DateTime.truncate(:second)
|
|
end
|
|
|
|
def to_datetime(datetime) when is_binary(datetime) do
|
|
datetime
|
|
|> NaiveDateTime.from_iso8601!()
|
|
|> to_datetime()
|
|
end
|
|
|
|
def collect_ids(collection) do
|
|
collection
|
|
|> Enum.map(& &1.id)
|
|
|> Enum.sort()
|
|
end
|
|
|
|
def refresh_record(%{id: id, __struct__: model} = _),
|
|
do: refresh_record(model, %{id: id})
|
|
|
|
def refresh_record(model, %{id: id} = _) do
|
|
Pleroma.Repo.get_by(model, id: id)
|
|
end
|
|
|
|
# Used for comparing json rendering during tests.
|
|
def render_json(view, template, assigns) do
|
|
assigns = Map.new(assigns)
|
|
|
|
view.render(template, assigns)
|
|
|> Jason.encode!()
|
|
|> Jason.decode!()
|
|
end
|
|
|
|
def stringify_keys(nil), do: nil
|
|
|
|
def stringify_keys(key) when key in [true, false], do: key
|
|
def stringify_keys(key) when is_atom(key), do: Atom.to_string(key)
|
|
|
|
def stringify_keys(map) when is_map(map) do
|
|
map
|
|
|> Enum.map(fn {k, v} -> {stringify_keys(k), stringify_keys(v)} end)
|
|
|> Enum.into(%{})
|
|
end
|
|
|
|
def stringify_keys([head | rest] = list) when is_list(list) do
|
|
[stringify_keys(head) | stringify_keys(rest)]
|
|
end
|
|
|
|
def stringify_keys(key), do: key
|
|
|
|
defmacro guards_config(config_path) do
|
|
quote do
|
|
initial_setting = Config.get(config_path)
|
|
|
|
Config.put(config_path, true)
|
|
on_exit(fn -> Config.put(config_path, initial_setting) end)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|