39f09e5c30
This allows failures to be passed back to the parent rather than the koan process just dying and the parent stuck waiting for the result.
97 lines
2.3 KiB
Elixir
97 lines
2.3 KiB
Elixir
defmodule Koans do
|
|
defp valid_name(name) do
|
|
Regex.match?(~r/([A-Z]|\.\.\.).+/, name)
|
|
end
|
|
|
|
defmacro koan(name, body) do
|
|
if not valid_name(name) do
|
|
raise "Name does not start with a capital ltter: #{name}"
|
|
end
|
|
|
|
compiled_name = String.to_atom(name)
|
|
number_of_args = Blanks.count(body)
|
|
compiled_body = Blanks.replace_line(body, &blank_line_replacement/1)
|
|
|
|
quote do
|
|
@koans unquote(compiled_name)
|
|
|
|
generate_test_method(unquote(compiled_name), unquote(number_of_args), unquote(body))
|
|
|
|
def unquote(compiled_name)() do
|
|
try do
|
|
unquote(compiled_body)
|
|
:ok
|
|
rescue
|
|
e in _ -> e
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
defmacro generate_test_method(_name, 0, _body), do: false
|
|
defmacro generate_test_method(name, 1, body) do
|
|
single_var = Blanks.replace(body, Macro.var(:answer, Koans))
|
|
quote do
|
|
def unquote(name)(answer) do
|
|
try do
|
|
unquote(single_var)
|
|
:ok
|
|
rescue
|
|
e in _ -> e
|
|
end
|
|
end
|
|
end
|
|
end
|
|
defmacro generate_test_method(name, number_of_args, body) do
|
|
answer_placeholders = create_vars(number_of_args)
|
|
multi_var = Blanks.replace(body, answer_placeholders)
|
|
quote do
|
|
def unquote(name)({:multiple, answers}) do
|
|
converted = List.to_tuple(answers)
|
|
try do
|
|
unquote(multi_var)
|
|
:ok
|
|
rescue
|
|
e in _ -> e
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
defp blank_line_replacement({:assert, _meta, [expr]}) do
|
|
code = Macro.escape(expr)
|
|
quote do: raise ExUnit.AssertionError, expr: unquote(code)
|
|
end
|
|
|
|
defp blank_line_replacement(line) do
|
|
code = Macro.escape(line)
|
|
quote do: raise ExUnit.AssertionError, expr: unquote(code)
|
|
end
|
|
|
|
defp create_vars(amount) do
|
|
for id <- 0..amount, do: quote do: elem(converted, unquote(id))
|
|
end
|
|
|
|
defmacro __using__(_opts) do
|
|
quote do
|
|
@compile :nowarn_unused_vars
|
|
Module.register_attribute(__MODULE__, :koans, accumulate: true)
|
|
|
|
require ExUnit.Assertions
|
|
import ExUnit.Assertions
|
|
import Koans
|
|
|
|
@before_compile Koans
|
|
end
|
|
end
|
|
|
|
defmacro __before_compile__(env) do
|
|
koans = Module.get_attribute(env.module, :koans) |> Enum.reverse
|
|
quote do
|
|
def all_koans do
|
|
unquote(koans)
|
|
end
|
|
end
|
|
end
|
|
end
|