Fix grammatical number of vulnerabilities in `ProjectVulnAnalysisCo…

…mpleteSubject` Signed-off-by: nscuro <[email protected]>
DependencyTrack · Sep 4, 2023 · f0f5a14 · f0f5a14
1 parent 757e553
commit f0f5a14
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 11 deletions.
diff --git a/src/main/java/org/dependencytrack/parser/hyades/NotificationModelConverter.java b/src/main/java/org/dependencytrack/parser/hyades/NotificationModelConverter.java
@@ -310,7 +310,7 @@ private static ComponentVulnAnalysisCompleteSubject convert(ComponentVulnAnalysi
         ComponentVulnAnalysisCompleteSubject.Builder builder = ComponentVulnAnalysisCompleteSubject.newBuilder();
         builder.setComponent(component);
         List<Vulnerability> vulnerabilities = componentVulnAnalysisComplete.getVulnerabilityList().stream().map(NotificationModelConverter::convert).toList();
-        builder.addAllVulnerability(vulnerabilities);
+        builder.addAllVulnerabilities(vulnerabilities);
         return builder.build();
     }
 

diff --git a/src/main/proto/org/hyades/notification/v1/notification.proto b/src/main/proto/org/hyades/notification/v1/notification.proto
@@ -198,12 +198,13 @@ message Vulnerability {
     string name = 2;
   }
 }
-message ComponentVulnAnalysisCompleteSubject{
+
+message ComponentVulnAnalysisCompleteSubject {
   Component component = 1;
-  repeated Vulnerability vulnerability = 2;
+  repeated Vulnerability vulnerabilities = 2;
 }
 
-message ProjectVulnAnalysisCompleteSubject{
+message ProjectVulnAnalysisCompleteSubject {
   Project project = 1;
   repeated ComponentVulnAnalysisCompleteSubject findings = 2;
   ProjectVulnAnalysisStatus status = 3;

diff --git a/src/test/java/org/dependencytrack/event/kafka/KafkaStreamsTopologyTest.java b/src/test/java/org/dependencytrack/event/kafka/KafkaStreamsTopologyTest.java
@@ -244,15 +244,15 @@ public void vulnScanResultProcessingTest() throws Exception {
                                 assertThat(subject.getFindingsList()).satisfiesExactlyInAnyOrder(
                                         finding -> {
                                             assertThat(finding.getComponent().getUuid()).isEqualTo(componentA.getUuid().toString());
-                                            assertThat(finding.getVulnerabilityCount()).isEqualTo(1);
-                                            assertThat(finding.getVulnerability(0).getVulnId()).isEqualTo("SNYK-001");
-                                            assertThat(finding.getVulnerability(0).getSource()).isEqualTo("SNYK");
+                                            assertThat(finding.getVulnerabilitiesCount()).isEqualTo(1);
+                                            assertThat(finding.getVulnerabilities(0).getVulnId()).isEqualTo("SNYK-001");
+                                            assertThat(finding.getVulnerabilities(0).getSource()).isEqualTo("SNYK");
                                         },
                                         finding -> {
                                             assertThat(finding.getComponent().getUuid()).isEqualTo(componentB.getUuid().toString());
-                                            assertThat(finding.getVulnerabilityCount()).isEqualTo(1);
-                                            assertThat(finding.getVulnerability(0).getVulnId()).isEqualTo("SONATYPE-001");
-                                            assertThat(finding.getVulnerability(0).getSource()).isEqualTo("OSSINDEX");
+                                            assertThat(finding.getVulnerabilitiesCount()).isEqualTo(1);
+                                            assertThat(finding.getVulnerabilities(0).getVulnId()).isEqualTo("SONATYPE-001");
+                                            assertThat(finding.getVulnerabilities(0).getSource()).isEqualTo("OSSINDEX");
                                         }
                                 );
                             }

diff --git a/src/test/java/org/dependencytrack/parser/hyades/NotificationModelConverterTest.java b/src/test/java/org/dependencytrack/parser/hyades/NotificationModelConverterTest.java
@@ -744,7 +744,7 @@ public void testConvertComponentVulnAnalysisCompleteSubject() throws Exception {
         final var subject = notification.getSubject().unpack(ProjectVulnAnalysisCompleteSubject.class);
         assertProject(subject.getProject());
         assertComponent(subject.getFindingsList().get(0).getComponent());
-        assertVulnerability(subject.getFindingsList().get(0).getVulnerability(0));
+        assertVulnerability(subject.getFindingsList().get(0).getVulnerabilities(0));
         assertThat(subject.getStatus()).isEqualTo(ProjectVulnAnalysisStatus.PROJECT_VULN_ANALYSIS_STATUS_COMPLETED);
     }
 }