85 lines
2.5 KiB
Elixir
85 lines
2.5 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.RepoTest do
|
|
use Pleroma.DataCase
|
|
import ExUnit.CaptureLog
|
|
import Pleroma.Factory
|
|
import Mock
|
|
|
|
alias Pleroma.User
|
|
|
|
describe "find_resource/1" do
|
|
test "returns user" do
|
|
user = insert(:user)
|
|
query = from(t in User, where: t.id == ^user.id)
|
|
assert Repo.find_resource(query) == {:ok, user}
|
|
end
|
|
|
|
test "returns not_found" do
|
|
query = from(t in User, where: t.id == ^"9gBuXNpD2NyDmmxxdw")
|
|
assert Repo.find_resource(query) == {:error, :not_found}
|
|
end
|
|
end
|
|
|
|
describe "get_assoc/2" do
|
|
test "get assoc from preloaded data" do
|
|
user = %User{name: "Agent Smith"}
|
|
token = %Pleroma.Web.OAuth.Token{insert(:oauth_token) | user: user}
|
|
assert Repo.get_assoc(token, :user) == {:ok, user}
|
|
end
|
|
|
|
test "get one-to-one assoc from repo" do
|
|
user = insert(:user, name: "Jimi Hendrix")
|
|
token = refresh_record(insert(:oauth_token, user: user))
|
|
|
|
assert Repo.get_assoc(token, :user) == {:ok, user}
|
|
end
|
|
|
|
test "get one-to-many assoc from repo" do
|
|
user = insert(:user)
|
|
notification = refresh_record(insert(:notification, user: user))
|
|
|
|
assert Repo.get_assoc(user, :notifications) == {:ok, [notification]}
|
|
end
|
|
|
|
test "return error if has not assoc " do
|
|
token = insert(:oauth_token, user: nil)
|
|
assert Repo.get_assoc(token, :user) == {:error, :not_found}
|
|
end
|
|
end
|
|
|
|
describe "check_migrations_applied!" do
|
|
setup_with_mocks([
|
|
{Ecto.Migrator, [],
|
|
[
|
|
with_repo: fn repo, fun -> passthrough([repo, fun]) end,
|
|
migrations: fn Pleroma.Repo ->
|
|
[
|
|
{:up, 20_191_128_153_944, "fix_missing_following_count"},
|
|
{:up, 20_191_203_043_610, "create_report_notes"},
|
|
{:down, 20_191_220_174_645, "add_scopes_to_pleroma_feo_auth_records"}
|
|
]
|
|
end
|
|
]}
|
|
]) do
|
|
:ok
|
|
end
|
|
|
|
clear_config([:i_am_aware_this_may_cause_data_loss, :disable_migration_check])
|
|
|
|
test "raises if it detects unapplied migrations" do
|
|
assert_raise Pleroma.Repo.UnappliedMigrationsError, fn ->
|
|
capture_log(&Repo.check_migrations_applied!/0)
|
|
end
|
|
end
|
|
|
|
test "doesn't do anything if disabled" do
|
|
Pleroma.Config.put([:i_am_aware_this_may_cause_data_loss, :disable_migration_check], true)
|
|
|
|
assert :ok == Repo.check_migrations_applied!()
|
|
end
|
|
end
|
|
end
|