git.squeep.com Git - akkoma/blob - test/web/twitter_api/twitter_api_test.exs

   1 defmodule Pleroma.Web.TwitterAPI.TwitterAPITest do
   2   use Pleroma.DataCase
   3   alias Pleroma.Builders.{UserBuilder, ActivityBuilder}
   4   alias Pleroma.Web.TwitterAPI.TwitterAPI
   5   alias Pleroma.{Activity, User, Object, Repo}
   6   alias Pleroma.Web.TwitterAPI.Representers.ActivityRepresenter
   7   alias Pleroma.Web.ActivityPub.ActivityPub
   8
   9   import Pleroma.Factory
  10
  11   test "create a status" do
  12     user = UserBuilder.build(%{ap_id: "142344"})
  13     _mentioned_user = UserBuilder.insert(%{nickname: "shp", ap_id: "shp"})
  14
  15     object_data = %{
  16       "type" => "Image",
  17       "url" => [
  18         %{
  19           "type" => "Link",
  20           "mediaType" => "image/jpg",
  21           "href" => "http://example.org/image.jpg"
  22         }
  23       ],
  24       "uuid" => 1
  25     }
  26
  27     object = Repo.insert!(%Object{data: object_data})
  28
  29     input = %{
  30       "status" => "Hello again, @shp.<script></script>",
  31       "media_ids" => [object.id]
  32     }
  33
  34     { :ok, activity = %Activity{} } = TwitterAPI.create_status(user, input)
  35
  36     assert get_in(activity.data, ["object", "content"]) == "Hello again, <a href='shp'>@shp</a>."
  37     assert get_in(activity.data, ["object", "type"]) == "Note"
  38     assert get_in(activity.data, ["object", "actor"]) == user.ap_id
  39     assert get_in(activity.data, ["actor"]) == user.ap_id
  40     assert Enum.member?(get_in(activity.data, ["to"]), User.ap_followers(user))
  41     assert Enum.member?(get_in(activity.data, ["to"]), "https://www.w3.org/ns/activitystreams#Public")
  42     assert Enum.member?(get_in(activity.data, ["to"]), "shp")
  43
  44     # Add a context + 'statusnet_conversation_id'
  45     assert is_binary(get_in(activity.data, ["context"]))
  46     assert is_binary(get_in(activity.data, ["object", "context"]))
  47     assert get_in(activity.data, ["object", "statusnetConversationId"]) == activity.id
  48     assert get_in(activity.data, ["statusnetConversationId"]) == activity.id
  49
  50     assert is_list(activity.data["object"]["attachment"])
  51   end
  52
  53   test "create a status that is a reply" do
  54     user = UserBuilder.build(%{ap_id: "some_cool_id"})
  55     input = %{
  56       "status" => "Hello again."
  57     }
  58
  59     { :ok, activity = %Activity{} } = TwitterAPI.create_status(user, input)
  60
  61     input = %{
  62       "status" => "Here's your (you).",
  63       "in_reply_to_status_id" => activity.id
  64     }
  65
  66     { :ok, reply = %Activity{} } = TwitterAPI.create_status(user, input)
  67
  68     assert get_in(reply.data, ["context"]) == get_in(activity.data, ["context"])
  69     assert get_in(reply.data, ["object", "context"]) == get_in(activity.data, ["object", "context"])
  70     assert get_in(reply.data, ["statusnetConversationId"]) == get_in(activity.data, ["statusnetConversationId"])
  71     assert get_in(reply.data, ["object", "statusnetConversationId"]) == get_in(activity.data, ["object", "statusnetConversationId"])
  72     assert get_in(reply.data, ["object", "inReplyTo"]) == get_in(activity.data, ["object", "id"])
  73     assert get_in(reply.data, ["object", "inReplyToStatusId"]) == activity.id
  74     assert Enum.member?(get_in(reply.data, ["to"]), "some_cool_id")
  75   end
  76
  77   test "fetch public statuses" do
  78     %{ public: activity, user: user } = ActivityBuilder.public_and_non_public
  79     {:ok, follower } = UserBuilder.insert(%{name: "dude", ap_id: "idididid", following: [User.ap_followers(user)]})
  80
  81     statuses = TwitterAPI.fetch_public_statuses(follower)
  82
  83     assert length(statuses) == 1
  84     assert Enum.at(statuses, 0) == ActivityRepresenter.to_map(activity, %{user: user, for: follower})
  85   end
  86
  87   test "fetch friends' statuses" do
  88     ActivityBuilder.public_and_non_public
  89
  90     {:ok, activity} = ActivityBuilder.insert(%{"to" => ["someguy/followers"]})
  91     {:ok, direct_activity} = ActivityBuilder.insert(%{"to" => ["some other id"]})
  92     {:ok, user} = UserBuilder.insert(%{ap_id: "some other id", following: ["someguy/followers"]})
  93
  94     statuses = TwitterAPI.fetch_friend_statuses(user)
  95
  96     activity_user = Repo.get_by(User, ap_id: activity.data["actor"])
  97
  98     assert length(statuses) == 2
  99     assert Enum.at(statuses, 0) == ActivityRepresenter.to_map(activity, %{user: activity_user})
 100     assert Enum.at(statuses, 1) == ActivityRepresenter.to_map(direct_activity, %{user: activity_user, mentioned: [user]})
 101   end
 102
 103   test "fetch a single status" do
 104     {:ok, activity} = ActivityBuilder.insert()
 105     {:ok, user} = UserBuilder.insert()
 106     actor = Repo.get_by!(User, ap_id: activity.data["actor"])
 107
 108     status = TwitterAPI.fetch_status(user, activity.id)
 109
 110     assert status == ActivityRepresenter.to_map(activity, %{for: user, user: actor})
 111   end
 112
 113   test "Follow another user" do
 114     { :ok, user } = UserBuilder.insert
 115     { :ok, following } = UserBuilder.insert(%{nickname: "guy"})
 116
 117     {:ok, user, following, activity } = TwitterAPI.follow(user, following.id)
 118
 119     user = Repo.get(User, user.id)
 120     follow = Repo.get(Activity, activity.id)
 121
 122     assert user.following == [User.ap_followers(following)]
 123     assert follow == activity
 124   end
 125
 126   test "Unfollow another user" do
 127     { :ok, following } = UserBuilder.insert(%{nickname: "guy"})
 128     { :ok, user } = UserBuilder.insert(%{following: [User.ap_followers(following)]})
 129
 130     {:ok, user, _following } = TwitterAPI.unfollow(user, following.id)
 131
 132     user = Repo.get(User, user.id)
 133
 134     assert user.following == []
 135   end
 136
 137   test "fetch statuses in a context using the conversation id" do
 138     {:ok, user} = UserBuilder.insert()
 139     {:ok, activity} = ActivityBuilder.insert(%{"statusnetConversationId" => 1, "context" => "2hu"})
 140     {:ok, activity_two} = ActivityBuilder.insert(%{"statusnetConversationId" => 1,"context" => "2hu"})
 141     {:ok, _activity_three} = ActivityBuilder.insert(%{"context" => "3hu"})
 142
 143     statuses = TwitterAPI.fetch_conversation(user, 1)
 144
 145     assert length(statuses) == 2
 146     assert Enum.at(statuses, 0)["id"] == activity.id
 147     assert Enum.at(statuses, 1)["id"] == activity_two.id
 148   end
 149
 150   test "upload a file" do
 151     file = %Plug.Upload{content_type: "image/jpg", path: Path.absname("test/fixtures/image.jpg"), filename: "an_image.jpg"}
 152
 153     response = TwitterAPI.upload(file)
 154
 155     assert is_binary(response)
 156   end
 157
 158   test "it can parse mentions and return the relevant users" do
 159     text = "@gsimg According to @archaeme , that is @daggsy."
 160
 161     {:ok, gsimg} = UserBuilder.insert(%{nickname: "gsimg"})
 162     {:ok, archaeme} = UserBuilder.insert(%{nickname: "archaeme"})
 163
 164     expected_result = [
 165       {"@gsimg", gsimg},
 166       {"@archaeme", archaeme}
 167     ]
 168
 169     assert TwitterAPI.parse_mentions(text) == expected_result
 170   end
 171
 172   test "it adds user links to an existing text" do
 173     text = "@gsimg According to @archaeme , that is @daggsy."
 174
 175     {:ok, _gsimg} = UserBuilder.insert(%{nickname: "gsimg", ap_id: "first_link" })
 176     {:ok, _archaeme} = UserBuilder.insert(%{nickname: "archaeme", ap_id: "second_link"})
 177
 178     mentions = TwitterAPI.parse_mentions(text)
 179     expected_text = "<a href='first_link'>@gsimg</a> According to <a href='second_link'>@archaeme</a> , that is @daggsy."
 180
 181     assert TwitterAPI.add_user_links(text, mentions) == expected_text
 182   end
 183
 184   test "it favorites a status, returns the updated status" do
 185     user = insert(:user)
 186     note_activity = insert(:note_activity)
 187     activity_user = Repo.get_by!(User, ap_id: note_activity.data["actor"])
 188
 189     {:ok, status} = TwitterAPI.favorite(user, note_activity)
 190     updated_activity = Activity.get_by_ap_id(note_activity.data["id"])
 191
 192     assert status == ActivityRepresenter.to_map(updated_activity, %{user: activity_user, for: user})
 193   end
 194
 195   test "it unfavorites a status, returns the updated status" do
 196     user = insert(:user)
 197     note_activity = insert(:note_activity)
 198     activity_user = Repo.get_by!(User, ap_id: note_activity.data["actor"])
 199     object = Object.get_by_ap_id(note_activity.data["object"]["id"])
 200
 201     {:ok, like_activity, object } = ActivityPub.like(user, object)
 202     updated_activity = Activity.get_by_ap_id(note_activity.data["id"])
 203     assert ActivityRepresenter.to_map(updated_activity, %{user: activity_user, for: user})["fave_num"] == 1
 204
 205     {:ok, status} = TwitterAPI.unfavorite(user, note_activity)
 206     updated_activity = Activity.get_by_ap_id(note_activity.data["id"])
 207
 208     assert status["fave_num"] == 0
 209   end
 210
 211   setup do
 212     Supervisor.terminate_child(Pleroma.Supervisor, ConCache)
 213     Supervisor.restart_child(Pleroma.Supervisor, ConCache)
 214     :ok
 215   end
 216 end