336 lines
		
	
	
	
		
			10 KiB
		
	
	
	
		
			Elixir
		
	
	
	
	
	
			
		
		
	
	
			336 lines
		
	
	
	
		
			10 KiB
		
	
	
	
		
			Elixir
		
	
	
	
	
	
# Pleroma: A lightweight social networking server
 | 
						|
# Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
 | 
						|
# SPDX-License-Identifier: AGPL-3.0-only
 | 
						|
defmodule Pleroma.Web.PleromaAPI.ChatControllerTest do
 | 
						|
  use Pleroma.Web.ConnCase, async: true
 | 
						|
 | 
						|
  alias Pleroma.Chat
 | 
						|
  alias Pleroma.Chat.MessageReference
 | 
						|
  alias Pleroma.Object
 | 
						|
  alias Pleroma.User
 | 
						|
  alias Pleroma.Web.ActivityPub.ActivityPub
 | 
						|
  alias Pleroma.Web.CommonAPI
 | 
						|
 | 
						|
  import Pleroma.Factory
 | 
						|
 | 
						|
  describe "POST /api/v1/pleroma/chats/:id/messages/:message_id/read" do
 | 
						|
    setup do: oauth_access(["write:chats"])
 | 
						|
 | 
						|
    test "it marks one message as read", %{conn: conn, user: user} do
 | 
						|
      other_user = insert(:user)
 | 
						|
 | 
						|
      {:ok, create} = CommonAPI.post_chat_message(other_user, user, "sup")
 | 
						|
      {:ok, _create} = CommonAPI.post_chat_message(other_user, user, "sup part 2")
 | 
						|
      {:ok, chat} = Chat.get_or_create(user.id, other_user.ap_id)
 | 
						|
      object = Object.normalize(create, false)
 | 
						|
      cm_ref = MessageReference.for_chat_and_object(chat, object)
 | 
						|
 | 
						|
      assert cm_ref.unread == true
 | 
						|
 | 
						|
      result =
 | 
						|
        conn
 | 
						|
        |> post("/api/v1/pleroma/chats/#{chat.id}/messages/#{cm_ref.id}/read")
 | 
						|
        |> json_response_and_validate_schema(200)
 | 
						|
 | 
						|
      assert result["unread"] == false
 | 
						|
 | 
						|
      cm_ref = MessageReference.for_chat_and_object(chat, object)
 | 
						|
 | 
						|
      assert cm_ref.unread == false
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  describe "POST /api/v1/pleroma/chats/:id/read" do
 | 
						|
    setup do: oauth_access(["write:chats"])
 | 
						|
 | 
						|
    test "given a `last_read_id`, it marks everything until then as read", %{
 | 
						|
      conn: conn,
 | 
						|
      user: user
 | 
						|
    } do
 | 
						|
      other_user = insert(:user)
 | 
						|
 | 
						|
      {:ok, create} = CommonAPI.post_chat_message(other_user, user, "sup")
 | 
						|
      {:ok, _create} = CommonAPI.post_chat_message(other_user, user, "sup part 2")
 | 
						|
      {:ok, chat} = Chat.get_or_create(user.id, other_user.ap_id)
 | 
						|
      object = Object.normalize(create, false)
 | 
						|
      cm_ref = MessageReference.for_chat_and_object(chat, object)
 | 
						|
 | 
						|
      assert cm_ref.unread == true
 | 
						|
 | 
						|
      result =
 | 
						|
        conn
 | 
						|
        |> put_req_header("content-type", "application/json")
 | 
						|
        |> post("/api/v1/pleroma/chats/#{chat.id}/read", %{"last_read_id" => cm_ref.id})
 | 
						|
        |> json_response_and_validate_schema(200)
 | 
						|
 | 
						|
      assert result["unread"] == 1
 | 
						|
 | 
						|
      cm_ref = MessageReference.for_chat_and_object(chat, object)
 | 
						|
 | 
						|
      assert cm_ref.unread == false
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  describe "POST /api/v1/pleroma/chats/:id/messages" do
 | 
						|
    setup do: oauth_access(["write:chats"])
 | 
						|
 | 
						|
    test "it posts a message to the chat", %{conn: conn, user: user} do
 | 
						|
      other_user = insert(:user)
 | 
						|
 | 
						|
      {:ok, chat} = Chat.get_or_create(user.id, other_user.ap_id)
 | 
						|
 | 
						|
      result =
 | 
						|
        conn
 | 
						|
        |> put_req_header("content-type", "application/json")
 | 
						|
        |> post("/api/v1/pleroma/chats/#{chat.id}/messages", %{"content" => "Hallo!!"})
 | 
						|
        |> json_response_and_validate_schema(200)
 | 
						|
 | 
						|
      assert result["content"] == "Hallo!!"
 | 
						|
      assert result["chat_id"] == chat.id |> to_string()
 | 
						|
    end
 | 
						|
 | 
						|
    test "it fails if there is no content", %{conn: conn, user: user} do
 | 
						|
      other_user = insert(:user)
 | 
						|
 | 
						|
      {:ok, chat} = Chat.get_or_create(user.id, other_user.ap_id)
 | 
						|
 | 
						|
      result =
 | 
						|
        conn
 | 
						|
        |> put_req_header("content-type", "application/json")
 | 
						|
        |> post("/api/v1/pleroma/chats/#{chat.id}/messages")
 | 
						|
        |> json_response_and_validate_schema(400)
 | 
						|
 | 
						|
      assert result
 | 
						|
    end
 | 
						|
 | 
						|
    test "it works with an attachment", %{conn: conn, user: user} do
 | 
						|
      file = %Plug.Upload{
 | 
						|
        content_type: "image/jpg",
 | 
						|
        path: Path.absname("test/fixtures/image.jpg"),
 | 
						|
        filename: "an_image.jpg"
 | 
						|
      }
 | 
						|
 | 
						|
      {:ok, upload} = ActivityPub.upload(file, actor: user.ap_id)
 | 
						|
 | 
						|
      other_user = insert(:user)
 | 
						|
 | 
						|
      {:ok, chat} = Chat.get_or_create(user.id, other_user.ap_id)
 | 
						|
 | 
						|
      result =
 | 
						|
        conn
 | 
						|
        |> put_req_header("content-type", "application/json")
 | 
						|
        |> post("/api/v1/pleroma/chats/#{chat.id}/messages", %{
 | 
						|
          "media_id" => to_string(upload.id)
 | 
						|
        })
 | 
						|
        |> json_response_and_validate_schema(200)
 | 
						|
 | 
						|
      assert result["attachment"]
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  describe "DELETE /api/v1/pleroma/chats/:id/messages/:message_id" do
 | 
						|
    setup do: oauth_access(["write:chats"])
 | 
						|
 | 
						|
    test "it deletes a message from the chat", %{conn: conn, user: user} do
 | 
						|
      recipient = insert(:user)
 | 
						|
 | 
						|
      {:ok, message} =
 | 
						|
        CommonAPI.post_chat_message(user, recipient, "Hello darkness my old friend")
 | 
						|
 | 
						|
      {:ok, other_message} = CommonAPI.post_chat_message(recipient, user, "nico nico ni")
 | 
						|
 | 
						|
      object = Object.normalize(message, false)
 | 
						|
 | 
						|
      chat = Chat.get(user.id, recipient.ap_id)
 | 
						|
 | 
						|
      cm_ref = MessageReference.for_chat_and_object(chat, object)
 | 
						|
 | 
						|
      # Deleting your own message removes the message and the reference
 | 
						|
      result =
 | 
						|
        conn
 | 
						|
        |> put_req_header("content-type", "application/json")
 | 
						|
        |> delete("/api/v1/pleroma/chats/#{chat.id}/messages/#{cm_ref.id}")
 | 
						|
        |> json_response_and_validate_schema(200)
 | 
						|
 | 
						|
      assert result["id"] == cm_ref.id
 | 
						|
      refute MessageReference.get_by_id(cm_ref.id)
 | 
						|
      assert %{data: %{"type" => "Tombstone"}} = Object.get_by_id(object.id)
 | 
						|
 | 
						|
      # Deleting other people's messages just removes the reference
 | 
						|
      object = Object.normalize(other_message, false)
 | 
						|
      cm_ref = MessageReference.for_chat_and_object(chat, object)
 | 
						|
 | 
						|
      result =
 | 
						|
        conn
 | 
						|
        |> put_req_header("content-type", "application/json")
 | 
						|
        |> delete("/api/v1/pleroma/chats/#{chat.id}/messages/#{cm_ref.id}")
 | 
						|
        |> json_response_and_validate_schema(200)
 | 
						|
 | 
						|
      assert result["id"] == cm_ref.id
 | 
						|
      refute MessageReference.get_by_id(cm_ref.id)
 | 
						|
      assert Object.get_by_id(object.id)
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  describe "GET /api/v1/pleroma/chats/:id/messages" do
 | 
						|
    setup do: oauth_access(["read:chats"])
 | 
						|
 | 
						|
    test "it paginates", %{conn: conn, user: user} do
 | 
						|
      recipient = insert(:user)
 | 
						|
 | 
						|
      Enum.each(1..30, fn _ ->
 | 
						|
        {:ok, _} = CommonAPI.post_chat_message(user, recipient, "hey")
 | 
						|
      end)
 | 
						|
 | 
						|
      chat = Chat.get(user.id, recipient.ap_id)
 | 
						|
 | 
						|
      result =
 | 
						|
        conn
 | 
						|
        |> get("/api/v1/pleroma/chats/#{chat.id}/messages")
 | 
						|
        |> json_response_and_validate_schema(200)
 | 
						|
 | 
						|
      assert length(result) == 20
 | 
						|
 | 
						|
      result =
 | 
						|
        conn
 | 
						|
        |> get("/api/v1/pleroma/chats/#{chat.id}/messages?max_id=#{List.last(result)["id"]}")
 | 
						|
        |> json_response_and_validate_schema(200)
 | 
						|
 | 
						|
      assert length(result) == 10
 | 
						|
    end
 | 
						|
 | 
						|
    test "it returns the messages for a given chat", %{conn: conn, user: user} do
 | 
						|
      other_user = insert(:user)
 | 
						|
      third_user = insert(:user)
 | 
						|
 | 
						|
      {:ok, _} = CommonAPI.post_chat_message(user, other_user, "hey")
 | 
						|
      {:ok, _} = CommonAPI.post_chat_message(user, third_user, "hey")
 | 
						|
      {:ok, _} = CommonAPI.post_chat_message(user, other_user, "how are you?")
 | 
						|
      {:ok, _} = CommonAPI.post_chat_message(other_user, user, "fine, how about you?")
 | 
						|
 | 
						|
      chat = Chat.get(user.id, other_user.ap_id)
 | 
						|
 | 
						|
      result =
 | 
						|
        conn
 | 
						|
        |> get("/api/v1/pleroma/chats/#{chat.id}/messages")
 | 
						|
        |> json_response_and_validate_schema(200)
 | 
						|
 | 
						|
      result
 | 
						|
      |> Enum.each(fn message ->
 | 
						|
        assert message["chat_id"] == chat.id |> to_string()
 | 
						|
      end)
 | 
						|
 | 
						|
      assert length(result) == 3
 | 
						|
 | 
						|
      # Trying to get the chat of a different user
 | 
						|
      result =
 | 
						|
        conn
 | 
						|
        |> assign(:user, other_user)
 | 
						|
        |> get("/api/v1/pleroma/chats/#{chat.id}/messages")
 | 
						|
 | 
						|
      assert result |> json_response(404)
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  describe "POST /api/v1/pleroma/chats/by-account-id/:id" do
 | 
						|
    setup do: oauth_access(["write:chats"])
 | 
						|
 | 
						|
    test "it creates or returns a chat", %{conn: conn} do
 | 
						|
      other_user = insert(:user)
 | 
						|
 | 
						|
      result =
 | 
						|
        conn
 | 
						|
        |> post("/api/v1/pleroma/chats/by-account-id/#{other_user.id}")
 | 
						|
        |> json_response_and_validate_schema(200)
 | 
						|
 | 
						|
      assert result["id"]
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  describe "GET /api/v1/pleroma/chats/:id" do
 | 
						|
    setup do: oauth_access(["read:chats"])
 | 
						|
 | 
						|
    test "it returns a chat", %{conn: conn, user: user} do
 | 
						|
      other_user = insert(:user)
 | 
						|
 | 
						|
      {:ok, chat} = Chat.get_or_create(user.id, other_user.ap_id)
 | 
						|
 | 
						|
      result =
 | 
						|
        conn
 | 
						|
        |> get("/api/v1/pleroma/chats/#{chat.id}")
 | 
						|
        |> json_response_and_validate_schema(200)
 | 
						|
 | 
						|
      assert result["id"] == to_string(chat.id)
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  describe "GET /api/v1/pleroma/chats" do
 | 
						|
    setup do: oauth_access(["read:chats"])
 | 
						|
 | 
						|
    test "it does not return chats with users you blocked", %{conn: conn, user: user} do
 | 
						|
      recipient = insert(:user)
 | 
						|
 | 
						|
      {:ok, _} = Chat.get_or_create(user.id, recipient.ap_id)
 | 
						|
 | 
						|
      result =
 | 
						|
        conn
 | 
						|
        |> get("/api/v1/pleroma/chats")
 | 
						|
        |> json_response_and_validate_schema(200)
 | 
						|
 | 
						|
      assert length(result) == 1
 | 
						|
 | 
						|
      User.block(user, recipient)
 | 
						|
 | 
						|
      result =
 | 
						|
        conn
 | 
						|
        |> get("/api/v1/pleroma/chats")
 | 
						|
        |> json_response_and_validate_schema(200)
 | 
						|
 | 
						|
      assert length(result) == 0
 | 
						|
    end
 | 
						|
 | 
						|
    test "it returns all chats", %{conn: conn, user: user} do
 | 
						|
      Enum.each(1..30, fn _ ->
 | 
						|
        recipient = insert(:user)
 | 
						|
        {:ok, _} = Chat.get_or_create(user.id, recipient.ap_id)
 | 
						|
      end)
 | 
						|
 | 
						|
      result =
 | 
						|
        conn
 | 
						|
        |> get("/api/v1/pleroma/chats")
 | 
						|
        |> json_response_and_validate_schema(200)
 | 
						|
 | 
						|
      assert length(result) == 30
 | 
						|
    end
 | 
						|
 | 
						|
    test "it return a list of chats the current user is participating in, in descending order of updates",
 | 
						|
         %{conn: conn, user: user} do
 | 
						|
      har = insert(:user)
 | 
						|
      jafnhar = insert(:user)
 | 
						|
      tridi = insert(:user)
 | 
						|
 | 
						|
      {:ok, chat_1} = Chat.get_or_create(user.id, har.ap_id)
 | 
						|
      :timer.sleep(1000)
 | 
						|
      {:ok, _chat_2} = Chat.get_or_create(user.id, jafnhar.ap_id)
 | 
						|
      :timer.sleep(1000)
 | 
						|
      {:ok, chat_3} = Chat.get_or_create(user.id, tridi.ap_id)
 | 
						|
      :timer.sleep(1000)
 | 
						|
 | 
						|
      # bump the second one
 | 
						|
      {:ok, chat_2} = Chat.bump_or_create(user.id, jafnhar.ap_id)
 | 
						|
 | 
						|
      result =
 | 
						|
        conn
 | 
						|
        |> get("/api/v1/pleroma/chats")
 | 
						|
        |> json_response_and_validate_schema(200)
 | 
						|
 | 
						|
      ids = Enum.map(result, & &1["id"])
 | 
						|
 | 
						|
      assert ids == [
 | 
						|
               chat_2.id |> to_string(),
 | 
						|
               chat_3.id |> to_string(),
 | 
						|
               chat_1.id |> to_string()
 | 
						|
             ]
 | 
						|
    end
 | 
						|
  end
 | 
						|
end
 |