test

Author: novaugust

test/style/pipes_test.exs

@@ -499,12 +499,14 @@ defmodule Styler.Style.PipesTest do
     end
 
     test "rewrites then/2 when the passed function is a named function reference" do
-      assert_style("a |> then(&fun/1) |> c", "a |> fun() |> c()")
-      assert_style("a |> then(&DateTime.from_is8601/1) |> c", "a |> DateTime.from_is8601() |> c()")
-      assert_style("a |> then(&DateTime.from_is8601/1)", "DateTime.from_is8601(a)")
-      assert_style("a |> then(&fun(&1)) |> c", "a |> fun() |> c()")
-      assert_style("a |> then(&fun(&1, d)) |> c", "a |> fun(d) |> c()")
-      assert_style("a |> then(&fun(d, &1)) |> c", "a |> then(&fun(d, &1)) |> c()")
+      assert_style "a |> then(&fun/1) |> c", "a |> fun() |> c()"
+      assert_style "a |> then(&DateTime.from_is8601/1) |> c", "a |> DateTime.from_is8601() |> c()"
+      assert_style "a |> then(&DateTime.from_is8601/1)", "DateTime.from_is8601(a)"
+      assert_style "a |> then(&fun(&1)) |> c", "a |> fun() |> c()"
+      assert_style "a |> then(&fun(&1, d)) |> c", "a |> fun(d) |> c()"
+
+      assert_style "a |> then(&fun(d, &1)) |> c()"
+      assert_style "a |> then(&fun(&1, d, %{foo: &1})) |> c()"
     end
 
     test "adds parens to 1-arity pipes" do