user_test.exs 17 KB
Newer Older
lain's avatar
lain committed
1 2
defmodule Pleroma.UserTest do
  alias Pleroma.Builders.UserBuilder
lain's avatar
lain committed
3
  alias Pleroma.{User, Repo, Activity}
lain's avatar
lain committed
4 5
  alias Pleroma.Web.OStatus
  alias Pleroma.Web.Websub.WebsubClientSubscription
6
  alias Pleroma.Web.CommonAPI
lain's avatar
lain committed
7 8
  use Pleroma.DataCase

lain's avatar
lain committed
9
  import Pleroma.Factory
lain's avatar
lain committed
10
  import Ecto.Query
lain's avatar
lain committed
11

lain's avatar
lain committed
12
  test "ap_id returns the activity pub id for the user" do
lain's avatar
lain committed
13
    user = UserBuilder.build()
lain's avatar
lain committed
14

lain's avatar
lain committed
15
    expected_ap_id = "#{Pleroma.Web.base_url()}/users/#{user.nickname}"
lain's avatar
lain committed
16 17 18 19 20

    assert expected_ap_id == User.ap_id(user)
  end

  test "ap_followers returns the followers collection for the user" do
lain's avatar
lain committed
21
    user = UserBuilder.build()
lain's avatar
lain committed
22 23 24 25 26

    expected_followers_collection = "#{User.ap_id(user)}/followers"

    assert expected_followers_collection == User.ap_followers(user)
  end
lain's avatar
lain committed
27

lain's avatar
lain committed
28
  test "follow takes a user and another user" do
lain's avatar
lain committed
29 30
    user = insert(:user)
    followed = insert(:user)
lain's avatar
lain committed
31

lain's avatar
lain committed
32
    {:ok, user} = User.follow(user, followed)
lain's avatar
lain committed
33 34 35

    user = Repo.get(User, user.id)

36 37 38
    followed = User.get_by_ap_id(followed.ap_id)
    assert followed.info["follower_count"] == 1

39
    assert User.ap_followers(followed) in user.following
lain's avatar
lain committed
40
  end
lain's avatar
lain committed
41

lain's avatar
lain committed
42 43 44 45 46 47 48
  test "can't follow a deactivated users" do
    user = insert(:user)
    followed = insert(:user, info: %{"deactivated" => true})

    {:error, _} = User.follow(user, followed)
  end

49 50 51 52 53 54 55 56 57
  test "can't follow a user who blocked us" do
    blocker = insert(:user)
    blockee = insert(:user)

    {:ok, blocker} = User.block(blocker, blockee)

    {:error, _} = User.follow(blockee, blocker)
  end

kaniini's avatar
kaniini committed
58 59 60 61 62 63 64 65 66
  test "local users do not automatically follow local locked accounts" do
    follower = insert(:user, info: %{"locked" => true})
    followed = insert(:user, info: %{"locked" => true})

    {:ok, follower} = User.maybe_direct_follow(follower, followed)

    refute User.following?(follower, followed)
  end

lain's avatar
lain committed
67 68 69 70
  # This is a somewhat useless test.
  # test "following a remote user will ensure a websub subscription is present" do
  #   user = insert(:user)
  #   {:ok, followed} = OStatus.make_user("shp@social.heldscal.la")
lain's avatar
lain committed
71

lain's avatar
lain committed
72
  #   assert followed.local == false
lain's avatar
lain committed
73

lain's avatar
lain committed
74 75
  #   {:ok, user} = User.follow(user, followed)
  #   assert User.ap_followers(followed) in user.following
lain's avatar
lain committed
76

lain's avatar
lain committed
77 78 79
  #   query = from w in WebsubClientSubscription,
  #   where: w.topic == ^followed.info["topic"]
  #   websub = Repo.one(query)
lain's avatar
lain committed
80

lain's avatar
lain committed
81 82
  #   assert websub
  # end
lain's avatar
lain committed
83

lain's avatar
lain committed
84
  test "unfollow takes a user and another user" do
lain's avatar
lain committed
85 86
    followed = insert(:user)
    user = insert(:user, %{following: [User.ap_followers(followed)]})
lain's avatar
lain committed
87

lain's avatar
lain committed
88
    {:ok, user, _activity} = User.unfollow(user, followed)
lain's avatar
lain committed
89 90 91 92 93

    user = Repo.get(User, user.id)

    assert user.following == []
  end
94

95 96 97 98 99
  test "unfollow doesn't unfollow yourself" do
    user = insert(:user)

    {:error, _} = User.unfollow(user, user)

eal's avatar
eal committed
100
    user = Repo.get(User, user.id)
101 102 103
    assert user.following == [user.ap_id]
  end

104
  test "test if a user is following another user" do
lain's avatar
lain committed
105 106
    followed = insert(:user)
    user = insert(:user, %{following: [User.ap_followers(followed)]})
107 108 109 110

    assert User.following?(user, followed)
    refute User.following?(followed, user)
  end
lain's avatar
lain committed
111 112 113 114 115 116 117 118 119 120 121

  describe "user registration" do
    @full_user_data %{
      bio: "A guy",
      name: "my name",
      nickname: "nick",
      password: "test",
      password_confirmation: "test",
      email: "email@example.com"
    }

vaartis's avatar
vaartis committed
122
    test "it requires an email, name, nickname and password, bio is optional" do
lain's avatar
lain committed
123
      @full_user_data
lain's avatar
lain committed
124 125
      |> Map.keys()
      |> Enum.each(fn key ->
lain's avatar
lain committed
126 127
        params = Map.delete(@full_user_data, key)
        changeset = User.register_changeset(%User{}, params)
vaartis's avatar
vaartis committed
128

lain's avatar
lain committed
129
        assert if key == :bio, do: changeset.valid?, else: not changeset.valid?
lain's avatar
lain committed
130 131 132 133 134 135 136 137 138 139
      end)
    end

    test "it sets the password_hash, ap_id and following fields" do
      changeset = User.register_changeset(%User{}, @full_user_data)

      assert changeset.valid?

      assert is_binary(changeset.changes[:password_hash])
      assert changeset.changes[:ap_id] == User.ap_id(%User{nickname: @full_user_data.nickname})
lain's avatar
lain committed
140 141 142 143 144

      assert changeset.changes[:following] == [
               User.ap_followers(%User{nickname: @full_user_data.nickname})
             ]

145
      assert changeset.changes.follower_address == "#{changeset.changes.ap_id}/followers"
lain's avatar
lain committed
146 147
    end
  end
lain's avatar
lain committed
148 149 150 151 152 153 154 155 156

  describe "fetching a user from nickname or trying to build one" do
    test "gets an existing user" do
      user = insert(:user)
      fetched_user = User.get_or_fetch_by_nickname(user.nickname)

      assert user == fetched_user
    end

157 158 159 160 161 162 163
    test "gets an existing user, case insensitive" do
      user = insert(:user, nickname: "nick")
      fetched_user = User.get_or_fetch_by_nickname("NICK")

      assert user == fetched_user
    end

lain's avatar
lain committed
164 165 166 167 168 169 170 171 172
    test "fetches an external user via ostatus if no user exists" do
      fetched_user = User.get_or_fetch_by_nickname("shp@social.heldscal.la")
      assert fetched_user.nickname == "shp@social.heldscal.la"
    end

    test "returns nil if no user could be fetched" do
      fetched_user = User.get_or_fetch_by_nickname("nonexistant@social.heldscal.la")
      assert fetched_user == nil
    end
lain's avatar
lain committed
173 174 175 176 177

    test "returns nil for nonexistant local user" do
      fetched_user = User.get_or_fetch_by_nickname("nonexistant")
      assert fetched_user == nil
    end
178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196

    test "updates an existing user, if stale" do
      a_week_ago = NaiveDateTime.add(NaiveDateTime.utc_now(), -604_800)

      orig_user =
        insert(
          :user,
          local: false,
          nickname: "admin@mastodon.example.org",
          ap_id: "http://mastodon.example.org/users/admin",
          last_refreshed_at: a_week_ago
        )

      assert orig_user.last_refreshed_at == a_week_ago

      user = User.get_or_fetch_by_ap_id("http://mastodon.example.org/users/admin")

      refute user.last_refreshed_at == orig_user.last_refreshed_at
    end
lain's avatar
lain committed
197
  end
198 199 200

  test "returns an ap_id for a user" do
    user = insert(:user)
lain's avatar
lain committed
201 202 203 204 205 206 207

    assert User.ap_id(user) ==
             Pleroma.Web.Router.Helpers.o_status_url(
               Pleroma.Web.Endpoint,
               :feed_redirect,
               user.nickname
             )
208 209 210 211
  end

  test "returns an ap_followers link for a user" do
    user = insert(:user)
lain's avatar
lain committed
212 213 214 215 216 217 218

    assert User.ap_followers(user) ==
             Pleroma.Web.Router.Helpers.o_status_url(
               Pleroma.Web.Endpoint,
               :feed_redirect,
               user.nickname
             ) <> "/followers"
219
  end
lain's avatar
lain committed
220 221 222 223 224 225 226

  describe "remote user creation changeset" do
    @valid_remote %{
      bio: "hello",
      name: "Someone",
      nickname: "a@b.de",
      ap_id: "http...",
lain's avatar
lain committed
227 228
      info: %{some: "info"},
      avatar: %{some: "avatar"}
lain's avatar
lain committed
229 230 231 232 233 234 235
    }

    test "it confirms validity" do
      cs = User.remote_user_creation(@valid_remote)
      assert cs.valid?
    end

236 237 238
    test "it sets the follower_adress" do
      cs = User.remote_user_creation(@valid_remote)
      # remote users get a fake local follower address
lain's avatar
lain committed
239 240
      assert cs.changes.follower_address ==
               User.ap_followers(%User{nickname: @valid_remote[:nickname]})
241 242
    end

lain's avatar
lain committed
243 244
    test "it enforces the fqn format for nicknames" do
      cs = User.remote_user_creation(%{@valid_remote | nickname: "bla"})
lain's avatar
lain committed
245 246
      assert cs.changes.local == false
      assert cs.changes.avatar
lain's avatar
lain committed
247 248 249 250
      refute cs.valid?
    end

    test "it has required fields" do
251
      [:name, :ap_id]
lain's avatar
lain committed
252
      |> Enum.each(fn field ->
lain's avatar
lain committed
253 254 255 256 257 258
        cs = User.remote_user_creation(Map.delete(@valid_remote, field))
        refute cs.valid?
      end)
    end

    test "it restricts some sizes" do
lain's avatar
lain committed
259
      [bio: 5000, name: 100]
lain's avatar
lain committed
260
      |> Enum.each(fn {field, size} ->
lain's avatar
lain committed
261 262 263 264 265 266 267 268 269 270
        string = String.pad_leading(".", size)
        cs = User.remote_user_creation(Map.put(@valid_remote, field, string))
        assert cs.valid?

        string = String.pad_leading(".", size + 1)
        cs = User.remote_user_creation(Map.put(@valid_remote, field, string))
        refute cs.valid?
      end)
    end
  end
lain's avatar
lain committed
271 272 273 274 275 276 277 278 279 280 281 282 283

  describe "followers and friends" do
    test "gets all followers for a given user" do
      user = insert(:user)
      follower_one = insert(:user)
      follower_two = insert(:user)
      not_follower = insert(:user)

      {:ok, follower_one} = User.follow(follower_one, user)
      {:ok, follower_two} = User.follow(follower_two, user)

      {:ok, res} = User.get_followers(user)

lain's avatar
lain committed
284 285 286
      assert Enum.member?(res, follower_one)
      assert Enum.member?(res, follower_two)
      refute Enum.member?(res, not_follower)
lain's avatar
lain committed
287 288 289 290 291 292 293 294 295 296 297 298 299
    end

    test "gets all friends (followed users) for a given user" do
      user = insert(:user)
      followed_one = insert(:user)
      followed_two = insert(:user)
      not_followed = insert(:user)

      {:ok, user} = User.follow(user, followed_one)
      {:ok, user} = User.follow(user, followed_two)

      {:ok, res} = User.get_friends(user)

300 301
      followed_one = User.get_by_ap_id(followed_one.ap_id)
      followed_two = User.get_by_ap_id(followed_two.ap_id)
lain's avatar
lain committed
302 303 304
      assert Enum.member?(res, followed_one)
      assert Enum.member?(res, followed_two)
      refute Enum.member?(res, not_followed)
lain's avatar
lain committed
305 306
    end
  end
307 308 309 310 311 312 313 314 315 316 317 318 319 320

  describe "updating note and follower count" do
    test "it sets the info->note_count property" do
      note = insert(:note)

      user = User.get_by_ap_id(note.data["actor"])

      assert user.info["note_count"] == nil

      {:ok, user} = User.update_note_count(user)

      assert user.info["note_count"] == 1
    end

321 322 323 324
    test "it increases the info->note_count property" do
      note = insert(:note)
      user = User.get_by_ap_id(note.data["actor"])

lain's avatar
lain committed
325
      assert user.info.note_count == 0
326 327 328

      {:ok, user} = User.increase_note_count(user)

lain's avatar
lain committed
329
      assert user.info.note_count == 1
330 331 332

      {:ok, user} = User.increase_note_count(user)

lain's avatar
lain committed
333
      assert user.info.note_count == 2
334 335
    end

336 337 338 339
    test "it decreases the info->note_count property" do
      note = insert(:note)
      user = User.get_by_ap_id(note.data["actor"])

lain's avatar
lain committed
340
      assert user.info.note_count == 0
341 342 343

      {:ok, user} = User.increase_note_count(user)

lain's avatar
lain committed
344
      assert user.info.note_count == 1
345 346 347

      {:ok, user} = User.decrease_note_count(user)

lain's avatar
lain committed
348
      assert user.info.note_count == 0
349 350 351

      {:ok, user} = User.decrease_note_count(user)

lain's avatar
lain committed
352
      assert user.info.note_count == 0
353 354
    end

355 356 357 358 359 360 361 362 363 364 365 366 367
    test "it sets the info->follower_count property" do
      user = insert(:user)
      follower = insert(:user)

      User.follow(follower, user)

      assert user.info["follower_count"] == nil

      {:ok, user} = User.update_follower_count(user)

      assert user.info["follower_count"] == 1
    end
  end
lain's avatar
lain committed
368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389

  describe "blocks" do
    test "it blocks people" do
      user = insert(:user)
      blocked_user = insert(:user)

      refute User.blocks?(user, blocked_user)

      {:ok, user} = User.block(user, blocked_user)

      assert User.blocks?(user, blocked_user)
    end

    test "it unblocks users" do
      user = insert(:user)
      blocked_user = insert(:user)

      {:ok, user} = User.block(user, blocked_user)
      {:ok, user} = User.unblock(user, blocked_user)

      refute User.blocks?(user, blocked_user)
    end
390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444

    test "blocks tear down cyclical follow relationships" do
      blocker = insert(:user)
      blocked = insert(:user)

      {:ok, blocker} = User.follow(blocker, blocked)
      {:ok, blocked} = User.follow(blocked, blocker)

      assert User.following?(blocker, blocked)
      assert User.following?(blocked, blocker)

      {:ok, blocker} = User.block(blocker, blocked)
      blocked = Repo.get(User, blocked.id)

      assert User.blocks?(blocker, blocked)

      refute User.following?(blocker, blocked)
      refute User.following?(blocked, blocker)
    end

    test "blocks tear down blocker->blocked follow relationships" do
      blocker = insert(:user)
      blocked = insert(:user)

      {:ok, blocker} = User.follow(blocker, blocked)

      assert User.following?(blocker, blocked)
      refute User.following?(blocked, blocker)

      {:ok, blocker} = User.block(blocker, blocked)
      blocked = Repo.get(User, blocked.id)

      assert User.blocks?(blocker, blocked)

      refute User.following?(blocker, blocked)
      refute User.following?(blocked, blocker)
    end

    test "blocks tear down blocked->blocker follow relationships" do
      blocker = insert(:user)
      blocked = insert(:user)

      {:ok, blocked} = User.follow(blocked, blocker)

      refute User.following?(blocker, blocked)
      assert User.following?(blocked, blocker)

      {:ok, blocker} = User.block(blocker, blocked)
      blocked = Repo.get(User, blocked.id)

      assert User.blocks?(blocker, blocked)

      refute User.following?(blocker, blocked)
      refute User.following?(blocked, blocker)
    end
lain's avatar
lain committed
445
  end
446

eal's avatar
eal committed
447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467
  describe "domain blocking" do
    test "blocks domains" do
      user = insert(:user)
      collateral_user = insert(:user, %{ap_id: "https://awful-and-rude-instance.com/user/bully"})

      {:ok, user} = User.block_domain(user, "awful-and-rude-instance.com")

      assert User.blocks?(user, collateral_user)
    end

    test "unblocks domains" do
      user = insert(:user)
      collateral_user = insert(:user, %{ap_id: "https://awful-and-rude-instance.com/user/bully"})

      {:ok, user} = User.block_domain(user, "awful-and-rude-instance.com")
      {:ok, user} = User.unblock_domain(user, "awful-and-rude-instance.com")

      refute User.blocks?(user, collateral_user)
    end
  end

468 469 470 471 472 473
  test "get recipients from activity" do
    actor = insert(:user)
    user = insert(:user, local: true)
    user_two = insert(:user, local: false)
    addressed = insert(:user, local: true)
    addressed_remote = insert(:user, local: false)
lain's avatar
lain committed
474 475 476 477 478

    {:ok, activity} =
      CommonAPI.post(actor, %{
        "status" => "hey @#{addressed.nickname} @#{addressed_remote.nickname}"
      })
479 480 481 482

    assert [addressed] == User.get_recipients_from_activity(activity)

    {:ok, user} = User.follow(user, actor)
lain's avatar
lain committed
483
    {:ok, _user_two} = User.follow(user_two, actor)
484 485 486 487 488
    recipients = User.get_recipients_from_activity(activity)
    assert length(recipients) == 2
    assert user in recipients
    assert addressed in recipients
  end
lain's avatar
lain committed
489

scarlett's avatar
scarlett committed
490
  test ".deactivate can de-activate then re-activate a user" do
lain's avatar
lain committed
491
    user = insert(:user)
lain's avatar
lain committed
492
    assert false == user.info.deactivated
lain's avatar
lain committed
493
    {:ok, user} = User.deactivate(user)
lain's avatar
lain committed
494
    assert true == user.info.deactivated
scarlett's avatar
scarlett committed
495
    {:ok, user} = User.deactivate(user, false)
lain's avatar
lain committed
496
    assert false == user.info.deactivated
lain's avatar
lain committed
497
  end
lain's avatar
lain committed
498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513

  test ".delete deactivates a user, all follow relationships and all create activities" do
    user = insert(:user)
    followed = insert(:user)
    follower = insert(:user)

    {:ok, user} = User.follow(user, followed)
    {:ok, follower} = User.follow(follower, user)

    {:ok, activity} = CommonAPI.post(user, %{"status" => "2hu"})
    {:ok, activity_two} = CommonAPI.post(follower, %{"status" => "3hu"})

    {:ok, _, _} = CommonAPI.favorite(activity_two.id, user)
    {:ok, _, _} = CommonAPI.favorite(activity.id, follower)
    {:ok, _, _} = CommonAPI.repeat(activity.id, follower)

514
    {:ok, _} = User.delete(user)
lain's avatar
lain committed
515 516 517 518 519 520 521 522 523 524

    followed = Repo.get(User, followed.id)
    follower = Repo.get(User, follower.id)
    user = Repo.get(User, user.id)

    assert user.info["deactivated"]

    refute User.following?(user, followed)
    refute User.following?(followed, follower)

lain's avatar
lain committed
525 526 527
    # TODO: Remove favorites, repeats, delete activities.

    refute Repo.get(Activity, activity.id)
lain's avatar
lain committed
528
  end
529 530 531 532

  test "get_public_key_for_ap_id fetches a user that's not in the db" do
    assert {:ok, _key} = User.get_public_key_for_ap_id("http://mastodon.example.org/users/admin")
  end
lain's avatar
lain committed
533 534 535

  test "insert or update a user from given data" do
    user = insert(:user, %{nickname: "nick@name.de"})
lain's avatar
lain committed
536
    data = %{ap_id: user.ap_id <> "xxx", name: user.name, nickname: user.nickname}
lain's avatar
lain committed
537 538 539

    assert {:ok, %User{}} = User.insert_or_update_user(data)
  end
540 541 542 543 544 545 546 547 548 549 550 551 552 553

  describe "per-user rich-text filtering" do
    test "html_filter_policy returns nil when rich-text is enabled" do
      user = insert(:user)

      assert nil == User.html_filter_policy(user)
    end

    test "html_filter_policy returns TwitterText scrubber when rich-text is disabled" do
      user = insert(:user, %{info: %{"no_rich_text" => true}})

      assert Pleroma.HTML.Scrubber.TwitterText == User.html_filter_policy(user)
    end
  end
554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580

  describe "caching" do
    test "invalidate_cache works" do
      user = insert(:user)
      user_info = User.get_cached_user_info(user)

      User.invalidate_cache(user)

      {:ok, nil} = Cachex.get(:user_cache, "ap_id:#{user.ap_id}")
      {:ok, nil} = Cachex.get(:user_cache, "nickname:#{user.nickname}")
      {:ok, nil} = Cachex.get(:user_cache, "user_info:#{user.id}")
    end

    test "User.delete() plugs any possible zombie objects" do
      user = insert(:user)

      {:ok, _} = User.delete(user)

      {:ok, cached_user} = Cachex.get(:user_cache, "ap_id:#{user.ap_id}")

      assert cached_user != user

      {:ok, cached_user} = Cachex.get(:user_cache, "nickname:#{user.ap_id}")

      assert cached_user != user
    end
  end
581 582 583 584 585 586 587 588 589 590 591 592

  describe "User.search" do
    test "finds a user, ranking by similarity" do
      user = insert(:user, %{name: "lain"})
      user_two = insert(:user, %{name: "ean"})
      user_three = insert(:user, %{name: "ebn", nickname: "lain@mastodon.social"})
      user_four = insert(:user, %{nickname: "lain@pleroma.soykaf.com"})

      assert user_four ==
               User.search("lain@ple") |> List.first() |> Map.put(:search_distance, nil)
    end
  end
lain's avatar
lain committed
593
end