1 files changed, 125 insertions, 0 deletions
diff --git a/test/lit/passes/inline-main.wast b/test/lit/passes/inline-main.wast
new file mode 100644
index 000000000..042e3277b
--- /dev/null
+++ b/test/lit/passes/inline-main.wast
@@ -0,0 +1,125 @@
+;; NOTE: Assertions have been generated by update_lit_checks.py --all-items and should not be edited.
+;; NOTE: This test was ported using port_test.py and could be cleaned up.
+
+;; RUN: foreach %s %t wasm-opt --inline-main -S -o - | filecheck %s
+
+(module
+ ;; CHECK:      (type $none_=>_i32 (func (result i32)))
+
+ ;; CHECK:      (type $i32_i32_=>_i32 (func (param i32 i32) (result i32)))
+
+ ;; CHECK:      (export "main" (func $main))
+ (export "main" (func $main))
+ ;; CHECK:      (func $__original_main (result i32)
+ ;; CHECK-NEXT:  (i32.const 0)
+ ;; CHECK-NEXT: )
+ (func $__original_main (result i32)
+  (i32.const 0)
+ )
+ ;; CHECK:      (func $main (param $0 i32) (param $1 i32) (result i32)
+ ;; CHECK-NEXT:  (block $__inlined_func$__original_main (result i32)
+ ;; CHECK-NEXT:   (i32.const 0)
+ ;; CHECK-NEXT:  )
+ ;; CHECK-NEXT: )
+ (func $main (param $0 i32) (param $1 i32) (result i32)
+  (call $__original_main)
+ )
+)
+(module
+ ;; CHECK:      (type $none_=>_i32 (func (result i32)))
+
+ ;; CHECK:      (type $i32_i32_=>_i32 (func (param i32 i32) (result i32)))
+
+ ;; CHECK:      (export "main" (func $main))
+ (export "main" (func $main))
+ ;; CHECK:      (func $__original_main (result i32)
+ ;; CHECK-NEXT:  (i32.const 0)
+ ;; CHECK-NEXT: )
+ (func $__original_main (result i32)
+  (i32.const 0)
+ )
+ ;; CHECK:      (func $main (param $0 i32) (param $1 i32) (result i32)
+ ;; CHECK-NEXT:  (i32.const 0)
+ ;; CHECK-NEXT: )
+ (func $main (param $0 i32) (param $1 i32) (result i32)
+  (i32.const 0)
+ )
+)
+(module
+ ;; CHECK:      (type $i32_i32_=>_i32 (func (param i32 i32) (result i32)))
+
+ ;; CHECK:      (export "main" (func $main))
+ (export "main" (func $main))
+ ;; CHECK:      (func $main (param $0 i32) (param $1 i32) (result i32)
+ ;; CHECK-NEXT:  (i32.const 0)
+ ;; CHECK-NEXT: )
+ (func $main (param $0 i32) (param $1 i32) (result i32)
+  (i32.const 0)
+ )
+)
+(module
+ ;; CHECK:      (type $none_=>_i32 (func (result i32)))
+
+ ;; CHECK:      (type $i32_i32_=>_i32 (func (param i32 i32) (result i32)))
+
+ ;; CHECK:      (export "main" (func $main))
+ (export "main" (func $main))
+ ;; CHECK:      (func $__original_main (result i32)
+ ;; CHECK-NEXT:  (i32.const 0)
+ ;; CHECK-NEXT: )
+ (func $__original_main (result i32)
+  (i32.const 0)
+ )
+ ;; CHECK:      (func $main (param $0 i32) (param $1 i32) (result i32)
+ ;; CHECK-NEXT:  (drop
+ ;; CHECK-NEXT:   (call $__original_main)
+ ;; CHECK-NEXT:  )
+ ;; CHECK-NEXT:  (call $__original_main)
+ ;; CHECK-NEXT: )
+ (func $main (param $0 i32) (param $1 i32) (result i32)
+  (drop (call $__original_main))
+  (call $__original_main)
+ )
+)
+(module
+ ;; CHECK:      (type $none_=>_i32 (func (result i32)))
+
+ ;; CHECK:      (func $__original_main (result i32)
+ ;; CHECK-NEXT:  (i32.const 0)
+ ;; CHECK-NEXT: )
+ (func $__original_main (result i32)
+  (i32.const 0)
+ )
+)
+(module
+ ;; CHECK:      (type $i32_i32_=>_i32 (func (param i32 i32) (result i32)))
+
+ ;; CHECK:      (type $none_=>_i32 (func (result i32)))
+
+ ;; CHECK:      (import "env" "main" (func $main (param i32 i32) (result i32)))
+ (import "env" "main" (func $main (param i32 i32) (result i32)))
+ ;; CHECK:      (export "main" (func $main))
+ (export "main" (func $main))
+ ;; CHECK:      (func $__original_main (result i32)
+ ;; CHECK-NEXT:  (i32.const 0)
+ ;; CHECK-NEXT: )
+ (func $__original_main (result i32)
+  (i32.const 0)
+ )
+)
+(module
+ ;; CHECK:      (type $none_=>_i32 (func (result i32)))
+
+ ;; CHECK:      (type $i32_i32_=>_i32 (func (param i32 i32) (result i32)))
+
+ ;; CHECK:      (import "env" "original_main" (func $__original_main (result i32)))
+ (import "env" "original_main" (func $__original_main (result i32)))
+ ;; CHECK:      (export "main" (func $main))
+ (export "main" (func $main))
+ ;; CHECK:      (func $main (param $0 i32) (param $1 i32) (result i32)
+ ;; CHECK-NEXT:  (call $__original_main)
+ ;; CHECK-NEXT: )
+ (func $main (param $0 i32) (param $1 i32) (result i32)
+  (call $__original_main)
+ )
+)