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(:banned_urls_cache) end)
  16   end
  17
  18   describe "Invalidation.Http" do
  19     test "perform request to clear cache" do
  20       Config.put([:media_proxy, :enabled], false)
  21       Config.put([:media_proxy, :invalidation, :enabled], true)
  22       Config.put([:media_proxy, :invalidation, :provider], Invalidation.Http)
  23
  24       Config.put([Invalidation.Http], method: :purge, headers: [{"x-refresh", 1}])
  25       image_url = "http://example.com/media/example.jpg"
  26       Pleroma.Web.MediaProxy.put_in_banned_urls(image_url)
  27
  28       mock(fn
  29         %{
  30           method: :purge,
  31           url: "http://example.com/media/example.jpg",
  32           headers: [{"x-refresh", 1}]
  33         } ->
  34           %Tesla.Env{status: 200}
  35       end)
  36
  37       assert capture_log(fn ->
  38                assert Pleroma.Web.MediaProxy.in_banned_urls(image_url)
  39                assert Invalidation.purge([image_url]) == {:ok, [image_url]}
  40                assert Pleroma.Web.MediaProxy.in_banned_urls(image_url)
  41              end) =~ "Running cache purge: [\"#{image_url}\"]"
  42     end
  43   end
  44
  45   describe "Invalidation.Script" do
  46     test "run script to clear cache" do
  47       Config.put([:media_proxy, :enabled], false)
  48       Config.put([:media_proxy, :invalidation, :enabled], true)
  49       Config.put([:media_proxy, :invalidation, :provider], Invalidation.Script)
  50       Config.put([Invalidation.Script], script_path: "purge-nginx")
  51
  52       image_url = "http://example.com/media/example.jpg"
  53       Pleroma.Web.MediaProxy.put_in_banned_urls(image_url)
  54
  55       with_mocks [{System, [], [cmd: fn _, _ -> {"ok", 0} end]}] do
  56         assert capture_log(fn ->
  57                  assert Pleroma.Web.MediaProxy.in_banned_urls(image_url)
  58                  assert Invalidation.purge([image_url]) == {:ok, [image_url]}
  59                  assert Pleroma.Web.MediaProxy.in_banned_urls(image_url)
  60                end) =~ "Running cache purge: [\"#{image_url}\"]"
  61       end
  62     end
  63   end
  64 end