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, UserRepresenter}
   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>\nThis is on another line.",
  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>.<br>This is on another line."
  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
  52     assert activity.data["object"] == Object.get_by_ap_id(activity.data["object"]["id"]).data
  53   end
  54
  55   test "create a status that is a reply" do
  56     user = UserBuilder.build(%{ap_id: "some_cool_id"})
  57     input = %{
  58       "status" => "Hello again."
  59     }
  60
  61     { :ok, activity = %Activity{} } = TwitterAPI.create_status(user, input)
  62
  63     input = %{
  64       "status" => "Here's your (you).",
  65       "in_reply_to_status_id" => activity.id
  66     }
  67
  68     { :ok, reply = %Activity{} } = TwitterAPI.create_status(user, input)
  69
  70     assert get_in(reply.data, ["context"]) == get_in(activity.data, ["context"])
  71     assert get_in(reply.data, ["object", "context"]) == get_in(activity.data, ["object", "context"])
  72     assert get_in(reply.data, ["statusnetConversationId"]) == get_in(activity.data, ["statusnetConversationId"])
  73     assert get_in(reply.data, ["object", "statusnetConversationId"]) == get_in(activity.data, ["object", "statusnetConversationId"])
  74     assert get_in(reply.data, ["object", "inReplyTo"]) == get_in(activity.data, ["object", "id"])
  75     assert get_in(reply.data, ["object", "inReplyToStatusId"]) == activity.id
  76     assert Enum.member?(get_in(reply.data, ["to"]), "some_cool_id")
  77   end
  78
  79   test "fetch public statuses" do
  80     %{ public: activity, user: user } = ActivityBuilder.public_and_non_public
  81
  82     follower = insert(:user, following: [User.ap_followers(user)])
  83
  84     statuses = TwitterAPI.fetch_public_statuses(follower)
  85
  86     assert length(statuses) == 1
  87     assert Enum.at(statuses, 0) == ActivityRepresenter.to_map(activity, %{user: user, for: follower})
  88   end
  89
  90   test "fetch friends' statuses" do
  91     user = insert(:user, %{following: ["someguy/followers"]})
  92     {:ok, activity} = ActivityBuilder.insert(%{"to" => ["someguy/followers"]})
  93     {:ok, direct_activity} = ActivityBuilder.insert(%{"to" => [user.ap_id]})
  94
  95     statuses = TwitterAPI.fetch_friend_statuses(user)
  96
  97     activity_user = Repo.get_by(User, ap_id: activity.data["actor"])
  98     direct_activity_user = Repo.get_by(User, ap_id: direct_activity.data["actor"])
  99
 100     assert length(statuses) == 2
 101     assert Enum.at(statuses, 0) == ActivityRepresenter.to_map(activity, %{user: activity_user})
 102     assert Enum.at(statuses, 1) == ActivityRepresenter.to_map(direct_activity, %{user: direct_activity_user, mentioned: [user]})
 103   end
 104
 105   test "get a user by params" do
 106     user1_result = {:ok, user1} = UserBuilder.insert(%{ap_id: "some id", email: "test@pleroma"})
 107     {:ok, user2} = UserBuilder.insert(%{ap_id: "some other id", nickname: "testname2", email: "test2@pleroma"})
 108
 109     assert {:error, "You need to specify screen_name or user_id"} == TwitterAPI.get_user(nil, nil)
 110     assert user1_result == TwitterAPI.get_user(nil, %{"user_id" => user1.id})
 111     assert user1_result == TwitterAPI.get_user(nil, %{"screen_name" => user1.nickname})
 112     assert user1_result == TwitterAPI.get_user(user1, nil)
 113     assert user1_result == TwitterAPI.get_user(user2, %{"user_id" => user1.id})
 114     assert user1_result == TwitterAPI.get_user(user2, %{"screen_name" => user1.nickname})
 115     assert {:error, "No user with such screen_name"} == TwitterAPI.get_user(nil, %{"screen_name" => "Satan"})
 116     assert {:error, "No user with such user_id"} == TwitterAPI.get_user(nil, %{"user_id" => 666})
 117   end
 118
 119   test "fetch user's statuses" do
 120     {:ok, user1} = UserBuilder.insert(%{ap_id: "some id", email: "test@pleroma"})
 121     {:ok, user2} = UserBuilder.insert(%{ap_id: "some other id", nickname: "testname2", email: "test2@pleroma"})
 122
 123     {:ok, status1} = ActivityBuilder.insert(%{"id" => 1}, %{user: user1})
 124     {:ok, status2} = ActivityBuilder.insert(%{"id" => 2}, %{user: user2})
 125
 126     user1_statuses = TwitterAPI.fetch_user_statuses(user1, %{"actor_id" => user1.ap_id})
 127
 128     assert length(user1_statuses) == 1
 129     assert Enum.at(user1_statuses, 0) == ActivityRepresenter.to_map(status1, %{user: user1})
 130
 131     user2_statuses = TwitterAPI.fetch_user_statuses(user1, %{"actor_id" => user2.ap_id})
 132
 133     assert length(user2_statuses) == 1
 134     assert Enum.at(user2_statuses, 0) == ActivityRepresenter.to_map(status2, %{user: user2})
 135   end
 136
 137   test "fetch a single status" do
 138     {:ok, activity} = ActivityBuilder.insert()
 139     {:ok, user} = UserBuilder.insert()
 140     actor = Repo.get_by!(User, ap_id: activity.data["actor"])
 141
 142     status = TwitterAPI.fetch_status(user, activity.id)
 143
 144     assert status == ActivityRepresenter.to_map(activity, %{for: user, user: actor})
 145   end
 146
 147   test "Follow another user" do
 148     user = insert(:user)
 149     followed = insert(:user)
 150
 151     { :ok, user, followed, activity } = TwitterAPI.follow(user, followed.id)
 152
 153     user = Repo.get(User, user.id)
 154     follow = Repo.get(Activity, activity.id)
 155
 156     assert user.following == [User.ap_followers(followed)]
 157     assert follow == activity
 158
 159     { :error, msg } = TwitterAPI.follow(user, followed.id)
 160     assert msg == "Could not follow user: #{followed.nickname} is already on your list."
 161   end
 162
 163   test "Unfollow another user" do
 164     followed = insert(:user)
 165     user = insert(:user, %{following: [User.ap_followers(followed)]})
 166
 167     { :ok, user, _followed } = TwitterAPI.unfollow(user, followed.id)
 168
 169     user = Repo.get(User, user.id)
 170
 171     assert user.following == []
 172     { :error, msg } = TwitterAPI.unfollow(user, followed.id)
 173     assert msg == "Not subscribed!"
 174   end
 175
 176   test "fetch statuses in a context using the conversation id" do
 177     {:ok, user} = UserBuilder.insert()
 178     {:ok, activity} = ActivityBuilder.insert(%{"statusnetConversationId" => 1, "context" => "2hu"})
 179     {:ok, activity_two} = ActivityBuilder.insert(%{"statusnetConversationId" => 1,"context" => "2hu"})
 180     {:ok, _activity_three} = ActivityBuilder.insert(%{"context" => "3hu"})
 181
 182     statuses = TwitterAPI.fetch_conversation(user, 1)
 183
 184     assert length(statuses) == 2
 185     assert Enum.at(statuses, 0)["id"] == activity.id
 186     assert Enum.at(statuses, 1)["id"] == activity_two.id
 187   end
 188
 189   test "upload a file" do
 190     file = %Plug.Upload{content_type: "image/jpg", path: Path.absname("test/fixtures/image.jpg"), filename: "an_image.jpg"}
 191
 192     response = TwitterAPI.upload(file)
 193
 194     assert is_binary(response)
 195   end
 196
 197   test "it can parse mentions and return the relevant users" do
 198     text = "@gsimg According to @archaeme , that is @daggsy."
 199
 200     gsimg = insert(:user, %{nickname: "gsimg"})
 201     archaeme = insert(:user, %{nickname: "archaeme"})
 202
 203     expected_result = [
 204       {"@gsimg", gsimg},
 205       {"@archaeme", archaeme}
 206     ]
 207
 208     assert TwitterAPI.parse_mentions(text) == expected_result
 209   end
 210
 211   test "it adds user links to an existing text" do
 212     text = "@gsimg According to @archaeme , that is @daggsy."
 213
 214     gsimg = insert(:user, %{nickname: "gsimg"})
 215     archaeme = insert(:user, %{nickname: "archaeme"})
 216
 217     mentions = TwitterAPI.parse_mentions(text)
 218     expected_text = "<a href='#{gsimg.ap_id}'>@gsimg</a> According to <a href='#{archaeme.ap_id}'>@archaeme</a> , that is @daggsy."
 219
 220     assert TwitterAPI.add_user_links(text, mentions) == expected_text
 221   end
 222
 223   test "it favorites a status, returns the updated status" do
 224     user = insert(:user)
 225     note_activity = insert(:note_activity)
 226     activity_user = Repo.get_by!(User, ap_id: note_activity.data["actor"])
 227
 228     {:ok, status} = TwitterAPI.favorite(user, note_activity)
 229     updated_activity = Activity.get_by_ap_id(note_activity.data["id"])
 230
 231     assert status == ActivityRepresenter.to_map(updated_activity, %{user: activity_user, for: user})
 232   end
 233
 234   test "it unfavorites a status, returns the updated status" do
 235     user = insert(:user)
 236     note_activity = insert(:note_activity)
 237     activity_user = Repo.get_by!(User, ap_id: note_activity.data["actor"])
 238     object = Object.get_by_ap_id(note_activity.data["object"]["id"])
 239
 240     {:ok, _like_activity, _object } = ActivityPub.like(user, object)
 241     updated_activity = Activity.get_by_ap_id(note_activity.data["id"])
 242     assert ActivityRepresenter.to_map(updated_activity, %{user: activity_user, for: user})["fave_num"] == 1
 243
 244     {:ok, status} = TwitterAPI.unfavorite(user, note_activity)
 245
 246     assert status["fave_num"] == 0
 247   end
 248
 249   test "it retweets a status and returns the retweet" do
 250     user = insert(:user)
 251     note_activity = insert(:note_activity)
 252     activity_user = Repo.get_by!(User, ap_id: note_activity.data["actor"])
 253
 254     {:ok, status} = TwitterAPI.retweet(user, note_activity)
 255     updated_activity = Activity.get_by_ap_id(note_activity.data["id"])
 256
 257     assert status == ActivityRepresenter.to_map(updated_activity, %{user: activity_user, for: user})
 258   end
 259
 260   test "it registers a new user and returns the user." do
 261     data = %{
 262       "nickname" => "lain",
 263       "email" => "lain@wired.jp",
 264       "fullname" => "lain iwakura",
 265       "bio" => "close the world.",
 266       "password" => "bear",
 267       "confirm" => "bear"
 268     }
 269
 270     {:ok, user} = TwitterAPI.register_user(data)
 271
 272     fetched_user = Repo.get_by(User, nickname: "lain")
 273     assert user == UserRepresenter.to_map(fetched_user)
 274   end
 275
 276   test "it returns the error on registration problems" do
 277     data = %{
 278       "nickname" => "lain",
 279       "email" => "lain@wired.jp",
 280       "fullname" => "lain iwakura",
 281       "bio" => "close the world.",
 282       "password" => "bear"
 283     }
 284
 285     {:error, error_object} = TwitterAPI.register_user(data)
 286
 287     assert is_binary(error_object[:error])
 288     refute Repo.get_by(User, nickname: "lain")
 289   end
 290
 291   setup do
 292     Supervisor.terminate_child(Pleroma.Supervisor, Cachex)
 293     Supervisor.restart_child(Pleroma.Supervisor, Cachex)
 294     :ok
 295   end
 296 end