[FIR] Fix suppressing diagnostics for some expressions

compiler/fir/checkers/src/org/jetbrains/kotlin/fir/analysis/collectors/AbstractDiagnosticCollectorVisitor.kt

@@ -27,7 +27,10 @@ abstract class AbstractDiagnosticCollectorVisitor(
     protected open fun shouldVisitDeclaration(declaration: FirDeclaration) = true
     protected open fun onDeclarationExit(declaration: FirDeclaration) {}
 
-    protected abstract fun visitNestedElements(element: FirElement)
+    protected open fun visitNestedElements(element: FirElement) {
+        element.acceptChildren(this, null)
+    }
+
     protected abstract fun checkElement(element: FirElement)
 
     override fun visitElement(element: FirElement, data: Nothing?) {
@@ -210,15 +213,13 @@ abstract class AbstractDiagnosticCollectorVisitor(
 
     private fun visitWithQualifiedAccess(qualifiedAccess: FirQualifiedAccess) {
         return withQualifiedAccess(qualifiedAccess) {
-            checkElement(qualifiedAccess)
+            visitElement(qualifiedAccess, null)
-            visitNestedElements(qualifiedAccess)
         }
     }
 
     private fun visitWithGetClassCall(getClassCall: FirGetClassCall) {
         return withGetClassCall(getClassCall) {
-            checkElement(getClassCall)
+            visitElement(getClassCall, null)
-            visitNestedElements(getClassCall)
         }
     }
 
@@ -303,4 +304,4 @@ abstract class AbstractDiagnosticCollectorVisitor(
             allErrorsSuppressed = AbstractDiagnosticCollector.SUPPRESS_ALL_ERRORS in arguments
         )
     }
-}
+}

compiler/fir/checkers/src/org/jetbrains/kotlin/fir/analysis/collectors/CheckerRunningDiagnosticCollectorVisitor.kt

@@ -14,10 +14,6 @@ open class CheckerRunningDiagnosticCollectorVisitor(
     protected val components: List<AbstractDiagnosticCollectorComponent>
 ) : AbstractDiagnosticCollectorVisitor(context) {
 
-    override fun visitNestedElements(element: FirElement) {
-        element.acceptChildren(this, null)
-    }
-
     override fun checkElement(element: FirElement) {
         components.forEach {
             element.accept(it, context)