java中如何测试从excel文件中读取数据的方法？

Question

上图为包含 Facebook 数据的 excel 文件。

FacebookDataExtraction 类从 Excel 文件中读取数据并将数据存储为行对象列表，如代码所示。 我已经使用 config.properties 文件来获取文件路径。 config.properties 文件内容为：FILE_NAME=D:/Refreshed_data_daily/all_hue_posts_in_excel.xlsx.

public class FacebookDataExtraction {

//private static final String FILE_NAME="D:/Refreshed_data_daily/all_hue_posts_in_excel.xlsx";
private static final String SHEET_NAME="nextv54plus_actions";
XSSFWorkbook workbook;

    public static void main(String[] args){

        FacebookDataExtraction obj= new FacebookDataExtraction();
        List<FacebookFields> displayList= new ArrayList<FacebookFields>();
        displayList=obj.readFromExcel();
        System.out.println("The Size of the list is:"+ displayList.size());
    }

    public List<FacebookFields> readFromExcel() {
        List<FacebookFields> fbList= new ArrayList<FacebookFields>();
        try
        {
            ReadPropertyFile data= new ReadPropertyFile("config.properties");
            FileInputStream fin= new FileInputStream(data.getPropertyFor("FILE_NAME"));
            workbook= new XSSFWorkbook(fin);
            int sheetIndex=0;
            for (Sheet sheet : workbook) {
                readSheet(sheet,sheetIndex ++, fbList);}

        }catch(FileNotFoundException e){
            e.printStackTrace();
        }
        catch(IOException e){
            e.printStackTrace();
        }
        return fbList;
    }

    private void readSheet(Sheet sheet, int sheetIndex , List<FacebookFields> fbList) {

        if(SHEET_NAME.equals(sheet.getSheetName())){
            workbook.removeSheetAt(sheetIndex);
            return;
        }
        for (Row row : sheet){
            if (row.getRowNum() > 0)
                fbList.add(readRow(row));}

    }

    private FacebookFields readRow(Row row) {

        FacebookFields record= new FacebookFields();
        for (Cell cell : row) {
            switch (cell.getColumnIndex()) {
            case 0: record.setName(cell.getStringCellValue()); 
            break; 
            case 1: record.setId(cell.getStringCellValue()); 
            break; 
            case 2: record.setDate(cell.getStringCellValue());
            break; 
            case 3: record.setMessage(cell.getStringCellValue());
            break; 
            case 4: record.setType(cell.getStringCellValue());
            break; 
            case 5: record.setPage(cell.getStringCellValue());
            break; 
            case 6: record.setLikeCount(String.valueOf(cell.getNumericCellValue()));
            break; 
            case 7: record.setCommentCount(String.valueOf(cell.getNumericCellValue())); 
            break; 
            case 8: record.setShareCount(String.valueOf(cell.getNumericCellValue())); 
            break; 
            }
        }

        return record;
    }

    public boolean containsData() {  

        List<FacebookFields> checkList= readFromExcel();    
        return !checkList.isEmpty() ;
    }

    }

FacebookFields 类（此处未显示）包含提取数据的设置方法！ 如何编写readRow()方法的测试用例或如何测试列的每个字段是否包含数据？

Answer 1

• 因此，您想测试将返回 FacebookFields 的 readRow() 方法。
• 由于它是私有的，你不能直接对这个私有方法进行单元测试，但你的主要目标似乎是测试这个方法。让我们在这里考虑一下重构/重新设计。
• 你可以创建一个类FacebookFieldRetriever，它有一个公共方法retrieveFromRow()，它返回FacebookFields。
• 因此，每次FacebookDataExtraction 遇到 Excel 工作表中的一行时，它都会创建一个 FacebookFieldRetriever 对象并调用 retrieveFromRow() 以获取从该行建模的 FacebookFields 对象。

• 我们已经通过了类设计阶段，让我们继续进行单元测试。所以，现在你有一个公共方法：retrieveFromRow()，你可以实际进行单元测试，假设你有一个行对象，其中包含包含来自 excel 工作表的实际值的单元格。 （基本上，我们专注于测试 switch-case 逻辑）。

• FacebookFieldRetriever 类

  class FacebookFieldRetriever{
public FacebookFields retrieveFromRow(Row row) {
FacebookFields record= new FacebookFields(); for (Cell cell : row) { switch (cell.getColumnIndex()) { case 0: record.setName(cell.getStringCellValue()); break; case 1: record.setId(cell.getStringCellValue()); break; case 2: record.setDate(cell.getStringCellValue()); break; case 3: record.setMessage(cell.getStringCellValue()); break; case 4: record.setType(cell.getStringCellValue()); break; case 5: record.setPage(cell.getStringCellValue()); break; case 6: record.setLikeCount(String.valueOf(cell.getNumericCellValue())); break; case 7: record.setCommentCount(String.valueOf(cell.getNumericCellValue())); break; case 8: record.setShareCount(String.valueOf(cell.getNumericCellValue())); break; } }
return record; }
}

• 下面的单元测试示例：

类 FacebookFieldRetrieverTest {

private final String TEST_NAME = "Mark Zuckerberg";
private final String TEST_id = "4";
private final String TEST_DATE = "2015-10-13;                                
private final String TEST_MSG = "Welcome to Facebook";                                 
private final String TEST_TYPE = "SomeType";                                
private final String TEST_PAGE = "SomePage";
private final String TEST_LIKECOUNT = 6;
private final String TEST_COMMENTCOUNT = 7;
private final String TEST_SHARECOUNT = 8 ;


public void testFacebookFieldRetriever()
{
XSSFWorkbook wb = new XSSFWorkbook();
Sheet sheettemp = wb.createSheet();
Row row = sheettemp.createRow(1);

Cell cell = row.createCell(0);
cell.setCellValue(TEST_NAME);

Cell cell1 = row.createCell(1);
cell.setCellValue(TEST_id);

Cell cell2 = row.createCell(2);
cell.setCellValue(TEST_DATE);

Cell cell3 = row.createCell(3);
cell.setCellValue(TEST_MSG);

Cell cell4 = row.createCell(4);
cell.setCellValue(TEST_TYPE);

Cell cell5 = row.createCell(5);
cell.setCellValue(TEST_PAGE);

Cell cell6 = row.createCell(6);
cell.setCellValue(TEST_LIKECOUNT);

Cell cell7 = row.createCell(7);
cell.setCellValue(TEST_COMMENTCOUNT);

Cell cell8 = row.createCell(8);
cell.setCellValue(TEST_SHARECOUNT);

FacebookFieldRetriever fbRetriever = new FacebookFieldRetriever();
FacebookFields fbFields = fbRetriever.retrieveFromRow(row);

assertEquals(fbFields.getName(), TEST_NAME);
assertEquals(fbFields.getId(), TEST_id);
assertEquals(fbFields.getMessage(), TEST_MSG);
assertEquals(fbFields.getLikeCount(), TEST_LIKECOUNT);
assertEquals(fbFields.getShareCount(), TEST_SHARECOUNT);
assertEquals(fbFields.getCommentCount(), TEST_COMMENTCOUNT);
assertEquals(fbFields.getPage(), TEST_PAGE);
assertEquals(fbFields.getType(), TEST_TYPE);
}
}

`

• 您可以按照上面给出的方式进行测试。希望对您有所帮助。

Answer 2

如果您的 Excel 示例中的第一行包含“预期值”，则可以开始以下测试。请注意，列表的 equals 可能有点粗略，尤其是因为我不确定它如何处理元素的内部顺序。

编辑我在FacebookDataExtraction 类中添加了一个构造函数参数，该参数将文件名添加到它应该从中提取数据的excel 文件中。

@Test
public void testWhetherListConatinsData(){ 
    List<FacebookFields> expectedList = new ArrayList<>();
    // Fill the expected list here:
    expectedList.add("expected value");
    // ... Add more if you want to...

    // Read in data.
    FacebookDataExtraction fbDataList= new FacebookDataExtraction("path/to/testdata/test_with_strange_first_name.xlsx");

    List<FacebookFields> listOfFields = fbDataList.readFromExcel();
    // Good check to start with.
    assertEquals(listOfFields.containsData(), true); 

    // Actual data check.
    assertEquals("Lists not the same", expectedList, listOfFields);
}