commit: f97c8e43792f76e84c436e75b0c0893d3c16d7d8
parent: 4cac385411b743fce039e0ebe93ce27323e6f231
Author: Roger Braun <roger@rogerbraun.net>
Date: Thu, 13 Apr 2017 15:49:42 +0200
Add utility functions for objects and activities.
Diffstat:
3 files changed, 41 insertions(+), 0 deletions(-)
diff --git a/lib/pleroma/activity.ex b/lib/pleroma/activity.ex
@@ -1,9 +1,21 @@
defmodule Pleroma.Activity do
use Ecto.Schema
+ alias Pleroma.{Repo, Activity}
+ import Ecto.Query
schema "activities" do
field :data, :map
timestamps()
end
+
+ def get_by_ap_id(ap_id) do
+ Repo.one(from activity in Activity,
+ where: fragment("? @> ?", activity.data, ^%{id: ap_id}))
+ end
+
+ def all_by_object_ap_id(ap_id) do
+ Repo.all(from activity in Activity,
+ where: fragment("? @> ?", activity.data, ^%{object: %{id: ap_id}}))
+ end
end
diff --git a/test/activity_test.exs b/test/activity_test.exs
@@ -0,0 +1,18 @@
+defmodule Pleroma.ActivityTest do
+ use Pleroma.DataCase
+ import Pleroma.Factory
+
+ test "returns an activity by it's AP id" do
+ activity = insert(:note_activity)
+ found_activity = Pleroma.Activity.get_by_ap_id(activity.data["id"])
+
+ assert activity == found_activity
+ end
+
+ test "returns activities by it's objects AP ids" do
+ activity = insert(:note_activity)
+ [found_activity] = Pleroma.Activity.all_by_object_ap_id(activity.data["object"]["id"])
+
+ assert activity == found_activity
+ end
+end
diff --git a/test/object_test.exs b/test/object_test.exs
@@ -0,0 +1,11 @@
+defmodule Pleroma.ObjectTest do
+ use Pleroma.DataCase
+ import Pleroma.Factory
+
+ test "returns an object by it's AP id" do
+ object = insert(:note)
+ found_object = Pleroma.Object.get_by_ap_id(object.data["id"])
+
+ assert object == found_object
+ end
+end