From 6b309c938149b03e2da9a9caf22b9a62e77fc74a Mon Sep 17 00:00:00 2001
From: Gaith Hallak <gaithhallak@gmail.com>
Date: Tue, 21 Jun 2022 20:19:24 +0400
Subject: [PATCH] Expand the tests for missing init function

---
 test/aeso_compiler_tests.erl                           | 10 ++++++++--
 .../missing_init_fun_alias_to_alias_to_type.aes        |  3 +++
 test/contracts/missing_init_fun_alias_to_type.aes      |  2 ++
 test/contracts/missing_init_fun_state_unit.aes         |  9 +++++++++
 test/contracts/missing_init_function.aes               |  3 ---
 5 files changed, 22 insertions(+), 5 deletions(-)
 create mode 100644 test/contracts/missing_init_fun_alias_to_alias_to_type.aes
 create mode 100644 test/contracts/missing_init_fun_alias_to_type.aes
 create mode 100644 test/contracts/missing_init_fun_state_unit.aes
 delete mode 100644 test/contracts/missing_init_function.aes

diff --git a/test/aeso_compiler_tests.erl b/test/aeso_compiler_tests.erl
index 561254c..cb4d047 100644
--- a/test/aeso_compiler_tests.erl
+++ b/test/aeso_compiler_tests.erl
@@ -208,6 +208,7 @@ compilable_contracts() ->
      "polymorphism_contract_interface_extensions",
      "polymorphism_contract_interface_same_decl_multi_interface",
      "polymorphism_contract_interface_same_name_same_type",
+     "missing_init_fun_state_unit",
      "test" % Custom general-purpose test file. Keep it last on the list.
     ].
 
@@ -1035,9 +1036,14 @@ failing_contracts() ->
                   [<<?Pos(3,12)
                      "The event type cannot be parameterized">>
                   ])
-    , ?TYPE_ERROR(missing_init_function,
+    , ?TYPE_ERROR(missing_init_fun_alias_to_type,
                   [<<?Pos(1,10)
-                     "Missing `init` function for the contract `MissingInitFunction`.\n"
+                     "Missing `init` function for the contract `AliasToType`.\n"
+                     "The `init` function can only be omitted if the state type is `unit`">>
+                  ])
+    , ?TYPE_ERROR(missing_init_fun_alias_to_alias_to_type,
+                  [<<?Pos(1,10)
+                     "Missing `init` function for the contract `AliasToAliasToType`.\n"
                      "The `init` function can only be omitted if the state type is `unit`">>
                   ])
     ].
diff --git a/test/contracts/missing_init_fun_alias_to_alias_to_type.aes b/test/contracts/missing_init_fun_alias_to_alias_to_type.aes
new file mode 100644
index 0000000..7514fd0
--- /dev/null
+++ b/test/contracts/missing_init_fun_alias_to_alias_to_type.aes
@@ -0,0 +1,3 @@
+contract AliasToAliasToType =
+  type alias = int * int
+  type state = alias
\ No newline at end of file
diff --git a/test/contracts/missing_init_fun_alias_to_type.aes b/test/contracts/missing_init_fun_alias_to_type.aes
new file mode 100644
index 0000000..3e0b828
--- /dev/null
+++ b/test/contracts/missing_init_fun_alias_to_type.aes
@@ -0,0 +1,2 @@
+contract AliasToType =
+  type state = int * int
\ No newline at end of file
diff --git a/test/contracts/missing_init_fun_state_unit.aes b/test/contracts/missing_init_fun_state_unit.aes
new file mode 100644
index 0000000..04603e2
--- /dev/null
+++ b/test/contracts/missing_init_fun_state_unit.aes
@@ -0,0 +1,9 @@
+contract AliasToAliasToUnit =
+  type alias = unit
+  type state = alias
+
+contract AliasToUnit =
+  type state = unit
+
+main contract ImplicitState =
+    type sometype = int
\ No newline at end of file
diff --git a/test/contracts/missing_init_function.aes b/test/contracts/missing_init_function.aes
deleted file mode 100644
index 49372fb..0000000
--- a/test/contracts/missing_init_function.aes
+++ /dev/null
@@ -1,3 +0,0 @@
-contract MissingInitFunction =
-  type state = int * int
-