fix implements visit

vyperlang · Feb 28, 2023 · b7d7d88 · b7d7d88
1 parent 7efc115
commit b7d7d88
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 3 deletions.
diff --git a/tests/parser/syntax/test_interfaces.py b/tests/parser/syntax/test_interfaces.py
@@ -91,6 +91,21 @@ def foo(): nonpayable
     """,
         StructureException,
     ),
+    (
+        """
+implements: 123
+    """,
+        StructureException,
+    ),
+    (
+        """
+struct Foo:
+    a: uint256
+
+implements: Foo
+    """,
+        StructureException,
+    ),
 ]
 
 

diff --git a/vyper/semantics/analysis/module.py b/vyper/semantics/analysis/module.py
@@ -155,10 +155,11 @@ def __init__(
             self_members[fn_name].recursive_calls = function_set
 
     def visit_ImplementsDecl(self, node):
-        interface_name = node.annotation.id
+        type_ = type_from_annotation(node.annotation)
+        if not isinstance(type_, InterfaceT):
+            raise StructureException("Invalid interface name", node.annotation)
 
-        other_iface = self.namespace[interface_name]
-        other_iface.validate_implements(node)
+        type_.validate_implements(node)
 
     def visit_VariableDecl(self, node):
         name = node.get("target.id")