git.squeep.com Git - akkoma/blob - test/web/mastodon_api/controllers/search_controller_test.exs

   1 # Pleroma: A lightweight social networking server
   2 # Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
   3 # SPDX-License-Identifier: AGPL-3.0-only
   4
   5 defmodule Pleroma.Web.MastodonAPI.SearchControllerTest do
   6   use Pleroma.Web.ConnCase
   7
   8   alias Pleroma.Object
   9   alias Pleroma.Web
  10   alias Pleroma.Web.CommonAPI
  11   import Pleroma.Factory
  12   import ExUnit.CaptureLog
  13   import Tesla.Mock
  14   import Mock
  15
  16   setup_all do
  17     mock_global(fn env -> apply(HttpRequestMock, :request, [env]) end)
  18     :ok
  19   end
  20
  21   describe ".search2" do
  22     test "it returns empty result if user or status search return undefined error", %{conn: conn} do
  23       with_mocks [
  24         {Pleroma.User, [], [search: fn _q, _o -> raise "Oops" end]},
  25         {Pleroma.Activity, [], [search: fn _u, _q, _o -> raise "Oops" end]}
  26       ] do
  27         capture_log(fn ->
  28           results =
  29             conn
  30             |> get("/api/v2/search?q=2hu")
  31             |> json_response_and_validate_schema(200)
  32
  33           assert results["accounts"] == []
  34           assert results["statuses"] == []
  35         end) =~
  36           "[error] Elixir.Pleroma.Web.MastodonAPI.SearchController search error: %RuntimeError{message: \"Oops\"}"
  37       end
  38     end
  39
  40     test "search", %{conn: conn} do
  41       user = insert(:user)
  42       user_two = insert(:user, %{nickname: "shp@shitposter.club"})
  43       user_three = insert(:user, %{nickname: "shp@heldscal.la", name: "I love 2hu"})
  44
  45       {:ok, activity} = CommonAPI.post(user, %{status: "This is about 2hu private 天子"})
  46
  47       {:ok, _activity} =
  48         CommonAPI.post(user, %{
  49           status: "This is about 2hu, but private",
  50           visibility: "private"
  51         })
  52
  53       {:ok, _} = CommonAPI.post(user_two, %{status: "This isn't"})
  54
  55       results =
  56         conn
  57         |> get("/api/v2/search?#{URI.encode_query(%{q: "2hu #private"})}")
  58         |> json_response_and_validate_schema(200)
  59
  60       [account | _] = results["accounts"]
  61       assert account["id"] == to_string(user_three.id)
  62
  63       assert results["hashtags"] == [
  64                %{"name" => "private", "url" => "#{Web.base_url()}/tag/private"}
  65              ]
  66
  67       [status] = results["statuses"]
  68       assert status["id"] == to_string(activity.id)
  69
  70       results =
  71         get(conn, "/api/v2/search?q=天子")
  72         |> json_response_and_validate_schema(200)
  73
  74       assert results["hashtags"] == [
  75                %{"name" => "天子", "url" => "#{Web.base_url()}/tag/天子"}
  76              ]
  77
  78       [status] = results["statuses"]
  79       assert status["id"] == to_string(activity.id)
  80     end
  81
  82     test "constructs hashtags from search query", %{conn: conn} do
  83       results =
  84         conn
  85         |> get("/api/v2/search?#{URI.encode_query(%{q: "some text with #explicit #hashtags"})}")
  86         |> json_response_and_validate_schema(200)
  87
  88       assert results["hashtags"] == [
  89                %{"name" => "explicit", "url" => "#{Web.base_url()}/tag/explicit"},
  90                %{"name" => "hashtags", "url" => "#{Web.base_url()}/tag/hashtags"}
  91              ]
  92
  93       results =
  94         conn
  95         |> get("/api/v2/search?#{URI.encode_query(%{q: "john doe JOHN DOE"})}")
  96         |> json_response_and_validate_schema(200)
  97
  98       assert results["hashtags"] == [
  99                %{"name" => "john", "url" => "#{Web.base_url()}/tag/john"},
 100                %{"name" => "doe", "url" => "#{Web.base_url()}/tag/doe"},
 101                %{"name" => "JohnDoe", "url" => "#{Web.base_url()}/tag/JohnDoe"}
 102              ]
 103
 104       results =
 105         conn
 106         |> get("/api/v2/search?#{URI.encode_query(%{q: "accident-prone"})}")
 107         |> json_response_and_validate_schema(200)
 108
 109       assert results["hashtags"] == [
 110                %{"name" => "accident", "url" => "#{Web.base_url()}/tag/accident"},
 111                %{"name" => "prone", "url" => "#{Web.base_url()}/tag/prone"},
 112                %{"name" => "AccidentProne", "url" => "#{Web.base_url()}/tag/AccidentProne"}
 113              ]
 114
 115       results =
 116         conn
 117         |> get("/api/v2/search?#{URI.encode_query(%{q: "https://shpposter.club/users/shpuld"})}")
 118         |> json_response_and_validate_schema(200)
 119
 120       assert results["hashtags"] == [
 121                %{"name" => "shpuld", "url" => "#{Web.base_url()}/tag/shpuld"}
 122              ]
 123
 124       results =
 125         conn
 126         |> get(
 127           "/api/v2/search?#{
 128             URI.encode_query(%{
 129               q:
 130                 "https://www.washingtonpost.com/sports/2020/06/10/" <>
 131                   "nascar-ban-display-confederate-flag-all-events-properties/"
 132             })
 133           }"
 134         )
 135         |> json_response_and_validate_schema(200)
 136
 137       assert results["hashtags"] == [
 138                %{"name" => "nascar", "url" => "#{Web.base_url()}/tag/nascar"},
 139                %{"name" => "ban", "url" => "#{Web.base_url()}/tag/ban"},
 140                %{"name" => "display", "url" => "#{Web.base_url()}/tag/display"},
 141                %{"name" => "confederate", "url" => "#{Web.base_url()}/tag/confederate"},
 142                %{"name" => "flag", "url" => "#{Web.base_url()}/tag/flag"},
 143                %{"name" => "all", "url" => "#{Web.base_url()}/tag/all"},
 144                %{"name" => "events", "url" => "#{Web.base_url()}/tag/events"},
 145                %{"name" => "properties", "url" => "#{Web.base_url()}/tag/properties"},
 146                %{
 147                  "name" => "NascarBanDisplayConfederateFlagAllEventsProperties",
 148                  "url" =>
 149                    "#{Web.base_url()}/tag/NascarBanDisplayConfederateFlagAllEventsProperties"
 150                }
 151              ]
 152     end
 153
 154     test "supports pagination of hashtags search results", %{conn: conn} do
 155       results =
 156         conn
 157         |> get(
 158           "/api/v2/search?#{
 159             URI.encode_query(%{q: "#some #text #with #hashtags", limit: 2, offset: 1})
 160           }"
 161         )
 162         |> json_response_and_validate_schema(200)
 163
 164       assert results["hashtags"] == [
 165                %{"name" => "text", "url" => "#{Web.base_url()}/tag/text"},
 166                %{"name" => "with", "url" => "#{Web.base_url()}/tag/with"}
 167              ]
 168     end
 169
 170     test "excludes a blocked users from search results", %{conn: conn} do
 171       user = insert(:user)
 172       user_smith = insert(:user, %{nickname: "Agent", name: "I love 2hu"})
 173       user_neo = insert(:user, %{nickname: "Agent Neo", name: "Agent"})
 174
 175       {:ok, act1} = CommonAPI.post(user, %{status: "This is about 2hu private 天子"})
 176       {:ok, act2} = CommonAPI.post(user_smith, %{status: "Agent Smith"})
 177       {:ok, act3} = CommonAPI.post(user_neo, %{status: "Agent Smith"})
 178       Pleroma.User.block(user, user_smith)
 179
 180       results =
 181         conn
 182         |> assign(:user, user)
 183         |> assign(:token, insert(:oauth_token, user: user, scopes: ["read"]))
 184         |> get("/api/v2/search?q=Agent")
 185         |> json_response_and_validate_schema(200)
 186
 187       status_ids = Enum.map(results["statuses"], fn g -> g["id"] end)
 188
 189       assert act3.id in status_ids
 190       refute act2.id in status_ids
 191       refute act1.id in status_ids
 192     end
 193   end
 194
 195   describe ".account_search" do
 196     test "account search", %{conn: conn} do
 197       user_two = insert(:user, %{nickname: "shp@shitposter.club"})
 198       user_three = insert(:user, %{nickname: "shp@heldscal.la", name: "I love 2hu"})
 199
 200       results =
 201         conn
 202         |> get("/api/v1/accounts/search?q=shp")
 203         |> json_response_and_validate_schema(200)
 204
 205       result_ids = for result <- results, do: result["acct"]
 206
 207       assert user_two.nickname in result_ids
 208       assert user_three.nickname in result_ids
 209
 210       results =
 211         conn
 212         |> get("/api/v1/accounts/search?q=2hu")
 213         |> json_response_and_validate_schema(200)
 214
 215       result_ids = for result <- results, do: result["acct"]
 216
 217       assert user_three.nickname in result_ids
 218     end
 219
 220     test "returns account if query contains a space", %{conn: conn} do
 221       insert(:user, %{nickname: "shp@shitposter.club"})
 222
 223       results =
 224         conn
 225         |> get("/api/v1/accounts/search?q=shp@shitposter.club xxx")
 226         |> json_response_and_validate_schema(200)
 227
 228       assert length(results) == 1
 229     end
 230   end
 231
 232   describe ".search" do
 233     test "it returns empty result if user or status search return undefined error", %{conn: conn} do
 234       with_mocks [
 235         {Pleroma.User, [], [search: fn _q, _o -> raise "Oops" end]},
 236         {Pleroma.Activity, [], [search: fn _u, _q, _o -> raise "Oops" end]}
 237       ] do
 238         capture_log(fn ->
 239           results =
 240             conn
 241             |> get("/api/v1/search?q=2hu")
 242             |> json_response_and_validate_schema(200)
 243
 244           assert results["accounts"] == []
 245           assert results["statuses"] == []
 246         end) =~
 247           "[error] Elixir.Pleroma.Web.MastodonAPI.SearchController search error: %RuntimeError{message: \"Oops\"}"
 248       end
 249     end
 250
 251     test "search", %{conn: conn} do
 252       user = insert(:user)
 253       user_two = insert(:user, %{nickname: "shp@shitposter.club"})
 254       user_three = insert(:user, %{nickname: "shp@heldscal.la", name: "I love 2hu"})
 255
 256       {:ok, activity} = CommonAPI.post(user, %{status: "This is about 2hu"})
 257
 258       {:ok, _activity} =
 259         CommonAPI.post(user, %{
 260           status: "This is about 2hu, but private",
 261           visibility: "private"
 262         })
 263
 264       {:ok, _} = CommonAPI.post(user_two, %{status: "This isn't"})
 265
 266       results =
 267         conn
 268         |> get("/api/v1/search?q=2hu")
 269         |> json_response_and_validate_schema(200)
 270
 271       [account | _] = results["accounts"]
 272       assert account["id"] == to_string(user_three.id)
 273
 274       assert results["hashtags"] == ["2hu"]
 275
 276       [status] = results["statuses"]
 277       assert status["id"] == to_string(activity.id)
 278     end
 279
 280     test "search fetches remote statuses and prefers them over other results", %{conn: conn} do
 281       capture_log(fn ->
 282         {:ok, %{id: activity_id}} =
 283           CommonAPI.post(insert(:user), %{
 284             status: "check out https://shitposter.club/notice/2827873"
 285           })
 286
 287         results =
 288           conn
 289           |> get("/api/v1/search?q=https://shitposter.club/notice/2827873")
 290           |> json_response_and_validate_schema(200)
 291
 292         [status, %{"id" => ^activity_id}] = results["statuses"]
 293
 294         assert status["uri"] ==
 295                  "tag:shitposter.club,2017-05-05:noticeId=2827873:objectType=comment"
 296       end)
 297     end
 298
 299     test "search doesn't show statuses that it shouldn't", %{conn: conn} do
 300       {:ok, activity} =
 301         CommonAPI.post(insert(:user), %{
 302           status: "This is about 2hu, but private",
 303           visibility: "private"
 304         })
 305
 306       capture_log(fn ->
 307         q = Object.normalize(activity).data["id"]
 308
 309         results =
 310           conn
 311           |> get("/api/v1/search?q=#{q}")
 312           |> json_response_and_validate_schema(200)
 313
 314         [] = results["statuses"]
 315       end)
 316     end
 317
 318     test "search fetches remote accounts", %{conn: conn} do
 319       user = insert(:user)
 320
 321       results =
 322         conn
 323         |> assign(:user, user)
 324         |> assign(:token, insert(:oauth_token, user: user, scopes: ["read"]))
 325         |> get("/api/v1/search?q=mike@osada.macgirvin.com&resolve=true")
 326         |> json_response_and_validate_schema(200)
 327
 328       [account] = results["accounts"]
 329       assert account["acct"] == "mike@osada.macgirvin.com"
 330     end
 331
 332     test "search doesn't fetch remote accounts if resolve is false", %{conn: conn} do
 333       results =
 334         conn
 335         |> get("/api/v1/search?q=mike@osada.macgirvin.com&resolve=false")
 336         |> json_response_and_validate_schema(200)
 337
 338       assert [] == results["accounts"]
 339     end
 340
 341     test "search with limit and offset", %{conn: conn} do
 342       user = insert(:user)
 343       _user_two = insert(:user, %{nickname: "shp@shitposter.club"})
 344       _user_three = insert(:user, %{nickname: "shp@heldscal.la", name: "I love 2hu"})
 345
 346       {:ok, _activity1} = CommonAPI.post(user, %{status: "This is about 2hu"})
 347       {:ok, _activity2} = CommonAPI.post(user, %{status: "This is also about 2hu"})
 348
 349       result =
 350         conn
 351         |> get("/api/v1/search?q=2hu&limit=1")
 352
 353       assert results = json_response_and_validate_schema(result, 200)
 354       assert [%{"id" => activity_id1}] = results["statuses"]
 355       assert [_] = results["accounts"]
 356
 357       results =
 358         conn
 359         |> get("/api/v1/search?q=2hu&limit=1&offset=1")
 360         |> json_response_and_validate_schema(200)
 361
 362       assert [%{"id" => activity_id2}] = results["statuses"]
 363       assert [] = results["accounts"]
 364
 365       assert activity_id1 != activity_id2
 366     end
 367
 368     test "search returns results only for the given type", %{conn: conn} do
 369       user = insert(:user)
 370       _user_two = insert(:user, %{nickname: "shp@heldscal.la", name: "I love 2hu"})
 371
 372       {:ok, _activity} = CommonAPI.post(user, %{status: "This is about 2hu"})
 373
 374       assert %{"statuses" => [_activity], "accounts" => [], "hashtags" => []} =
 375                conn
 376                |> get("/api/v1/search?q=2hu&type=statuses")
 377                |> json_response_and_validate_schema(200)
 378
 379       assert %{"statuses" => [], "accounts" => [_user_two], "hashtags" => []} =
 380                conn
 381                |> get("/api/v1/search?q=2hu&type=accounts")
 382                |> json_response_and_validate_schema(200)
 383     end
 384
 385     test "search uses account_id to filter statuses by the author", %{conn: conn} do
 386       user = insert(:user, %{nickname: "shp@shitposter.club"})
 387       user_two = insert(:user, %{nickname: "shp@heldscal.la", name: "I love 2hu"})
 388
 389       {:ok, activity1} = CommonAPI.post(user, %{status: "This is about 2hu"})
 390       {:ok, activity2} = CommonAPI.post(user_two, %{status: "This is also about 2hu"})
 391
 392       results =
 393         conn
 394         |> get("/api/v1/search?q=2hu&account_id=#{user.id}")
 395         |> json_response_and_validate_schema(200)
 396
 397       assert [%{"id" => activity_id1}] = results["statuses"]
 398       assert activity_id1 == activity1.id
 399       assert [_] = results["accounts"]
 400
 401       results =
 402         conn
 403         |> get("/api/v1/search?q=2hu&account_id=#{user_two.id}")
 404         |> json_response_and_validate_schema(200)
 405
 406       assert [%{"id" => activity_id2}] = results["statuses"]
 407       assert activity_id2 == activity2.id
 408     end
 409   end
 410 end