Init tests

Author: Premwoik

lib/gradient/elixir_fmt.ex

@@ -17,15 +17,19 @@ defmodule Gradient.ElixirFmt do
   def print_error(error, opts) do
     file = Keyword.get(opts, :filename)
     fmt_loc = Keyword.get(opts, :fmt_location, :verbose)
-    opts = Keyword.put(opts, :fmt_type_fun, &ElixirType.pretty_print/1)
 
     case file do
       nil -> :ok
       _ when fmt_loc == :brief -> :io.format("~s:", [file])
       _ -> :io.format("~s: ", [file])
     end
 
-    :io.put_chars(format_type_error(error, opts))
+    :io.put_chars(format_error(error, opts))
+  end
+
+  def format_error(error, opts) do
+    opts = Keyword.put(opts, :fmt_type_fun, &ElixirType.pretty_print/1)
+    format_type_error(error, opts)
   end
 
   @impl Gradient.Fmt

test/examples/type/Elixir.WrongRet.beam

@@ -0,0 +1,49 @@
+defmodule WrongRet do
+  @spec ret_wrong_atom() :: atom()
+  def ret_wrong_atom, do: 1
+
+  @spec ret_wrong_atom2() :: atom()
+  def ret_wrong_atom2, do: {:ok, []}
+
+  @spec ret_wrong_atom3() :: atom()
+  def ret_wrong_atom3, do: %{a: 1}
+
+  @spec ret_wrong_atom4() :: atom()
+  def ret_wrong_atom4, do: false
+
+  @spec ret_wrong_integer() :: integer()
+  def ret_wrong_integer, do: 1.0
+
+  @spec ret_wrong_integer2() :: integer()
+  def ret_wrong_integer2, do: :ok
+
+  @spec ret_wrong_integer3() :: integer()
+  def ret_wrong_integer3, do: true
+
+  @spec ret_wrong_integer4() :: integer()
+  def ret_wrong_integer4, do: [1, 2, 3]
+
+  @spec ret_out_of_range_int() :: 1..10
+  def ret_out_of_range_int, do: 12
+
+  @spec ret_wrong_boolean() :: boolean()
+  def ret_wrong_boolean, do: :ok
+
+  @spec ret_wrong_boolean2() :: boolean()
+  def ret_wrong_boolean2, do: "1234"
+
+  @spec ret_wrong_boolean3() :: boolean()
+  def ret_wrong_boolean3, do: 1
+
+  @spec ret_wrong_boolean4() :: boolean()
+  def ret_wrong_boolean4, do: [a: 1, b: 2]
+
+  @spec ret_wrong_keyword() :: keyword()
+  def ret_wrong_keyword, do: [1, 2, 3]
+
+  @spec ret_wrong_tuple() :: tuple()
+  def ret_wrong_tuple, do: %{a: 1, b: 2}
+
+  @spec ret_wrong_map() :: map()
+  def ret_wrong_map, do: {:a, 1, 2}
+end

test/examples/type/wrong_ret.ex

@@ -3,6 +3,8 @@ defmodule Gradient.ElixirFmtTest do
   doctest Gradient.ElixirFmt
 
   alias Gradient.ElixirFmt
+  alias Gradient.TestHelpers
+  alias Gradient.AstSpecifier
 
   @example_module_path "test/examples/simple_app.ex"
 
@@ -18,6 +20,28 @@ defmodule Gradient.ElixirFmtTest do
     assert res == expected
   end
 
+  describe "types format" do
+    test "wrong return type" do
+      {_tokens, ast} = load("/type/Elixir.WrongRet.beam", "/type/wrong_ret.ex")
+      opts = []
+      errors = type_check_file(ast, opts)
+      for e <- errors do
+        :io.put_chars(e)
+      end
+
+    end
+  end
+
+  def type_check_file(ast, opts) do
+    forms = AstSpecifier.specify(ast)
+    opts = Keyword.put(opts, :return_errors, true)
+    opts = Keyword.put(opts, :forms, forms)
+
+    forms
+    |> :gradualizer.type_check_forms(opts)
+    |> Enum.map(fn {_, err} -> ElixirFmt.format_error(err, opts) end)
+  end
+
   @tag :skip
   test "format_expr_type_error/4" do
     opts = [forms: basic_erlang_forms()]