git.squeep.com Git - akkoma/blob - test/web/websub/websub_controller_test.exs

   1 defmodule Pleroma.Web.Websub.WebsubControllerTest do
   2   use Pleroma.Web.ConnCase
   3   import Pleroma.Factory
   4   alias Pleroma.Web.Websub.WebsubClientSubscription
   5   alias Pleroma.{Repo, Activity}
   6   alias Pleroma.Web.Websub
   7
   8   test "websub subscription request", %{conn: conn} do
   9     user = insert(:user)
  10
  11     path = Pleroma.Web.OStatus.pubsub_path(user)
  12
  13     data = %{
  14       "hub.callback": "http://example.org/sub",
  15       "hub.mode": "subscribe",
  16       "hub.topic": Pleroma.Web.OStatus.feed_path(user),
  17       "hub.secret": "a random secret",
  18       "hub.lease_seconds": "100"
  19     }
  20
  21     conn = conn
  22     |> post(path, data)
  23
  24     assert response(conn, 202) == "Accepted"
  25   end
  26
  27   test "websub subscription confirmation", %{conn: conn} do
  28     websub = insert(:websub_client_subscription)
  29
  30     params = %{
  31       "hub.mode" => "subscribe",
  32       "hub.topic" => websub.topic,
  33       "hub.challenge" => "some challenge",
  34       "hub.lease_seconds" => "100"
  35     }
  36
  37     conn = conn
  38     |> get("/push/subscriptions/#{websub.id}", params)
  39
  40     websub = Repo.get(WebsubClientSubscription, websub.id)
  41
  42     assert response(conn, 200) == "some challenge"
  43     assert websub.state == "accepted"
  44     assert_in_delta NaiveDateTime.diff(websub.valid_until, NaiveDateTime.utc_now), 100, 5
  45   end
  46
  47   test "handles incoming feed updates", %{conn: conn} do
  48     websub = insert(:websub_client_subscription)
  49     doc = "some stuff"
  50     signature = Websub.sign(websub.secret, doc)
  51
  52     conn = conn
  53     |> put_req_header("x-hub-signature", "sha1=" <> signature)
  54     |> put_req_header("content-type", "application/atom+xml")
  55     |> post("/push/subscriptions/#{websub.id}", doc)
  56
  57     assert response(conn, 200) == "OK"
  58
  59     assert length(Repo.all(Activity)) == 1
  60   end
  61
  62   test "rejects incoming feed updates with the wrong signature", %{conn: conn} do
  63     websub = insert(:websub_client_subscription)
  64     doc = "some stuff"
  65     signature = Websub.sign("wrong secret", doc)
  66
  67     conn = conn
  68     |> put_req_header("x-hub-signature", "sha1=" <> signature)
  69     |> put_req_header("content-type", "application/atom+xml")
  70     |> post("/push/subscriptions/#{websub.id}", doc)
  71
  72     assert response(conn, 500) == "Error"
  73
  74     assert length(Repo.all(Activity)) == 0
  75   end
  76 end