git.squeep.com Git - akkoma/blob - test/web/mastodon_api/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         conn = get(conn, "/api/v2/search", %{"q" => "2hu"})
  28
  29         assert results = json_response(conn, 200)
  30
  31         assert results["accounts"] == []
  32         assert results["statuses"] == []
  33       end
  34     end
  35
  36     test "search", %{conn: conn} do
  37       user = insert(:user)
  38       user_two = insert(:user, %{nickname: "shp@shitposter.club"})
  39       user_three = insert(:user, %{nickname: "shp@heldscal.la", name: "I love 2hu"})
  40
  41       {:ok, activity} = CommonAPI.post(user, %{"status" => "This is about 2hu private"})
  42
  43       {:ok, _activity} =
  44         CommonAPI.post(user, %{
  45           "status" => "This is about 2hu, but private",
  46           "visibility" => "private"
  47         })
  48
  49       {:ok, _} = CommonAPI.post(user_two, %{"status" => "This isn't"})
  50
  51       conn = get(conn, "/api/v2/search", %{"q" => "2hu #private"})
  52
  53       assert results = json_response(conn, 200)
  54
  55       [account | _] = results["accounts"]
  56       assert account["id"] == to_string(user_three.id)
  57
  58       assert results["hashtags"] == [
  59                %{"name" => "private", "url" => "#{Web.base_url()}/tag/private"}
  60              ]
  61
  62       [status] = results["statuses"]
  63       assert status["id"] == to_string(activity.id)
  64     end
  65   end
  66
  67   describe ".account_search" do
  68     test "account search", %{conn: conn} do
  69       user = insert(:user)
  70       user_two = insert(:user, %{nickname: "shp@shitposter.club"})
  71       user_three = insert(:user, %{nickname: "shp@heldscal.la", name: "I love 2hu"})
  72
  73       results =
  74         conn
  75         |> assign(:user, user)
  76         |> get("/api/v1/accounts/search", %{"q" => "shp"})
  77         |> json_response(200)
  78
  79       result_ids = for result <- results, do: result["acct"]
  80
  81       assert user_two.nickname in result_ids
  82       assert user_three.nickname in result_ids
  83
  84       results =
  85         conn
  86         |> assign(:user, user)
  87         |> get("/api/v1/accounts/search", %{"q" => "2hu"})
  88         |> json_response(200)
  89
  90       result_ids = for result <- results, do: result["acct"]
  91
  92       assert user_three.nickname in result_ids
  93     end
  94   end
  95
  96   describe ".search" do
  97     test "it returns empty result if user or status search return undefined error", %{conn: conn} do
  98       with_mocks [
  99         {Pleroma.User, [], [search: fn _q, _o -> raise "Oops" end]},
 100         {Pleroma.Activity, [], [search: fn _u, _q, _o -> raise "Oops" end]}
 101       ] do
 102         conn =
 103           conn
 104           |> get("/api/v1/search", %{"q" => "2hu"})
 105
 106         assert results = json_response(conn, 200)
 107
 108         assert results["accounts"] == []
 109         assert results["statuses"] == []
 110       end
 111     end
 112
 113     test "search", %{conn: conn} do
 114       user = insert(:user)
 115       user_two = insert(:user, %{nickname: "shp@shitposter.club"})
 116       user_three = insert(:user, %{nickname: "shp@heldscal.la", name: "I love 2hu"})
 117
 118       {:ok, activity} = CommonAPI.post(user, %{"status" => "This is about 2hu"})
 119
 120       {:ok, _activity} =
 121         CommonAPI.post(user, %{
 122           "status" => "This is about 2hu, but private",
 123           "visibility" => "private"
 124         })
 125
 126       {:ok, _} = CommonAPI.post(user_two, %{"status" => "This isn't"})
 127
 128       conn =
 129         conn
 130         |> get("/api/v1/search", %{"q" => "2hu"})
 131
 132       assert results = json_response(conn, 200)
 133
 134       [account | _] = results["accounts"]
 135       assert account["id"] == to_string(user_three.id)
 136
 137       assert results["hashtags"] == []
 138
 139       [status] = results["statuses"]
 140       assert status["id"] == to_string(activity.id)
 141     end
 142
 143     test "search fetches remote statuses", %{conn: conn} do
 144       capture_log(fn ->
 145         conn =
 146           conn
 147           |> get("/api/v1/search", %{"q" => "https://shitposter.club/notice/2827873"})
 148
 149         assert results = json_response(conn, 200)
 150
 151         [status] = results["statuses"]
 152
 153         assert status["uri"] ==
 154                  "tag:shitposter.club,2017-05-05:noticeId=2827873:objectType=comment"
 155       end)
 156     end
 157
 158     test "search doesn't show statuses that it shouldn't", %{conn: conn} do
 159       {:ok, activity} =
 160         CommonAPI.post(insert(:user), %{
 161           "status" => "This is about 2hu, but private",
 162           "visibility" => "private"
 163         })
 164
 165       capture_log(fn ->
 166         conn =
 167           conn
 168           |> get("/api/v1/search", %{"q" => Object.normalize(activity).data["id"]})
 169
 170         assert results = json_response(conn, 200)
 171
 172         [] = results["statuses"]
 173       end)
 174     end
 175
 176     test "search fetches remote accounts", %{conn: conn} do
 177       user = insert(:user)
 178
 179       conn =
 180         conn
 181         |> assign(:user, user)
 182         |> get("/api/v1/search", %{"q" => "shp@social.heldscal.la", "resolve" => "true"})
 183
 184       assert results = json_response(conn, 200)
 185       [account] = results["accounts"]
 186       assert account["acct"] == "shp@social.heldscal.la"
 187     end
 188
 189     test "search doesn't fetch remote accounts if resolve is false", %{conn: conn} do
 190       conn =
 191         conn
 192         |> get("/api/v1/search", %{"q" => "shp@social.heldscal.la", "resolve" => "false"})
 193
 194       assert results = json_response(conn, 200)
 195       assert [] == results["accounts"]
 196     end
 197
 198     test "search with limit and offset", %{conn: conn} do
 199       user = insert(:user)
 200       _user_two = insert(:user, %{nickname: "shp@shitposter.club"})
 201       _user_three = insert(:user, %{nickname: "shp@heldscal.la", name: "I love 2hu"})
 202
 203       {:ok, _activity1} = CommonAPI.post(user, %{"status" => "This is about 2hu"})
 204       {:ok, _activity2} = CommonAPI.post(user, %{"status" => "This is also about 2hu"})
 205
 206       result =
 207         conn
 208         |> get("/api/v1/search", %{"q" => "2hu", "limit" => 1})
 209
 210       assert results = json_response(result, 200)
 211       assert [%{"id" => activity_id1}] = results["statuses"]
 212       assert [_] = results["accounts"]
 213
 214       results =
 215         conn
 216         |> get("/api/v1/search", %{"q" => "2hu", "limit" => 1, "offset" => 1})
 217         |> json_response(200)
 218
 219       assert [%{"id" => activity_id2}] = results["statuses"]
 220       assert [] = results["accounts"]
 221
 222       assert activity_id1 != activity_id2
 223     end
 224
 225     test "search returns results only for the given type", %{conn: conn} do
 226       user = insert(:user)
 227       _user_two = insert(:user, %{nickname: "shp@heldscal.la", name: "I love 2hu"})
 228
 229       {:ok, _activity} = CommonAPI.post(user, %{"status" => "This is about 2hu"})
 230
 231       assert %{"statuses" => [_activity], "accounts" => [], "hashtags" => []} =
 232                conn
 233                |> get("/api/v1/search", %{"q" => "2hu", "type" => "statuses"})
 234                |> json_response(200)
 235
 236       assert %{"statuses" => [], "accounts" => [_user_two], "hashtags" => []} =
 237                conn
 238                |> get("/api/v1/search", %{"q" => "2hu", "type" => "accounts"})
 239                |> json_response(200)
 240     end
 241
 242     test "search uses account_id to filter statuses by the author", %{conn: conn} do
 243       user = insert(:user, %{nickname: "shp@shitposter.club"})
 244       user_two = insert(:user, %{nickname: "shp@heldscal.la", name: "I love 2hu"})
 245
 246       {:ok, activity1} = CommonAPI.post(user, %{"status" => "This is about 2hu"})
 247       {:ok, activity2} = CommonAPI.post(user_two, %{"status" => "This is also about 2hu"})
 248
 249       results =
 250         conn
 251         |> get("/api/v1/search", %{"q" => "2hu", "account_id" => user.id})
 252         |> json_response(200)
 253
 254       assert [%{"id" => activity_id1}] = results["statuses"]
 255       assert activity_id1 == activity1.id
 256       assert [_] = results["accounts"]
 257
 258       results =
 259         conn
 260         |> get("/api/v1/search", %{"q" => "2hu", "account_id" => user_two.id})
 261         |> json_response(200)
 262
 263       assert [%{"id" => activity_id2}] = results["statuses"]
 264       assert activity_id2 == activity2.id
 265     end
 266   end
 267 end