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, UserView}
   5   alias Pleroma.Web.CommonAPI.Utils
   6   alias Pleroma.{Activity, User, Object, Repo}
   7   alias Pleroma.Web.TwitterAPI.Representers.ActivityRepresenter
   8   alias Pleroma.Web.ActivityPub.ActivityPub
   9
  10   import Pleroma.Factory
  11
  12   test "create a status" do
  13     user = insert(:user)
  14     _mentioned_user = UserBuilder.insert(%{nickname: "shp", ap_id: "shp"})
  15
  16     object_data = %{
  17       "type" => "Image",
  18       "url" => [
  19         %{
  20           "type" => "Link",
  21           "mediaType" => "image/jpg",
  22           "href" => "http://example.org/image.jpg"
  23         }
  24       ],
  25       "uuid" => 1
  26     }
  27
  28     object = Repo.insert!(%Object{data: object_data})
  29
  30     input = %{
  31       "status" => "Hello again, @shp.<script></script>\nThis is on another :moominmamma: line. #2hu #epic #phantasmagoric",
  32       "media_ids" => [object.id]
  33     }
  34
  35     { :ok, activity = %Activity{} } = TwitterAPI.create_status(user, input)
  36
  37     assert get_in(activity.data, ["object", "content"]) == "Hello again, <span><a href='shp'>@<span>shp</span></a></span>.&lt;script&gt;&lt;/script&gt;<br>This is on another :moominmamma: line. #2hu #epic #phantasmagoric<br><a href=\"http://example.org/image.jpg\" class='attachment'>image.jpg</a>"
  38     assert get_in(activity.data, ["object", "type"]) == "Note"
  39     assert get_in(activity.data, ["object", "actor"]) == user.ap_id
  40     assert get_in(activity.data, ["actor"]) == user.ap_id
  41     assert Enum.member?(get_in(activity.data, ["to"]), User.ap_followers(user))
  42     assert Enum.member?(get_in(activity.data, ["to"]), "https://www.w3.org/ns/activitystreams#Public")
  43     assert Enum.member?(get_in(activity.data, ["to"]), "shp")
  44     assert activity.local == true
  45
  46     assert %{"moominmamma" => "http://localhost:4001/finmoji/128px/moominmamma-128.png"} = activity.data["object"]["emoji"]
  47
  48     # hashtags
  49     assert activity.data["object"]["tag"] == ["2hu", "epic", "phantasmagoric"]
  50
  51     # Add a context
  52     assert is_binary(get_in(activity.data, ["context"]))
  53     assert is_binary(get_in(activity.data, ["object", "context"]))
  54
  55     assert is_list(activity.data["object"]["attachment"])
  56
  57     assert activity.data["object"] == Object.get_by_ap_id(activity.data["object"]["id"]).data
  58
  59     user = User.get_by_ap_id(user.ap_id)
  60
  61     assert user.info["note_count"] == 1
  62   end
  63
  64   test "create a status that is a reply" do
  65     user = insert(:user)
  66     input = %{
  67       "status" => "Hello again."
  68     }
  69
  70     { :ok, activity = %Activity{} } = TwitterAPI.create_status(user, input)
  71
  72     input = %{
  73       "status" => "Here's your (you).",
  74       "in_reply_to_status_id" => activity.id
  75     }
  76
  77     { :ok, reply = %Activity{} } = TwitterAPI.create_status(user, input)
  78
  79     assert get_in(reply.data, ["context"]) == get_in(activity.data, ["context"])
  80     assert get_in(reply.data, ["object", "context"]) == get_in(activity.data, ["object", "context"])
  81     assert get_in(reply.data, ["object", "inReplyTo"]) == get_in(activity.data, ["object", "id"])
  82     assert get_in(reply.data, ["object", "inReplyToStatusId"]) == activity.id
  83     assert Enum.member?(get_in(reply.data, ["to"]), user.ap_id)
  84   end
  85
  86   test "fetch public statuses, excluding remote ones." do
  87     %{ public: activity, user: user } = ActivityBuilder.public_and_non_public
  88     insert(:note_activity, %{local: false})
  89
  90     follower = insert(:user, following: [User.ap_followers(user)])
  91
  92     statuses = TwitterAPI.fetch_public_statuses(follower)
  93
  94     assert length(statuses) == 1
  95     assert Enum.at(statuses, 0) == ActivityRepresenter.to_map(activity, %{user: user, for: follower})
  96   end
  97
  98   test "fetch whole known network statuses" do
  99     %{ public: activity, user: user } = ActivityBuilder.public_and_non_public
 100     insert(:note_activity, %{local: false})
 101
 102     follower = insert(:user, following: [User.ap_followers(user)])
 103
 104     statuses = TwitterAPI.fetch_public_and_external_statuses(follower)
 105
 106     assert length(statuses) == 2
 107     assert Enum.at(statuses, 0) == ActivityRepresenter.to_map(activity, %{user: user, for: follower})
 108   end
 109
 110   test "fetch friends' statuses" do
 111     user = insert(:user, %{following: ["someguy/followers"]})
 112     {:ok, activity} = ActivityBuilder.insert(%{"to" => ["someguy/followers"]})
 113     {:ok, direct_activity} = ActivityBuilder.insert(%{"to" => [user.ap_id]})
 114
 115     statuses = TwitterAPI.fetch_friend_statuses(user)
 116
 117     activity_user = Repo.get_by(User, ap_id: activity.data["actor"])
 118     direct_activity_user = Repo.get_by(User, ap_id: direct_activity.data["actor"])
 119
 120     assert length(statuses) == 2
 121     assert Enum.at(statuses, 0) == ActivityRepresenter.to_map(activity, %{user: activity_user})
 122     assert Enum.at(statuses, 1) == ActivityRepresenter.to_map(direct_activity, %{user: direct_activity_user, mentioned: [user]})
 123   end
 124
 125   test "fetch user's mentions" do
 126     user = insert(:user)
 127     {:ok, activity} = ActivityBuilder.insert(%{"to" => [user.ap_id]})
 128     activity_user = Repo.get_by(User, ap_id: activity.data["actor"])
 129
 130     statuses = TwitterAPI.fetch_mentions(user)
 131
 132     assert length(statuses) == 1
 133     assert Enum.at(statuses, 0) == ActivityRepresenter.to_map(activity, %{user: activity_user, mentioned: [user]})
 134   end
 135
 136   test "get a user by params" do
 137     user1_result = {:ok, user1} = UserBuilder.insert(%{ap_id: "some id", email: "test@pleroma"})
 138     {:ok, user2} = UserBuilder.insert(%{ap_id: "some other id", nickname: "testname2", email: "test2@pleroma"})
 139
 140     assert {:error, "You need to specify screen_name or user_id"} == TwitterAPI.get_user(nil, nil)
 141     assert user1_result == TwitterAPI.get_user(nil, %{"user_id" => user1.id})
 142     assert user1_result == TwitterAPI.get_user(nil, %{"user_id" => user1.nickname})
 143     assert user1_result == TwitterAPI.get_user(nil, %{"screen_name" => user1.nickname})
 144     assert user1_result == TwitterAPI.get_user(user1, nil)
 145     assert user1_result == TwitterAPI.get_user(user2, %{"user_id" => user1.id})
 146     assert user1_result == TwitterAPI.get_user(user2, %{"screen_name" => user1.nickname})
 147     assert {:error, "No user with such screen_name"} == TwitterAPI.get_user(nil, %{"screen_name" => "Satan"})
 148     assert {:error, "No user with such user_id"} == TwitterAPI.get_user(nil, %{"user_id" => 666})
 149   end
 150
 151   test "fetch user's statuses" do
 152     {:ok, user1} = UserBuilder.insert(%{ap_id: "some id", email: "test@pleroma"})
 153     {:ok, user2} = UserBuilder.insert(%{ap_id: "some other id", nickname: "testname2", email: "test2@pleroma"})
 154
 155     {:ok, status1} = ActivityBuilder.insert(%{"id" => 1}, %{user: user1})
 156     {:ok, status2} = ActivityBuilder.insert(%{"id" => 2}, %{user: user2})
 157
 158     user1_statuses = TwitterAPI.fetch_user_statuses(user1, %{"actor_id" => user1.ap_id})
 159
 160     assert length(user1_statuses) == 1
 161     assert Enum.at(user1_statuses, 0) == ActivityRepresenter.to_map(status1, %{user: user1})
 162
 163     user2_statuses = TwitterAPI.fetch_user_statuses(user1, %{"actor_id" => user2.ap_id})
 164
 165     assert length(user2_statuses) == 1
 166     assert Enum.at(user2_statuses, 0) == ActivityRepresenter.to_map(status2, %{user: user2})
 167   end
 168
 169   test "fetch a single status" do
 170     {:ok, activity} = ActivityBuilder.insert()
 171     {:ok, user} = UserBuilder.insert()
 172     actor = Repo.get_by!(User, ap_id: activity.data["actor"])
 173
 174     status = TwitterAPI.fetch_status(user, activity.id)
 175
 176     assert status == ActivityRepresenter.to_map(activity, %{for: user, user: actor})
 177   end
 178
 179   test "Follow another user using user_id" do
 180     user = insert(:user)
 181     followed = insert(:user)
 182
 183     {:ok, user, followed, _activity } = TwitterAPI.follow(user, %{"user_id" => followed.id})
 184     assert User.ap_followers(followed) in user.following
 185
 186     { :error, msg } = TwitterAPI.follow(user, %{"user_id" => followed.id})
 187     assert msg == "Could not follow user: #{followed.nickname} is already on your list."
 188   end
 189
 190   test "Follow another user using screen_name" do
 191     user = insert(:user)
 192     followed = insert(:user)
 193
 194     {:ok, user, followed, _activity } = TwitterAPI.follow(user, %{"screen_name" => followed.nickname})
 195     assert User.ap_followers(followed) in user.following
 196
 197     followed = User.get_by_ap_id(followed.ap_id)
 198     assert followed.info["follower_count"] == 1
 199
 200     { :error, msg } = TwitterAPI.follow(user, %{"screen_name" => followed.nickname})
 201     assert msg == "Could not follow user: #{followed.nickname} is already on your list."
 202   end
 203
 204   test "Unfollow another user using user_id" do
 205     unfollowed = insert(:user)
 206     user = insert(:user, %{following: [User.ap_followers(unfollowed)]})
 207     ActivityPub.follow(user, unfollowed)
 208
 209     {:ok, user, unfollowed } = TwitterAPI.unfollow(user, %{"user_id" => unfollowed.id})
 210     assert user.following == []
 211
 212     { :error, msg } = TwitterAPI.unfollow(user, %{"user_id" => unfollowed.id})
 213     assert msg == "Not subscribed!"
 214   end
 215
 216   test "Unfollow another user using screen_name" do
 217     unfollowed = insert(:user)
 218     user = insert(:user, %{following: [User.ap_followers(unfollowed)]})
 219
 220     ActivityPub.follow(user, unfollowed)
 221
 222     {:ok, user, unfollowed } = TwitterAPI.unfollow(user, %{"screen_name" => unfollowed.nickname})
 223     assert user.following == []
 224
 225     { :error, msg } = TwitterAPI.unfollow(user, %{"screen_name" => unfollowed.nickname})
 226     assert msg == "Not subscribed!"
 227   end
 228
 229   test "Block another user using user_id" do
 230     user = insert(:user)
 231     blocked = insert(:user)
 232
 233     {:ok, user, blocked} = TwitterAPI.block(user, %{"user_id" => blocked.id})
 234     assert User.blocks?(user, blocked)
 235   end
 236
 237   test "Block another user using screen_name" do
 238     user = insert(:user)
 239     blocked = insert(:user)
 240
 241     {:ok, user, blocked} = TwitterAPI.block(user, %{"screen_name" => blocked.nickname})
 242     assert User.blocks?(user, blocked)
 243   end
 244
 245   test "Unblock another user using user_id" do
 246     unblocked = insert(:user)
 247     user = insert(:user)
 248     User.block(user, unblocked)
 249
 250     {:ok, user, unblocked} = TwitterAPI.unblock(user, %{"user_id" => unblocked.id})
 251     assert user.info["blocks"] == []
 252   end
 253
 254   test "Unblock another user using screen_name" do
 255     unblocked = insert(:user)
 256     user = insert(:user)
 257     User.block(user, unblocked)
 258
 259     {:ok, user, unblocked} = TwitterAPI.unblock(user, %{"screen_name" => unblocked.nickname})
 260     assert user.info["blocks"] == []
 261   end
 262
 263   test "fetch statuses in a context using the conversation id" do
 264     {:ok, user} = UserBuilder.insert()
 265     {:ok, activity} = ActivityBuilder.insert(%{"type" => "Create", "context" => "2hu"})
 266     {:ok, activity_two} = ActivityBuilder.insert(%{"type" => "Create", "context" => "2hu"})
 267     {:ok, _activity_three} = ActivityBuilder.insert(%{"type" => "Create", "context" => "3hu"})
 268
 269     {:ok, object} = Object.context_mapping("2hu") |> Repo.insert
 270
 271     statuses = TwitterAPI.fetch_conversation(user, object.id)
 272
 273     assert length(statuses) == 2
 274     assert Enum.at(statuses, 1)["id"] == activity.id
 275     assert Enum.at(statuses, 0)["id"] == activity_two.id
 276   end
 277
 278   test "upload a file" do
 279     file = %Plug.Upload{content_type: "image/jpg", path: Path.absname("test/fixtures/image.jpg"), filename: "an_image.jpg"}
 280
 281     response = TwitterAPI.upload(file)
 282
 283     assert is_binary(response)
 284   end
 285
 286   test "it adds user links to an existing text" do
 287     text = "@gsimg According to @archaeme, that is @daggsy. Also hello @archaeme@archae.me"
 288
 289     gsimg = insert(:user, %{nickname: "gsimg"})
 290     archaeme = insert(:user, %{nickname: "archaeme"})
 291     archaeme_remote = insert(:user, %{nickname: "archaeme@archae.me"})
 292
 293     mentions = Pleroma.Formatter.parse_mentions(text)
 294     expected_text = "<span><a href='#{gsimg.ap_id}'>@<span>gsimg</span></a></span> According to <span><a href='#{archaeme.ap_id}'>@<span>archaeme</span></a></span>, that is @daggsy. Also hello <span><a href='#{archaeme_remote.ap_id}'>@<span>archaeme</span></a></span>"
 295
 296     assert Utils.add_user_links(text, mentions) == expected_text
 297   end
 298
 299   test "it favorites a status, returns the updated status" do
 300     user = insert(:user)
 301     note_activity = insert(:note_activity)
 302     activity_user = Repo.get_by!(User, ap_id: note_activity.data["actor"])
 303
 304     {:ok, status} = TwitterAPI.fav(user, note_activity.id)
 305     updated_activity = Activity.get_by_ap_id(note_activity.data["id"])
 306
 307     assert status == ActivityRepresenter.to_map(updated_activity, %{user: activity_user, for: user})
 308   end
 309
 310   test "it unfavorites a status, returns the updated status" do
 311     user = insert(:user)
 312     note_activity = insert(:note_activity)
 313     activity_user = Repo.get_by!(User, ap_id: note_activity.data["actor"])
 314     object = Object.get_by_ap_id(note_activity.data["object"]["id"])
 315
 316     {:ok, _like_activity, _object } = ActivityPub.like(user, object)
 317     updated_activity = Activity.get_by_ap_id(note_activity.data["id"])
 318     assert ActivityRepresenter.to_map(updated_activity, %{user: activity_user, for: user})["fave_num"] == 1
 319
 320     {:ok, status} = TwitterAPI.unfav(user, note_activity.id)
 321
 322     assert status["fave_num"] == 0
 323   end
 324
 325   test "it retweets a status and returns the retweet" do
 326     user = insert(:user)
 327     note_activity = insert(:note_activity)
 328     activity_user = Repo.get_by!(User, ap_id: note_activity.data["actor"])
 329
 330     {:ok, status} = TwitterAPI.repeat(user, note_activity.id)
 331     updated_activity = Activity.get_by_ap_id(note_activity.data["id"])
 332
 333     assert status == ActivityRepresenter.to_map(updated_activity, %{user: activity_user, for: user})
 334   end
 335
 336   test "it registers a new user and returns the user." do
 337     data = %{
 338       "nickname" => "lain",
 339       "email" => "lain@wired.jp",
 340       "fullname" => "lain iwakura",
 341       "bio" => "close the world.",
 342       "password" => "bear",
 343       "confirm" => "bear"
 344     }
 345
 346     {:ok, user} = TwitterAPI.register_user(data)
 347
 348     fetched_user = Repo.get_by(User, nickname: "lain")
 349     assert UserView.render("show.json", %{user: user}) == UserView.render("show.json", %{user: fetched_user})
 350   end
 351
 352   test "it returns the error on registration problems" do
 353     data = %{
 354       "nickname" => "lain",
 355       "email" => "lain@wired.jp",
 356       "fullname" => "lain iwakura",
 357       "bio" => "close the world.",
 358       "password" => "bear"
 359     }
 360
 361     {:error, error_object} = TwitterAPI.register_user(data)
 362
 363     assert is_binary(error_object[:error])
 364     refute Repo.get_by(User, nickname: "lain")
 365   end
 366
 367   test "it assigns an integer conversation_id" do
 368     note_activity = insert(:note_activity)
 369     user = User.get_cached_by_ap_id(note_activity.data["actor"])
 370     status = ActivityRepresenter.to_map(note_activity, %{user: user})
 371
 372     assert is_number(status["statusnet_conversation_id"])
 373   end
 374
 375   setup do
 376     Supervisor.terminate_child(Pleroma.Supervisor, Cachex)
 377     Supervisor.restart_child(Pleroma.Supervisor, Cachex)
 378     :ok
 379   end
 380
 381   describe "context_to_conversation_id" do
 382     test "creates a mapping object" do
 383       conversation_id = TwitterAPI.context_to_conversation_id("random context")
 384       object = Object.get_by_ap_id("random context")
 385
 386       assert conversation_id == object.id
 387     end
 388
 389     test "returns an existing mapping for an existing object" do
 390       {:ok, object} = Object.context_mapping("random context") |> Repo.insert
 391       conversation_id = TwitterAPI.context_to_conversation_id("random context")
 392
 393       assert conversation_id == object.id
 394     end
 395   end
 396
 397   describe "fetching a user by uri" do
 398     test "fetches a user by uri" do
 399       id = "https://mastodon.social/users/lambadalambda"
 400       user = insert(:user)
 401       {:ok, represented} = TwitterAPI.get_external_profile(user, id)
 402       remote = User.get_by_ap_id(id)
 403
 404       assert represented["id"] == UserView.render("show.json", %{user: remote, for: user})["id"]
 405
 406       # Also fetches the feed.
 407       # assert Activity.get_create_activity_by_object_ap_id("tag:mastodon.social,2017-04-05:objectId=1641750:objectType=Status")
 408     end
 409   end
 410 end