抟扶

  博客园 :: 首页 :: 博问 :: 闪存 :: 新随笔 :: 联系 :: 订阅 订阅 :: 管理 ::
  1 package testNG;
  2 
  3 import java.awt.Desktop;
  4 import java.io.File;
  5 import java.io.FileInputStream;
  6 import java.io.FileOutputStream;
  7 import java.io.FileWriter;
  8 import java.io.IOException;
  9 import java.io.FileNotFoundException; 
 10 import java.io.Writer;
 11 import java.text.SimpleDateFormat;
 12 import java.util.Date;
 13 import java.util.regex.Matcher;
 14 import java.util.regex.Pattern;
 15 
 16 import javax.xml.crypto.Data;
 17 
 18 import org.apache.commons.io.FileUtils;
 19 import org.apache.poi.ss.usermodel.Cell; 
 20 import org.apache.poi.xssf.usermodel.XSSFRow;
 21 import org.apache.poi.xssf.usermodel.XSSFSheet;
 22 import org.apache.poi.xssf.usermodel.XSSFWorkbook;
 23 import org.openqa.selenium.By;
 24 import org.openqa.selenium.OutputType;
 25 import org.openqa.selenium.TakesScreenshot;
 26 import org.openqa.selenium.WebDriver;
 27 import org.openqa.selenium.chrome.ChromeDriver;
 28 import org.openqa.selenium.firefox.FirefoxDriver;
 29 import org.openqa.selenium.ie.InternetExplorerDriver;
 30 import org.openqa.selenium.remote.DesiredCapabilities;
 31 
 32  
 33 @SuppressWarnings("unused")
 34 
 35 public class BaseClass {
 36     /***
 37      * 基础类,包括方法:启动webdriver、获取基础数据、获取测试数据、返回当前时间、写入结果文件、
 38      * 截图、打开结果文件、去除空格换行及制表符的字符串
 39      * 
 40      * @param args
 41      * @throws IOException
 42      * @throws InterruptedException
 43      */
 44     public static WebDriver startBrowser(String browserType){
 45         /**
 46           * FunName:        startBrowser
 47           * Description :   启动浏览器,支持ie/fireFox/chrome
 48           * @param:                 String browserType
 49           * @return WebDriver:  driver
 50           * @Author:        Bingo
 51           * @Create Date:   2015-07-21
 52           **/        
 53         WebDriver wd = null;
 54         if (browserType == "ie"){
 55              System.setProperty("webdriver.ie.driver", "drivers\\IEDriverServer.exe");//ie驱动路径
 56              DesiredCapabilities dc = DesiredCapabilities.internetExplorer();
 57              dc.setCapability(InternetExplorerDriver.INTRODUCE_FLAKINESS_BY_IGNORING_SECURITY_DOMAINS, true);
 58              //关闭IE浏览器保护模式
 59              dc.setCapability("ignoreProtectedModeSettings", true);
 60              wd = new InternetExplorerDriver();
 61         }else if(browserType == "fireFox"){
 62             System.setProperty("webdriver.firefox.bin", //firefox的安装路径
 63                      "D:/Program Files/Mozilla Firefox/firefox.exe");
 64             wd = new FirefoxDriver();
 65         }else if(browserType == "chrome"){
 66             System.setProperty("webdriver.chrome.driver", "drivers\\chromedriver.exe");//驱动路径
 67             wd = new ChromeDriver();
 68         }        
 69         return wd;
 70         
 71     }
 72     
 73         
 74     //读取测试URL、测试数据文件路径和结果文件路径
 75     public String getBaseData(String testInfo, String baseDataPath) throws IOException{
 76         /**
 77           * FunName:        getBaseData
 78           * Description :   获取测试URL、测试数据文件路径和结果文件路径
 79           * @param:                 String testInfo
 80           * @return String:  返回基础数据字符串;
 81           * @Author:        Bingo
 82           * @Create Date:   2015-06-29
 83           **/
 84         String testString =  null;
 85         FileInputStream file = new FileInputStream(baseDataPath);
 86         @SuppressWarnings("resource")
 87         XSSFWorkbook wb = new XSSFWorkbook(file);
 88         XSSFSheet sheet = wb.getSheetAt(0);
 89         XSSFRow rowinf = sheet.getRow(1);
 90         Cell cell = null;
 91         switch(testInfo){
 92             case "inPath"                   : cell = rowinf.getCell(0);break;
 93             case "outPath"                 : cell = rowinf.getCell(1);break;
 94             case "baseURL"                  : cell = rowinf.getCell(2);break;
 95             case "firefoxPath"            : cell = rowinf.getCell(3);break;
 96             case "screenShotDir"           : cell = rowinf.getCell(4);break;
 97             case "snapshotDir"            : cell = rowinf.getCell(5);break;
 98             case "xiaoWeiIssuerURL"       : cell = rowinf.getCell(6);break;
 99             case "Tesseract-OCRPath"       : cell = rowinf.getCell(7);break;
100             default: System.out.println("no testInfo found!");
101         }
102         cell.setCellType(Cell.CELL_TYPE_STRING);  
103         testString= cell.getStringCellValue(); 
104          
105         return testString;
106     }
107 
108     
109     //读取信息,被测试方法调用,参数为方法名、列名colNm、文件路径inPath
110     public String getData_xlsx(String funNm, String colNm, String inPath) throws FileNotFoundException, IOException{
111         /**
112           * FunName:        getData_xlsx
113           * Description :   读取信息,供被测试方法调用
114           * @param:                 String funNm, String colNm, String inPath
115           * @return String:  返回测试数据;
116           * @Author:        Bingo
117           * @Create Date:   2015-06-29
118           **/
119         String str1 = null;
120         FileInputStream file = new FileInputStream(inPath);
121         @SuppressWarnings("resource")
122         XSSFWorkbook wb = new XSSFWorkbook(file);
123         XSSFSheet sheet = null;
124         Cell cell = null;
125         //根据方法名,读取对应的sheet
126         if (funNm == "Login"){
127             sheet = wb.getSheetAt(1);
128         } else if (funNm == "QueryApp"){
129             sheet = wb.getSheetAt(2);
130         }
131         XSSFRow row1 = sheet.getRow(1);
132         //根据不同的列名,读取对应的列
133         if(colNm == "orgz"||colNm == "appNo")
134             cell = row1.getCell(0); 
135             
136         else if(colNm == "userId"||colNm == "holName")
137             cell = row1.getCell(1);
138         else if(colNm == "pwd"||colNm == "cerType")
139             cell = row1.getCell(2);
140         else if(colNm == "cerNo"||colNm == "xiaoWeiUserId")
141             cell = row1.getCell(3);
142         else if(colNm == "xiaoWeiUserPwd")
143             cell = row1.getCell(4);    
144         else System.out.println(nowDate()+" err funcName or colNume");
145         cell.setCellType(Cell.CELL_TYPE_STRING);  
146         str1= cell.getStringCellValue();
147         return str1;
148     }
149     //写入测试结果文件,参数为方法名、是否通过、文件路径和执行时间
150     public void outCome_xlsx(String funNm, String outCome, String outPath,String date) throws IOException
151     {
152         /**
153           * FunName:        outCome_xlsx
154           * Description :   写入测试结果
155           * @param:                 String funNm, String outCome, String outPath,String date
156           * @return void:        无返回数据;
157           * @Author:        Bingo
158           * @Create Date:   2015-06-29
159           **/
160         FileInputStream file = new FileInputStream(outPath);
161         @SuppressWarnings("resource")
162         XSSFWorkbook wb = new XSSFWorkbook(file);
163         XSSFSheet sheet = wb.getSheetAt(0);
164         int trLength = sheet.getLastRowNum();  
165         XSSFRow newrow = sheet.createRow((short)(trLength+1));
166         newrow.createCell(0).setCellValue(funNm);  
167         newrow.createCell(1).setCellValue(outCome);
168         newrow.createCell(2).setCellValue(date);
169         FileOutputStream fout = new FileOutputStream(outPath); 
170         wb.write(fout);
171         file.close();
172         fout.close();
173  
174     }    
175         
176 //        public File createLogFile(String fileName, String logDir){
177 //            File file = null;
178 //            
179 //            return file;
180 //        }
181 //        
182         
183         //写入日志文件信息,文件名为yyyy-mm-dd.log
184         public static File writeLog(String logMsg) throws IOException
185         {
186             /**
187               * FunName:        writeLog
188               * Description :   记录日志
189               * @param:                 File file, String msg
190               * @return file:        返回文件;
191               * @Author:        Bingo
192               * @Create Date:   2015-07-21
193               **/
194             //FileInputStream file = new FileInputStream(outPath);
195             String logDir = System.getProperty("user.dir")+"\\log\\";
196             String time = null;
197             SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd");//设置日期格式
198             time = sdf.format(new Date()).toString();
199             File file = new File(logDir+time+".log");
200             if(!file.exists()) {  
201                 file.createNewFile();
202             }  
203             @SuppressWarnings("resource")
204             Writer txtWriter = new FileWriter(file,true);  
205             txtWriter.write(nowDate() +"\t"+logMsg+"\n");  
206             txtWriter.flush();
207             return file;         
208         }
209             
210         public void openOutCome(String outPath){
211         /**
212           * FunName:        openOutCome
213           * Description :   打开测试结果文件
214           * @param:                 文件URL
215           * @return void:        无返回数据;
216           * @Author:        Bingo
217           * @Create Date:   2015-06-30
218           **/
219         Desktop desk=Desktop.getDesktop();  
220         try  
221         {  
222             File file1=new File(outPath);//创建一个java文件系统  
223             if (!file1.canWrite())//判断文件是否被占用
224             {
225                 desk.open(file1); //调用open(File f)方法打开文件  
226             }else {
227                 FileInputStream in=new FileInputStream(file1);
228                 in.close();
229                 desk.open(file1);
230             }
231         }catch(Exception e)  
232         {  
233             System.out.println(e.toString());  
234         }
235         
236     }
237     
238     public void screenShot(String screenOutDir,WebDriver driver,String funNm) throws InterruptedException{
239         /**
240           * FunName:        openOutCome
241           * Description :   截图
242           * @param:                 文件URL
243           * @return void:        无返回数据;
244           * @Author:        Bingo
245           * @Create Date:   2015-06-30
246           **/
247         //Thread.sleep(3000);
248         SimpleDateFormat sdf = new SimpleDateFormat("yyyyMMdd HHmmss-SSS");
249         String time = sdf.format(new Date());
250         
251         String fileName = funNm+time +".png";
252         try {// 执行屏幕截图,默认会把截图保存到temp目录
253              File source_file = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE);  
254          // 这里将截图另存到我们需要保存的目录,例如screenshot\20150406 165210-333.png    
255              FileUtils.copyFile(source_file, new File(screenOutDir + "/" + fileName));  
256         } catch (IOException e) {
257              e.printStackTrace();
258           }        
259     }
260     
261     public  static String replaceBlank(String str) {
262         /**
263          * 返回STR中去除空格、换行制表符的内容
264          * 
265          */
266         
267         String dest = "";
268         if (str!=null) {
269             Pattern p = Pattern.compile("\\s*|\t|\r|\n");
270             Matcher m = p.matcher(str);
271             dest = m.replaceAll("");
272         }
273         return dest;
274     }
275     //获取当前系统时间
276     public static String nowDate(){
277         String nowDate = null;
278         SimpleDateFormat df = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");//设置日期格式
279         nowDate =df.format(new Date()).toString();
280         return nowDate;
281     }
282     
283     //判断字符串是否纯数字
284     public boolean isDigitStr(String s){
285         for(int i =0; i<s.length()-1;i++){
286             if(!Character.isDigit(s.charAt(i))){
287                 return false;
288             }
289         }
290         return true;
291     }
292     
293     //根据byType及值,返回by,访问方式:driver.findElement(getBy("id","kw"))
294     public static By getBy(String byType, String byValue){
295         By by = null;
296         switch(byType){
297             case "id": by = By.id(byValue);break;
298             case "xpath": by = By.xpath(byValue);break;
299             case "name": by = By.xpath(byValue);break;
300             case "className": by = By.className(byValue);break;
301             case "linkText": by = By.linkText(byValue);break;
302             case "cssSelector": by = By.cssSelector(byValue);break;
303             case "tagName": by = By.tagName(byValue);break;
304             case "partialLinkText": by = By.partialLinkText(byValue);break;
305             default:break;
306             
307         }
308         return by;
309     }
310     public static void println(Object obj){
311         /**
312           * FunName:        println
313           * Description :   简版打印,代替System.out.print(object)
314           * @param:                 object
315           * @return String:  none
316           * @Author:        Bingoreo
317           * @Create Date:   2015-07-15
318           **/
319         System.out.println(obj);
320     }
321 
322     public static void print(Object obj){
323         /**
324           * FunName:        println
325           * Description :   简版打印,代替System.out.print(object)
326           * @param:                 object
327           * @return String:  none
328           * @Author:        Bingoreo
329           * @Create Date:   2015-07-15
330           **/
331         System.out.print(obj);
332         
333     }
334 
335     public static void println(){
336         /**
337           * FunName:        println
338           * Description :   简版打印,代替System.out.println()
339           * @param:                 none
340           * @return String:  none
341           * @Author:        Bingoreo
342           * @Create Date:   2015-07-15
343           **/
344         System.out.println();
345     }
346     public static void main(String[] args) throws IOException, InterruptedException {
347         // TODO 自动生成的方法存根
348 //        BaseClass baseClass = new BaseClass();
349 //        String baseDatePath = "about:blank";
350 //        String inPath = baseClass.getBaseData("inPath",baseDatePath);
351 //        String outPath = baseClass.getBaseData("outPath",baseDatePath);
352 //        String baseURL = baseClass.getBaseData("baseURL",baseDatePath);
353 //        String firefoxPath = baseClass.getBaseData("firefoxPath",baseDatePath);
354 //        String orgz = baseClass.getData_xlsx("Login", "orgz", inPath);
355 //        String userId = baseClass.getData_xlsx("Login", "userId", inPath);
356 //        String pwd = baseClass.getData_xlsx("Login", "pwd", inPath);
357 //        String appNo = baseClass.getData_xlsx("QueryApp", "appNo", inPath);
358 //        String holName = baseClass.getData_xlsx("QueryApp", "holName", inPath);
359 //        String cerType = baseClass.getData_xlsx("QueryApp", "cerType", inPath);
360 //        String cerNo = baseClass.getData_xlsx("QueryApp", "cerNo", inPath);
361 //        SimpleDateFormat df = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");//设置日期格式
362 //        String date =df.format(new Date());
363 //        baseClass.outCome_xlsx("测试方法", "是否通过", outPath, date);
364 //        System.out.println("inPath:"+inPath+"\noutPath:"+outPath+"\nbaseURL:"+baseURL+"\nfirefoxPath"+firefoxPath);
365 //        System.out.println("\norgz:"+orgz+"\nuserId:"+userId+"\npwdL:"+pwd+"\nappNo"+appNo);
366 //        System.out.println("\nholName:"+holName+"\ncerType:"+cerType+"\ncerNo:"+cerNo);
367 //        baseClass.openOutCome(outPath);
368 //        WebDriver driver = new FirefoxDriver();
369 //        driver.get("https://www.baidu.com");
370 //        driver.findElement(getBy("id","kw")).sendKeys("11");
371 //        Thread.sleep(3000);
372 //        String screenShotDir = baseClass.getBaseData("screenShotDir",baseDatePath);
373 //        Thread.sleep(3000);
374 //        baseClass.screenShot(screenShotDir, driver,"testFunNm");
375 //        driver.close();
376 //        System.out.println(System.getProperty("user.dir")) ;
377 //        writeLog("4st msg");
378 //        writeLog("5nd msg");
379 //        writeLog("6rd msg");
380         WebDriver ieDriver = startBrowser("ie");
381         ieDriver.get("about:blank");
382 //        startBrowser("chrome");
383         WebDriver fireFoxdriver = startBrowser("fireFox");
384         fireFoxdriver.get("http://www.baidu.com");
385         println("tt");
386         
387     }
388     
389 } 

 

posted on 2015-07-20 18:27  抟扶  阅读(696)  评论(0编辑  收藏  举报