[java-inspections] IDEA-247103 "Assert x != null" quick fix does nothing

GitOrigin-RevId: f1a0fa854bceb37349dfb893cf1bf2e0df301726

java/java-impl/src/com/intellij/codeInspection/AddAssertStatementFix.java

@@ -42,7 +42,7 @@ public class AddAssertStatementFix implements LocalQuickFix {
 
   @Override
   public void applyFix(@NotNull Project project, @NotNull ProblemDescriptor descriptor) {
-    PsiExpression element = PsiTreeUtil.getParentOfType(descriptor.getPsiElement(), PsiExpression.class);
+    PsiExpression element = PsiTreeUtil.getNonStrictParentOfType(descriptor.getPsiElement(), PsiExpression.class);
     if (element == null) return;
     CodeBlockSurrounder surrounder = CodeBlockSurrounder.forExpression(element);
     if (surrounder == null) return;

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/addAssert/afterInForDeclaration.java

@@ -0,0 +1,14 @@
+// "Assert 'list != null'" "true"
+import java.util.List;
+
+class A{
+  void test(List<String> list){
+    if(list == null) {
+      System.out.println("oops");
+    }
+      assert list != null;
+      for(String s : list) {
+      
+    }
+  }
+}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/addAssert/beforeInForDeclaration.java

@@ -0,0 +1,13 @@
+// "Assert 'list != null'" "true"
+import java.util.List;
+
+class A{
+  void test(List<String> list){
+    if(list == null) {
+      System.out.println("oops");
+    }
+    for(String s : li<caret>st) {
+      
+    }
+  }
+}