Skip to content

Commit 77e0f14

Browse files
author
Grace Calianese
committed
checkstyle and UnknownField implementation change
1 parent 125d6c6 commit 77e0f14

3 files changed

Lines changed: 15 additions & 79 deletions

File tree

api/src/test/java/com/basistech/rosette/api/RosetteAPITest.java

Lines changed: 9 additions & 37 deletions
Original file line numberDiff line numberDiff line change
@@ -241,43 +241,15 @@ private static Stream<Arguments> testMatchRecordMissingFieldParameters() throws
241241
@MethodSource("testMatchRecordMissingFieldParameters")
242242
void testMatchRecordMissingField(String testFilename, String responseStr, int statusCode) throws IOException {
243243
setStatusCodeResponse(responseStr, statusCode);
244-
readValueRecordMatcher(testFilename);
245-
assertEquals("{\n" +
246-
" \"results\": [\n" +
247-
" {\n" +
248-
" \"score\": 0.0,\n" +
249-
" \"left\": {\n" +
250-
" \"dob2\": \"1993/04/16\",\n" +
251-
" \"dob\": \"1993-04-16\",\n" +
252-
" \"primaryName\": {\n" +
253-
" \"data\": \"Ethan R\",\n" +
254-
" \"language\": \"eng\",\n" +
255-
" \"entityType\": \"PERSON\"\n" +
256-
" },\n" +
257-
" \"addr\": \"123 Roadlane Ave\"\n" +
258-
" },\n" +
259-
" \"right\": {\n" +
260-
" \"dob\": \"1993-04-16\",\n" +
261-
" \"primaryName\": \"Seth R\"\n" +
262-
" },\n" +
263-
" \"error\": \"Field 'primaryName' not found in field mapping\"\n" +
264-
" },\n" +
265-
" {\n" +
266-
" \"score\": 0.0,\n" +
267-
" \"left\": {\n" +
268-
" \"dob\": \"1993-04-16\",\n" +
269-
" \"primaryName\": \"Evan R\"\n" +
270-
" },\n" +
271-
" \"right\": {\n" +
272-
" \"dob2\": \"1993/04/16\",\n" +
273-
" \"dob\": \"1993-04-16\",\n" +
274-
" \"primaryName\": \"Ivan R\",\n" +
275-
" \"addr\": \"123 Roadlane Ave\"\n" +
276-
" },\n" +
277-
" \"error\": \"Field 'primaryName' not found in field mapping\"\n" +
278-
" }\n" +
279-
" ]\n" +
280-
"}", responseStr);
244+
readValueRecordMatcher(testFilename);
245+
assertEquals("{\"results\":[{\"score\":0.0,\"left\":{\"dob2\":\"1993/04/16\","
246+
+ "\"dob\":\"1993-04-16\",\"primaryName\":{\"data\":\"Ethan R\",\"language\":\"eng\","
247+
+ "\"entityType\":\"PERSON\"},\"addr\":\"123 Roadlane Ave\"},\"right\":{\"dob\":\"1993-04-16\","
248+
+ "\"primaryName\":\"Seth R\"},\"error\":\"Field 'primaryName' not found in field mapping\"},"
249+
+ "{\"score\":0.0,\"left\":{\"dob\":\"1993-04-16\",\"primaryName\":\"Evan R\"},"
250+
+ "\"right\":{\"dob2\":\"1993/04/16\",\"dob\":\"1993-04-16\",\"primaryName\":\"Ivan R\","
251+
+ "\"addr\":\"123 Roadlane Ave\"},\"error\":\"Field 'primaryName' not found in field mapping\"}]}",
252+
responseStr);
281253
}
282254

283255
private static Stream<Arguments> testMatchRecordNullFieldParameters() throws IOException {
Lines changed: 1 addition & 36 deletions
Original file line numberDiff line numberDiff line change
@@ -1,36 +1 @@
1-
{
2-
"results": [
3-
{
4-
"score": 0.0,
5-
"left": {
6-
"dob2": "1993/04/16",
7-
"dob": "1993-04-16",
8-
"primaryName": {
9-
"data": "Ethan R",
10-
"language": "eng",
11-
"entityType": "PERSON"
12-
},
13-
"addr": "123 Roadlane Ave"
14-
},
15-
"right": {
16-
"dob": "1993-04-16",
17-
"primaryName": "Seth R"
18-
},
19-
"error": "Field 'primaryName' not found in field mapping"
20-
},
21-
{
22-
"score": 0.0,
23-
"left": {
24-
"dob": "1993-04-16",
25-
"primaryName": "Evan R"
26-
},
27-
"right": {
28-
"dob2": "1993/04/16",
29-
"dob": "1993-04-16",
30-
"primaryName": "Ivan R",
31-
"addr": "123 Roadlane Ave"
32-
},
33-
"error": "Field 'primaryName' not found in field mapping"
34-
}
35-
]
36-
}
1+
{"results":[{"score":0.0,"left":{"dob2":"1993/04/16","dob":"1993-04-16","primaryName":{"data":"Ethan R","language":"eng","entityType":"PERSON"},"addr":"123 Roadlane Ave"},"right":{"dob":"1993-04-16","primaryName":"Seth R"},"error":"Field 'primaryName' not found in field mapping"},{"score":0.0,"left":{"dob":"1993-04-16","primaryName":"Evan R"},"right":{"dob2":"1993/04/16","dob":"1993-04-16","primaryName":"Ivan R","addr":"123 Roadlane Ave"},"error":"Field 'primaryName' not found in field mapping"}]}

model/src/main/java/com/basistech/rosette/apimodel/recordsimilarity/records/UnknownField.java

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -16,8 +16,10 @@
1616

1717
package com.basistech.rosette.apimodel.recordsimilarity.records;
1818

19+
import java.util.ArrayList;
1920
import java.util.Iterator;
2021
import java.util.LinkedHashMap;
22+
import java.util.List;
2123
import java.util.Map;
2224

2325
import com.fasterxml.jackson.annotation.JsonCreator;
@@ -36,19 +38,16 @@ public UnknownField(JsonNode data) {
3638
this.data = data;
3739
}
3840

39-
//There's probably a better way to do this
4041
@JsonValue
4142
public Object getData() {
4243
if (data == null) {
4344
return "";
4445
} else {
4546
if (data.isObject()) {
46-
Iterator<Map.Entry<String, JsonNode>> fields = data.fields();
47+
List<Map.Entry<String, JsonNode>> list = new ArrayList<>();
48+
data.fields().forEachRemaining(list::add);
4749
Map<String, JsonNode> map = new LinkedHashMap<>();
48-
while (fields.hasNext()) {
49-
Map.Entry<String, JsonNode> fieldEntry = fields.next();
50-
map.put(fieldEntry.getKey(), fieldEntry.getValue());
51-
}
50+
list.stream().forEach(fieldEntry -> map.put(fieldEntry.getKey(), fieldEntry.getValue()));
5251
try {
5352
ObjectMapper mapper = new ObjectMapper();
5453
String jsonString = mapper.writeValueAsString(map);

0 commit comments

Comments
 (0)