// Find the right place to put the data.
for (TestMethod testMethod : inputTestData.getTestMethod()) {
List<TestRecord> originalTestRecords = testMethod.getTestRecord();
for (TestRecord originalTestRecord : originalTestRecords) {
if (originalTestRecord.getId().equals(testRecord.get(RECORD_POSITION))) {
OutputData outputData = new OutputData();
Entry outputEntry = new Entry();
outputEntry.setKey(ACTUAL_RESULT);
outputEntry.setValue(testRecord.get(ACTUAL_RESULT).toString());
Entry durationEntry = new Entry();
durationEntry.setKey(DURATION);
durationEntry.setValue(testRecord.get(DURATION).toString());
outputData.getEntry().add(outputEntry);
outputData.getEntry().add(durationEntry);
if(testRecord.get(TEST_STATUS) != null){
Entry statusEntry = new Entry();
statusEntry.setKey(TEST_STATUS);
statusEntry.setValue(testRecord.get(TEST_STATUS).toString());
outputData.getEntry().add(statusEntry);
}
originalTestRecord.setOutputData(outputData);
outputDataAdded = true;
break;
}
}
if (outputDataAdded) {
break;
}
}
}else{
//Method did not return any data. So only write the duration as output
for (TestMethod testMethod : inputTestData.getTestMethod()) {
List<TestRecord> originalTestRecords = testMethod.getTestRecord();
for (TestRecord originalTestRecord : originalTestRecords) {
if (originalTestRecord.getId().equals(testRecord.get(RECORD_POSITION))) {
OutputData outputData = new OutputData();
Entry durationEntry = new Entry();
durationEntry.setKey(DURATION);
durationEntry.setValue(testRecord.get(DURATION).toString());
outputData.getEntry().add(durationEntry);
originalTestRecord.setOutputData(outputData);
outputDataAdded = true;
break;
}
}