WIP

Author: edgurgel

test/ham/utils_test.exs

@@ -2,15 +2,24 @@ defmodule Ham.UtilsTest do
   use ExUnit.Case, async: true
   alias Ham.Utils
 
+  defp fetch_test_type(type_name) do
+    {:ok, types} =
+      Code.Typespec.fetch_types(Ham.Test.TestModule)
+
+    Enum.find_value(types, fn
+      {:type, {^type_name, type, _}} -> type
+      _ -> nil
+    end)
+  end
+
   describe "type_to_string/1" do
     test "complex type" do
-      {:ok, [type: {:complex_with_name, type_with_name, _}, type: {:complex, type, _}]} =
-        Code.Typespec.fetch_types(Complex)
-
-      assert Utils.type_to_string(type) ==
+      assert :complex |> fetch_test_type() |> Utils.type_to_string() ==
                "{:ok, binary(), req :: binary()} | {:more, binary(), req :: binary()}"
+    end
 
-      assert Utils.type_to_string(type_with_name) ==
+    test "complex type with annotation" do
+      assert :complex_annotated |> fetch_test_type() |> Utils.type_to_string() ==
                "{:ok, binary(), req :: binary()} | {:more, binary(), req :: binary()} (\"result\")"
     end
   end

test/support/module.ex

@@ -244,17 +244,13 @@ defmodule Ham.Test.TestModule do
   @spec map_type_with_underscore() :: :test.req()
   def map_type_with_underscore, do: %{method: "GET"}
 
-  @spec foo_multiple_options_annotated ::
+  @type complex :: {:ok, binary(), req :: binary()} | {:more, binary(), req :: binary()}
+  @type complex_annotated ::
           result :: {:ok, binary(), req :: binary()} | {:more, binary(), req :: binary()}
+
+  @spec foo_multiple_options_annotated :: complex_annotated
   def foo_multiple_options_annotated, do: {:ok, "a", "b"}
 
-  @spec foo_multiple_options ::
-          result :: {:ok, binary(), req :: binary()} | {:more, binary(), req :: binary()}
+  @spec foo_multiple_options :: complex
   def foo_multiple_options, do: {:ok, "a", "b"}
 end
-
-defmodule Complex do
-  @type complex :: {:ok, binary(), req :: binary()} | {:more, binary(), req :: binary()}
-  @type complex_with_name ::
-          result :: {:ok, binary(), req :: binary()} | {:more, binary(), req :: binary()}
-end