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-2019 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 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(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         get(conn, "/api/v2/search", %{"q" => "2hu #private"})
  57         |> json_response(200)
  58
  59       [account | _] = results["accounts"]
  60       assert account["id"] == to_string(user_three.id)
  61
  62       assert results["hashtags"] == [
  63                %{"name" => "private", "url" => "#{Web.base_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(200)
  72
  73       [status] = results["statuses"]
  74       assert status["id"] == to_string(activity.id)
  75     end
  76   end
  77
  78   describe ".account_search" do
  79     test "account search", %{conn: conn} do
  80       user_two = insert(:user, %{nickname: "shp@shitposter.club"})
  81       user_three = insert(:user, %{nickname: "shp@heldscal.la", name: "I love 2hu"})
  82
  83       results =
  84         conn
  85         |> get("/api/v1/accounts/search", %{"q" => "shp"})
  86         |> json_response(200)
  87
  88       result_ids = for result <- results, do: result["acct"]
  89
  90       assert user_two.nickname in result_ids
  91       assert user_three.nickname in result_ids
  92
  93       results =
  94         conn
  95         |> get("/api/v1/accounts/search", %{"q" => "2hu"})
  96         |> json_response(200)
  97
  98       result_ids = for result <- results, do: result["acct"]
  99
 100       assert user_three.nickname in result_ids
 101     end
 102
 103     test "returns account if query contains a space", %{conn: conn} do
 104       insert(:user, %{nickname: "shp@shitposter.club"})
 105
 106       results =
 107         conn
 108         |> get("/api/v1/accounts/search", %{"q" => "shp@shitposter.club xxx "})
 109         |> json_response(200)
 110
 111       assert length(results) == 1
 112     end
 113   end
 114
 115   describe ".search" do
 116     test "it returns empty result if user or status search return undefined error", %{conn: conn} do
 117       with_mocks [
 118         {Pleroma.User, [], [search: fn _q, _o -> raise "Oops" end]},
 119         {Pleroma.Activity, [], [search: fn _u, _q, _o -> raise "Oops" end]}
 120       ] do
 121         capture_log(fn ->
 122           results =
 123             conn
 124             |> get("/api/v1/search", %{"q" => "2hu"})
 125             |> json_response(200)
 126
 127           assert results["accounts"] == []
 128           assert results["statuses"] == []
 129         end) =~
 130           "[error] Elixir.Pleroma.Web.MastodonAPI.SearchController search error: %RuntimeError{message: \"Oops\"}"
 131       end
 132     end
 133
 134     test "search", %{conn: conn} do
 135       user = insert(:user)
 136       user_two = insert(:user, %{nickname: "shp@shitposter.club"})
 137       user_three = insert(:user, %{nickname: "shp@heldscal.la", name: "I love 2hu"})
 138
 139       {:ok, activity} = CommonAPI.post(user, %{"status" => "This is about 2hu"})
 140
 141       {:ok, _activity} =
 142         CommonAPI.post(user, %{
 143           "status" => "This is about 2hu, but private",
 144           "visibility" => "private"
 145         })
 146
 147       {:ok, _} = CommonAPI.post(user_two, %{"status" => "This isn't"})
 148
 149       conn =
 150         conn
 151         |> get("/api/v1/search", %{"q" => "2hu"})
 152
 153       assert results = json_response(conn, 200)
 154
 155       [account | _] = results["accounts"]
 156       assert account["id"] == to_string(user_three.id)
 157
 158       assert results["hashtags"] == []
 159
 160       [status] = results["statuses"]
 161       assert status["id"] == to_string(activity.id)
 162     end
 163
 164     test "search fetches remote statuses and prefers them over other results", %{conn: conn} do
 165       capture_log(fn ->
 166         {:ok, %{id: activity_id}} =
 167           CommonAPI.post(insert(:user), %{
 168             "status" => "check out https://shitposter.club/notice/2827873"
 169           })
 170
 171         conn =
 172           conn
 173           |> get("/api/v1/search", %{"q" => "https://shitposter.club/notice/2827873"})
 174
 175         assert results = json_response(conn, 200)
 176
 177         [status, %{"id" => ^activity_id}] = results["statuses"]
 178
 179         assert status["uri"] ==
 180                  "tag:shitposter.club,2017-05-05:noticeId=2827873:objectType=comment"
 181       end)
 182     end
 183
 184     test "search doesn't show statuses that it shouldn't", %{conn: conn} do
 185       {:ok, activity} =
 186         CommonAPI.post(insert(:user), %{
 187           "status" => "This is about 2hu, but private",
 188           "visibility" => "private"
 189         })
 190
 191       capture_log(fn ->
 192         conn =
 193           conn
 194           |> get("/api/v1/search", %{"q" => Object.normalize(activity).data["id"]})
 195
 196         assert results = json_response(conn, 200)
 197
 198         [] = results["statuses"]
 199       end)
 200     end
 201
 202     test "search fetches remote accounts", %{conn: conn} do
 203       user = insert(:user)
 204
 205       conn =
 206         conn
 207         |> assign(:user, user)
 208         |> assign(:token, insert(:oauth_token, user: user, scopes: ["read"]))
 209         |> get("/api/v1/search", %{"q" => "mike@osada.macgirvin.com", "resolve" => "true"})
 210
 211       assert results = json_response(conn, 200)
 212       [account] = results["accounts"]
 213       assert account["acct"] == "mike@osada.macgirvin.com"
 214     end
 215
 216     test "search doesn't fetch remote accounts if resolve is false", %{conn: conn} do
 217       conn =
 218         conn
 219         |> get("/api/v1/search", %{"q" => "mike@osada.macgirvin.com", "resolve" => "false"})
 220
 221       assert results = json_response(conn, 200)
 222       assert [] == results["accounts"]
 223     end
 224
 225     test "search with limit and offset", %{conn: conn} do
 226       user = insert(:user)
 227       _user_two = insert(:user, %{nickname: "shp@shitposter.club"})
 228       _user_three = insert(:user, %{nickname: "shp@heldscal.la", name: "I love 2hu"})
 229
 230       {:ok, _activity1} = CommonAPI.post(user, %{"status" => "This is about 2hu"})
 231       {:ok, _activity2} = CommonAPI.post(user, %{"status" => "This is also about 2hu"})
 232
 233       result =
 234         conn
 235         |> get("/api/v1/search", %{"q" => "2hu", "limit" => 1})
 236
 237       assert results = json_response(result, 200)
 238       assert [%{"id" => activity_id1}] = results["statuses"]
 239       assert [_] = results["accounts"]
 240
 241       results =
 242         conn
 243         |> get("/api/v1/search", %{"q" => "2hu", "limit" => 1, "offset" => 1})
 244         |> json_response(200)
 245
 246       assert [%{"id" => activity_id2}] = results["statuses"]
 247       assert [] = results["accounts"]
 248
 249       assert activity_id1 != activity_id2
 250     end
 251
 252     test "search returns results only for the given type", %{conn: conn} do
 253       user = insert(:user)
 254       _user_two = insert(:user, %{nickname: "shp@heldscal.la", name: "I love 2hu"})
 255
 256       {:ok, _activity} = CommonAPI.post(user, %{"status" => "This is about 2hu"})
 257
 258       assert %{"statuses" => [_activity], "accounts" => [], "hashtags" => []} =
 259                conn
 260                |> get("/api/v1/search", %{"q" => "2hu", "type" => "statuses"})
 261                |> json_response(200)
 262
 263       assert %{"statuses" => [], "accounts" => [_user_two], "hashtags" => []} =
 264                conn
 265                |> get("/api/v1/search", %{"q" => "2hu", "type" => "accounts"})
 266                |> json_response(200)
 267     end
 268
 269     test "search uses account_id to filter statuses by the author", %{conn: conn} do
 270       user = insert(:user, %{nickname: "shp@shitposter.club"})
 271       user_two = insert(:user, %{nickname: "shp@heldscal.la", name: "I love 2hu"})
 272
 273       {:ok, activity1} = CommonAPI.post(user, %{"status" => "This is about 2hu"})
 274       {:ok, activity2} = CommonAPI.post(user_two, %{"status" => "This is also about 2hu"})
 275
 276       results =
 277         conn
 278         |> get("/api/v1/search", %{"q" => "2hu", "account_id" => user.id})
 279         |> json_response(200)
 280
 281       assert [%{"id" => activity_id1}] = results["statuses"]
 282       assert activity_id1 == activity1.id
 283       assert [_] = results["accounts"]
 284
 285       results =
 286         conn
 287         |> get("/api/v1/search", %{"q" => "2hu", "account_id" => user_two.id})
 288         |> json_response(200)
 289
 290       assert [%{"id" => activity_id2}] = results["statuses"]
 291       assert activity_id2 == activity2.id
 292     end
 293   end
 294 end