Add more tests

Author: GearsDatapacks

compiler-core/src/erlang/tests/inlining.rs

@@ -204,3 +204,40 @@ fn do_side_effects() {
 "#
     );
 }
+
+#[test]
+fn inline_anonymous_function_call() {
+    assert_erl!(
+        "
+pub fn main() {
+  fn(a, b) { #(a, b) }(42, False)
+}
+"
+    );
+}
+
+#[test]
+fn inline_anonymous_function_in_pipe() {
+    assert_erl!(
+        "
+pub fn main() {
+  1 |> fn(x) { x + 1 } |> fn(y) { y * y }
+}
+"
+    );
+}
+
+#[test]
+fn inline_function_capture_in_pipe() {
+    // The function capture is desugared to an anonymous function, so it should
+    // be turned into a direct call to `add`
+    assert_erl!(
+        "
+pub fn main() {
+  1 |> add(4, _)
+}
+
+fn add(a, b) { a + b }
+"
+    );
+}

compiler-core/src/erlang/tests/snapshots/gleam_core__erlang__tests__inlining__inline_anonymous_function_call.snap

@@ -0,0 +1,25 @@
+---
+source: compiler-core/src/erlang/tests/inlining.rs
+expression: "\npub fn main() {\n  fn(a, b) { #(a, b) }(42, False)\n}\n"
+---
+----- SOURCE CODE
+
+pub fn main() {
+  fn(a, b) { #(a, b) }(42, False)
+}
+
+
+----- COMPILED ERLANG
+-module(my@mod).
+-compile([no_auto_import, nowarn_unused_vars, nowarn_unused_function, nowarn_nomatch]).
+-define(FILEPATH, "project/test/my/mod.gleam").
+-export([main/0]).
+
+-file("project/test/my/mod.gleam", 2).
+-spec main() -> {integer(), boolean()}.
+main() ->
+    begin
+        A = 42,
+        B = false,
+        {A, B}
+    end.

compiler-core/src/erlang/tests/snapshots/gleam_core__erlang__tests__inlining__inline_anonymous_function_in_pipe.snap

@@ -0,0 +1,29 @@
+---
+source: compiler-core/src/erlang/tests/inlining.rs
+expression: "\npub fn main() {\n  1 |> fn(x) { x + 1 } |> fn(y) { y * y }\n}\n"
+---
+----- SOURCE CODE
+
+pub fn main() {
+  1 |> fn(x) { x + 1 } |> fn(y) { y * y }
+}
+
+
+----- COMPILED ERLANG
+-module(my@mod).
+-compile([no_auto_import, nowarn_unused_vars, nowarn_unused_function, nowarn_nomatch]).
+-define(FILEPATH, "project/test/my/mod.gleam").
+-export([main/0]).
+
+-file("project/test/my/mod.gleam", 2).
+-spec main() -> integer().
+main() ->
+    _pipe = 1,
+    _pipe@1 = begin
+        X = _pipe,
+        X + 1
+    end,
+    begin
+        Y = _pipe@1,
+        Y * Y
+    end.

compiler-core/src/erlang/tests/snapshots/gleam_core__erlang__tests__inlining__inline_function_capture_in_pipe.snap

@@ -0,0 +1,32 @@
+---
+source: compiler-core/src/erlang/tests/inlining.rs
+expression: "\npub fn main() {\n  1 |> add(4, _)\n}\n\nfn add(a, b) { a + b }\n"
+---
+----- SOURCE CODE
+
+pub fn main() {
+  1 |> add(4, _)
+}
+
+fn add(a, b) { a + b }
+
+
+----- COMPILED ERLANG
+-module(my@mod).
+-compile([no_auto_import, nowarn_unused_vars, nowarn_unused_function, nowarn_nomatch]).
+-define(FILEPATH, "project/test/my/mod.gleam").
+-export([main/0]).
+
+-file("project/test/my/mod.gleam", 6).
+-spec add(integer(), integer()) -> integer().
+add(A, B) ->
+    A + B.
+
+-file("project/test/my/mod.gleam", 2).
+-spec main() -> integer().
+main() ->
+    _pipe = 1,
+    begin
+        _capture = _pipe,
+        add(4, _capture)
+    end.

compiler-core/src/javascript/tests/inlining.rs

@@ -204,3 +204,40 @@ fn do_side_effects() {
 "#
     );
 }
+
+#[test]
+fn inline_anonymous_function_call() {
+    assert_js!(
+        "
+pub fn main() {
+  fn(a, b) { #(a, b) }(42, False)
+}
+"
+    );
+}
+
+#[test]
+fn inline_anonymous_function_in_pipe() {
+    assert_js!(
+        "
+pub fn main() {
+  1 |> fn(x) { x + 1 } |> fn(y) { y * y }
+}
+"
+    );
+}
+
+#[test]
+fn inline_function_capture_in_pipe() {
+    // The function capture is desugared to an anonymous function, so it should
+    // be turned into a direct call to `add`
+    assert_js!(
+        "
+pub fn main() {
+  1 |> add(4, _)
+}
+
+fn add(a, b) { a + b }
+"
+    );
+}

compiler-core/src/javascript/tests/snapshots/gleam_core__javascript__tests__inlining__inline_anonymous_function_call.snap

@@ -0,0 +1,19 @@
+---
+source: compiler-core/src/javascript/tests/inlining.rs
+expression: "\npub fn main() {\n  fn(a, b) { #(a, b) }(42, False)\n}\n"
+---
+----- SOURCE CODE
+
+pub fn main() {
+  fn(a, b) { #(a, b) }(42, False)
+}
+
+
+----- COMPILED JAVASCRIPT
+export function main() {
+  {
+    let a = 42;
+    let b = false;
+    return [a, b];
+  }
+}

compiler-core/src/javascript/tests/snapshots/gleam_core__javascript__tests__inlining__inline_anonymous_function_in_pipe.snap

@@ -0,0 +1,25 @@
+---
+source: compiler-core/src/javascript/tests/inlining.rs
+expression: "\npub fn main() {\n  1 |> fn(x) { x + 1 } |> fn(y) { y * y }\n}\n"
+---
+----- SOURCE CODE
+
+pub fn main() {
+  1 |> fn(x) { x + 1 } |> fn(y) { y * y }
+}
+
+
+----- COMPILED JAVASCRIPT
+export function main() {
+  let _pipe = 1;
+  let _block;
+  {
+    let x = _pipe;
+    _block = x + 1;
+  }
+  let _pipe$1 = _block;
+  {
+    let y = _pipe$1;
+    return y * y;
+  }
+}

compiler-core/src/javascript/tests/snapshots/gleam_core__javascript__tests__inlining__inline_function_capture_in_pipe.snap

@@ -0,0 +1,25 @@
+---
+source: compiler-core/src/javascript/tests/inlining.rs
+expression: "\npub fn main() {\n  1 |> add(4, _)\n}\n\nfn add(a, b) { a + b }\n"
+---
+----- SOURCE CODE
+
+pub fn main() {
+  1 |> add(4, _)
+}
+
+fn add(a, b) { a + b }
+
+
+----- COMPILED JAVASCRIPT
+function add(a, b) {
+  return a + b;
+}
+
+export function main() {
+  let _pipe = 1;
+  {
+    let _capture = _pipe;
+    return add(4, _capture);
+  }
+}