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