SLLS-295 republish all taints on CaYC toggle

SonarSource · Jan 10, 2025 · 2b91c28 · 2b91c28
1 parent c4ca2fb
commit 2b91c28
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/src/main/java/org/sonarsource/sonarlint/ls/CleanAsYouCodeManager.java b/src/main/java/org/sonarsource/sonarlint/ls/CleanAsYouCodeManager.java
@@ -43,6 +43,7 @@ public void onChange(@Nullable WorkspaceSettings oldValue, WorkspaceSettings new
     if (oldValue != null && oldValue.isFocusOnNewCode() != newValue.isFocusOnNewCode()) {
       backendServiceFacade.getBackendService().toggleCleanAsYouCode();
       openFilesCache.getAll().forEach(f -> diagnosticPublisher.publishDiagnostics(f.getUri(), false));
+      diagnosticPublisher.publishTaints();
     }
   }
 }
diff --git a/src/main/java/org/sonarsource/sonarlint/ls/DiagnosticPublisher.java b/src/main/java/org/sonarsource/sonarlint/ls/DiagnosticPublisher.java
@@ -82,6 +82,10 @@ public void publishTaints(URI f) {
     client.publishTaintVulnerabilities(createPublishTaintsParams(f));
   }
 
+  public void publishTaints() {
+    taintVulnerabilitiesCache.getTaintVulnerabilitiesPerFile().forEach((uri, taints) -> client.publishTaintVulnerabilities(createPublishTaintsParams(uri)));
+  }
+
   public void publishHotspots(URI f) {
     client.publishSecurityHotspots(createPublishSecurityHotspotsParams(f));
   }

diff --git a/src/test/java/org/sonarsource/sonarlint/ls/CleanAsYouCodeManagerTest.java b/src/test/java/org/sonarsource/sonarlint/ls/CleanAsYouCodeManagerTest.java
@@ -69,5 +69,6 @@ void shouldNotifyBackendAndRepublishDiagnosticsOnToggle() {
     verify(backendService).toggleCleanAsYouCode();
     verify(diagnosticPublisher).publishDiagnostics(dummyFile1, false);
     verify(diagnosticPublisher).publishDiagnostics(dummyFile2, false);
+    verify(diagnosticPublisher).publishTaints();
   }
 }