git.squeep.com Git - akkoma/blob - test/support/gun_mock.ex

   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.GunMock do
   6   @behaviour Pleroma.Gun
   7
   8   alias Pleroma.Gun
   9   alias Pleroma.GunMock
  10
  11   @impl Gun
  12   def open('some-domain.com', 443, _) do
  13     {:ok, conn_pid} = Task.start_link(fn -> Process.sleep(1_000) end)
  14
  15     Registry.register(GunMock, conn_pid, %{
  16       origin_scheme: "https",
  17       origin_host: 'some-domain.com',
  18       origin_port: 443
  19     })
  20
  21     {:ok, conn_pid}
  22   end
  23
  24   @impl Gun
  25   def open(ip, port, _)
  26       when ip in [{10_755, 10_368, 61_708, 131, 64_206, 45_068, 0, 9_694}, {127, 0, 0, 1}] and
  27              port in [80, 443] do
  28     {:ok, conn_pid} = Task.start_link(fn -> Process.sleep(1_000) end)
  29
  30     scheme = if port == 443, do: "https", else: "http"
  31
  32     Registry.register(GunMock, conn_pid, %{
  33       origin_scheme: scheme,
  34       origin_host: ip,
  35       origin_port: port
  36     })
  37
  38     {:ok, conn_pid}
  39   end
  40
  41   @impl Gun
  42   def open('localhost', 1234, %{
  43         protocols: [:socks],
  44         proxy: {:socks5, 'localhost', 1234},
  45         socks_opts: %{host: 'proxy-socks.com', port: 80, version: 5}
  46       }) do
  47     {:ok, conn_pid} = Task.start_link(fn -> Process.sleep(1_000) end)
  48
  49     Registry.register(GunMock, conn_pid, %{
  50       origin_scheme: "http",
  51       origin_host: 'proxy-socks.com',
  52       origin_port: 80
  53     })
  54
  55     {:ok, conn_pid}
  56   end
  57
  58   @impl Gun
  59   def open('localhost', 1234, %{
  60         protocols: [:socks],
  61         proxy: {:socks4, 'localhost', 1234},
  62         socks_opts: %{
  63           host: 'proxy-socks.com',
  64           port: 443,
  65           protocols: [:http2],
  66           tls_opts: [],
  67           transport: :tls,
  68           version: 4
  69         }
  70       }) do
  71     {:ok, conn_pid} = Task.start_link(fn -> Process.sleep(1_000) end)
  72
  73     Registry.register(GunMock, conn_pid, %{
  74       origin_scheme: "https",
  75       origin_host: 'proxy-socks.com',
  76       origin_port: 443
  77     })
  78
  79     {:ok, conn_pid}
  80   end
  81
  82   @impl Gun
  83   def open('gun-not-up.com', 80, _opts), do: {:error, :timeout}
  84
  85   @impl Gun
  86   def open('example.com', port, _) when port in [443, 115] do
  87     {:ok, conn_pid} = Task.start_link(fn -> Process.sleep(1_000) end)
  88
  89     Registry.register(GunMock, conn_pid, %{
  90       origin_scheme: "https",
  91       origin_host: 'example.com',
  92       origin_port: 443
  93     })
  94
  95     {:ok, conn_pid}
  96   end
  97
  98   @impl Gun
  99   def open(domain, 80, _) do
 100     {:ok, conn_pid} = Task.start_link(fn -> Process.sleep(1_000) end)
 101
 102     Registry.register(GunMock, conn_pid, %{
 103       origin_scheme: "http",
 104       origin_host: domain,
 105       origin_port: 80
 106     })
 107
 108     {:ok, conn_pid}
 109   end
 110
 111   @impl Gun
 112   def open({127, 0, 0, 1}, 8123, _) do
 113     Task.start_link(fn -> Process.sleep(1_000) end)
 114   end
 115
 116   @impl Gun
 117   def open('localhost', 9050, _) do
 118     Task.start_link(fn -> Process.sleep(1_000) end)
 119   end
 120
 121   @impl Gun
 122   def await_up(_pid, _timeout), do: {:ok, :http}
 123
 124   @impl Gun
 125   def set_owner(_pid, _owner), do: :ok
 126
 127   @impl Gun
 128   def connect(pid, %{host: _, port: 80}) do
 129     ref = make_ref()
 130     Registry.register(GunMock, ref, pid)
 131     ref
 132   end
 133
 134   @impl Gun
 135   def connect(pid, %{host: _, port: 443, protocols: [:http2], transport: :tls}) do
 136     ref = make_ref()
 137     Registry.register(GunMock, ref, pid)
 138     ref
 139   end
 140
 141   @impl Gun
 142   def await(pid, ref) do
 143     [{_, ^pid}] = Registry.lookup(GunMock, ref)
 144     {:response, :fin, 200, []}
 145   end
 146
 147   @impl Gun
 148   def info(pid) do
 149     [{_, info}] = Registry.lookup(GunMock, pid)
 150     info
 151   end
 152
 153   @impl Gun
 154   def close(_pid), do: :ok
 155 end