Fix tests.

[akkoma] / lib / pleroma / user.ex

diff --git a/lib/pleroma/user.ex b/lib/pleroma/user.ex
index a503a5b3f1dd073354a0abb0505a451298212bad..e92b85f52c2baea89ecc9c96f4bf8f4064e07d5a 100644 (file)
--- a/lib/pleroma/user.ex
+++ b/lib/pleroma/user.ex
@@ -417,7 +417,7 @@ defmodule Pleroma.User do
         _ ->
           case OStatus.make_user(ap_id) do
             {:ok, user} -> user
-            _ -> {:error, "Could not fetch by ap id"}
+            _ -> {:error, "Could not fetch by AP id"}
           end
       end
     end
@@ -458,4 +458,12 @@ defmodule Pleroma.User do
 
   def ap_enabled?(%User{info: info}), do: info["ap_enabled"]
   def ap_enabled?(_), do: false
+
+  def get_or_fetch(uri_or_nickname) do
+    if String.starts_with?(uri_or_nickname, "http") do
+      get_or_fetch_by_ap_id(uri_or_nickname)
+    else
+      get_or_fetch_by_nickname(uri_or_nickname)
+    end
+  end
 end