reverse proxy clients integration tests

tesla reverse proxy client fixes

4 files changed, 96 insertions, 0 deletions

diff --git a/test/reverse_proxy/client/hackney_test.exs b/test/reverse_proxy/client/hackney_test.exs
new file mode 100644
index 000000000..577e0b0b2
--- /dev/null
+++ b/test/reverse_proxy/client/hackney_test.exs
@@ -0,0 +1,7 @@
+defmodule Pleroma.ReverseProxy.Client.HackneyTest do
+  use Pleroma.ReverseProxyClientCase, client: Pleroma.ReverseProxy.Client.Hackney
+
+  defp check_ref(ref) do
+    assert is_reference(ref)
+  end
+end
diff --git a/test/reverse_proxy/client/tesla_test.exs b/test/reverse_proxy/client/tesla_test.exs
new file mode 100644
index 000000000..029a25d0f
--- /dev/null
+++ b/test/reverse_proxy/client/tesla_test.exs
@@ -0,0 +1,9 @@
+defmodule Pleroma.ReverseProxy.Client.TeslaTest do
+  use Pleroma.ReverseProxyClientCase, client: Pleroma.ReverseProxy.Client.Tesla
+
+  defp check_ref(%{pid: pid, stream: stream} = ref) do
+    assert is_pid(pid)
+    assert is_reference(stream)
+    assert ref[:fin]
+  end
+end
diff --git a/test/reverse_proxy_test.exs b/test/reverse_proxy/reverse_proxy_test.exs
index df559cb66..96667ab04 100644
--- a/test/reverse_proxy_test.exs
+++ b/test/reverse_proxy/reverse_proxy_test.exs
@@ -297,4 +297,8 @@ defmodule Pleroma.ReverseProxyTest do
       assert {"content-disposition", "attachment; filename=\"filename.jpg\""} in conn.resp_headers
     end
   end
+
+  describe "integration tests" do
+    @describetag :integration
+  end
 end
diff --git a/test/support/reverse_proxy_client_case.ex b/test/support/reverse_proxy_client_case.ex
new file mode 100644
index 000000000..40cd59ea2
--- /dev/null
+++ b/test/support/reverse_proxy_client_case.ex
@@ -0,0 +1,76 @@
+defmodule Pleroma.ReverseProxyClientCase do
+  defmacro __using__(client: client) do
+    quote do
+      use ExUnit.Case
+      @moduletag :integration
+      @client unquote(client)
+
+      setup do
+        Application.put_env(:tesla, :adapter, Tesla.Adapter.Gun)
+        on_exit(fn -> Application.put_env(:tesla, :adapter, Tesla.Mock) end)
+      end
+
+      test "get response body stream" do
+        {:ok, status, headers, ref} =
+          @client.request(
+            :get,
+            "http://httpbin.org/stream-bytes/10",
+            [{"accept", "application/octet-stream"}],
+            "",
+            []
+          )
+
+        assert status == 200
+        assert headers != []
+
+        {:ok, response, ref} = @client.stream_body(ref)
+        check_ref(ref)
+        assert is_binary(response)
+        assert byte_size(response) == 10
+
+        assert :done == @client.stream_body(ref)
+      end
+
+      test "head response" do
+        {:ok, status, headers} = @client.request(:head, "http://httpbin.org/get", [], "", [])
+
+        assert status == 200
+        assert headers != []
+      end
+
+      test "get error response" do
+        case @client.request(
+               :get,
+               "http://httpbin.org/status/500",
+               [],
+               "",
+               []
+             ) do
+          {:ok, status, headers, ref} ->
+            assert status == 500
+            assert headers != []
+            check_ref(ref)
+
+            assert :ok = @client.close(ref)
+
+          {:ok, status, headers} ->
+            assert headers != []
+        end
+      end
+
+      test "head error response" do
+        {:ok, status, headers} =
+          @client.request(
+            :head,
+            "http://httpbin.org/status/500",
+            [],
+            "",
+            []
+          )
+
+        assert status == 500
+        assert headers != []
+      end
+    end
+  end
+end