git.squeep.com Git - akkoma/blob - test/web/media_proxy/invalidation_test.exs

   1 defmodule Pleroma.Web.MediaProxy.InvalidationTest do
   2   use ExUnit.Case
   3   use Pleroma.Tests.Helpers
   4
   5   alias Pleroma.Config
   6   alias Pleroma.Web.MediaProxy.Invalidation
   7
   8   import ExUnit.CaptureLog
   9   import Mock
  10   import Tesla.Mock
  11
  12   setup do: clear_config([:media_proxy])
  13
  14   setup do
  15     on_exit(fn -> Cachex.clear(:deleted_urls_cache) end)
  16     :ok
  17   end
  18
  19   describe "Invalidation.Http" do
  20     test "perform request to clear cache" do
  21       Config.put([:media_proxy, :enabled], false)
  22       Config.put([:media_proxy, :invalidation, :enabled], true)
  23       Config.put([:media_proxy, :invalidation, :provider], Invalidation.Http)
  24
  25       Config.put([Invalidation.Http], method: :purge, headers: [{"x-refresh", 1}])
  26       image_url = "http://example.com/media/example.jpg"
  27       Pleroma.Web.MediaProxy.put_in_deleted_urls(image_url)
  28
  29       mock(fn
  30         %{
  31           method: :purge,
  32           url: "http://example.com/media/example.jpg",
  33           headers: [{"x-refresh", 1}]
  34         } ->
  35           %Tesla.Env{status: 200}
  36       end)
  37
  38       assert capture_log(fn ->
  39                assert Pleroma.Web.MediaProxy.in_deleted_urls(image_url)
  40                assert Invalidation.purge([image_url]) == {:ok, [image_url]}
  41                assert Pleroma.Web.MediaProxy.in_deleted_urls(image_url)
  42              end) =~ "Running cache purge: [\"#{image_url}\"]"
  43     end
  44   end
  45
  46   describe "Invalidation.Script" do
  47     test "run script to clear cache" do
  48       Config.put([:media_proxy, :enabled], false)
  49       Config.put([:media_proxy, :invalidation, :enabled], true)
  50       Config.put([:media_proxy, :invalidation, :provider], Invalidation.Script)
  51       Config.put([Invalidation.Script], script_path: "purge-nginx")
  52
  53       image_url = "http://example.com/media/example.jpg"
  54       Pleroma.Web.MediaProxy.put_in_deleted_urls(image_url)
  55
  56       with_mocks [{System, [], [cmd: fn _, _ -> {"ok", 0} end]}] do
  57         assert capture_log(fn ->
  58                  assert Pleroma.Web.MediaProxy.in_deleted_urls(image_url)
  59                  assert Invalidation.purge([image_url]) == {:ok, [image_url]}
  60                  assert Pleroma.Web.MediaProxy.in_deleted_urls(image_url)
  61                end) =~ "Running cache purge: [\"#{image_url}\"]"
  62       end
  63     end
  64   end
  65 end