2023-06-03 00:18:02 +12:00
|
|
|
defmodule Ash.Test.Actions.StreamTest do
|
2023-04-27 09:12:21 +12:00
|
|
|
@moduledoc false
|
|
|
|
use ExUnit.Case, async: true
|
|
|
|
|
|
|
|
defmodule Post do
|
|
|
|
@moduledoc false
|
|
|
|
use Ash.Resource, data_layer: Ash.DataLayer.Ets
|
|
|
|
|
|
|
|
ets do
|
|
|
|
private? true
|
|
|
|
end
|
|
|
|
|
|
|
|
actions do
|
|
|
|
defaults [:create, :update, :destroy]
|
|
|
|
|
|
|
|
read :read do
|
|
|
|
primary? true
|
|
|
|
pagination keyset?: true
|
|
|
|
end
|
|
|
|
|
|
|
|
read :read_with_no_pagination
|
|
|
|
end
|
|
|
|
|
|
|
|
attributes do
|
|
|
|
uuid_primary_key :id
|
|
|
|
attribute :title, :string, allow_nil?: false
|
|
|
|
|
|
|
|
timestamps()
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
defmodule Registry do
|
|
|
|
@moduledoc false
|
|
|
|
use Ash.Registry
|
|
|
|
|
|
|
|
entries do
|
|
|
|
entry Post
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
defmodule Api do
|
|
|
|
@moduledoc false
|
|
|
|
use Ash.Api
|
|
|
|
|
|
|
|
resources do
|
|
|
|
registry Registry
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
test "records can be streamed" do
|
|
|
|
1..10
|
2023-05-03 05:17:52 +12:00
|
|
|
|> Stream.map(&%{title: "title#{&1}"})
|
|
|
|
|> Api.bulk_create!(Post, :create)
|
2023-04-27 09:12:21 +12:00
|
|
|
|
|
|
|
count =
|
|
|
|
Post
|
|
|
|
|> Api.stream!(batch_size: 5)
|
|
|
|
|> Enum.count()
|
|
|
|
|
|
|
|
assert count == 10
|
|
|
|
end
|
|
|
|
end
|