Glauber Campinho
3fcc9241db
Optimizes :maps.put/3 to a VM operation instead of a remote call, including nested calls to it. It does the same for :maps.merge/2 with a simple map on the right side. Closes #7353
303 lines
9.2 KiB
Erlang
303 lines
9.2 KiB
Erlang
-module(control_test).
|
|
-include_lib("eunit/include/eunit.hrl").
|
|
|
|
eval(Content) ->
|
|
{Value, Binding, _, _} = elixir:eval(Content, []),
|
|
{Value, Binding}.
|
|
|
|
to_erl(String) ->
|
|
Forms = elixir:'string_to_quoted!'(String, 1, <<"nofile">>, []),
|
|
{Expr, _, _} = elixir:quoted_to_erl(Forms, elixir:env_for_eval([])),
|
|
Expr.
|
|
|
|
% Booleans
|
|
|
|
booleans_test() ->
|
|
{nil, _} = eval("nil"),
|
|
{true, _} = eval("true"),
|
|
{false, _} = eval("false").
|
|
|
|
% If
|
|
|
|
if_else_kv_args_test() ->
|
|
{1, _} = eval("if(true, do: 1)"),
|
|
{nil, _} = eval("if(false, do: 1)"),
|
|
{2, _} = eval("if(false, do: 1, else: 2)").
|
|
|
|
if_else_kv_blocks_test() ->
|
|
{2, _} = eval("if(false) do\n1\nelse\n2\nend"),
|
|
{2, _} = eval("if(false) do 1 else 2 end"),
|
|
{2, _} = eval("if(false) do 1;else 2; end").
|
|
|
|
multi_line_if_test() ->
|
|
{1, _} = eval("if true\ndo\n1\nelse\n2\nend").
|
|
|
|
% Try
|
|
|
|
try_test() ->
|
|
{2, _} = eval("try do\n:foo.bar\ncatch\n:error, :undef -> 2\nend").
|
|
|
|
try_else_test() ->
|
|
{true, _} = eval("try do\n1\nelse 2 -> false\n1 -> true\nrescue\nErlangError -> nil\nend"),
|
|
{true, _} = eval("try do\n1\nelse {_x, _y} -> false\n_x -> true\nrescue\nErlangError -> nil\nend"),
|
|
{true, _} = eval("try do\n{1, 2}\nelse {3, 4} -> false\n_ -> true\nrescue\nErlangError -> nil\nend").
|
|
|
|
% Receive
|
|
|
|
receive_test() ->
|
|
{10, _} = eval("send self(), :foo\nreceive do\n:foo -> 10\nend"),
|
|
{20, _} = eval("send self(), :bar\nreceive do\n:foo -> 10\n_ -> 20\nend"),
|
|
{30, _} = eval("receive do\nafter 1 -> 30\nend").
|
|
|
|
% Case
|
|
|
|
case_test() ->
|
|
{true, []} = eval("case 1 do\n2 -> false\n1 -> true\nend"),
|
|
{true, []} = eval("case 1 do\n{_x, _y} -> false\n_x -> true\nend"),
|
|
{true, []} = eval("case {1, 2} do; {3, 4} -> false\n_ -> true\nend").
|
|
|
|
case_with_do_ambiguity_test() ->
|
|
{true, _} = eval("case Atom.to_charlist(true) do\n_ -> true\nend").
|
|
|
|
case_with_match_do_ambiguity_test() ->
|
|
{true, _} = eval("case x = Atom.to_charlist(true) do\n_ -> true\nend").
|
|
|
|
case_with_unary_do_ambiguity_test() ->
|
|
{false, _} = eval("! case Atom.to_charlist(true) do\n_ -> true\nend").
|
|
|
|
% Comparison
|
|
|
|
equal_test() ->
|
|
{true, _} = eval(":a == :a"),
|
|
{true, _} = eval("1 == 1"),
|
|
{true, _} = eval("{1, 2} == {1, 2}"),
|
|
{false, _} = eval("1 == 2"),
|
|
{false, _} = eval("{1, 2} == {1, 3}").
|
|
|
|
not_equal_test() ->
|
|
{false, _} = eval(":a != :a"),
|
|
{false, _} = eval("1 != 1"),
|
|
{false, _} = eval("{1, 2} != {1, 2}"),
|
|
{true, _} = eval("1 != 2"),
|
|
{true, _} = eval("{1, 2} != {1, 3}").
|
|
|
|
not_exclamation_mark_test() ->
|
|
{false, _} = eval("! :a"),
|
|
{false, _} = eval("!true"),
|
|
{false, _} = eval("!1"),
|
|
{false, _} = eval("![]"),
|
|
{true, _} = eval("!nil"),
|
|
{true, _} = eval("!false").
|
|
|
|
notnot_exclamation_mark_test() ->
|
|
{true, _} = eval("!! :a"),
|
|
{true, _} = eval("!!true"),
|
|
{true, _} = eval("!!1"),
|
|
{true, _} = eval("!![]"),
|
|
{false, _} = eval("!!nil"),
|
|
{false, _} = eval("!!false").
|
|
|
|
less_greater_test() ->
|
|
{true, _} = eval("1 < 2"),
|
|
{true, _} = eval("1 < :a"),
|
|
{false, _} = eval("1 < 1.0"),
|
|
{false, _} = eval("1 < 1"),
|
|
{true, _} = eval("1 <= 1.0"),
|
|
{true, _} = eval("1 <= 1"),
|
|
{true, _} = eval("1 <= :a"),
|
|
{false, _} = eval("1 > 2"),
|
|
{false, _} = eval("1 > :a"),
|
|
{false, _} = eval("1 > 1.0"),
|
|
{false, _} = eval("1 > 1"),
|
|
{true, _} = eval("1 >= 1.0"),
|
|
{true, _} = eval("1 >= 1"),
|
|
{false, _} = eval("1 >= :a").
|
|
|
|
integer_and_float_test() ->
|
|
{true, _} = eval("1 == 1"),
|
|
{false, _} = eval("1 != 1"),
|
|
{true, _} = eval("1 == 1.0"),
|
|
{false, _} = eval("1 != 1.0"),
|
|
{true, _} = eval("1 === 1"),
|
|
{false, _} = eval("1 !== 1"),
|
|
{false, _} = eval("1 === 1.0"),
|
|
{true, _} = eval("1 !== 1.0").
|
|
|
|
not_test() ->
|
|
{false, _} = eval("not true"),
|
|
{true, _} = eval("not false"),
|
|
?assertError(badarg, eval("not 1")).
|
|
|
|
rearrange_left_not_in_right_test() ->
|
|
{true, _} = eval("false not in []"),
|
|
{false, _} = eval("true not in [true]").
|
|
|
|
andand_test() ->
|
|
F = fun() ->
|
|
eval("defmodule Bar do\ndef foo, do: true\ndef bar, do: false\n def baz(x), do: x == 1\nend"),
|
|
{true, _} = eval("Kernel.&&(true, true)"),
|
|
{true, _} = eval("true && true"),
|
|
{false, _} = eval("true && false"),
|
|
{false, _} = eval("false && true"),
|
|
{false, _} = eval("false && false"),
|
|
{nil, _} = eval("true && nil"),
|
|
{nil, _} = eval("nil && true"),
|
|
{false, _} = eval("false && nil"),
|
|
{true, _} = eval("Bar.foo && Bar.foo"),
|
|
{false, _} = eval("Bar.foo && Bar.bar"),
|
|
{true, _} = eval("Bar.foo && Bar.baz 1"),
|
|
{false, _} = eval("Bar.foo && Bar.baz 2"),
|
|
{true, _} = eval("1 == 1 && 2 < 3"),
|
|
{3, _} = eval("Bar.foo && 1 + 2"),
|
|
{false, _} = eval("Bar.bar && :erlang.error(:bad)"),
|
|
{2, _} = eval("1 && 2"),
|
|
{nil, _} = eval("nil && 2")
|
|
end,
|
|
test_helper:run_and_remove(F, ['Elixir.Bar']).
|
|
|
|
andand_with_literal_test() ->
|
|
{[nil, nil, nil], _} = eval("[nil && 2, nil && 3, nil && 4]").
|
|
|
|
oror_test() ->
|
|
F = fun() ->
|
|
eval("defmodule Bar do\ndef foo, do: true\ndef bar, do: false\n def baz(x), do: x == 1\nend"),
|
|
{true, _} = eval("Kernel.||(false, true)"),
|
|
{true, _} = eval("true || true"),
|
|
{true, _} = eval("true || false"),
|
|
{true, _} = eval("false || true"),
|
|
{false, _} = eval("false || false"),
|
|
{false, _} = eval("nil || false"),
|
|
{nil, _} = eval("false || nil"),
|
|
{true, _} = eval("false || nil || true"),
|
|
{true, _} = eval("Bar.foo || Bar.foo"),
|
|
{true, _} = eval("Bar.foo || Bar.bar"),
|
|
{false, _} = eval("Bar.bar || Bar.bar"),
|
|
{true, _} = eval("Bar.bar || Bar.baz 1"),
|
|
{false, _} = eval("Bar.bar || Bar.baz 2"),
|
|
{false, _} = eval("1 == 2 || 2 > 3"),
|
|
{3, _} = eval("Bar.bar || 1 + 2"),
|
|
{true, _} = eval("Bar.foo || :erlang.error(:bad)"),
|
|
{1, _} = eval("1 || 2"),
|
|
{2, _} = eval("nil || 2"),
|
|
{true, _} = eval("false && false || true")
|
|
end,
|
|
test_helper:run_and_remove(F, ['Elixir.Bar']).
|
|
|
|
cond_line_test() ->
|
|
{'case', 1, _,
|
|
[{clause, 2, _, _, _},
|
|
{clause, 3, _, _, _}]
|
|
} = to_erl("cond do\n 1 -> :ok\n 2 -> :ok\nend").
|
|
|
|
% Optimized
|
|
|
|
optimized_if_test() ->
|
|
{'case', _, _,
|
|
[{clause, _, [{atom, _, false}], [], [{atom, _, else}]},
|
|
{clause, _, [{atom, _, true}], [], [{atom, _, do}]}]
|
|
} = to_erl("if is_list([]), do: :do, else: :else").
|
|
|
|
optimized_andand_test() ->
|
|
{'case', _, _,
|
|
[{clause, _,
|
|
[{var, _, Var}],
|
|
[[{op, _, 'orelse', _, _}]],
|
|
[{var, _, Var}]},
|
|
{clause, _, [{var, _, '_'}], [], [{atom, 0, done}]}]
|
|
} = to_erl("is_list([]) && :done").
|
|
|
|
optimized_oror_test() ->
|
|
{'case', _, _,
|
|
[{clause, 1,
|
|
[{var, 1, _}],
|
|
[[{op, 1, 'orelse', _, _}]],
|
|
[{atom, 0, done}]},
|
|
{clause, 1, [{var, 1, Var}], [], [{var, 1, Var}]}]
|
|
} = to_erl("is_list([]) || :done").
|
|
|
|
optimized_and_test() ->
|
|
{'case',_, _,
|
|
[{clause, _, [{atom, 0, false}], [], [{atom, 0, false}]},
|
|
{clause, _, [{atom, 0, true}], [], [{atom, 0, done}]}]
|
|
} = to_erl("is_list([]) and :done").
|
|
|
|
optimized_or_test() ->
|
|
{'case', _, _,
|
|
[{clause, _, [{atom, 0, false}], [], [{atom, 0, done}]},
|
|
{clause, _, [{atom, 0, true}], [], [{atom, 0, true}]}]
|
|
} = to_erl("is_list([]) or :done").
|
|
|
|
no_after_in_try_test() ->
|
|
{'try', _, [_], [], [_], []} = to_erl("try do :foo.bar() catch _ -> :ok end").
|
|
|
|
optimized_inspect_interpolation_test() ->
|
|
{bin, _,
|
|
[{bin_element, _,
|
|
{call, _, {remote, _,{atom, _, 'Elixir.Kernel'}, {atom, _, inspect}}, [_]},
|
|
default, [binary]}]} = to_erl("\"#{inspect(1)}\"").
|
|
|
|
optimized_map_put_test() ->
|
|
{map, _,
|
|
[{map_field_assoc, _, {atom, _, a}, {integer, _, 1}},
|
|
{map_field_assoc, _, {atom, _, b}, {integer, _, 2}}]
|
|
} = to_erl("Map.put(%{a: 1}, :b, 2)").
|
|
|
|
optimized_map_put_variable_test() ->
|
|
{block, _,
|
|
[_,
|
|
{map, _, {var, _, _},
|
|
[{map_field_assoc, _, {atom, _, a}, {integer, _, 1}}]
|
|
}]
|
|
} = to_erl("x = %{}; Map.put(x, :a, 1)").
|
|
|
|
optimized_nested_map_put_variable_test() ->
|
|
{block, _,
|
|
[_,
|
|
{map, _, {var, _, _},
|
|
[{map_field_assoc, _, {atom, _, a}, {integer, _, 1}},
|
|
{map_field_assoc, _, {atom, _, b}, {integer, _, 2}}]
|
|
}]
|
|
} = to_erl("x = %{}; Map.put(Map.put(x, :a, 1), :b, 2)").
|
|
|
|
optimized_map_merge_test() ->
|
|
{map, _,
|
|
[{map_field_assoc, _, {atom, _, a}, {integer, _, 1}},
|
|
{map_field_assoc, _, {atom, _, b}, {integer, _, 2}},
|
|
{map_field_assoc, _, {atom, _, c}, {integer, _, 3}}]
|
|
} = to_erl("Map.merge(%{a: 1, b: 2}, %{c: 3})").
|
|
|
|
optimized_map_merge_variable_test() ->
|
|
{block, _,
|
|
[_,
|
|
{map, _, {var, _, _},
|
|
[{map_field_assoc, _, {atom, _, a}, {integer, _, 1}}]
|
|
}]
|
|
} = to_erl("x = %{}; Map.merge(x, %{a: 1})").
|
|
|
|
optimized_map_update_and_merge_test() ->
|
|
{block, _,
|
|
[_,
|
|
{map, _, {var, _, _},
|
|
[{map_field_exact, _, {atom, _, a}, {integer, _, 2}},
|
|
{map_field_assoc, _, {atom, _, b}, {integer, _, 3}}]
|
|
}]
|
|
} = to_erl("x = %{a: 1}; Map.merge(%{x | a: 2}, %{b: 3})"),
|
|
{block, _,
|
|
[_,
|
|
{call, _, {remote, _, {atom, _, maps}, {atom, _, merge}},
|
|
[{map, _,
|
|
[{map_field_assoc, _, {atom, _, a}, {integer, _, 2}}]},
|
|
{map, _, {var, _, _},
|
|
[{map_field_exact, _, {atom, _, b}, {integer, _, 3}}]}]
|
|
}]
|
|
} = to_erl("x = %{a: 1}; Map.merge(%{a: 2}, %{x | b: 3})").
|
|
|
|
optimized_nested_map_merge_variable_test() ->
|
|
{block, _,
|
|
[_,
|
|
{map, _, {var, _, _},
|
|
[{map_field_assoc, _, {atom, _, a}, {integer, _, 1}},
|
|
{map_field_assoc, _, {atom, _, b}, {integer, _, 2}}]
|
|
}]
|
|
} = to_erl("x = %{}; Map.merge(Map.merge(x, %{a: 1}), %{b: 2})").
|