git.squeep.com Git - akkoma/blob - test/web/salmon/salmon_test.exs

   1 defmodule Pleroma.Web.Salmon.SalmonTest do
   2   use Pleroma.DataCase
   3   alias Pleroma.Web.Salmon
   4   alias Pleroma.{Repo, Activity, User}
   5   import Pleroma.Factory
   6   import Tesla.Mock
   7
   8   @magickey "RSA.pu0s-halox4tu7wmES1FVSx6u-4wc0YrUFXcqWXZG4-27UmbCOpMQftRCldNRfyA-qLbz-eqiwQhh-1EwUvjsD4cYbAHNGHwTvDOyx5AKthQUP44ykPv7kjKGh3DWKySJvcs9tlUG87hlo7AvnMo9pwRS_Zz2CacQ-MKaXyDepk=.AQAB"
   9
  10   @wrong_magickey "RSA.pu0s-halox4tu7wmES1FVSx6u-4wc0YrUFXcqWXZG4-27UmbCOpMQftRCldNRfyA-qLbz-eqiwQhh-1EwUvjsD4cYbAHNGHwTvDOyx5AKthQUP44ykPv7kjKGh3DWKySJvcs9tlUG87hlo7AvnMo9pwRS_Zz2CacQ-MKaXyDepk=.AQAA"
  11
  12   @magickey_friendica "RSA.AMwa8FUs2fWEjX0xN7yRQgegQffhBpuKNC6fa5VNSVorFjGZhRrlPMn7TQOeihlc9lBz2OsHlIedbYn2uJ7yCs0.AQAB"
  13
  14   setup do
  15     mock(fn env -> apply(HttpRequestMock, :request, [env]) end)
  16     :ok
  17   end
  18
  19   test "decodes a salmon" do
  20     {:ok, salmon} = File.read("test/fixtures/salmon.xml")
  21     {:ok, doc} = Salmon.decode_and_validate(@magickey, salmon)
  22     assert Regex.match?(~r/xml/, doc)
  23   end
  24
  25   test "errors on wrong magic key" do
  26     {:ok, salmon} = File.read("test/fixtures/salmon.xml")
  27     assert Salmon.decode_and_validate(@wrong_magickey, salmon) == :error
  28   end
  29
  30   test "generates an RSA private key pem" do
  31     {:ok, key} = Salmon.generate_rsa_pem()
  32     assert is_binary(key)
  33     assert Regex.match?(~r/RSA/, key)
  34   end
  35
  36   test "it encodes a magic key from a public key" do
  37     key = Salmon.decode_key(@magickey)
  38     magic_key = Salmon.encode_key(key)
  39
  40     assert @magickey == magic_key
  41   end
  42
  43   test "it decodes a friendica public key" do
  44     _key = Salmon.decode_key(@magickey_friendica)
  45   end
  46
  47   test "returns a public and private key from a pem" do
  48     pem = File.read!("test/fixtures/private_key.pem")
  49     {:ok, private, public} = Salmon.keys_from_pem(pem)
  50
  51     assert elem(private, 0) == :RSAPrivateKey
  52     assert elem(public, 0) == :RSAPublicKey
  53   end
  54
  55   test "encodes an xml payload with a private key" do
  56     doc = File.read!("test/fixtures/incoming_note_activity.xml")
  57     pem = File.read!("test/fixtures/private_key.pem")
  58     {:ok, private, public} = Salmon.keys_from_pem(pem)
  59
  60     # Let's try a roundtrip.
  61     {:ok, salmon} = Salmon.encode(private, doc)
  62     {:ok, decoded_doc} = Salmon.decode_and_validate(Salmon.encode_key(public), salmon)
  63
  64     assert doc == decoded_doc
  65   end
  66
  67   test "it gets a magic key" do
  68     salmon = File.read!("test/fixtures/salmon2.xml")
  69     {:ok, key} = Salmon.fetch_magic_key(salmon)
  70
  71     assert key ==
  72              "RSA.uzg6r1peZU0vXGADWxGJ0PE34WvmhjUmydbX5YYdOiXfODVLwCMi1umGoqUDm-mRu4vNEdFBVJU1CpFA7dKzWgIsqsa501i2XqElmEveXRLvNRWFB6nG03Q5OUY2as8eE54BJm0p20GkMfIJGwP6TSFb-ICp3QjzbatuSPJ6xCE=.AQAB"
  73   end
  74
  75   test "it pushes an activity to remote accounts it's addressed to" do
  76     user_data = %{
  77       info: %{
  78         "salmon" => "http://example.org/salmon"
  79       },
  80       local: false
  81     }
  82
  83     mentioned_user = insert(:user, user_data)
  84     note = insert(:note)
  85
  86     activity_data = %{
  87       "id" => Pleroma.Web.ActivityPub.Utils.generate_activity_id(),
  88       "type" => "Create",
  89       "actor" => note.data["actor"],
  90       "to" => note.data["to"] ++ [mentioned_user.ap_id],
  91       "object" => note.data,
  92       "published_at" => DateTime.utc_now() |> DateTime.to_iso8601(),
  93       "context" => note.data["context"]
  94     }
  95
  96     {:ok, activity} = Repo.insert(%Activity{data: activity_data, recipients: activity_data["to"]})
  97     user = Repo.get_by(User, ap_id: activity.data["actor"])
  98     {:ok, user} = Pleroma.Web.WebFinger.ensure_keys_present(user)
  99
 100     poster = fn url, _data, _headers, _options ->
 101       assert url == "http://example.org/salmon"
 102     end
 103
 104     Salmon.publish(user, activity, poster)
 105   end
 106 end