package dataprovider;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.Map;
import java.util.TreeMap;

import org.testng.annotations.DataProvider;

import com.codoid.products.exception.FilloException;
import com.codoid.products.fillo.Connection;
import com.codoid.products.fillo.Fillo;
import com.codoid.products.fillo.Recordset;

public class providingdata {

@DataProvider(name="studentDataprovider")
public Object[][] studentDataprovider(){
Object[][]data = new Object[][]{{1,"sai","testing"},
{2,"nookesh","manuval"},{3,"kishore","java"}};
return data;
}

@DataProvider(name="studentMapDataprovider")

public Object[][] studentMapDataprovider() {

Object [][] Mapdata = new Object[2][1];

HashMap<String,Object> data =new HashMap<String,Object>();
data.put("stid", 1);
data.put("name", "sai");
data.put("course", "testing");
Mapdata[0][0]=data;
HashMap<String,Object> data1 =new HashMap<String,Object>();
data.put("stid", 2);
data.put("name", "SURESH");
data.put("course", "JAVA");

Mapdata[1][0]=data1;
return Mapdata;
}
public static Map<String,String> getExceltestdata(String Module,String key){
String query = String.format("select * from "+Module+"where
testcaseId='"+key+"'");
String testDataLoc ="C:\\Users\\nooke\\eclipse-workspace\\seleniump\\data\\
excelsheet1.xlsx";
return excelDataprovider(query,testDataLoc).get("gettestdata");

public static Map<String,TreeMap< String, String>> excelDataprovider(String

Query, String Location) {

System.out.println(Query);
System.out.println(Location);
Map<String,TreeMap< String,String>> testdata=new TreeMap<
String,TreeMap<String,String>>();
TreeMap< String,String> data= null;
Fillo Filo = new Fillo();
 try {
Connection conn = Filo.getConnection(Location);

Recordset recordset =conn.executeQuery(Query);

ArrayList<String> fieldnames = recordset.getFieldNames();

while(recordset.next()) {
data = new
TreeMap<String,String>(String.CASE_INSENSITIVE_ORDER);
for(String field:fieldnames) {
data.put(field, recordset.getField(field));}
}
testdata.put("gettestdata", data);
}catch (FilloException e) {
e.printStackTrace();

return testdata;
}
}