git.squeep.com Git - akkoma/blob - test/http/request_builder_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.HTTP.RequestBuilderTest do
   6   use ExUnit.Case, async: true
   7   use Pleroma.Tests.Helpers
   8   alias Pleroma.Config
   9   alias Pleroma.HTTP.Request
  10   alias Pleroma.HTTP.RequestBuilder
  11
  12   describe "headers/2" do
  13     clear_config([:http, :send_user_agent])
  14     clear_config([:http, :user_agent])
  15
  16     test "don't send pleroma user agent" do
  17       assert RequestBuilder.headers(%Request{}, []) == %Request{headers: []}
  18     end
  19
  20     test "send pleroma user agent" do
  21       Config.put([:http, :send_user_agent], true)
  22       Config.put([:http, :user_agent], :default)
  23
  24       assert RequestBuilder.headers(%Request{}, []) == %Request{
  25                headers: [{"user-agent", Pleroma.Application.user_agent()}]
  26              }
  27     end
  28
  29     test "send custom user agent" do
  30       Config.put([:http, :send_user_agent], true)
  31       Config.put([:http, :user_agent], "totally-not-pleroma")
  32
  33       assert RequestBuilder.headers(%Request{}, []) == %Request{
  34                headers: [{"user-agent", "totally-not-pleroma"}]
  35              }
  36     end
  37   end
  38
  39   describe "add_optional_params/3" do
  40     test "don't add if keyword is empty" do
  41       assert RequestBuilder.add_optional_params(%{}, %{}, []) == %{}
  42     end
  43
  44     test "add query parameter" do
  45       assert RequestBuilder.add_optional_params(
  46                %Request{},
  47                %{query: :query, body: :body, another: :val},
  48                [
  49                  {:query, "param1=val1&param2=val2"},
  50                  {:body, "some body"}
  51                ]
  52              ) == %Request{query: "param1=val1&param2=val2", body: "some body"}
  53     end
  54   end
  55
  56   describe "add_param/4" do
  57     test "add file parameter" do
  58       %Request{
  59         body: %Tesla.Multipart{
  60           boundary: _,
  61           content_type_params: [],
  62           parts: [
  63             %Tesla.Multipart.Part{
  64               body: %File.Stream{
  65                 line_or_bytes: 2048,
  66                 modes: [:raw, :read_ahead, :read, :binary],
  67                 path: "some-path/filename.png",
  68                 raw: true
  69               },
  70               dispositions: [name: "filename.png", filename: "filename.png"],
  71               headers: []
  72             }
  73           ]
  74         }
  75       } = RequestBuilder.add_param(%Request{}, :file, "filename.png", "some-path/filename.png")
  76     end
  77
  78     test "add key to body" do
  79       %{
  80         body: %Tesla.Multipart{
  81           boundary: _,
  82           content_type_params: [],
  83           parts: [
  84             %Tesla.Multipart.Part{
  85               body: "\"someval\"",
  86               dispositions: [name: "somekey"],
  87               headers: [{"content-type", "application/json"}]
  88             }
  89           ]
  90         }
  91       } = RequestBuilder.add_param(%{}, :body, "somekey", "someval")
  92     end
  93
  94     test "add form parameter" do
  95       assert RequestBuilder.add_param(%{}, :form, "somename", "someval") == %{
  96                body: %{"somename" => "someval"}
  97              }
  98     end
  99
 100     test "add for location" do
 101       assert RequestBuilder.add_param(%{}, :some_location, "somekey", "someval") == %{
 102                some_location: [{"somekey", "someval"}]
 103              }
 104     end
 105   end
 106 end