git.squeep.com Git - akkoma/blob - test/repo_test.exs

   1 # Pleroma: A lightweight social networking server
   2 # Copyright © 2017-2019 Pleroma Authors <https://pleroma.social/>
   3 # SPDX-License-Identifier: AGPL-3.0-only
   4
   5 defmodule Pleroma.RepoTest do
   6   use Pleroma.DataCase
   7   import ExUnit.CaptureLog
   8   import Pleroma.Factory
   9   import Mock
  10
  11   alias Pleroma.User
  12
  13   describe "find_resource/1" do
  14     test "returns user" do
  15       user = insert(:user)
  16       query = from(t in User, where: t.id == ^user.id)
  17       assert Repo.find_resource(query) == {:ok, user}
  18     end
  19
  20     test "returns not_found" do
  21       query = from(t in User, where: t.id == ^"9gBuXNpD2NyDmmxxdw")
  22       assert Repo.find_resource(query) == {:error, :not_found}
  23     end
  24   end
  25
  26   describe "get_assoc/2" do
  27     test "get assoc from preloaded data" do
  28       user = %User{name: "Agent Smith"}
  29       token = %Pleroma.Web.OAuth.Token{insert(:oauth_token) | user: user}
  30       assert Repo.get_assoc(token, :user) == {:ok, user}
  31     end
  32
  33     test "get one-to-one assoc from repo" do
  34       user = insert(:user, name: "Jimi Hendrix")
  35       token = refresh_record(insert(:oauth_token, user: user))
  36
  37       assert Repo.get_assoc(token, :user) == {:ok, user}
  38     end
  39
  40     test "get one-to-many assoc from repo" do
  41       user = insert(:user)
  42       notification = refresh_record(insert(:notification, user: user))
  43
  44       assert Repo.get_assoc(user, :notifications) == {:ok, [notification]}
  45     end
  46
  47     test "return error if has not assoc " do
  48       token = insert(:oauth_token, user: nil)
  49       assert Repo.get_assoc(token, :user) == {:error, :not_found}
  50     end
  51   end
  52
  53   describe "check_migrations_applied!" do
  54     setup_with_mocks([
  55       {Ecto.Migrator, [],
  56        [
  57          with_repo: fn repo, fun -> passthrough([repo, fun]) end,
  58          migrations: fn Pleroma.Repo ->
  59            [
  60              {:up, 20_191_128_153_944, "fix_missing_following_count"},
  61              {:up, 20_191_203_043_610, "create_report_notes"},
  62              {:down, 20_191_220_174_645, "add_scopes_to_pleroma_feo_auth_records"}
  63            ]
  64          end
  65        ]}
  66     ]) do
  67       :ok
  68     end
  69
  70     test "raises if it detects unapplied migrations" do
  71       assert_raise Pleroma.Repo.UnappliedMigrationsError, fn ->
  72         capture_log(&Repo.check_migrations_applied!/0)
  73       end
  74     end
  75
  76     test "doesn't do anything if disabled" do
  77       disable_migration_check =
  78         Pleroma.Config.get([:i_am_aware_this_may_cause_data_loss, :disable_migration_check])
  79
  80       Pleroma.Config.put([:i_am_aware_this_may_cause_data_loss, :disable_migration_check], true)
  81
  82       on_exit(fn ->
  83         Pleroma.Config.put(
  84           [:i_am_aware_this_may_cause_data_loss, :disable_migration_check],
  85           disable_migration_check
  86         )
  87       end)
  88
  89       assert :ok == Repo.check_migrations_applied!()
  90     end
  91   end
  92 end