how to read data from excel file in java using selenium webdriver code example
Example 1: read excel selenium
# How to read & write to Excel file
import XL_utility # Importing from the second file
from selenium import webdriver
import unittest
import time
class Test_Excel(unittest.TestCase):
def setUp(self):
self.driver = webdriver.Chrome("path for browser driver...")
self.driver.implicitly_wait(10)
self.driver.get("web adress...")
self.driver.maximize_window()
return self.driver
def test_read_excel_file(self):
path = "file location for your excel file..."
print(path.title())
rows = XL_utility.getRowCount(path, "Sheet1")
for r in range(2, rows + 1):
username = XL_utility.readData(path, "Sheet1", r, 1)
password = XL_utility.readData(path, "Sheet1", r, 2)
self.driver.find_element_by_name("username").send_keys(username)
self.driver.find_element_by_name("password").send_keys(password)
self.driver.find_element_by_class_name("auth-form__submit").click()
time.sleep(10)
if self.driver.title == "title when you're logged in":
print("Test is passed!")
XL_utility.writeData(path, "Sheet1", r, 3, "Test passed")
else:
print("Test failed")
XL_utility.writeData(path, "Sheet1", r, 3, "Test failed")
self.driver.back()
return self.driver
if __name__ == "__main__":
unittest.main()
# This following code should be placed in a new python file...
import openpyxl
def getRowCount(file, sheetName):
workbook = openpyxl.load_workbook(file)
sheet = workbook.get_sheet_by_name(sheetName)
return sheet.max_row
def getColumnCount(file, sheetName):
workbook = openpyxl.load_workbook(file)
sheet = workbook.get_sheet_by_name(sheetName)
return sheet.max_column
def readData(file, sheetName, rownum, columnno):
workbook = openpyxl.load_workbook(file)
sheet = workbook.get_sheet_by_name(sheetName)
return sheet.cell(row=rownum, column=columnno).value
def writeData(file, sheetName, rownum, columnno, data):
workbook = openpyxl.load_workbook(file)
sheet = workbook.get_sheet_by_name(sheetName)
sheet.cell(row=rownum, column=columnno).value = data
workbook.save(file)
Example 2: how to read data from excel
I used Apache POI libraries to read and write from
excel file, I add the Apache poi dependencies to my pom.xml file.
In order to connect I use following classes
-FileInputStream from Java. It is used to create connection
to the file. We pass the file path as constructor to it.
-WorkBook is a class that represents the excel file. We create
workbook is a class from Apache POI that represents the excel
file. We create Workbook object using the FileInputStream
object.
-Sheet represents a single sheet from the excel file. We create
sheet using Workbook object. We can create worksheet using
the 0 based index.
public String readExcel(String path, String sheetName, int rowNum, int colNum){
try {
FileInputStream file = new FileInputStream(path);
Workbook book = WorkbookFactory.create(file);
Sheet sheet = book.getSheet(sheetName);
Row row = sheet.getRow(rowNum);
Cell cell = row.getCell(colNum);
String cellData = cell.toString();
return cellData;
} catch (Exception e) {
throw new RuntimeException(e);
}
}
How to get row and column numbers:
int rowCount = workSheet.getLastRowNum()+1; ==> why we add '+1'? Because
row num starts from 0.
int colCount = workSheet.getRow(0).getLastCellNum();
String sheetName = workSheet.getSheetName();
The data we get from excel can be converted to different formats such as set,
list, map
-------------------EXCEL WRITING AND SAVING---------------------------------
The first creation part is same then go to a cell where you want to write.
Row row = sheet.getRow(0)
Cell resultCell = row.getCell(2)
Let's imagine you have values in index 0 and index 1.
Now you want to create a cell on index 2. First, check if it is null to
avoid problems.
if(resultCell==null){
resultCell = row.createCell(2);
}
resultCell.setValue("Germany");
in order to save:
FileOutputStream fileOutputStream = new FileOutputStream("src/test/resources/
Countries.xlsx");
workbook.write(fileOutputStream);