logo

pleroma

My custom branche(s) on git.pleroma.social/pleroma/pleroma
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:

Mlib/pleroma/activity.ex12++++++++++++
Atest/activity_test.exs18++++++++++++++++++
Atest/object_test.exs11+++++++++++
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