# 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 Pleroma.Factory 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 end