fix: fix incorrect error message on unsupported class declaration

algorandfoundation · Apr 5, 2024 · 8eb5a78 · 8eb5a78
1 parent 6beb29d
commit 8eb5a78
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/src/puya/awst_build/module.py b/src/puya/awst_build/module.py
@@ -229,8 +229,9 @@ def visit_class_def(self, cdef: mypy.nodes.ClassDef) -> StatementResult:
                 return [lambda ctx: ContractASTConverter.convert(ctx, cdef, class_options)]
         else:
             self._error(
-                f"not a subclass of {constants.CONTRACT_BASE_ALIAS}"
-                f" or a direct subclass of {constants.STRUCT_BASE_ALIAS}",
+                f"Unsupported class declaration."
+                f" Contract classes must inherit either directly"
+                f" or indirectly from {constants.CONTRACT_BASE_ALIAS}.",
                 location=cdef,
             )
         return []

diff --git a/tests/test_expected_output/module.test b/tests/test_expected_output/module.test
@@ -54,7 +54,7 @@ class AllowedContract(Contract):
     def clear_state_program(self) -> UInt64:
         return UInt64(1)
 
-class NotAllowed: ## E: not a subclass of algopy.Contract or a direct subclass of algopy.Struct
+class NotAllowed: ## E: Unsupported class declaration. Contract classes must inherit either directly or indirectly from algopy.Contract.
     pass
 
 ## case: all_ignored