TmsDataProvider.java

package com.xxx.mobile.util;

 

import org.testng.annotations.DataProvider;

import org.testng.annotations.Test;

 

import com.xxx.mobile.testcase.Constant;

import com.xxx.util.DbHandler;

import java.sql.ResultSet;

import java.sql.PreparedStatement;

import org.apache.log4j.Logger;

 

public class TmsDataProvider {

 

               static Logger logger = Logger.getLogger(TmsDataProvider.class);

               

               @DataProvider(name=Constant.certDataProvider)

               public static Object[][] credentialDataProvider(){

                               String sqlCmd="select name, filepath, password from "+Constant.certDataProvider;    

                               return defaultDataProvider(sqlCmd);

               }

               

               private static Object[][] defaultDataProvider(String sqlCmd){

 

                 // DbHandler db = new DbHandler(Constant.serverIP, Constant.portNumber,Constant.user, Constant.password,

       //       Constant.databaseName, false);

                  DbHandler db = new DbHandler("xx.xx.xx.xx","1433", "sa", "xxx", "x",false);

                  PreparedStatement pstmt;

                  ResultSet rs;

                                

                  Object[][] retval =null;                

                  try {

                                  db.getConnection();   

                                  //System.out.println(sqlCmd);

                                  pstmt=db.con.prepareStatement(sqlCmd, ResultSet.TYPE_SCROLL_SENSITIVE,ResultSet.CONCUR_UPDATABLE);

                                  //db.execQuery(sqlCmd);

                                  rs = pstmt.executeQuery();

                                  rs.last();                            

                                  int rsnum = rs.getRow();                             

                                  //System.out.println(rsnum);

                                  int columnCount = rs.getMetaData().getColumnCount();          

                                  //System.out.println(columnCount);

                                  retval = new Object[rsnum][columnCount];

                                  rs.beforeFirst();            

                                  int j = 0;                              

                                  while (rs.next()) {                                                                                                                        

                                                               for(int i = 1; i <= columnCount; i++) {

                                                                               retval[j][i-1] =rs.getString(i);                                     

                                                                   }

                                                               j++;

                                                               }                              

                                

                               } catch (Exception e) {   

                                                               logger.error(e);

                               } finally {

                                                               db.destroy();

                                               }

                 

               return retval;                   

                 

}

                

                 @Test(dataProvider=Constant.certDataProvider,dataProviderClass=com.xxx.mobile.util.TmsDataProvider.class)

                 public static void testCredentails(String name, String filepath, Stringpassword){                                             

                                 System.out.println(name);

                                 System.out.println(filepath);

                                 System.out.println(password);

                 }

                

                 public static void main(String[] args) throws Exception {

                                                                               

                 }

}