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

   1 # Pleroma: A lightweight social networking server
   2 # Copyright © 2017-2021 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.Web.CommonAPI
   9   alias Pleroma.Web.Endpoint
  10   import Pleroma.Factory
  11   import ExUnit.CaptureLog
  12   import Tesla.Mock
  13   import Mock
  14
  15   setup_all do
  16     mock_global(fn env -> apply(HttpRequestMock, :request, [env]) end)
  17     :ok
  18   end
  19
  20   describe ".search2" do
  21     test "it returns empty result if user or status search return undefined error", %{conn: conn} do
  22       with_mocks [
  23         {Pleroma.User, [], [search: fn _q, _o -> raise "Oops" end]},
  24         {Pleroma.Activity, [], [search: fn _u, _q, _o -> raise "Oops" end]}
  25       ] do
  26         capture_log(fn ->
  27           results =
  28             conn
  29             |> get("/api/v2/search?q=2hu")
  30             |> json_response_and_validate_schema(200)
  31
  32           assert results["accounts"] == []
  33           assert results["statuses"] == []
  34         end) =~
  35           "[error] Elixir.Pleroma.Web.MastodonAPI.SearchController search error: %RuntimeError{message: \"Oops\"}"
  36       end
  37     end
  38
  39     test "search", %{conn: conn} do
  40       user = insert(:user)
  41       user_two = insert(:user, %{nickname: "shp@shitposter.club"})
  42       user_three = insert(:user, %{nickname: "shp@heldscal.la", name: "I love 2hu"})
  43
  44       {:ok, activity} = CommonAPI.post(user, %{status: "This is about 2hu private 天子"})
  45
  46       {:ok, _activity} =
  47         CommonAPI.post(user, %{
  48           status: "This is about 2hu, but private",
  49           visibility: "private"
  50         })
  51
  52       {:ok, _} = CommonAPI.post(user_two, %{status: "This isn't"})
  53
  54       results =
  55         conn
  56         |> get("/api/v2/search?#{URI.encode_query(%{q: "2hu #private"})}")
  57         |> json_response_and_validate_schema(200)
  58
  59       [account | _] = results["accounts"]
  60       assert account["id"] == to_string(user_three.id)
  61
  62       assert results["hashtags"] == [
  63                %{"name" => "private", "url" => "#{Endpoint.url()}/tag/private"}
  64              ]
  65
  66       [status] = results["statuses"]
  67       assert status["id"] == to_string(activity.id)
  68
  69       results =
  70         get(conn, "/api/v2/search?q=天子")
  71         |> json_response_and_validate_schema(200)
  72
  73       assert results["hashtags"] == [
  74                %{"name" => "天子", "url" => "#{Endpoint.url()}/tag/天子"}
  75              ]
  76
  77       [status] = results["statuses"]
  78       assert status["id"] == to_string(activity.id)
  79     end
  80
  81     @tag capture_log: true
  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" => "#{Endpoint.url()}/tag/explicit"},
  90                %{"name" => "hashtags", "url" => "#{Endpoint.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" => "#{Endpoint.url()}/tag/john"},
 100                %{"name" => "doe", "url" => "#{Endpoint.url()}/tag/doe"},
 101                %{"name" => "JohnDoe", "url" => "#{Endpoint.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" => "#{Endpoint.url()}/tag/accident"},
 111                %{"name" => "prone", "url" => "#{Endpoint.url()}/tag/prone"},
 112                %{"name" => "AccidentProne", "url" => "#{Endpoint.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" => "#{Endpoint.url()}/tag/shpuld"}
 122              ]
 123
 124       results =
 125         conn
 126         |> get(
 127           "/api/v2/search?#{URI.encode_query(%{q: "https://www.washingtonpost.com/sports/2020/06/10/" <> "nascar-ban-display-confederate-flag-all-events-properties/"})}"
 128         )
 129         |> json_response_and_validate_schema(200)
 130
 131       assert results["hashtags"] == [
 132                %{"name" => "nascar", "url" => "#{Endpoint.url()}/tag/nascar"},
 133                %{"name" => "ban", "url" => "#{Endpoint.url()}/tag/ban"},
 134                %{"name" => "display", "url" => "#{Endpoint.url()}/tag/display"},
 135                %{"name" => "confederate", "url" => "#{Endpoint.url()}/tag/confederate"},
 136                %{"name" => "flag", "url" => "#{Endpoint.url()}/tag/flag"},
 137                %{"name" => "all", "url" => "#{Endpoint.url()}/tag/all"},
 138                %{"name" => "events", "url" => "#{Endpoint.url()}/tag/events"},
 139                %{"name" => "properties", "url" => "#{Endpoint.url()}/tag/properties"},
 140                %{
 141                  "name" => "NascarBanDisplayConfederateFlagAllEventsProperties",
 142                  "url" =>
 143                    "#{Endpoint.url()}/tag/NascarBanDisplayConfederateFlagAllEventsProperties"
 144                }
 145              ]
 146     end
 147
 148     test "supports pagination of hashtags search results", %{conn: conn} do
 149       results =
 150         conn
 151         |> get(
 152           "/api/v2/search?#{URI.encode_query(%{q: "#some #text #with #hashtags", limit: 2, offset: 1})}"
 153         )
 154         |> json_response_and_validate_schema(200)
 155
 156       assert results["hashtags"] == [
 157                %{"name" => "text", "url" => "#{Endpoint.url()}/tag/text"},
 158                %{"name" => "with", "url" => "#{Endpoint.url()}/tag/with"}
 159              ]
 160     end
 161
 162     test "excludes a blocked users from search results", %{conn: conn} do
 163       user = insert(:user)
 164       user_smith = insert(:user, %{nickname: "Agent", name: "I love 2hu"})
 165       user_neo = insert(:user, %{nickname: "Agent Neo", name: "Agent"})
 166
 167       {:ok, act1} = CommonAPI.post(user, %{status: "This is about 2hu private 天子"})
 168       {:ok, act2} = CommonAPI.post(user_smith, %{status: "Agent Smith"})
 169       {:ok, act3} = CommonAPI.post(user_neo, %{status: "Agent Smith"})
 170       Pleroma.User.block(user, user_smith)
 171
 172       results =
 173         conn
 174         |> assign(:user, user)
 175         |> assign(:token, insert(:oauth_token, user: user, scopes: ["read"]))
 176         |> get("/api/v2/search?q=Agent")
 177         |> json_response_and_validate_schema(200)
 178
 179       status_ids = Enum.map(results["statuses"], fn g -> g["id"] end)
 180
 181       assert act3.id in status_ids
 182       refute act2.id in status_ids
 183       refute act1.id in status_ids
 184     end
 185   end
 186
 187   describe ".account_search" do
 188     test "account search", %{conn: conn} do
 189       user_two = insert(:user, %{nickname: "shp@shitposter.club"})
 190       user_three = insert(:user, %{nickname: "shp@heldscal.la", name: "I love 2hu"})
 191
 192       results =
 193         conn
 194         |> get("/api/v1/accounts/search?q=shp")
 195         |> json_response_and_validate_schema(200)
 196
 197       result_ids = for result <- results, do: result["acct"]
 198
 199       assert user_two.nickname in result_ids
 200       assert user_three.nickname in result_ids
 201
 202       results =
 203         conn
 204         |> get("/api/v1/accounts/search?q=2hu")
 205         |> json_response_and_validate_schema(200)
 206
 207       result_ids = for result <- results, do: result["acct"]
 208
 209       assert user_three.nickname in result_ids
 210     end
 211
 212     test "returns account if query contains a space", %{conn: conn} do
 213       insert(:user, %{nickname: "shp@shitposter.club"})
 214
 215       results =
 216         conn
 217         |> get("/api/v1/accounts/search?q=shp@shitposter.club xxx")
 218         |> json_response_and_validate_schema(200)
 219
 220       assert length(results) == 1
 221     end
 222   end
 223 end