Read and Write csv file in Selenium
This Tutorial represent a complete idea about how to read and write test data/result in csv file in selenium automation.
What is CSV file ?
A comma-separated values (CSV) file is a delimited text file that uses a comma to separate values.
Each line of the file is a data record. Each record consists of one or more fields, separated by commas.
We can handle the CSV file using the below packages :
- Apache Commons CSV
- Open CSV
Below are the steps used to integrate selenium project with Apache Commons CSV
If Maven Project is used for automation then we have to add below dependency in pom.xml file
<dependency>Â Â Â
<groupId>org.apache.commons</groupId>Â Â Â
<artifactId>commons-csv</artifactId>Â Â Â
<version>1.5</version>
</dependency>
And if Generic automation project is used then jar file can be downloaded from below location
https://commons.apache.org/proper/commons-csv/download_csv.cgi
We can read the CSV file using Apache Commons CSV library ; there could be two kinds of CSV files.
- CSV file with Header
- CSV file without Header
How to read csv file without header with Apache Commons CSV?
 Open Notepad and enter below values in format mention below and Save file as  .csv format
Ram,Kumar,Infosys
Shyam,Gahlot,TCS
Denial,W,Wipro
Steps:
- Create the CSV file.
- Create a new BufferReader object and pass the path of CSV file
- Parse the BufferReader object into comma-separated values so create an object for CSV parser and passing reader reference(whatever going to parse), and the delimiter for parsing
- Now iterate each row and column as data is loaded in csvParse object.
//Code Snippet
package com.test;
import org.testng.annotations.Test;
import org.apache.commons.csv.CSVFormat;
import org.apache.commons.csv.CSVParser;
import org.apache.commons.csv.CSVRecord;
import java.io.IOException;
import java.io.Reader;
import java.nio.file.Files;
import java.nio.file.Paths;
public class CSVReader {
@Test
public void readCSV() throws IOException {
String path = "C:\\File\\abc.csv";
// Create a new BufferReader object and pass the path of CSV file
Reader reader = Files.newBufferedReader(Paths.get(path));
// parse the file into csv values
CSVParser parse = new CSVParser(reader, CSVFormat.DEFAULT);
for (CSVRecord csvRecord : parse) {
// Accessing Values by Column Index
String firstname = csvRecord.get(0);
String lastName = csvRecord.get(1);
String company = csvRecord.get(2);
// print the value to console
System.out.println("Record No - " + csvRecord.getRecordNumber());
System.out.println("FirstName : " + firstname);
System.out.println("Last Name : " + lastName);
System.out.println("Company name : " + company);
}
}
}
How to read csv file with header with Apache Commons CSV?
Open Notepad and enter below values in format mention below and Save file as  .csv format
FirstName, LastName, CompanyName
Ram,Kumar,Infosys
Shyam,Gahlot,TCS
Denial,W,Wipro
Below is the complete program for reading .csv file
package com.test;
import org.testng.annotations.Test;
import org.apache.commons.csv.CSVFormat;
import org.apache.commons.csv.CSVParser;
import org.apache.commons.csv.CSVRecord;
import java.io.IOException;
import java.io.Reader;
import java.nio.file.Files;
import java.nio.file.Paths;
public class CSVReader {
@Test
public void readCSV() throws IOException {
String path = "C:\\File\\abc.csv";
// read the file
Reader reader = Files.newBufferedReader(Paths.get(path));
CSVParser csvParser = new CSVParser(reader, CSVFormat.DEFAULT
.withHeader("FirstName", "LastName", "CompanyName")
.withIgnoreHeaderCase()
.withTrim());
for (CSVRecord csvRecord : csvParser) {
// Accessing values by the names assigned to each column
String fname = csvRecord.get("FirstName");
String lname = csvRecord.get("LastName");
String company = csvRecord.get("CompanyName");
System.out.println("Record No - " + csvRecord.getRecordNumber());
System.out.println("FirstName : " + fname);
System.out.println("LastName : " + lname);
System.out.println("CompanyName : " + company);
}
}
}
Recommended Articles Â