Fixed tests

Shoobx · Jan 26, 2019 · c315612 · c315612
1 parent b61291c
commit c315612
Showing 1 changed file with 11 additions and 8 deletions.
diff --git a/src/mypy_zope/plugin.py b/src/mypy_zope/plugin.py
@@ -21,7 +21,7 @@
 
 from mypy.nodes import (
     Decorator, Var, Argument, FuncDef, CallExpr, RefExpr, Expression,
-    ClassDef, Statement, Block, IndexExpr,
+    ClassDef, Statement, Block, IndexExpr, MemberExpr,
     MDEF, ARG_POS, ARG_OPT
 )
 
@@ -104,19 +104,22 @@ def get_method_signature_hook(self, fullname: str
     def get_method_hook(self, fullname: str
                         ) -> Optional[Callable[[MethodContext], Type]]:
         # print(f"get_method_hook: {fullname}")
-        def analyze(method_ctx: MethodContext) -> Type:
-            methodname = fullname.split('.')[-1]
-            if methodname in ('providedBy', 'implementedBy'):
-                if methodname == 'providedBy':
+
+        methodname = fullname.split('.')[-1]
+        if methodname in ('providedBy', 'implementedBy'):
+            def analyze(method_ctx: MethodContext) -> Type:
+                assert isinstance(method_ctx.context, CallExpr)
+                assert isinstance(method_ctx.context.callee, MemberExpr)
+                if method_ctx.context.callee.name == 'providedBy':
                     method_ctx.context.callee.fullname = 'builtins.isinstance'
                 else:
                     method_ctx.context.callee.fullname = 'builtins.issubclass'
                 method_ctx.context.args = [method_ctx.args[0][0],
                                            method_ctx.context.callee.expr]
 
-            return method_ctx.default_return_type
-
-        return analyze
+                return method_ctx.default_return_type
+            return analyze
+        return None
 
     def get_attribute_hook(self, fullname: str
                            ) -> Optional[Callable[[AttributeContext], Type]]: