Fork of Pleroma with site-specific changes and feature branches https://git.pleroma.social/pleroma/pleroma
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

85 lines
2.5KB

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