X-Git-Url: http://git.squeep.com/?a=blobdiff_plain;f=test%2Fsupport%2Fbuilders%2Fuser_builder.ex;h=f58e1b0ad370df0422fd623657b78334bb992664;hb=6c06529cd4549fca3ba4b1de72838240463f8306;hp=0028d42e0cadff4477dd7a449be3d742941216da;hpb=9a8850eb9ec9eba9350fcb6aadf70fd966623c94;p=akkoma

diff --git a/test/support/builders/user_builder.ex b/test/support/builders/user_builder.ex
index 0028d42e0..f58e1b0ad 100644
--- a/test/support/builders/user_builder.ex
+++ b/test/support/builders/user_builder.ex
@@ -1,8 +1,9 @@
 defmodule Pleroma.Builders.UserBuilder do
-  alias Pleroma.{User, Repo}
+  alias Pleroma.Repo
+  alias Pleroma.User
 
-  def build do
-    %User{
+  def build(data \\ %{}) do
+    user = %User{
       email: "test@example.org",
       name: "Test Name",
       nickname: "testname",
@@ -10,9 +11,13 @@ defmodule Pleroma.Builders.UserBuilder do
       bio: "A tester.",
       ap_id: "some id"
     }
+
+    Map.merge(user, data)
   end
 
-  def insert do
-    Repo.insert(build())
+  def insert(data \\ %{}) do
+    {:ok, user} = Repo.insert(build(data))
+    User.invalidate_cache(user)
+    {:ok, user}
   end
 end