Road to data driven testing in webdriver java TestNg Part 1

In this post I am going to see you, how to perform data driven testing in java webdriver. In this post I read data from external excel file and use those data in test script.
For this I use java “poi” jar file to read excel file. Using poi java library I created below java class to read data from excel file and return data as a array list.

package com.webdriver.test;

import java.io.File;
import java.io.FileInputStream;
import java.util.ArrayList;
import java.util.List;
import org.apache.poi.xssf.usermodel.XSSFCell;
import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;

public class ReadXLS {

     public List getData() {

           String path = "DataSheet.xlsx";
           List dataList = new ArrayList();
           FileInputStream fis = null;
           try {
               fis = new FileInputStream(new File(path));
               XSSFWorkbook workbook = new XSSFWorkbook(fis);
               XSSFSheet sheet = workbook.getSheet("TestData");
               java.util.Iterator rows = sheet.rowIterator();

               while (rows.hasNext()) {
                   XSSFRow row = ((XSSFRow) rows.next());
                   // int r=row.getRowNum();
                   java.util.Iterator cells = row .cellIterator();
                   int i = 0;
                   String[] testData= new String[3];
                   while (cells.hasNext()) {

                       XSSFCell cell = (XSSFCell) cells.next();
                       String value = cell.getStringCellValue();
                       if (!value.equals(null)) {
                            testData [i] = value;
                            i++;
                       }
                   }
                   dataList.add(testData);
               }
           }
           catch (Exception e) {
               e.printStackTrace();
           }
           return dataList;
     }

}

Note: you need to add all poi jar file to class path of your project.

Here I take a example where I search and verify country by using wikipedia. For country I have created a data sheet ”DataSheet.xlsx” as below. 

Country	Description
India	This article is about the Republic of India
Brazil	Brazil bounded by the Atlantic Ocean on the east
Canada	Canada is a North American country
England	England is a country that is part of the United Kingdom

Below is java webdriver test case for wikipedia country search. In this test I call “ReadXLS” function to read data from xls file.

package com.webdriver.test;

import java.util.List;
import java.util.concurrent.TimeUnit;

import org.openqa.selenium.By;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.firefox.FirefoxDriver;
import org.testng.Assert;
import org.testng.annotations.AfterSuite;
import org.testng.annotations.BeforeSuite;
import org.testng.annotations.Test;

public class SearchCountry {

    private WebDriver driver;
    private String baseUrl;

    @BeforeSuite
    public void setUp() throws Exception {
         driver = new FirefoxDriver();
         baseUrl = "http://www.wikipedia.org/";
         driver.manage().timeouts().implicitlyWait(30, TimeUnit.SECONDS);
    }

    @Test
    public void testSearchCountry() throws Exception {
         driver.get(baseUrl + "/wiki/Main_Page");
         ReadXLS readXls = new ReadXLS();
         List dataList = readXls.getData();
         for (int i = 1; i < dataList.size(); i++) {
             String[] testCase = new String[5];
             String[] test = (String[]) dataList.get(i);
             String countryName = test[0];
             String countryDesc = test[1];
             driver.findElement(By.id("searchInput")).clear();
             driver.findElement(By.id("searchInput")).sendKeys(countryName);
             driver.findElement(By.id("searchButton")).click();
             String str = driver.findElement(
             By.xpath("//h1[@id='firstHeading']/span")).getText();
             System.out.println(countryDesc);
             Assert.assertTrue(str.contains(countryName));
         }
    }
                   

    @AfterSuite
    public void tearDown() throws Exception {
        driver.quit();
    }
}

When we execute above test case all country search and description should be print on console.