Add a regression test for generator raised StopIteration #4723

pylint-dev · Jul 19, 2021 · 4b0ea31 · 4b0ea31
1 parent f437752
commit 4b0ea31
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 0 deletions.
diff --git a/ChangeLog b/ChangeLog
@@ -84,6 +84,11 @@ Release date: TBA
 
   Closes #4685
 
+* Fix a crash when a StopIteration was raised when inferring
+  a faulty function in a context manager.
+
+  Closes #4723
+
 
 What's New in Pylint 2.9.3?
 ===========================

diff --git a/tests/functional/r/regression/regression_4723.py b/tests/functional/r/regression/regression_4723.py
@@ -0,0 +1,21 @@
+"""Latest version of astroid (2.6.3) causes RuntimeError:
+generator raised StopIteration #4723"""
+# pylint: disable=invalid-name,missing-docstring,too-few-public-methods
+
+import contextlib
+
+
+class A:
+    @contextlib.contextmanager
+    def get(self):
+        yield self
+
+
+class B(A):
+    def play():  # [no-method-argument]
+        pass
+
+
+def func():
+    with B().get() as b:
+        b.play()
diff --git a/tests/functional/r/regression/regression_4723.txt b/tests/functional/r/regression/regression_4723.txt
@@ -0,0 +1 @@
+no-method-argument:15:4:B.play:Method has no argument:HIGH