![[Selenium Java] Bài 28: Đọc data test từ File Excel trong Selenium Java](/uploads/lesson/thumb_post_lesson_selenium_java_new.jpg)
NỘI DUNG BÀI HỌC
✅ Ghi data mới vào lại File Excel
Cài đặt thư viện Apache Poi
Sử dụng Maven để cài thư viện Apache POI vào pom.xml của dự án Maven của bạn. Hiện tại Anh Tester đang sử dụng Apache Poi phiên bản 5.x.x, tất cả các phiên bản mới nhất có thể được tìm thấy tại đây
<!-- https://mvnrepository.com/artifact/org.apache.poi/poi -->
<dependency>
<groupId>org.apache.poi</groupId>
<artifactId>poi</artifactId>
<version>5.2.5</version>
</dependency>
<!-- https://mvnrepository.com/artifact/org.apache.poi/poi-ooxml -->
<dependency>
<groupId>org.apache.poi</groupId>
<artifactId>poi-ooxml</artifactId>
<version>5.2.5</version>
</dependency>(Cập nhật ngày: 25/11/2023)
Bản Apache POI Common mới này cần có thư viện Commons IO bổ trợ mới chạy êm được:
<!-- https://mvnrepository.com/artifact/commons-io/commons-io -->
<dependency>
<groupId>commons-io</groupId>
<artifactId>commons-io</artifactId>
<version>2.16.1</version>
</dependency>(Cập nhật ngày: 08/04/2024)
Tiếp theo tạo class để lưu các hàm xử lý Excel. Cụ thể An đặt là ExcelHelper. Bỏ trong package helpers chung với PropertiesHelper ở bài trước.
Hàm set Excel file theo Sheet cần dùng
Tại đây đặt tên hàm là setExcelFile với 2 đối số là đường dẫn đến Excel Path và Sheet Name
import java.awt.Color;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.util.HashMap;
import java.util.Map;
import org.apache.poi.ss.usermodel.*;
public class ExcelHelper {
private FileInputStream fis;
private FileOutputStream fileOut;
private Workbook wb;
private Sheet sh;
private Cell cell;
private Row row;
private CellStyle cellstyle;
private Color mycolor;
private String excelFilePath;
private Map<String, Integer> columns = new HashMap<>();
public void setExcelFile(String ExcelPath, String SheetName){
try {
File f = new File(ExcelPath);
if (!f.exists()) {
System.out.println("File doesn't exist.");
}
fis = new FileInputStream(ExcelPath);
wb = WorkbookFactory.create(fis);
sh = wb.getSheet(SheetName);
if (sh == null) {
throw new Exception("Sheet name doesn't exist.");
}
this.excelFilePath = ExcelPath;
//adding all the column header names to the map 'columns'
sh.getRow(0).forEach(cell ->{
columns.put(cell.getStringCellValue(), cell.getColumnIndex());
});
} catch (Exception e) {
System.out.println(e.getMessage());
}
}
}
Hàm get Cell data từ Excel File
Hàm getCellData(String columnName, int rowIndex) là hàm chúng ta sẽ gọi ra để dùng. Còn hàm getCellData(int columnIndex, int rowIndex) để xử lý thôi. Kiểu viết lại cho nó gọn ý mà.
public String getCellData(int columnIndex, int rowIndex) {
try {
cell = sh.getRow(rowIndex).getCell(columnIndex);
String CellData = null;
switch (cell.getCellType()) {
case STRING:
CellData = cell.getStringCellValue();
break;
case NUMERIC:
if (DateUtil.isCellDateFormatted(cell)) {
CellData = String.valueOf(cell.getDateCellValue());
} else {
CellData = String.valueOf((long) cell.getNumericCellValue());
}
break;
case BOOLEAN:
CellData = Boolean.toString(cell.getBooleanCellValue());
break;
case BLANK:
CellData = "";
break;
}
return CellData;
} catch (Exception e) {
return "";
}
}
//Gọi ra hàm này nè
public String getCellData(String columnName, int rowIndex) {
return getCellData(columns.get(columnName), rowIndex);
}
🔆 Gọi ra dùng lại thì gọi hàm setExcelFile trước để khai báo File Excel cần dùng xong đến gọi getCellData để lấy data theo từng ô
Chuẩn bị file Excel như sau:![[Selenium Java] Bài 24: Đọc File Excel để lấy data test | Anh Tester](/uploads/lesson/selenium_java/excel_file_read_write.png)
Ví dụ đọc từ file testData.xlsx tại Sheet1 với các giá trị bên dưới (nhớ chú ý link kẻo sai)
- cột username dòng 1 (dòng và cột bắt đầu từ 0) nghĩa là thứ tự 2 của excel á
- cột password dòng 1
- cột pin dòng 1
public static void main(String []args) throws Exception {
ExcelHelper excel = new ExcelHelper();
excel.setExcelFile("./testData.xlsx", "Sheet1");
System.out.println(excel.getCellData("userName", 1));
System.out.println(excel.getCellData("password", 1));
System.out.println(excel.getCellData("pin", 1));
}
Hàm ghi data vào lại File Excel theo từng Cell
Bạn chỉ cần truyền tham số tuần tự cho hàm setCellData(String text, int columnIndex, int rowIndex)
(giá trị cần ghi - vị trí dòng - vị trí cột)
Hoặc hàm setCellData(String text, String columnName, int rowIndex) lấy theo tên cột
//set by column index
public void setCellData(String text, int columnIndex, int rowIndex) {
try {
row = sh.getRow(rowIndex);
if (row == null) {
row = sh.createRow(rowIndex);
}
cell = row.getCell(columnIndex);
if (cell == null) {
cell = row.createCell(columnIndex);
}
cell.setCellValue(text);
XSSFCellStyle style = (XSSFCellStyle) wb.createCellStyle();
style.setFillPattern(FillPatternType.NO_FILL);
style.setAlignment(HorizontalAlignment.CENTER);
style.setVerticalAlignment(VerticalAlignment.CENTER);
cell.setCellStyle(style);
fileOut = new FileOutputStream(excelFilePath);
wb.write(fileOut);
fileOut.flush();
fileOut.close();
} catch (Exception e) {
e.getMessage();
}
}
//set by column name
public void setCellData(String text, String columnName, int rowIndex) {
try {
row = sh.getRow(rowIndex);
if (row == null) {
row = sh.createRow(rowIndex);
}
cell = row.getCell(columns.get(columnName));
if (cell == null) {
cell = row.createCell(columns.get(columnName));
}
cell.setCellValue(text);
XSSFCellStyle style = (XSSFCellStyle) wb.createCellStyle();
style.setFillPattern(FillPatternType.NO_FILL);
style.setAlignment(HorizontalAlignment.CENTER);
style.setVerticalAlignment(VerticalAlignment.CENTER);
cell.setCellStyle(style);
fileOut = new FileOutputStream(excelFilePath);
wb.write(fileOut);
fileOut.flush();
fileOut.close();
} catch (Exception e) {
e.getMessage();
}
}
🔆 Gọi hàm setCellData ra dùng lại:
public static void main(String []args) throws Exception {
ExcelHelper excel = new ExcelHelper();
excel.setExcelFile("./testData.xlsx", "Sheet1");
//Ghi giá trị "pass" vào dòng 1 cột 3 (dòng và cột bắt đầu tính từ 0)
excel.setCellData("pass", 3, 1);
}
🔆 Kết quả như bên dưới:
![[Selenium Java] Bài 24: Đọc File Excel để lấy data test | Anh Tester](/uploads/lesson/selenium_java/kq_excel_write.png)
Yeah xong rồi đó. Mở rộng hơn thì chúng ra có thể lấy ra nhiều dòng thì sẽ dùng vòng lặp FOR và DataProvider của TestNG cung cấp.
DataProvider sẽ học tiếp theo ở bài sau nhé !!
Share
