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.Web.TwitterAPI.Utils
   6   alias Pleroma.{Activity, User, Object, Repo}
   7   alias Pleroma.Web.TwitterAPI.Representers.{ActivityRepresenter, UserRepresenter}
   8   alias Pleroma.Web.ActivityPub.ActivityPub
   9
  10   import Pleroma.Factory
  11
  12   test "create a status" do
  13     user = UserBuilder.build(%{ap_id: "142344"})
  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 line.",
  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, <a href='shp'>@shp</a>.<br>This is on another line.<br><a href='http://example.org/image.jpg' class='attachment'>http://example.org/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     # Add a context
  47     assert is_binary(get_in(activity.data, ["context"]))
  48     assert is_binary(get_in(activity.data, ["object", "context"]))
  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, ["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, excluding remote ones." do
  78     %{ public: activity, user: user } = ActivityBuilder.public_and_non_public
  79     insert(:note_activity, %{local: false})
  80
  81     follower = insert(:user, following: [User.ap_followers(user)])
  82
  83     statuses = TwitterAPI.fetch_public_statuses(follower)
  84
  85     assert length(statuses) == 1
  86     assert Enum.at(statuses, 0) == ActivityRepresenter.to_map(activity, %{user: user, for: follower})
  87   end
  88
  89   test "fetch whole known network statuses" do
  90     %{ public: activity, user: user } = ActivityBuilder.public_and_non_public
  91     insert(:note_activity, %{local: false})
  92
  93     follower = insert(:user, following: [User.ap_followers(user)])
  94
  95     statuses = TwitterAPI.fetch_public_and_external_statuses(follower)
  96
  97     assert length(statuses) == 2
  98     assert Enum.at(statuses, 0) == ActivityRepresenter.to_map(activity, %{user: user, for: follower})
  99   end
 100
 101   test "fetch friends' statuses" do
 102     user = insert(:user, %{following: ["someguy/followers"]})
 103     {:ok, activity} = ActivityBuilder.insert(%{"to" => ["someguy/followers"]})
 104     {:ok, direct_activity} = ActivityBuilder.insert(%{"to" => [user.ap_id]})
 105
 106     statuses = TwitterAPI.fetch_friend_statuses(user)
 107
 108     activity_user = Repo.get_by(User, ap_id: activity.data["actor"])
 109     direct_activity_user = Repo.get_by(User, ap_id: direct_activity.data["actor"])
 110
 111     assert length(statuses) == 2
 112     assert Enum.at(statuses, 0) == ActivityRepresenter.to_map(activity, %{user: activity_user})
 113     assert Enum.at(statuses, 1) == ActivityRepresenter.to_map(direct_activity, %{user: direct_activity_user, mentioned: [user]})
 114   end
 115
 116   test "fetch user's mentions" do
 117     user = insert(:user)
 118     {:ok, activity} = ActivityBuilder.insert(%{"to" => [user.ap_id]})
 119     activity_user = Repo.get_by(User, ap_id: activity.data["actor"])
 120
 121     statuses = TwitterAPI.fetch_mentions(user)
 122
 123     assert length(statuses) == 1
 124     assert Enum.at(statuses, 0) == ActivityRepresenter.to_map(activity, %{user: activity_user, mentioned: [user]})
 125   end
 126
 127   test "get a user by params" do
 128     user1_result = {:ok, user1} = UserBuilder.insert(%{ap_id: "some id", email: "test@pleroma"})
 129     {:ok, user2} = UserBuilder.insert(%{ap_id: "some other id", nickname: "testname2", email: "test2@pleroma"})
 130
 131     assert {:error, "You need to specify screen_name or user_id"} == TwitterAPI.get_user(nil, nil)
 132     assert user1_result == TwitterAPI.get_user(nil, %{"user_id" => user1.id})
 133     assert user1_result == TwitterAPI.get_user(nil, %{"screen_name" => user1.nickname})
 134     assert user1_result == TwitterAPI.get_user(user1, nil)
 135     assert user1_result == TwitterAPI.get_user(user2, %{"user_id" => user1.id})
 136     assert user1_result == TwitterAPI.get_user(user2, %{"screen_name" => user1.nickname})
 137     assert {:error, "No user with such screen_name"} == TwitterAPI.get_user(nil, %{"screen_name" => "Satan"})
 138     assert {:error, "No user with such user_id"} == TwitterAPI.get_user(nil, %{"user_id" => 666})
 139   end
 140
 141   test "fetch user's statuses" do
 142     {:ok, user1} = UserBuilder.insert(%{ap_id: "some id", email: "test@pleroma"})
 143     {:ok, user2} = UserBuilder.insert(%{ap_id: "some other id", nickname: "testname2", email: "test2@pleroma"})
 144
 145     {:ok, status1} = ActivityBuilder.insert(%{"id" => 1}, %{user: user1})
 146     {:ok, status2} = ActivityBuilder.insert(%{"id" => 2}, %{user: user2})
 147
 148     user1_statuses = TwitterAPI.fetch_user_statuses(user1, %{"actor_id" => user1.ap_id})
 149
 150     assert length(user1_statuses) == 1
 151     assert Enum.at(user1_statuses, 0) == ActivityRepresenter.to_map(status1, %{user: user1})
 152
 153     user2_statuses = TwitterAPI.fetch_user_statuses(user1, %{"actor_id" => user2.ap_id})
 154
 155     assert length(user2_statuses) == 1
 156     assert Enum.at(user2_statuses, 0) == ActivityRepresenter.to_map(status2, %{user: user2})
 157   end
 158
 159   test "fetch a single status" do
 160     {:ok, activity} = ActivityBuilder.insert()
 161     {:ok, user} = UserBuilder.insert()
 162     actor = Repo.get_by!(User, ap_id: activity.data["actor"])
 163
 164     status = TwitterAPI.fetch_status(user, activity.id)
 165
 166     assert status == ActivityRepresenter.to_map(activity, %{for: user, user: actor})
 167   end
 168
 169   test "Follow another user using user_id" do
 170     user = insert(:user)
 171     followed = insert(:user)
 172
 173     {:ok, user, followed, _activity } = TwitterAPI.follow(user, %{"user_id" => followed.id})
 174     assert user.following == [User.ap_followers(followed)]
 175
 176     { :error, msg } = TwitterAPI.follow(user, %{"user_id" => followed.id})
 177     assert msg == "Could not follow user: #{followed.nickname} is already on your list."
 178   end
 179
 180   test "Follow another user using screen_name" do
 181     user = insert(:user)
 182     followed = insert(:user)
 183
 184     {:ok, user, followed, _activity } = TwitterAPI.follow(user, %{"screen_name" => followed.nickname})
 185     assert user.following == [User.ap_followers(followed)]
 186
 187     { :error, msg } = TwitterAPI.follow(user, %{"screen_name" => followed.nickname})
 188     assert msg == "Could not follow user: #{followed.nickname} is already on your list."
 189   end
 190
 191   test "Unfollow another user using user_id" do
 192     unfollowed = insert(:user)
 193     user = insert(:user, %{following: [User.ap_followers(unfollowed)]})
 194     ActivityPub.follow(user, unfollowed)
 195
 196     {:ok, user, unfollowed } = TwitterAPI.unfollow(user, %{"user_id" => unfollowed.id})
 197     assert user.following == []
 198
 199     { :error, msg } = TwitterAPI.unfollow(user, %{"user_id" => unfollowed.id})
 200     assert msg == "Not subscribed!"
 201   end
 202
 203   test "Unfollow another user using screen_name" do
 204     unfollowed = insert(:user)
 205     user = insert(:user, %{following: [User.ap_followers(unfollowed)]})
 206
 207     ActivityPub.follow(user, unfollowed)
 208
 209     {:ok, user, unfollowed } = TwitterAPI.unfollow(user, %{"screen_name" => unfollowed.nickname})
 210     assert user.following == []
 211
 212     { :error, msg } = TwitterAPI.unfollow(user, %{"screen_name" => unfollowed.nickname})
 213     assert msg == "Not subscribed!"
 214   end
 215
 216   test "fetch statuses in a context using the conversation id" do
 217     {:ok, user} = UserBuilder.insert()
 218     {:ok, activity} = ActivityBuilder.insert(%{"context" => "2hu"})
 219     {:ok, activity_two} = ActivityBuilder.insert(%{"context" => "2hu"})
 220     {:ok, _activity_three} = ActivityBuilder.insert(%{"context" => "3hu"})
 221
 222     {:ok, object} = Object.context_mapping("2hu") |> Repo.insert
 223
 224     statuses = TwitterAPI.fetch_conversation(user, object.id)
 225
 226     assert length(statuses) == 2
 227     assert Enum.at(statuses, 0)["id"] == activity.id
 228     assert Enum.at(statuses, 1)["id"] == activity_two.id
 229   end
 230
 231   test "upload a file" do
 232     file = %Plug.Upload{content_type: "image/jpg", path: Path.absname("test/fixtures/image.jpg"), filename: "an_image.jpg"}
 233
 234     response = TwitterAPI.upload(file)
 235
 236     assert is_binary(response)
 237   end
 238
 239   test "it can parse mentions and return the relevant users" do
 240     text = "@gsimg According to @archaeme, that is @daggsy. Also hello @archaeme@archae.me"
 241
 242     gsimg = insert(:user, %{nickname: "gsimg"})
 243     archaeme = insert(:user, %{nickname: "archaeme"})
 244     archaeme_remote = insert(:user, %{nickname: "archaeme@archae.me"})
 245
 246     expected_result = [
 247       {"@gsimg", gsimg},
 248       {"@archaeme", archaeme},
 249       {"@archaeme@archae.me", archaeme_remote},
 250     ]
 251
 252     assert TwitterAPI.parse_mentions(text) == expected_result
 253   end
 254
 255   test "it adds user links to an existing text" do
 256     text = "@gsimg According to @archaeme, that is @daggsy. Also hello @archaeme@archae.me"
 257
 258     gsimg = insert(:user, %{nickname: "gsimg"})
 259     archaeme = insert(:user, %{nickname: "archaeme"})
 260     archaeme_remote = insert(:user, %{nickname: "archaeme@archae.me"})
 261
 262     mentions = TwitterAPI.parse_mentions(text)
 263     expected_text = "<a href='#{gsimg.ap_id}'>@gsimg</a> According to <a href='#{archaeme.ap_id}'>@archaeme</a>, that is @daggsy. Also hello <a href='#{archaeme_remote.ap_id}'>@archaeme@archae.me</a>"
 264
 265     assert Utils.add_user_links(text, mentions) == expected_text
 266   end
 267
 268   test "it favorites a status, returns the updated status" do
 269     user = insert(:user)
 270     note_activity = insert(:note_activity)
 271     activity_user = Repo.get_by!(User, ap_id: note_activity.data["actor"])
 272
 273     {:ok, status} = TwitterAPI.favorite(user, note_activity)
 274     updated_activity = Activity.get_by_ap_id(note_activity.data["id"])
 275
 276     assert status == ActivityRepresenter.to_map(updated_activity, %{user: activity_user, for: user})
 277   end
 278
 279   test "it unfavorites a status, returns the updated status" do
 280     user = insert(:user)
 281     note_activity = insert(:note_activity)
 282     activity_user = Repo.get_by!(User, ap_id: note_activity.data["actor"])
 283     object = Object.get_by_ap_id(note_activity.data["object"]["id"])
 284
 285     {:ok, _like_activity, _object } = ActivityPub.like(user, object)
 286     updated_activity = Activity.get_by_ap_id(note_activity.data["id"])
 287     assert ActivityRepresenter.to_map(updated_activity, %{user: activity_user, for: user})["fave_num"] == 1
 288
 289     {:ok, status} = TwitterAPI.unfavorite(user, note_activity)
 290
 291     assert status["fave_num"] == 0
 292   end
 293
 294   test "it retweets a status and returns the retweet" do
 295     user = insert(:user)
 296     note_activity = insert(:note_activity)
 297     activity_user = Repo.get_by!(User, ap_id: note_activity.data["actor"])
 298
 299     {:ok, status} = TwitterAPI.retweet(user, note_activity)
 300     updated_activity = Activity.get_by_ap_id(note_activity.data["id"])
 301
 302     assert status == ActivityRepresenter.to_map(updated_activity, %{user: activity_user, for: user})
 303   end
 304
 305   test "it registers a new user and returns the user." do
 306     data = %{
 307       "nickname" => "lain",
 308       "email" => "lain@wired.jp",
 309       "fullname" => "lain iwakura",
 310       "bio" => "close the world.",
 311       "password" => "bear",
 312       "confirm" => "bear"
 313     }
 314
 315     {:ok, user} = TwitterAPI.register_user(data)
 316
 317     fetched_user = Repo.get_by(User, nickname: "lain")
 318     assert user == UserRepresenter.to_map(fetched_user)
 319   end
 320
 321   test "it returns the error on registration problems" do
 322     data = %{
 323       "nickname" => "lain",
 324       "email" => "lain@wired.jp",
 325       "fullname" => "lain iwakura",
 326       "bio" => "close the world.",
 327       "password" => "bear"
 328     }
 329
 330     {:error, error_object} = TwitterAPI.register_user(data)
 331
 332     assert is_binary(error_object[:error])
 333     refute Repo.get_by(User, nickname: "lain")
 334   end
 335
 336   test "it assigns an integer conversation_id" do
 337     note_activity = insert(:note_activity)
 338     user = User.get_cached_by_ap_id(note_activity.data["actor"])
 339     status = ActivityRepresenter.to_map(note_activity, %{user: user})
 340
 341     assert is_number(status["statusnet_conversation_id"])
 342   end
 343
 344   setup do
 345     Supervisor.terminate_child(Pleroma.Supervisor, Cachex)
 346     Supervisor.restart_child(Pleroma.Supervisor, Cachex)
 347     :ok
 348   end
 349
 350   describe "context_to_conversation_id" do
 351     test "creates a mapping object" do
 352       conversation_id = TwitterAPI.context_to_conversation_id("random context")
 353       object = Object.get_by_ap_id("random context")
 354
 355       assert conversation_id == object.id
 356     end
 357
 358     test "returns an existing mapping for an existing object" do
 359       {:ok, object} = Object.context_mapping("random context") |> Repo.insert
 360       conversation_id = TwitterAPI.context_to_conversation_id("random context")
 361
 362       assert conversation_id == object.id
 363     end
 364   end
 365
 366   describe "fetching a user by uri" do
 367     test "fetches a user by uri" do
 368       user = insert(:user)
 369
 370       {:ok, represented} = TwitterAPI.get_external_profile(user, user.ap_id)
 371       assert represented = UserRepresenter.to_map(user, %{for: user})
 372     end
 373   end
 374 end