Skip to content

Commit cc18cd2

Browse files
author
Pedro Galvão
committed
Removed skip condition in ImportVulnerability and changed tests accordingly
1 parent e6e3346 commit cc18cd2

File tree

4 files changed

+38
-21
lines changed

4 files changed

+38
-21
lines changed

kb-importer/src/main/java/org/eclipse/steady/kb/command/Import.java

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,16 @@ public class Import implements Command {
5959

6060
private static final Logger log = org.apache.logging.log4j.LogManager.getLogger();
6161

62+
private BackendConnector backendConnector;
63+
64+
public Import() {
65+
this.backendConnector = BackendConnector.getInstance();
66+
}
67+
68+
public Import(BackendConnector backendConnector) {
69+
this.backendConnector = backendConnector;
70+
}
71+
6272
/** {@inheritDoc} */
6373
@Override
6474
public Command.NAME getCommandName() {
@@ -121,7 +131,7 @@ private void importVuln(HashMap<String, Object> args, String dirPath) {
121131
String vulnId = vuln.getVulnId();
122132
boolean bugExists = false;
123133
try {
124-
bugExists = BackendConnector.getInstance().isBugExisting(vulnId);
134+
bugExists = this.backendConnector.isBugExisting(vulnId);
125135
} catch (BackendConnectionException e) {
126136
log.error("Can't connect to the Backend");
127137
return;
@@ -142,7 +152,7 @@ private void importVuln(HashMap<String, Object> args, String dirPath) {
142152
for (Task task : importTasks) {
143153
try {
144154
args.put(DIRECTORY_OPTION, dirPath);
145-
task.execute(vuln, args, BackendConnector.getInstance());
155+
task.execute(vuln, args, backendConnector);
146156
} catch (Exception e) {
147157
log.error(
148158
"Got ["

kb-importer/src/main/java/org/eclipse/steady/kb/task/ImportVulnerability.java

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -65,11 +65,6 @@ public void execute(
6565
String vulnId = vuln.getVulnId();
6666
this.backendConnector = _backendConnector;
6767

68-
if (backendConnector.isBugExisting(vuln.getVulnId())) {
69-
log.info("Bug [{}] already exists in backend, analysis will be skipped", vuln.getVulnId());
70-
return;
71-
}
72-
7368
List<Commit> commits = new ArrayList<Commit>();
7469
File file = new File((String) args.get(Import.DIRECTORY_OPTION));
7570

kb-importer/src/test/java/org/eclipse/steady/kb/command/ImportTest.java

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,12 +17,23 @@
1717
* SPDX-FileCopyrightText: Copyright (c) 2018-2020 SAP SE or an SAP affiliate company and Eclipse Steady contributors
1818
*/
1919
package org.eclipse.steady.kb.command;
20+
import org.eclipse.steady.kb.task.MockBackConnector;
21+
import org.eclipse.steady.shared.json.model.AffectedLibrary;
22+
import org.eclipse.steady.kb.model.Vulnerability;
23+
import org.eclipse.steady.kb.util.Metadata;
2024

25+
import com.fasterxml.jackson.databind.ObjectMapper;
2126
import static org.junit.Assert.assertEquals;
2227
import static org.junit.Assert.assertTrue;
28+
import static org.junit.Assert.assertNull;
2329
import java.util.HashMap;
30+
import java.util.List;
2431
import org.apache.commons.cli.Options;
32+
33+
import java.io.IOException;
2534
import org.eclipse.steady.kb.exception.ValidationException;
35+
import com.google.gson.JsonSyntaxException;
36+
import org.eclipse.steady.backend.BackendConnectionException;
2637
import org.junit.Test;
2738

2839
public class ImportTest {
@@ -52,4 +63,19 @@ public void validationFail() throws ValidationException {
5263
args.put("d", "invalidDir");
5364
command.validate(args);
5465
}
66+
67+
@Test
68+
public void testImportSkipExistingBug()
69+
throws JsonSyntaxException, IOException, BackendConnectionException {
70+
Vulnerability vuln = new Vulnerability();
71+
vuln.setVulnId("CVE-TEST01");
72+
MockBackConnector mockBackendConnector = new MockBackConnector();
73+
HashMap<String, Object> args = new HashMap<String, Object>();
74+
args.put("o", false);
75+
args.put("v", false);
76+
Import command = new Import(mockBackendConnector);
77+
command.run(args);
78+
assertNull(mockBackendConnector.getUploadJson());
79+
}
80+
5581
}

kb-importer/src/test/java/org/eclipse/steady/kb/task/TestImportVulnerability.java

Lines changed: 0 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -58,20 +58,6 @@ public void testImportVuln() throws JsonSyntaxException, IOException, BackendCon
5858
assertNotNull(bug.getDescription());
5959
}
6060

61-
@Test
62-
public void testImportVulnSkipExistingBug()
63-
throws JsonSyntaxException, IOException, BackendConnectionException {
64-
Vulnerability vuln = new Vulnerability();
65-
vuln.setVulnId("CVE-TEST01");
66-
MockBackConnector mockBackendConnector = new MockBackConnector();
67-
HashMap<String, Object> args = new HashMap<String, Object>();
68-
args.put("o", false);
69-
args.put("v", false);
70-
ImportVulnerability importVuln = new ImportVulnerability();
71-
importVuln.execute(vuln, args, mockBackendConnector);
72-
assertNull(mockBackendConnector.getUploadJson());
73-
}
74-
7561
@AfterClass
7662
public static void cleanup() {
7763
try {

0 commit comments

Comments
 (0)