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