X-Git-Url: http://git.squeep.com/?a=blobdiff_plain;f=test%2Frepo_test.exs;h=155791be22fd1fb2905ca0c82d97d9d1f14d4f01;hb=9c672ecbb5d4477cd16d2139a2cb66d3923ac5c8;hp=72c0b5071ccb99dfecfbef9b4c18f4e3111284a8;hpb=f879d07fa1a046b5aa33de8e445b1ca803fa1d04;p=akkoma

diff --git a/test/repo_test.exs b/test/repo_test.exs
index 72c0b5071..155791be2 100644
--- a/test/repo_test.exs
+++ b/test/repo_test.exs
@@ -49,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