Test gradient task with options

Author: Premwoik

lib/gradient.ex

@@ -67,6 +67,7 @@ defmodule Gradient do
   defp maybe_specify_forms(forms, opts) do
     unless opts[:no_specify] do
       IO.puts("Specifying froms...")
+
       forms
       |> put_code_path(opts)
       |> AstSpecifier.specify()

test/examples/type/Elixir.SWrongRet.beam

@@ -0,0 +1,7 @@
+defmodule SWrongRet do
+  @spec ret_wrong_atom() :: atom()
+  def ret_wrong_atom, do: 1
+
+  @spec ret_wrong_atom2() :: atom()
+  def ret_wrong_atom2, do: {:ok, []}
+end

test/examples/type/s_wrong_ret.ex

@@ -0,0 +1,175 @@
+defmodule Mix.Tasks.GradientTest do
+  use ExUnit.Case
+
+  import ExUnit.CaptureIO
+
+  @examples_path "test/examples/"
+  @type_path Path.join([@examples_path, "type"])
+
+  @s_wrong_ret_beam "Elixir.SWrongRet.beam"
+  @s_wrong_ret_ex "s_wrong_ret.ex"
+
+  test "--no-compile option" do
+    build_dir = Path.join(@type_path, "_build")
+    run_task(@type_path, [@s_wrong_ret_beam])
+    assert was_dir?(build_dir)
+
+    run_task(@type_path, ["--no-compile", "--", @s_wrong_ret_beam])
+    assert not was_dir?(build_dir)
+  end
+
+  test "path to the beam file" do
+    output = run_task(@type_path, ["--no-compile", "--", @s_wrong_ret_beam])
+    assert 3 == String.split(output, @s_wrong_ret_ex) |> length()
+  end
+
+  test "path to the ex file" do
+    output = run_task(@type_path, ["--no-compile", "--", @s_wrong_ret_ex])
+    assert 3 == String.split(output, @s_wrong_ret_ex) |> length()
+  end
+
+  test "no_fancy option" do
+    output = run_task(@type_path, ["--no-compile", "--", @s_wrong_ret_beam])
+    assert String.contains?(output, "The integer on line")
+    assert String.contains?(output, "The tuple on line")
+
+    output = run_task(@type_path, ["--no-compile", "--no-fancy", "--", @s_wrong_ret_beam])
+    assert String.contains?(output, "The integer \e[33m1\e[0m on line")
+    assert String.contains?(output, "The tuple \e[33m{:ok, []}\e[0m on line")
+  end
+
+  describe "colors" do
+    test "no_colors option" do
+      output = run_task(@type_path, ["--no-compile", "--", @s_wrong_ret_beam])
+      assert String.contains?(output, IO.ANSI.cyan())
+      assert String.contains?(output, IO.ANSI.red())
+
+      output = run_task(@type_path, ["--no-compile", "--no-colors", "--", @s_wrong_ret_beam])
+      assert not String.contains?(output, IO.ANSI.cyan())
+      assert not String.contains?(output, IO.ANSI.red())
+    end
+
+    test "--expr-color and --type-color option" do
+      output =
+        run_task(@type_path, [
+          "--no-compile",
+          "--no-fancy",
+          "--expr-color",
+          "green",
+          "--type-color",
+          "magenta",
+          "--",
+          @s_wrong_ret_beam
+        ])
+
+      assert String.contains?(output, IO.ANSI.green())
+      assert String.contains?(output, IO.ANSI.magenta())
+    end
+
+    test "--underscore_color option" do
+      output =
+        run_task(@type_path, [
+          "--no-compile",
+          "--underscore-color",
+          "green",
+          "--",
+          @s_wrong_ret_beam
+        ])
+
+      assert String.contains?(output, IO.ANSI.green())
+      assert not String.contains?(output, IO.ANSI.red())
+    end
+  end
+
+  test "--no-gradualizer-check option" do
+    output =
+      run_task(@type_path, ["--no-compile", "--no-gradualizer-check", "--", @s_wrong_ret_beam])
+
+    assert String.contains?(output, "No problems found!")
+  end
+
+  test "--no-ex-check option" do
+    beam = "Elixir.SpecAfterSpec.beam"
+    ex_spec_error_msg = "The spec convert_a/1 on line"
+
+    output = run_task(@examples_path, ["--no-compile", "--", beam])
+    assert String.contains?(output, ex_spec_error_msg)
+
+    output = run_task(@examples_path, ["--no-compile", "--no-ex-check", "--", beam])
+    assert not String.contains?(output, ex_spec_error_msg)
+  end
+
+  test "--no-specify option" do
+    info = "Specifying froms..."
+
+    output = run_task(@type_path, ["--no-compile", "--", @s_wrong_ret_beam])
+    assert String.contains?(output, info)
+
+    output = run_task(@type_path, ["--no-compile", "--no-specify", "--", @s_wrong_ret_beam])
+    assert not String.contains?(output, info)
+  end
+
+  test "--stop-on-first-error option" do
+    output =
+      run_task(@type_path, ["--no-compile", "--stop-on-first-error", "--", @s_wrong_ret_beam])
+
+    assert 2 == String.split(output, @s_wrong_ret_ex) |> length()
+  end
+
+  test "--fmt-location option" do
+    output =
+      run_task(@type_path, ["--no-compile", "--fmt-location", "none", "--", @s_wrong_ret_beam])
+
+    assert String.contains?(output, "s_wrong_ret.ex: The integer is expected to have type")
+
+    output =
+      run_task(@type_path, ["--no-compile", "--fmt-location", "brief", "--", @s_wrong_ret_beam])
+
+    assert String.contains?(output, "s_wrong_ret.ex:3: The integer is expected to have type")
+
+    output =
+      run_task(@type_path, ["--no-compile", "--fmt-location", "verbose", "--", @s_wrong_ret_beam])
+
+    assert String.contains?(
+             output,
+             "s_wrong_ret.ex: The integer on line 3 is expected to have type"
+           )
+  end
+
+  test "--no-deps option" do
+    info = "Loading deps..."
+
+    output = run_task(@type_path, ["--no-compile", "--", @s_wrong_ret_beam])
+    assert String.contains?(output, info)
+
+    output = run_task(@type_path, ["--no-compile", "--no-deps", "--", @s_wrong_ret_beam])
+    assert not String.contains?(output, info)
+  end
+
+  test "--infer option" do
+    # FIXME provide implementation
+  end
+
+  def run_task(rel_path, args) do
+    run_in_path(rel_path, fn ->
+      capture_io(fn -> Mix.Tasks.Gradient.run(args) end)
+    end)
+  end
+
+  def run_in_path(path, fun) do
+    cwd = File.cwd!()
+    File.cd(path)
+    res = fun.()
+    File.cd(cwd)
+    res
+  end
+
+  def was_dir?(dir) do
+    if File.dir?(dir) do
+      :os.cmd(String.to_charlist("rm -Rf " <> dir))
+      true
+    else
+      false
+    end
+  end
+end