use of org.eclipse.sw360.datahandler.thrift.vulnerabilities.Vulnerability in project sw360portal by sw360.
the class VulnerabilityConnectorTest method testaddOrUpdateVulnerabilitiesAndSetIds.
@Test
public void testaddOrUpdateVulnerabilitiesAndSetIds() {
Vulnerability v1 = new Vulnerability().setLastExternalUpdate("1999-12-31").setId("idv1").setExternalId("eid1");
Vulnerability v2 = new Vulnerability().setLastExternalUpdate("2000-01-01").setId("idv2").setExternalId("eid2");
Vulnerability v3 = new Vulnerability().setLastExternalUpdate("2000-01-01").setId("idv3").setExternalId("eid1");
Vulnerability v4 = new Vulnerability().setLastExternalUpdate("2000-01-01").setId("idv4").setExternalId("eid2");
Vulnerability v5 = new Vulnerability().setLastExternalUpdate("1999-12-31").setId("idv5").setExternalId("eid5");
List<Vulnerability> vulnerabilityList = Arrays.asList(v3, v4, v5);
when(vulnerabilityDatabaseHandler.getByExternalId(Vulnerability.class, "eid1")).thenReturn(v1);
when(vulnerabilityDatabaseHandler.getByExternalId(Vulnerability.class, "eid2")).thenReturn(v2);
when(vulnerabilityDatabaseHandler.getByExternalId(Vulnerability.class, "eid5")).thenReturn(null);
when(vulnerabilityDatabaseHandler.add(v5)).thenReturn(RequestStatus.SUCCESS);
when(vulnerabilityDatabaseHandler.update(v1)).thenReturn(RequestStatus.SUCCESS);
Map<UpdateType, List<Vulnerability>> resultMap = vulnerabilityConnector.addOrUpdateVulnerabilitiesAndSetIds(vulnerabilityList);
assertThat(resultMap.get(UpdateType.UPDATED).get(0).getId(), is("idv1"));
assertThat(resultMap.get(UpdateType.NEW).get(0).getId(), is("idv5"));
assertThat(resultMap.get(UpdateType.OLD).get(0).getId(), is("idv2"));
assertThat(resultMap.get(UpdateType.FAILED), is(Collections.EMPTY_LIST));
}
use of org.eclipse.sw360.datahandler.thrift.vulnerabilities.Vulnerability in project sw360portal by sw360.
the class VulnerabilityConnectorTest method testIsMoreRecentSecondEmpty.
@Test
public void testIsMoreRecentSecondEmpty() throws Exception {
Vulnerability v1 = new Vulnerability().setLastExternalUpdate("2016");
Vulnerability v2 = new Vulnerability().setLastExternalUpdate("");
assertThat(vulnerabilityConnector.isMoreRecent(v1, v2), is(true));
}
use of org.eclipse.sw360.datahandler.thrift.vulnerabilities.Vulnerability in project sw360portal by sw360.
the class VulnerabilityConnectorTest method testAddOrUpdateVulnerabilityAndSetIdUpdateVulnerability.
@Test
public void testAddOrUpdateVulnerabilityAndSetIdUpdateVulnerability() throws Exception {
Vulnerability v1 = new Vulnerability().setLastExternalUpdate("1999-12-31").setId("idv1").setExternalId("eid");
Vulnerability v2 = new Vulnerability().setLastExternalUpdate("2000-01-01").setId("idv2").setExternalId("eid");
when(vulnerabilityDatabaseHandler.getByExternalId(Vulnerability.class, "eid")).thenReturn(v1);
when(vulnerabilityDatabaseHandler.update(v1)).thenReturn(RequestStatus.SUCCESS);
Map<UpdateType, List<Vulnerability>> resultMap = vulnerabilityConnector.addOrUpdateVulnerabilityAndSetId(v2, statusToVulnerabilityMap);
assertThat(resultMap.get(UpdateType.UPDATED).get(0).getId(), is("idv1"));
assertThat(resultMap.get(UpdateType.UPDATED).get(0).getLastExternalUpdate(), is("2000-01-01"));
assertThat(resultMap.get(UpdateType.NEW), is(Collections.EMPTY_LIST));
assertThat(resultMap.get(UpdateType.OLD), is(Collections.EMPTY_LIST));
assertThat(resultMap.get(UpdateType.FAILED), is(Collections.EMPTY_LIST));
}
use of org.eclipse.sw360.datahandler.thrift.vulnerabilities.Vulnerability in project sw360portal by sw360.
the class VulnerabilityConnectorTest method testIsMoreRecentEmpty.
@Test
public void testIsMoreRecentEmpty() throws Exception {
Vulnerability v1 = new Vulnerability();
Vulnerability v2 = new Vulnerability();
assertThat(vulnerabilityConnector.isMoreRecent(v1, v2), is(true));
}
use of org.eclipse.sw360.datahandler.thrift.vulnerabilities.Vulnerability in project sw360portal by sw360.
the class VulnerabilityConnectorTest method testIsMoreRecentFirstEmpty.
@Test
public void testIsMoreRecentFirstEmpty() throws Exception {
Vulnerability v1 = new Vulnerability().setLastExternalUpdate("");
Vulnerability v2 = new Vulnerability().setLastExternalUpdate("2016");
assertThat(vulnerabilityConnector.isMoreRecent(v1, v2), is(true));
}
Aggregations