akkoma/test/web/websub/websub_controller_test.exs

defmodule Pleroma.Web.Websub.WebsubControllerTest do
  use Pleroma.Web.ConnCase
  import Pleroma.Factory
  alias Pleroma.Web.Websub.WebsubClientSubscription
  alias Pleroma.{Repo, Activity}
  alias Pleroma.Web.Websub

  test "websub subscription request", %{conn: conn} do
    user = insert(:user)

    path = Pleroma.Web.OStatus.pubsub_path(user)

    data = %{
      "hub.callback": "http://example.org/sub",
      "hub.mode": "subscribe",
      "hub.topic": Pleroma.Web.OStatus.feed_path(user),
      "hub.secret": "a random secret",
      "hub.lease_seconds": "100"
    }

    conn = conn
    |> post(path, data)

    assert response(conn, 202) == "Accepted"
  end

  test "websub subscription confirmation", %{conn: conn} do
    websub = insert(:websub_client_subscription)

    params = %{
      "hub.mode" => "subscribe",
      "hub.topic" => websub.topic,
      "hub.challenge" => "some challenge",
      "hub.lease_seconds" => "100"
    }

    conn = conn
    |> get("/push/subscriptions/#{websub.id}", params)

    websub = Repo.get(WebsubClientSubscription, websub.id)

    assert response(conn, 200) == "some challenge"
    assert websub.state == "accepted"
    assert_in_delta NaiveDateTime.diff(websub.valid_until, NaiveDateTime.utc_now), 100, 5
  end

  test "handles incoming feed updates", %{conn: conn} do
    websub = insert(:websub_client_subscription)
    doc = "some stuff"
    signature = Websub.sign(websub.secret, doc)

    conn = conn
    |> put_req_header("x-hub-signature", "sha1=" <> signature)
    |> put_req_header("content-type", "application/atom+xml")
    |> post("/push/subscriptions/#{websub.id}", doc)

    assert response(conn, 200) == "OK"

    assert length(Repo.all(Activity)) == 1
  end

  test "rejects incoming feed updates with the wrong signature", %{conn: conn} do
    websub = insert(:websub_client_subscription)
    doc = "some stuff"
    signature = Websub.sign("wrong secret", doc)

    conn = conn
    |> put_req_header("x-hub-signature", "sha1=" <> signature)
    |> put_req_header("content-type", "application/atom+xml")
    |> post("/push/subscriptions/#{websub.id}", doc)

    assert response(conn, 500) == "Error"

    assert length(Repo.all(Activity)) == 0
  end
end

defmodule Pleroma.Web.OStatusMock do
  import Pleroma.Factory
  def handle_incoming(_doc) do
    insert(:note_activity)
  end
end