akkoma/test/pleroma/translators/deepl_test.exs

defmodule Pleroma.Akkoma.Translators.DeepLTest do
  use Pleroma.DataCase, async: true

  alias Pleroma.Akkoma.Translators.DeepL

  describe "translating with deepl" do
    setup do
      clear_config([:deepl, :api_key], "deepl_api_key")
    end

    test "should list supported languages" do
      clear_config([:deepl, :tier], :free)

      Tesla.Mock.mock(fn
        %{method: :get, url: "https://api-free.deepl.com/v2/languages?type=target"} = env ->
          auth_header = Enum.find(env.headers, fn {k, _v} -> k == "authorization" end)
          assert {"authorization", "DeepL-Auth-Key deepl_api_key"} = auth_header

          %Tesla.Env{
            status: 200,
            body:
              Jason.encode!([
                %{
                  "language" => "BG",
                  "name" => "Bulgarian",
                  "supports_formality" => false
                },
                %{
                  "language" => "CS",
                  "name" => "Czech",
                  "supports_formality" => false
                }
              ])
          }
      end)

      assert {:ok, [%{code: "BG", name: "Bulgarian"}, %{code: "CS", name: "Czech"}]} =
               DeepL.languages()
    end

    test "should work with the free tier" do
      clear_config([:deepl, :tier], :free)

      Tesla.Mock.mock(fn
        %{method: :post, url: "https://api-free.deepl.com/v2/translate"} = env ->
          auth_header = Enum.find(env.headers, fn {k, _v} -> k == "authorization" end)
          assert {"authorization", "DeepL-Auth-Key deepl_api_key"} = auth_header

          %Tesla.Env{
            status: 200,
            body:
              Jason.encode!(%{
                translations: [
                  %{
                    "text" => "I will crush you",
                    "detected_source_language" => "ja"
                  }
                ]
              })
          }
      end)

      assert {:ok, "ja", "I will crush you"} = DeepL.translate("ギュギュ握りつぶしちゃうぞ", nil, "en")
    end

    test "should work with the pro tier" do
      clear_config([:deepl, :tier], :pro)

      Tesla.Mock.mock(fn
        %{method: :post, url: "https://api.deepl.com/v2/translate"} = env ->
          auth_header = Enum.find(env.headers, fn {k, _v} -> k == "authorization" end)
          assert {"authorization", "DeepL-Auth-Key deepl_api_key"} = auth_header

          %Tesla.Env{
            status: 200,
            body:
              Jason.encode!(%{
                translations: [
                  %{
                    "text" => "I will crush you",
                    "detected_source_language" => "ja"
                  }
                ]
              })
          }
      end)

      assert {:ok, "ja", "I will crush you"} = DeepL.translate("ギュギュ握りつぶしちゃうぞ", nil, "en")
    end

    test "should assign source language if set" do
      clear_config([:deepl, :tier], :pro)

      Tesla.Mock.mock(fn
        %{method: :post, url: "https://api.deepl.com/v2/translate"} = env ->
          auth_header = Enum.find(env.headers, fn {k, _v} -> k == "authorization" end)
          assert {"authorization", "DeepL-Auth-Key deepl_api_key"} = auth_header
          assert String.contains?(env.body, "source_lang=ja")

          %Tesla.Env{
            status: 200,
            body:
              Jason.encode!(%{
                translations: [
                  %{
                    "text" => "I will crush you",
                    "detected_source_language" => "ja"
                  }
                ]
              })
          }
      end)

      assert {:ok, "ja", "I will crush you"} = DeepL.translate("ギュギュ握りつぶしちゃうぞ", "ja", "en")
    end

    test "should gracefully fail if the API errors" do
      clear_config([:deepl, :tier], :free)

      Tesla.Mock.mock(fn
        %{method: :post, url: "https://api-free.deepl.com/v2/translate"} ->
          %Tesla.Env{
            status: 403,
            body: ""
          }
      end)

      assert {:error, "DeepL request failed (code 403)"} =
               DeepL.translate("ギュギュ握りつぶしちゃうぞ", nil, "en")
    end
  end
end