Add test on recursion

SonarSource · May 8, 2024 · a88d53e · a88d53e
1 parent 48b661c
commit a88d53e
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/python-frontend/src/test/java/org/sonar/python/semantic/v2/TypeInferenceV2Test.java b/python-frontend/src/test/java/org/sonar/python/semantic/v2/TypeInferenceV2Test.java
@@ -665,6 +665,19 @@ def foo():
       """).typeV2().unwrappedType()).isEqualTo(INT_TYPE);
   }
 
+  @Test
+  void recursive_function() {
+    FileInput fileInput = inferTypes("""
+      def my_recursive_func(n):
+        ...
+        my_recursive_func(n-1)
+      """);
+    FunctionDef functionDef = ((FunctionDef) fileInput.statements().statements().get(0));
+    FunctionType functionType = (FunctionType) functionDef.name().typeV2();
+    CallExpression call = (CallExpression) PythonTestUtils.getAllDescendant(fileInput, tree -> tree.is(Tree.Kind.CALL_EXPR)).get(0);
+    assertThat(call.callee().typeV2()).isEqualTo(functionType);
+  }
+
   @Test
   void flow_insensitive_when_try_except() {
     FileInput fileInput = inferTypes("""