Use conversation mapping objects to get / retrieve context from TwAPI.
[akkoma] / 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 "fetch user's mentions" do
106 user = insert(:user)
107 {:ok, activity} = ActivityBuilder.insert(%{"to" => [user.ap_id]})
108 activity_user = Repo.get_by(User, ap_id: activity.data["actor"])
109
110 statuses = TwitterAPI.fetch_mentions(user)
111
112 assert length(statuses) == 1
113 assert Enum.at(statuses, 0) == ActivityRepresenter.to_map(activity, %{user: activity_user, mentioned: [user]})
114 end
115
116 test "get a user by params" do
117 user1_result = {:ok, user1} = UserBuilder.insert(%{ap_id: "some id", email: "test@pleroma"})
118 {:ok, user2} = UserBuilder.insert(%{ap_id: "some other id", nickname: "testname2", email: "test2@pleroma"})
119
120 assert {:error, "You need to specify screen_name or user_id"} == TwitterAPI.get_user(nil, nil)
121 assert user1_result == TwitterAPI.get_user(nil, %{"user_id" => user1.id})
122 assert user1_result == TwitterAPI.get_user(nil, %{"screen_name" => user1.nickname})
123 assert user1_result == TwitterAPI.get_user(user1, nil)
124 assert user1_result == TwitterAPI.get_user(user2, %{"user_id" => user1.id})
125 assert user1_result == TwitterAPI.get_user(user2, %{"screen_name" => user1.nickname})
126 assert {:error, "No user with such screen_name"} == TwitterAPI.get_user(nil, %{"screen_name" => "Satan"})
127 assert {:error, "No user with such user_id"} == TwitterAPI.get_user(nil, %{"user_id" => 666})
128 end
129
130 test "fetch user's statuses" do
131 {:ok, user1} = UserBuilder.insert(%{ap_id: "some id", email: "test@pleroma"})
132 {:ok, user2} = UserBuilder.insert(%{ap_id: "some other id", nickname: "testname2", email: "test2@pleroma"})
133
134 {:ok, status1} = ActivityBuilder.insert(%{"id" => 1}, %{user: user1})
135 {:ok, status2} = ActivityBuilder.insert(%{"id" => 2}, %{user: user2})
136
137 user1_statuses = TwitterAPI.fetch_user_statuses(user1, %{"actor_id" => user1.ap_id})
138
139 assert length(user1_statuses) == 1
140 assert Enum.at(user1_statuses, 0) == ActivityRepresenter.to_map(status1, %{user: user1})
141
142 user2_statuses = TwitterAPI.fetch_user_statuses(user1, %{"actor_id" => user2.ap_id})
143
144 assert length(user2_statuses) == 1
145 assert Enum.at(user2_statuses, 0) == ActivityRepresenter.to_map(status2, %{user: user2})
146 end
147
148 test "fetch a single status" do
149 {:ok, activity} = ActivityBuilder.insert()
150 {:ok, user} = UserBuilder.insert()
151 actor = Repo.get_by!(User, ap_id: activity.data["actor"])
152
153 status = TwitterAPI.fetch_status(user, activity.id)
154
155 assert status == ActivityRepresenter.to_map(activity, %{for: user, user: actor})
156 end
157
158 test "Follow another user using user_id" do
159 user = insert(:user)
160 followed = insert(:user)
161
162 {:ok, user, followed, _activity } = TwitterAPI.follow(user, %{"user_id" => followed.id})
163 assert user.following == [User.ap_followers(followed)]
164
165 { :error, msg } = TwitterAPI.follow(user, %{"user_id" => followed.id})
166 assert msg == "Could not follow user: #{followed.nickname} is already on your list."
167 end
168
169 test "Follow another user using screen_name" do
170 user = insert(:user)
171 followed = insert(:user)
172
173 {:ok, user, followed, _activity } = TwitterAPI.follow(user, %{"screen_name" => followed.nickname})
174 assert user.following == [User.ap_followers(followed)]
175
176 { :error, msg } = TwitterAPI.follow(user, %{"screen_name" => followed.nickname})
177 assert msg == "Could not follow user: #{followed.nickname} is already on your list."
178 end
179
180 test "Unfollow another user using user_id" do
181 unfollowed = insert(:user)
182 user = insert(:user, %{following: [User.ap_followers(unfollowed)]})
183
184 {:ok, user, unfollowed } = TwitterAPI.unfollow(user, %{"user_id" => unfollowed.id})
185 assert user.following == []
186
187 { :error, msg } = TwitterAPI.unfollow(user, %{"user_id" => unfollowed.id})
188 assert msg == "Not subscribed!"
189 end
190
191 test "Unfollow another user using screen_name" do
192 unfollowed = insert(:user)
193 user = insert(:user, %{following: [User.ap_followers(unfollowed)]})
194
195 {:ok, user, unfollowed } = TwitterAPI.unfollow(user, %{"screen_name" => unfollowed.nickname})
196 assert user.following == []
197
198 { :error, msg } = TwitterAPI.unfollow(user, %{"screen_name" => unfollowed.nickname})
199 assert msg == "Not subscribed!"
200 end
201
202 test "fetch statuses in a context using the conversation id" do
203 {:ok, user} = UserBuilder.insert()
204 {:ok, activity} = ActivityBuilder.insert(%{"context" => "2hu"})
205 {:ok, activity_two} = ActivityBuilder.insert(%{"context" => "2hu"})
206 {:ok, _activity_three} = ActivityBuilder.insert(%{"context" => "3hu"})
207
208 {:ok, object} = Object.context_mapping("2hu") |> Repo.insert
209
210 statuses = TwitterAPI.fetch_conversation(user, object.id)
211
212 assert length(statuses) == 2
213 assert Enum.at(statuses, 0)["id"] == activity.id
214 assert Enum.at(statuses, 1)["id"] == activity_two.id
215 end
216
217 test "upload a file" do
218 file = %Plug.Upload{content_type: "image/jpg", path: Path.absname("test/fixtures/image.jpg"), filename: "an_image.jpg"}
219
220 response = TwitterAPI.upload(file)
221
222 assert is_binary(response)
223 end
224
225 test "it can parse mentions and return the relevant users" do
226 text = "@gsimg According to @archaeme , that is @daggsy."
227
228 gsimg = insert(:user, %{nickname: "gsimg"})
229 archaeme = insert(:user, %{nickname: "archaeme"})
230
231 expected_result = [
232 {"@gsimg", gsimg},
233 {"@archaeme", archaeme}
234 ]
235
236 assert TwitterAPI.parse_mentions(text) == expected_result
237 end
238
239 test "it adds user links to an existing text" do
240 text = "@gsimg According to @archaeme , that is @daggsy."
241
242 gsimg = insert(:user, %{nickname: "gsimg"})
243 archaeme = insert(:user, %{nickname: "archaeme"})
244
245 mentions = TwitterAPI.parse_mentions(text)
246 expected_text = "<a href='#{gsimg.ap_id}'>@gsimg</a> According to <a href='#{archaeme.ap_id}'>@archaeme</a> , that is @daggsy."
247
248 assert TwitterAPI.add_user_links(text, mentions) == expected_text
249 end
250
251 test "it favorites a status, returns the updated status" do
252 user = insert(:user)
253 note_activity = insert(:note_activity)
254 activity_user = Repo.get_by!(User, ap_id: note_activity.data["actor"])
255
256 {:ok, status} = TwitterAPI.favorite(user, note_activity)
257 updated_activity = Activity.get_by_ap_id(note_activity.data["id"])
258
259 assert status == ActivityRepresenter.to_map(updated_activity, %{user: activity_user, for: user})
260 end
261
262 test "it unfavorites a status, returns the updated status" do
263 user = insert(:user)
264 note_activity = insert(:note_activity)
265 activity_user = Repo.get_by!(User, ap_id: note_activity.data["actor"])
266 object = Object.get_by_ap_id(note_activity.data["object"]["id"])
267
268 {:ok, _like_activity, _object } = ActivityPub.like(user, object)
269 updated_activity = Activity.get_by_ap_id(note_activity.data["id"])
270 assert ActivityRepresenter.to_map(updated_activity, %{user: activity_user, for: user})["fave_num"] == 1
271
272 {:ok, status} = TwitterAPI.unfavorite(user, note_activity)
273
274 assert status["fave_num"] == 0
275 end
276
277 test "it retweets a status and returns the retweet" do
278 user = insert(:user)
279 note_activity = insert(:note_activity)
280 activity_user = Repo.get_by!(User, ap_id: note_activity.data["actor"])
281
282 {:ok, status} = TwitterAPI.retweet(user, note_activity)
283 updated_activity = Activity.get_by_ap_id(note_activity.data["id"])
284
285 assert status == ActivityRepresenter.to_map(updated_activity, %{user: activity_user, for: user})
286 end
287
288 test "it registers a new user and returns the user." do
289 data = %{
290 "nickname" => "lain",
291 "email" => "lain@wired.jp",
292 "fullname" => "lain iwakura",
293 "bio" => "close the world.",
294 "password" => "bear",
295 "confirm" => "bear"
296 }
297
298 {:ok, user} = TwitterAPI.register_user(data)
299
300 fetched_user = Repo.get_by(User, nickname: "lain")
301 assert user == UserRepresenter.to_map(fetched_user)
302 end
303
304 test "it returns the error on registration problems" do
305 data = %{
306 "nickname" => "lain",
307 "email" => "lain@wired.jp",
308 "fullname" => "lain iwakura",
309 "bio" => "close the world.",
310 "password" => "bear"
311 }
312
313 {:error, error_object} = TwitterAPI.register_user(data)
314
315 assert is_binary(error_object[:error])
316 refute Repo.get_by(User, nickname: "lain")
317 end
318
319 test "it assigns an integer conversation_id" do
320 note_activity = insert(:note_activity)
321 user = User.get_cached_by_ap_id(note_activity.data["actor"])
322 status = ActivityRepresenter.to_map(note_activity, %{user: user})
323
324 assert is_number(status["statusnet_conversation_id"])
325 end
326
327 setup do
328 Supervisor.terminate_child(Pleroma.Supervisor, Cachex)
329 Supervisor.restart_child(Pleroma.Supervisor, Cachex)
330 :ok
331 end
332
333 describe "context_to_conversation_id" do
334 test "creates a mapping object" do
335 conversation_id = TwitterAPI.context_to_conversation_id("random context")
336 object = Object.get_by_ap_id("random context")
337
338 assert conversation_id == object.id
339 end
340
341 test "returns an existing mapping for an existing object" do
342 {:ok, object} = Object.context_mapping("random context") |> Repo.insert
343 conversation_id = TwitterAPI.context_to_conversation_id("random context")
344
345 assert conversation_id == object.id
346 end
347 end
348 end