Stop wrapping ExUnit

test/blanks_test.exs

@@ -51,4 +51,38 @@ defmodule BlanksTest do
 
     assert Blanks.count(ast) == 2
   end
+
+  test "replaces whole line containing blank" do
+    ast = quote do
+      1 + 2
+      2 + ___
+    end
+
+    expected_result = quote do
+      1 + 2
+      true
+    end
+
+    actual_result = Blanks.replace_line(ast, fn(_) -> true end)
+
+    assert actual_result == expected_result
+  end
+
+  test "replacement fn can access line" do
+    ast = quote do
+      1 + 2
+      2 + ___
+    end
+
+    expected_result = quote do
+      1 + 2
+      some_fun(2 + ___)
+    end
+
+    actual_result = Blanks.replace_line(ast, fn(line) ->
+      quote do: some_fun(unquote(line))
+    end)
+
+    assert actual_result == expected_result
+  end
 end

test/executor_test.exs

@@ -8,6 +8,6 @@ defmodule ExecuteTest do
   test "stops at the first failing koan" do
     {:failed, %{error: _, file: file, line: line}, SampleKoan, _name} = Execute.run_module(SampleKoan)
     assert file == 'test/support/sample_koan.ex'
-    assert line == 5
+    assert line == 4
   end
 end