Support --code-path option

Author: Premwoik

lib/mix/tasks/gradient.ex

@@ -11,6 +11,7 @@ defmodule Mix.Tasks.Gradient do
     * `--no-gradualizer-check` - do not perform the Gradualizer checks
     * `--no-specify` - do not specify missing lines in AST what can
       result in less precise error messages
+    * `--code-path` -  provide a path to the .ex file containing code for analyzed .beam
 
     * `--no-deps` - do not import dependencies to the Gradualizer
     * `--stop_on_first_error` - stop type checking at the first error
@@ -40,6 +41,7 @@ defmodule Mix.Tasks.Gradient do
     no_ex_check: :boolean,
     no_gradualizer_check: :boolean,
     no_specify: :boolean,
+    code_path: :string,
     # checker options
     no_deps: :boolean,
     stop_on_first_error: :boolean,

test/mix/tasks/gradient_test.exs

@@ -9,7 +9,6 @@ defmodule Mix.Tasks.GradientTest do
   @s_wrong_ret_beam "Elixir.SWrongRet.beam"
   @s_wrong_ret_ex "s_wrong_ret.ex"
 
-  #@tag :skip
   test "--no-compile option" do
     info = "Compiling project..."
 
@@ -155,6 +154,13 @@ defmodule Mix.Tasks.GradientTest do
     # FIXME provide implementation
   end
 
+  test "--code-path option" do
+    ex_file = "wrong_ret.ex"
+    output = run_task(@type_path, ["--no-compile", "--code-path", ex_file, "--", @s_wrong_ret_beam])
+    assert not String.contains?(output, @s_wrong_ret_ex)
+    assert String.contains?(output, ex_file)
+  end
+
   def run_task(rel_path, args) do
     run_in_path(rel_path, fn ->
       capture_io(fn -> Mix.Tasks.Gradient.run(args) end)
@@ -168,13 +174,4 @@ defmodule Mix.Tasks.GradientTest do
     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