List<Object> employeeRow = new ArrayList<Object>();
List<Object> managerRow = new ArrayList<Object>();
List<Object> crRow = new ArrayList<Object>();
MockResultSet rs = new MockResultSet("test");
for (SelectColumn column : columns) {
rs.addColumn(column.getColumnName(md.getDbEntity(), null));
employeeRow.add(employeeRowMap.get(column.getDataRowKey()));
managerRow.add(managerRowMap.get(column.getDataRowKey()));
crRow.add(crRowMap.get(column.getDataRowKey()));
}
rs.addRow(employeeRow);
rs.addRow(managerRow);
rs.addRow(crRow);
RowReader<Object> reader = select.getRowReader(rs);
rs.next();
DataRow employeeRowRead = (DataRow) reader.readRow(rs);
assertEquals("Employee", employeeRowRead.getEntityName());
assertEquals(
"Invalid row read: " + employeeRowRead,
employeeRowMap,
employeeRowRead);
rs.next();
DataRow managerRowRead = (DataRow) reader.readRow(rs);
assertEquals("Manager", managerRowRead.getEntityName());
assertEquals("Invalid row read: " + managerRowRead, managerRowMap, managerRowRead);
rs.next();
DataRow crRowRead = (DataRow) reader.readRow(rs);
assertEquals("CustomerRepresentative", crRowRead.getEntityName());
assertEquals("Invalid row read: " + crRowRead, crRowMap, crRowRead);
}