77 lines
1.7 KiB
Elixir
77 lines
1.7 KiB
Elixir
defmodule BlankAssertions do
|
|
defmacro assert(expr) do
|
|
if contains_blank?(expr) do
|
|
code = Macro.escape(expr)
|
|
quote do
|
|
raise ExUnit.AssertionError, expr: unquote(code)
|
|
end
|
|
else
|
|
quote do
|
|
ExUnit.Assertions.assert(unquote(expr))
|
|
end
|
|
end
|
|
end
|
|
|
|
defmacro refute(expr) do
|
|
if contains_blank?(expr) do
|
|
code = Macro.escape(expr)
|
|
quote do
|
|
raise ExUnit.AssertionError, expr: unquote(code)
|
|
end
|
|
else
|
|
quote do
|
|
ExUnit.Assertions.refute(unquote(expr))
|
|
end
|
|
end
|
|
end
|
|
|
|
defmacro assert_receive(expr) do
|
|
code = Macro.escape(expr)
|
|
if contains_blank?(expr) do
|
|
quote do
|
|
raise ExUnit.AssertionError, expr: {:assert_receive, [], [unquote(code)]}
|
|
end
|
|
else
|
|
quote do
|
|
receive do
|
|
unquote(expr) -> :ok
|
|
after
|
|
100 -> {:messages, messages} = Process.info(self(), :messages)
|
|
mailbox = Enum.map_join(messages, ", ", &inspect/1)
|
|
message = inspect(unquote(expr))
|
|
ExUnit.Assertions.flunk("message #{message} not found in [#{mailbox}]")
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
def assert(value, opts) do
|
|
ExUnit.Assertions.assert(value, opts)
|
|
end
|
|
|
|
def refute(value, opts) do
|
|
ExUnit.Assertions.refute(value, opts)
|
|
end
|
|
|
|
def flunk(message \\ "Flunked!") do
|
|
ExUnit.Assertions.flunk(message)
|
|
end
|
|
|
|
defp contains_blank?(expr) do
|
|
{_, blank} = Macro.prewalk(expr, false, &blank?/2)
|
|
blank
|
|
end
|
|
|
|
defp blank?(node, true) do
|
|
{node, true}
|
|
end
|
|
|
|
defp blank?({expr, _, _} = node, _acc) do
|
|
{node, expr == :__}
|
|
end
|
|
|
|
defp blank?(expr, _acc) do
|
|
{expr, expr == :__}
|
|
end
|
|
end
|