git.squeep.com Git - akkoma/blob - test/reverse_proxy/client/tesla_test.exs

   1 # Pleroma: A lightweight social networking server
   2 # Copyright © 2017-2019 Pleroma Authors <https://pleroma.social/>
   3 # SPDX-License-Identifier: AGPL-3.0-only
   4
   5 defmodule Pleroma.ReverseProxy.Client.TeslaTest do
   6   use ExUnit.Case
   7   use Pleroma.Tests.Helpers
   8   alias Pleroma.ReverseProxy.Client
   9   @moduletag :integration
  10
  11   clear_config_all(Pleroma.Gun) do
  12     Pleroma.Config.put(Pleroma.Gun, Pleroma.Gun.API)
  13   end
  14
  15   setup do
  16     Application.put_env(:tesla, :adapter, Tesla.Adapter.Gun)
  17
  18     on_exit(fn ->
  19       Application.put_env(:tesla, :adapter, Tesla.Mock)
  20     end)
  21   end
  22
  23   test "get response body stream" do
  24     {:ok, status, headers, ref} =
  25       Client.Tesla.request(
  26         :get,
  27         "http://httpbin.org/stream-bytes/10",
  28         [{"accept", "application/octet-stream"}],
  29         "",
  30         []
  31       )
  32
  33     assert status == 200
  34     assert headers != []
  35
  36     {:ok, response, ref} = Client.Tesla.stream_body(ref)
  37     check_ref(ref)
  38     assert is_binary(response)
  39     assert byte_size(response) == 10
  40
  41     assert :done == Client.Tesla.stream_body(ref)
  42     assert :ok = Client.Tesla.close(ref)
  43   end
  44
  45   test "head response" do
  46     {:ok, status, headers} = Client.Tesla.request(:head, "https://httpbin.org/get", [], "")
  47
  48     assert status == 200
  49     assert headers != []
  50   end
  51
  52   test "get error response" do
  53     {:ok, status, headers, _body} =
  54       Client.Tesla.request(
  55         :get,
  56         "https://httpbin.org/status/500",
  57         [],
  58         ""
  59       )
  60
  61     assert status == 500
  62     assert headers != []
  63   end
  64
  65   describe "client error" do
  66     setup do
  67       adapter = Application.get_env(:tesla, :adapter)
  68       Application.put_env(:tesla, :adapter, Tesla.Adapter.Hackney)
  69
  70       on_exit(fn -> Application.put_env(:tesla, :adapter, adapter) end)
  71       :ok
  72     end
  73
  74     test "adapter doesn't support reading body in chunks" do
  75       assert_raise RuntimeError,
  76                    "Elixir.Tesla.Adapter.Hackney doesn't support reading body in chunks",
  77                    fn ->
  78                      Client.Tesla.request(
  79                        :get,
  80                        "http://httpbin.org/stream-bytes/10",
  81                        [{"accept", "application/octet-stream"}],
  82                        ""
  83                      )
  84                    end
  85     end
  86   end
  87
  88   defp check_ref(%{pid: pid, stream: stream} = ref) do
  89     assert is_pid(pid)
  90     assert is_reference(stream)
  91     assert ref[:fin]
  92   end
  93 end