fixed test cases

This commit is contained in:
Jeremy Long
2017-11-26 10:26:37 -05:00
parent f51edf52e7
commit 72c121797f
2 changed files with 27 additions and 11 deletions

View File

@@ -125,7 +125,11 @@ public class EscapeTool {
if (text == null || text.isEmpty()) {
return "\"\"";
}
return StringEscapeUtils.escapeCsv(text.trim().replace("\n", " "));
final String str = text.trim().replace("\n", " ");
if (str.length()==0) {
return "\"\"";
}
return StringEscapeUtils.escapeCsv(str);
}
/**
@@ -151,6 +155,9 @@ public class EscapeTool {
sb.append(id.getValue());
}
}
if (sb.length()==0) {
return "\"\"";
}
return StringEscapeUtils.escapeCsv(sb.toString());
}
@@ -177,6 +184,9 @@ public class EscapeTool {
sb.append(id.getValue());
}
}
if (sb.length()==0) {
return "\"\"";
}
return StringEscapeUtils.escapeCsv(sb.toString());
}
@@ -203,6 +213,9 @@ public class EscapeTool {
sb.append(id.getConfidence());
}
}
if (sb.length()==0) {
return "\"\"";
}
return StringEscapeUtils.escapeCsv(sb.toString());
}
@@ -229,6 +242,9 @@ public class EscapeTool {
sb.append(id.getValue());
}
}
if (sb.length()==0) {
return "\"\"";
}
return StringEscapeUtils.escapeCsv(sb.toString());
}
}

View File

@@ -126,7 +126,7 @@ public class EscapeToolTest {
public void testCsv() {
String text = null;
EscapeTool instance = new EscapeTool();
String expResult = null;
String expResult = "\"\"";
String result = instance.csv(text);
assertEquals(expResult, result);
@@ -164,22 +164,22 @@ public class EscapeToolTest {
assertEquals(expResult, result);
ids = new HashSet<>();
ids.add(new Identifier("gav", "somegroup:something:1.0", ""));
ids.add(new Identifier("maven", "somegroup:something:1.0", ""));
expResult = "somegroup:something:1.0";
result = instance.csvIdentifiers(ids);
assertEquals(expResult, result);
ids = new HashSet<>();
ids.add(new Identifier("cpe", "cpe:/a:somegroup:something:1.0", ""));
ids.add(new Identifier("gav", "somegroup:something:1.0", ""));
ids.add(new Identifier("maven", "somegroup:something:1.0", ""));
expResult = "somegroup:something:1.0";
result = instance.csvIdentifiers(ids);
assertEquals(expResult, result);
ids = new HashSet<>();
ids.add(new Identifier("cpe", "cpe:/a:somegroup:something:1.0", ""));
ids.add(new Identifier("gav", "somegroup:something:1.0", ""));
ids.add(new Identifier("gav", "somegroup2:something:1.2", ""));
ids.add(new Identifier("maven", "somegroup:something:1.0", ""));
ids.add(new Identifier("maven", "somegroup2:something:1.2", ""));
expResult = "\"somegroup:something:1.0, somegroup2:something:1.2\"";
String expResult2 = "\"somegroup2:something:1.2, somegroup:something:1.0\"";
result = instance.csvIdentifiers(ids);
@@ -203,7 +203,7 @@ public class EscapeToolTest {
assertEquals(expResult, result);
ids = new HashSet<>();
ids.add(new Identifier("gav", "somegroup:something:1.0", ""));
ids.add(new Identifier("maven", "somegroup:something:1.0", ""));
expResult = "\"\"";
result = instance.csvCpe(ids);
assertEquals(expResult, result);
@@ -216,14 +216,14 @@ public class EscapeToolTest {
ids = new HashSet<>();
ids.add(new Identifier("cpe", "cpe:/a:somegroup:something:1.0", ""));
ids.add(new Identifier("gav", "somegroup:something:1.0", ""));
ids.add(new Identifier("maven", "somegroup:something:1.0", ""));
expResult = "cpe:/a:somegroup:something:1.0";
result = instance.csvCpe(ids);
assertEquals(expResult, result);
ids = new HashSet<>();
ids.add(new Identifier("cpe", "cpe:/a:somegroup:something:1.0", ""));
ids.add(new Identifier("gav", "somegroup:something:1.0", ""));
ids.add(new Identifier("maven", "somegroup:something:1.0", ""));
ids.add(new Identifier("cpe", "cpe:/a:somegroup2:something:1.2", ""));
expResult = "\"cpe:/a:somegroup:something:1.0, cpe:/a:somegroup2:something:1.2\"";
String expResult2 = "\"cpe:/a:somegroup2:something:1.2, cpe:/a:somegroup:something:1.0\"";
@@ -248,7 +248,7 @@ public class EscapeToolTest {
assertEquals(expResult, result);
ids = new HashSet<>();
ids.add(new Identifier("gav", "somegroup:something:1.0", ""));
ids.add(new Identifier("maven", "somegroup:something:1.0", ""));
expResult = "\"\"";
result = instance.csvCpeConfidence(ids);
assertEquals(expResult, result);
@@ -268,7 +268,7 @@ public class EscapeToolTest {
Identifier i2 = new Identifier("cpe", "cpe:/a:somegroup:something2:1.0", "");
i2.setConfidence(Confidence.MEDIUM);
ids.add(i2);
Identifier i3 = new Identifier("gav", "somegroup:something:1.0", "");
Identifier i3 = new Identifier("maven", "somegroup:something:1.0", "");
i3.setConfidence(Confidence.LOW);
ids.add(i3);