2021-05-10 09:24:48 +12:00
|
|
|
defmodule Ash.Test.Actions.LoadTest do
|
2020-06-02 17:47:25 +12:00
|
|
|
@moduledoc false
|
2021-01-29 08:47:59 +13:00
|
|
|
use ExUnit.Case, async: false
|
2019-12-12 11:45:59 +13:00
|
|
|
|
2021-04-19 11:41:49 +12:00
|
|
|
import Ash.Changeset
|
2020-10-08 18:22:55 +13:00
|
|
|
require Ash.Query
|
|
|
|
|
2019-12-12 11:45:59 +13:00
|
|
|
defmodule Author do
|
2020-06-02 17:47:25 +12:00
|
|
|
@moduledoc false
|
2020-06-22 15:26:47 +12:00
|
|
|
use Ash.Resource,
|
|
|
|
data_layer: Ash.DataLayer.Ets,
|
|
|
|
authorizers: [
|
|
|
|
Ash.Test.Authorizer
|
|
|
|
]
|
2020-06-14 18:39:11 +12:00
|
|
|
|
|
|
|
ets do
|
|
|
|
private?(true)
|
|
|
|
end
|
2019-12-12 11:45:59 +13:00
|
|
|
|
|
|
|
actions do
|
2022-04-04 17:48:37 +12:00
|
|
|
defaults [:create, :read, :update, :destroy]
|
2019-12-12 11:45:59 +13:00
|
|
|
end
|
|
|
|
|
|
|
|
attributes do
|
2021-01-13 09:40:55 +13:00
|
|
|
uuid_primary_key :id
|
2019-12-12 11:45:59 +13:00
|
|
|
attribute :name, :string
|
|
|
|
end
|
|
|
|
|
|
|
|
relationships do
|
2021-05-10 09:24:48 +12:00
|
|
|
has_many :posts, Ash.Test.Actions.LoadTest.Post, destination_field: :author_id
|
2021-05-07 16:28:51 +12:00
|
|
|
|
2021-05-10 09:24:48 +12:00
|
|
|
has_one :latest_post, Ash.Test.Actions.LoadTest.Post,
|
2021-05-07 16:28:51 +12:00
|
|
|
destination_field: :author_id,
|
2021-05-07 17:07:01 +12:00
|
|
|
sort: [inserted_at: :desc]
|
2019-12-12 11:45:59 +13:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2022-02-21 16:46:39 +13:00
|
|
|
defmodule PostsInSameCategory do
|
|
|
|
use Ash.Resource.ManualRelationship
|
|
|
|
|
|
|
|
def load(posts, _, %{query: destination_query, api: api}) do
|
|
|
|
categories = Enum.map(posts, & &1.category)
|
|
|
|
|
|
|
|
other_posts =
|
|
|
|
destination_query
|
|
|
|
|> Ash.Query.filter(category in ^categories)
|
|
|
|
|> api.read!()
|
|
|
|
|> Enum.group_by(& &1.category)
|
|
|
|
|
|
|
|
{:ok,
|
|
|
|
Map.new(posts, fn post ->
|
|
|
|
related_posts =
|
|
|
|
other_posts
|
|
|
|
|> Map.get(post.category, [])
|
|
|
|
|> Enum.reject(&(&1.id == post.id))
|
|
|
|
|
|
|
|
{Map.take(post, [:id]), related_posts}
|
|
|
|
end)}
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2019-12-12 11:45:59 +13:00
|
|
|
defmodule Post do
|
2020-06-02 17:47:25 +12:00
|
|
|
@moduledoc false
|
2020-06-14 18:39:11 +12:00
|
|
|
use Ash.Resource, data_layer: Ash.DataLayer.Ets
|
|
|
|
|
|
|
|
ets do
|
|
|
|
private?(true)
|
|
|
|
end
|
2019-12-12 11:45:59 +13:00
|
|
|
|
|
|
|
actions do
|
2022-04-04 17:48:37 +12:00
|
|
|
defaults [:create, :read, :update, :destroy]
|
2019-12-12 11:45:59 +13:00
|
|
|
end
|
|
|
|
|
|
|
|
attributes do
|
2021-01-13 09:40:55 +13:00
|
|
|
uuid_primary_key :id
|
2019-12-12 11:45:59 +13:00
|
|
|
attribute :title, :string
|
|
|
|
attribute :contents, :string
|
2022-02-21 16:46:39 +13:00
|
|
|
attribute :category, :string
|
2021-05-07 16:28:51 +12:00
|
|
|
timestamps()
|
2019-12-12 11:45:59 +13:00
|
|
|
end
|
|
|
|
|
|
|
|
relationships do
|
2020-06-22 16:34:44 +12:00
|
|
|
belongs_to :author, Author
|
2020-06-22 15:26:47 +12:00
|
|
|
|
2022-02-21 16:46:39 +13:00
|
|
|
has_many :posts_in_same_category, __MODULE__ do
|
|
|
|
manual PostsInSameCategory
|
|
|
|
end
|
|
|
|
|
2022-04-04 17:48:37 +12:00
|
|
|
has_many :ratings, Ash.Test.Actions.LoadTest.Rating do
|
|
|
|
api Ash.Test.Actions.LoadTest.Api2
|
|
|
|
end
|
|
|
|
|
2021-05-10 09:24:48 +12:00
|
|
|
many_to_many :categories, Ash.Test.Actions.LoadTest.Category,
|
|
|
|
through: Ash.Test.Actions.LoadTest.PostCategory,
|
2020-06-22 15:26:47 +12:00
|
|
|
destination_field_on_join_table: :category_id,
|
|
|
|
source_field_on_join_table: :post_id
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
defmodule PostCategory do
|
|
|
|
@moduledoc false
|
|
|
|
use Ash.Resource, data_layer: Ash.DataLayer.Ets
|
|
|
|
|
|
|
|
ets do
|
|
|
|
private?(true)
|
|
|
|
end
|
|
|
|
|
|
|
|
actions do
|
2022-04-04 17:48:37 +12:00
|
|
|
defaults [:create, :read, :update, :destroy]
|
2020-06-22 15:26:47 +12:00
|
|
|
end
|
|
|
|
|
|
|
|
relationships do
|
2021-01-13 09:40:55 +13:00
|
|
|
belongs_to :post, Post, primary_key?: true, required?: true
|
|
|
|
|
2021-05-10 09:24:48 +12:00
|
|
|
belongs_to :category, Ash.Test.Actions.LoadTest.Category,
|
2021-01-13 09:40:55 +13:00
|
|
|
primary_key?: true,
|
|
|
|
required?: true
|
2020-06-22 15:26:47 +12:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
defmodule Category do
|
2021-04-19 11:41:49 +12:00
|
|
|
@moduledoc false
|
2020-06-22 15:26:47 +12:00
|
|
|
use Ash.Resource, data_layer: Ash.DataLayer.Ets
|
|
|
|
|
|
|
|
ets do
|
|
|
|
private?(true)
|
|
|
|
end
|
|
|
|
|
|
|
|
actions do
|
2022-04-04 17:48:37 +12:00
|
|
|
defaults [:create, :read, :update, :destroy]
|
2020-06-22 15:26:47 +12:00
|
|
|
end
|
|
|
|
|
|
|
|
attributes do
|
2021-01-13 09:40:55 +13:00
|
|
|
uuid_primary_key :id
|
2020-06-22 15:26:47 +12:00
|
|
|
attribute :name, :string
|
|
|
|
end
|
|
|
|
|
|
|
|
relationships do
|
|
|
|
many_to_many :posts, Post,
|
|
|
|
through: PostCategory,
|
|
|
|
destination_field_on_join_table: :post_id,
|
|
|
|
source_field_on_join_table: :category_id
|
2019-12-12 11:45:59 +13:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2022-04-04 17:48:37 +12:00
|
|
|
defmodule Rating do
|
|
|
|
use Ash.Resource,
|
|
|
|
data_layer: Ash.DataLayer.Ets
|
|
|
|
|
2022-06-21 09:01:28 +12:00
|
|
|
ets do
|
|
|
|
private? true
|
|
|
|
end
|
|
|
|
|
2022-04-04 17:48:37 +12:00
|
|
|
attributes do
|
|
|
|
uuid_primary_key :id
|
|
|
|
attribute :rating, :integer
|
|
|
|
end
|
|
|
|
|
|
|
|
actions do
|
|
|
|
defaults [:create, :read, :update, :destroy]
|
|
|
|
end
|
|
|
|
|
|
|
|
relationships do
|
|
|
|
belongs_to :post, Post do
|
|
|
|
api Ash.Test.Actions.LoadTest.Category
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2021-10-07 19:41:02 +13:00
|
|
|
defmodule Registry do
|
|
|
|
@moduledoc false
|
|
|
|
use Ash.Registry
|
|
|
|
|
|
|
|
entries do
|
|
|
|
entry(Author)
|
|
|
|
entry(Post)
|
|
|
|
entry(Category)
|
|
|
|
entry(PostCategory)
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2022-04-04 17:48:37 +12:00
|
|
|
defmodule Registry2 do
|
|
|
|
@moduledoc false
|
|
|
|
use Ash.Registry
|
|
|
|
|
|
|
|
entries do
|
|
|
|
entry(Rating)
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2019-12-12 11:45:59 +13:00
|
|
|
defmodule Api do
|
2020-06-02 17:47:25 +12:00
|
|
|
@moduledoc false
|
2019-12-12 11:45:59 +13:00
|
|
|
use Ash.Api
|
|
|
|
|
2020-06-14 18:39:11 +12:00
|
|
|
resources do
|
2021-10-07 19:41:02 +13:00
|
|
|
registry Registry
|
2020-06-14 18:39:11 +12:00
|
|
|
end
|
2019-12-12 11:45:59 +13:00
|
|
|
end
|
|
|
|
|
2022-04-04 17:48:37 +12:00
|
|
|
defmodule Api2 do
|
|
|
|
@moduledoc false
|
|
|
|
use Ash.Api
|
|
|
|
|
|
|
|
resources do
|
|
|
|
registry Registry2
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2020-06-22 15:26:47 +12:00
|
|
|
setup do
|
2021-01-29 08:47:59 +13:00
|
|
|
start_supervised(
|
|
|
|
{Ash.Test.Authorizer,
|
|
|
|
strict_check: :authorized,
|
|
|
|
check: {:error, Ash.Error.Forbidden.exception([])},
|
|
|
|
strict_check_context: [:query]}
|
|
|
|
)
|
2020-06-22 15:26:47 +12:00
|
|
|
|
|
|
|
:ok
|
|
|
|
end
|
|
|
|
|
2021-05-10 09:24:48 +12:00
|
|
|
describe "loads" do
|
2022-02-21 16:46:39 +13:00
|
|
|
test "it allows loading manual relationships" do
|
|
|
|
post1 =
|
|
|
|
Post
|
|
|
|
|> new(%{title: "post1", category: "foo"})
|
|
|
|
|> Api.create!()
|
|
|
|
|
|
|
|
Post
|
|
|
|
|> new(%{title: "post2", category: "bar"})
|
|
|
|
|> Api.create!()
|
|
|
|
|
|
|
|
post3 =
|
|
|
|
Post
|
|
|
|
|> new(%{title: "post2", category: "foo"})
|
|
|
|
|> Api.create!()
|
|
|
|
|
|
|
|
post3_id = post3.id
|
|
|
|
|
|
|
|
assert [%{id: ^post3_id}] =
|
|
|
|
post1
|
|
|
|
|> Api.load!(:posts_in_same_category)
|
|
|
|
|> Map.get(:posts_in_same_category)
|
|
|
|
end
|
|
|
|
|
2021-05-10 09:24:48 +12:00
|
|
|
test "it allows loading related data" do
|
2020-07-12 18:25:53 +12:00
|
|
|
author =
|
|
|
|
Author
|
2020-07-15 17:38:42 +12:00
|
|
|
|> new(%{name: "zerg"})
|
2020-07-12 18:25:53 +12:00
|
|
|
|> Api.create!()
|
2019-12-12 11:45:59 +13:00
|
|
|
|
2019-12-20 18:30:27 +13:00
|
|
|
post1 =
|
2020-07-12 18:25:53 +12:00
|
|
|
Post
|
2020-07-15 17:38:42 +12:00
|
|
|
|> new(%{title: "post1"})
|
2020-07-12 18:25:53 +12:00
|
|
|
|> replace_relationship(:author, author)
|
|
|
|
|> Api.create!()
|
2019-12-12 11:45:59 +13:00
|
|
|
|
2019-12-20 18:30:27 +13:00
|
|
|
post2 =
|
2020-07-12 18:25:53 +12:00
|
|
|
Post
|
2020-07-15 17:38:42 +12:00
|
|
|
|> new(%{title: "post2"})
|
2020-07-12 18:25:53 +12:00
|
|
|
|> replace_relationship(:author, author)
|
|
|
|
|> Api.create!()
|
2019-12-12 11:45:59 +13:00
|
|
|
|
2020-05-14 03:54:44 +12:00
|
|
|
[author] =
|
|
|
|
Author
|
2020-08-26 16:16:08 +12:00
|
|
|
|> Ash.Query.load(posts: [:author])
|
2020-10-08 18:22:55 +13:00
|
|
|
|> Ash.Query.filter(posts.id == ^post1.id)
|
2020-06-22 15:26:47 +12:00
|
|
|
|> Api.read!(authorize?: true)
|
2019-12-12 11:45:59 +13:00
|
|
|
|
2019-12-20 18:30:27 +13:00
|
|
|
assert Enum.sort(Enum.map(author.posts, &Map.get(&1, :id))) ==
|
|
|
|
Enum.sort([post1.id, post2.id])
|
2019-12-20 17:19:34 +13:00
|
|
|
|
2019-12-20 18:30:27 +13:00
|
|
|
for post <- author.posts do
|
|
|
|
assert post.author.id == author.id
|
|
|
|
end
|
|
|
|
end
|
2020-06-22 15:26:47 +12:00
|
|
|
|
2022-05-19 05:11:30 +12:00
|
|
|
test "loading something already loaded still loads it unless lazy?: true" do
|
|
|
|
author =
|
|
|
|
Author
|
|
|
|
|> new(%{name: "zerg"})
|
|
|
|
|> Api.create!()
|
|
|
|
|
|
|
|
post1 =
|
|
|
|
Post
|
|
|
|
|> new(%{title: "post1"})
|
|
|
|
|> replace_relationship(:author, author)
|
|
|
|
|> Api.create!()
|
|
|
|
|
|
|
|
post2 =
|
|
|
|
Post
|
|
|
|
|> new(%{title: "post2"})
|
|
|
|
|> replace_relationship(:author, author)
|
|
|
|
|> Api.create!()
|
|
|
|
|
|
|
|
[author] =
|
|
|
|
Author
|
|
|
|
|> Ash.Query.load(posts: [:author])
|
|
|
|
|> Ash.Query.filter(posts.id == ^post1.id)
|
|
|
|
|> Api.read!(authorize?: true)
|
|
|
|
|
|
|
|
assert Enum.sort(Enum.map(author.posts, &Map.get(&1, :id))) ==
|
|
|
|
Enum.sort([post1.id, post2.id])
|
|
|
|
|
|
|
|
for post <- author.posts do
|
|
|
|
assert post.author.id == author.id
|
|
|
|
end
|
|
|
|
|
|
|
|
post3 =
|
|
|
|
Post
|
|
|
|
|> new(%{title: "post3"})
|
|
|
|
|> replace_relationship(:author, author)
|
|
|
|
|> Api.create!()
|
|
|
|
|
|
|
|
author =
|
|
|
|
author
|
|
|
|
|> Api.load!([posts: [:author]], authorize?: true)
|
|
|
|
|
|
|
|
assert Enum.sort(Enum.map(author.posts, &Map.get(&1, :id))) ==
|
|
|
|
Enum.sort([post1.id, post2.id, post3.id])
|
|
|
|
|
|
|
|
for post <- author.posts do
|
|
|
|
assert post.author.id == author.id
|
|
|
|
end
|
|
|
|
|
|
|
|
Post
|
|
|
|
|> new(%{title: "post4"})
|
|
|
|
|> replace_relationship(:author, author)
|
|
|
|
|> Api.create!()
|
|
|
|
|
|
|
|
author =
|
|
|
|
author
|
|
|
|
|> Api.load!([posts: [:author]], authorize?: true, lazy?: true)
|
|
|
|
|
|
|
|
assert Enum.sort(Enum.map(author.posts, &Map.get(&1, :id))) ==
|
|
|
|
Enum.sort([post1.id, post2.id, post3.id])
|
|
|
|
|
|
|
|
for post <- author.posts do
|
|
|
|
assert post.author.id == author.id
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2022-04-04 17:48:37 +12:00
|
|
|
test "it allows loading across APIs" do
|
|
|
|
author =
|
|
|
|
Author
|
|
|
|
|> new(%{name: "zerg"})
|
|
|
|
|> Api.create!()
|
|
|
|
|
|
|
|
post =
|
|
|
|
Post
|
|
|
|
|> new(%{title: "post1"})
|
|
|
|
|> replace_relationship(:author, author)
|
|
|
|
|> Api.create!()
|
|
|
|
|
|
|
|
rating =
|
|
|
|
Rating
|
|
|
|
|> new(%{rating: 10})
|
|
|
|
|> replace_relationship(:post, post)
|
|
|
|
|> Api2.create!()
|
|
|
|
|
|
|
|
assert [loaded_rating] =
|
|
|
|
author
|
|
|
|
|> Api.load!(posts: :ratings)
|
|
|
|
|> Map.get(:posts)
|
|
|
|
|> Enum.at(0)
|
|
|
|
|> Map.get(:ratings)
|
|
|
|
|
|
|
|
assert loaded_rating.id == rating.id
|
|
|
|
end
|
|
|
|
|
2021-05-10 09:24:48 +12:00
|
|
|
test "it allows loading many to many relationships" do
|
2020-07-12 18:25:53 +12:00
|
|
|
category1 =
|
|
|
|
Category
|
2020-07-15 17:38:42 +12:00
|
|
|
|> new(%{name: "lame"})
|
2020-07-12 18:25:53 +12:00
|
|
|
|> Api.create!()
|
|
|
|
|
|
|
|
category2 =
|
|
|
|
Category
|
2020-07-15 17:38:42 +12:00
|
|
|
|> new(%{name: "cool"})
|
2020-07-12 18:25:53 +12:00
|
|
|
|> Api.create!()
|
2020-06-22 15:26:47 +12:00
|
|
|
|
|
|
|
post =
|
2020-07-12 18:25:53 +12:00
|
|
|
Post
|
2020-07-15 17:38:42 +12:00
|
|
|
|> new(%{title: "post1"})
|
2020-07-12 18:25:53 +12:00
|
|
|
|> replace_relationship(:categories, [category1, category2])
|
|
|
|
|> Api.create!()
|
2020-06-22 15:26:47 +12:00
|
|
|
|
|
|
|
[post] =
|
|
|
|
Post
|
2020-08-26 16:16:08 +12:00
|
|
|
|> Ash.Query.load(:categories)
|
2020-10-08 18:22:55 +13:00
|
|
|
|> Ash.Query.filter(id == ^post.id)
|
2020-06-22 15:26:47 +12:00
|
|
|
|> Api.read!(authorize?: true)
|
|
|
|
|
|
|
|
assert [%{id: id1}, %{id: id2}] = post.categories
|
|
|
|
|
|
|
|
assert Enum.sort([category1.id, category2.id]) == Enum.sort([id1, id2])
|
|
|
|
end
|
|
|
|
|
2021-05-10 09:24:48 +12:00
|
|
|
test "it allows loading nested many to many relationships" do
|
2020-07-12 18:25:53 +12:00
|
|
|
category1 =
|
|
|
|
Category
|
2020-07-15 17:38:42 +12:00
|
|
|
|> new(%{name: "lame"})
|
2020-07-12 18:25:53 +12:00
|
|
|
|> Api.create!()
|
|
|
|
|
|
|
|
category2 =
|
|
|
|
Category
|
2020-07-15 17:38:42 +12:00
|
|
|
|> new(%{name: "cool"})
|
2020-07-12 18:25:53 +12:00
|
|
|
|> Api.create!()
|
2020-06-22 15:26:47 +12:00
|
|
|
|
|
|
|
post =
|
2020-07-12 18:25:53 +12:00
|
|
|
Post
|
2020-07-15 17:38:42 +12:00
|
|
|
|> new(%{title: "post1"})
|
2020-07-12 18:25:53 +12:00
|
|
|
|> replace_relationship(:categories, [category1, category2])
|
|
|
|
|> Api.create!()
|
2020-06-22 15:26:47 +12:00
|
|
|
|
|
|
|
[post] =
|
|
|
|
Post
|
2020-08-26 16:16:08 +12:00
|
|
|
|> Ash.Query.load(categories: :posts)
|
2020-10-08 18:22:55 +13:00
|
|
|
|> Ash.Query.filter(id == ^post.id)
|
2020-06-22 15:26:47 +12:00
|
|
|
|> Api.read!(authorize?: true)
|
|
|
|
|
|
|
|
post_id = post.id
|
|
|
|
|
|
|
|
assert [%{posts: [%{id: ^post_id}]}, %{posts: [%{id: ^post_id}]}] = post.categories
|
|
|
|
end
|
2021-05-07 16:28:51 +12:00
|
|
|
|
|
|
|
test "it loads sorted relationships in the proper order" do
|
|
|
|
author =
|
|
|
|
Author
|
|
|
|
|> new(%{name: "zerg"})
|
|
|
|
|> Api.create!()
|
|
|
|
|
|
|
|
_post1 =
|
|
|
|
Post
|
|
|
|
|> new(%{title: "post1"})
|
|
|
|
|> replace_relationship(:author, author)
|
|
|
|
|> Api.create!()
|
|
|
|
|
|
|
|
:timer.sleep(2)
|
|
|
|
|
|
|
|
post2 =
|
|
|
|
Post
|
|
|
|
|> new(%{title: "post2"})
|
|
|
|
|> replace_relationship(:author, author)
|
|
|
|
|> Api.create!()
|
|
|
|
|
|
|
|
[author] =
|
|
|
|
Author
|
|
|
|
|> Ash.Query.load(:latest_post)
|
|
|
|
|> Api.read!()
|
|
|
|
|
|
|
|
assert author.latest_post.id == post2.id
|
|
|
|
end
|
2019-12-12 11:45:59 +13:00
|
|
|
end
|
|
|
|
end
|