added tests

This commit is contained in:
Maksim Pechnikov 2020-06-23 06:56:17 +03:00
parent b0a40fc2e4
commit 2715c40e1d
2 changed files with 47 additions and 9 deletions

View File

@ -24,7 +24,9 @@ def verify! do
defp handle_result(:ok), do: :ok defp handle_result(:ok), do: :ok
defp handle_result({:error, message}), do: raise(VerifyError, message: message) defp handle_result({:error, message}), do: raise(VerifyError, message: message)
defp check_migrations_applied!(:ok) do # Checks for pending migrations.
#
def check_migrations_applied!(:ok) do
unless Pleroma.Config.get( unless Pleroma.Config.get(
[:i_am_aware_this_may_cause_data_loss, :disable_migration_check], [:i_am_aware_this_may_cause_data_loss, :disable_migration_check],
false false
@ -58,8 +60,10 @@ defp check_migrations_applied!(:ok) do
end end
end end
defp check_migrations_applied!(result), do: result def check_migrations_applied!(result), do: result
# Checks for settings of RUM indexes.
#
defp check_rum!(:ok) do defp check_rum!(:ok) do
{_, res, _} = {_, res, _} =
Ecto.Migrator.with_repo(Pleroma.Repo, fn repo -> Ecto.Migrator.with_repo(Pleroma.Repo, fn repo ->

View File

@ -2,18 +2,15 @@
# Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/> # Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
# SPDX-License-Identifier: AGPL-3.0-only # SPDX-License-Identifier: AGPL-3.0-only
defmodule Pleroma.RepoTest do defmodule Pleroma.ApplicationRequirementsTest do
use Pleroma.DataCase use Pleroma.DataCase
import ExUnit.CaptureLog import ExUnit.CaptureLog
import Mock import Mock
describe "check_rum!" do describe "check_rum!" do
setup_with_mocks([ setup_with_mocks([
{Ecto.Migrator, [], {Pleroma.ApplicationRequirements, [:passthrough],
[ [check_migrations_applied!: fn _ -> :ok end]}
with_repo: fn repo, fun -> passthrough([repo, fun]) end,
migrations: fn Pleroma.Repo -> [] end
]}
]) do ]) do
:ok :ok
end end
@ -29,6 +26,43 @@ test "raises if rum is enabled and detects unapplied rum migrations" do
capture_log(&Pleroma.ApplicationRequirements.verify!/0) capture_log(&Pleroma.ApplicationRequirements.verify!/0)
end end
end end
test "raises if rum is disabled and detects rum migrations" do
Pleroma.Config.put([:database, :rum_enabled], false)
with_mocks([
{
Pleroma.Repo,
[:passthrough],
[exists?: fn _, _ -> true end]
}
]) do
assert_raise Pleroma.ApplicationRequirements.VerifyError,
"RUM Migrations detected",
fn ->
capture_log(&Pleroma.ApplicationRequirements.verify!/0)
end
end
end
test "doesn't do anything if rum enabled and applied migrations" do
Pleroma.Config.put([:database, :rum_enabled], true)
with_mocks([
{
Pleroma.Repo,
[:passthrough],
[exists?: fn _, _ -> true end]
}
]) do
assert Pleroma.ApplicationRequirements.verify!() == :ok
end
end
test "doesn't do anything if rum disabled" do
Pleroma.Config.put([:database, :rum_enabled], false)
assert Pleroma.ApplicationRequirements.verify!() == :ok
end
end end
describe "check_migrations_applied!" do describe "check_migrations_applied!" do