summaryrefslogtreecommitdiff
path: root/test/repo_test.exs
diff options
context:
space:
mode:
authorIvan Tashkinov <ivantashkinov@gmail.com>2020-09-17 17:14:20 +0300
committerIvan Tashkinov <ivantashkinov@gmail.com>2020-09-17 17:14:20 +0300
commitd9fb5bc08ad67b55d0cd25c1a0d7d3a740758427 (patch)
tree6303695147d01ab5e71e556f2b8f7725536e475f /test/repo_test.exs
parent7cdbd91d83c02a79c22783ca489ef82e82b31a51 (diff)
parentcd93aa2aed6fda516d4fc2aabb44c5e178a6543d (diff)
downloadpleroma-d9fb5bc08ad67b55d0cd25c1a0d7d3a740758427.tar.gz
pleroma-d9fb5bc08ad67b55d0cd25c1a0d7d3a740758427.zip
Merge remote-tracking branch 'remotes/origin/develop' into media-preview-proxy-nostream
Diffstat (limited to 'test/repo_test.exs')
-rw-r--r--test/repo_test.exs32
1 files changed, 31 insertions, 1 deletions
diff --git a/test/repo_test.exs b/test/repo_test.exs
index 92e827c95..155791be2 100644
--- a/test/repo_test.exs
+++ b/test/repo_test.exs
@@ -37,7 +37,9 @@ defmodule Pleroma.RepoTest do
test "get one-to-many assoc from repo" do
user = insert(:user)
- notification = refresh_record(insert(:notification, user: user))
+
+ notification =
+ refresh_record(insert(:notification, user: user, activity: insert(:note_activity)))
assert Repo.get_assoc(user, :notifications) == {:ok, [notification]}
end
@@ -47,4 +49,32 @@ defmodule Pleroma.RepoTest do
assert Repo.get_assoc(token, :user) == {:error, :not_found}
end
end
+
+ describe "chunk_stream/3" do
+ test "fetch records one-by-one" do
+ users = insert_list(50, :user)
+
+ {fetch_users, 50} =
+ from(t in User)
+ |> Repo.chunk_stream(5)
+ |> Enum.reduce({[], 0}, fn %User{} = user, {acc, count} ->
+ {acc ++ [user], count + 1}
+ end)
+
+ assert users == fetch_users
+ end
+
+ test "fetch records in bulk" do
+ users = insert_list(50, :user)
+
+ {fetch_users, 10} =
+ from(t in User)
+ |> Repo.chunk_stream(5, :batches)
+ |> Enum.reduce({[], 0}, fn users, {acc, count} ->
+ {acc ++ users, count + 1}
+ end)
+
+ assert users == fetch_users
+ end
+ end
end