Java自动化测试常用的工具代码

package com.auto.Test;

import java.awt.Dimension;

import java.awt.Rectangle;

import java.awt.Robot;

import java.awt.Toolkit;

import java.awt.image.BufferedImage;

import java.io.File;

import javax.imageio.ImageIO;

public class snap {

private String fileName;

private String defaultName="GuiCamera";

static int serialNum=0;

private String imageFormat;//图像文件的格式

private String defaultImageFormat="jpg";

Dimension d=Toolkit.getDefaultToolkit().getScreenSize();

public  snap(){

fileName=defaultName;

imageFormat=defaultImageFormat;

}

public snap(String s,String format) {

fileName=s;

imageFormat=format;

}

/**

* 对屏幕进行拍照

*

* **/

public void snapshot(){

try {

//拷贝屏幕到一个BufferedImage对象screenshot

BufferedImage screenshot=(new Robot()).createScreenCapture(

new Rectangle(0,0,(int)d.getWidth(),(int)d.getHeight()));

serialNum++;

//根据文件前缀变量和文件格式变量，自动生成文件名

String name=fileName+String.valueOf(serialNum)+"."+imageFormat;

System.out.println(name);

File f=new File(name);

System.out.println("Save File-"+name);

//将screenshot对象写入图像文件

ImageIO.write(screenshot, imageFormat, f);

System.out.println("..Finished");

} catch (Exception e) {

System.out.println(e);

}

}

/*

public static void main(String[] args) {

//"C:\\sally\\bootstrap栅格"是根据自己随意找的一个图片形式，"png"是图片的格式

snap cam=new snap("C:\\sally\\bootstrap栅格","png");

cam.snapshot();

}

*/

}

public class snapWithSize {

public void snap(int height,int width,String filename) {

try {

Robot robot=new Robot();

//根据指定的区域抓取屏幕的指定区域，1300是长度，800是宽度。

BufferedImage bi=robot.createScreenCapture(new Rectangle(height,width));

//把抓取到的内容写到一个jpg文件中

ImageIO.write(bi, "jpg", new File(filename+".png"));

} catch (AWTException e) {

e.printstacktrace();

} catch (IOException e) {

e.printStackTrace();

}

}

}

package CommUtils;  

import java.awt.Rectangle;  

import java.awt.Robot;   

import java.awt.image.BufferedImage;  

import java.io.File;  

import javax.imageio.ImageIO;  

public class snapFree {  

private String fileName;  

private String defaultName="GuiCamera";  

static int serialNum=0;  

private String imageFormat;//图像文件的格式  

private String defaultImageFormat="jpg";   

public  snapFree(){  

fileName=defaultName;  

imageFormat=defaultImageFormat;  

}  

public snapFree(String s,String format) {  

fileName=s;  

imageFormat=format;  

}  

/**

* 对屏幕进行拍照

*  

* **/  

public void snapshot(int x,int y,int width,int Height){  

try {  

//拷贝屏幕到一个BufferedImage对象screenshot  

BufferedImage screenshot=(new Robot()).createScreenCapture(  

       new Rectangle(x,y,width,Height));  

serialNum++;  

//根据文件前缀变量和文件格式变量，自动生成文件名  

String name=fileName+String.valueOf(serialNum)+"."+imageFormat;  

System.out.println(name);  

File f=new File(name);  

System.out.println("Save File-"+name);  

//将screenshot对象写入图像文件  

ImageIO.write(screenshot, imageFormat, f);  

System.out.println("..Finished");  

} catch (Exception e) {  

System.out.println(e);  

}  

}  

public static void main(String[] args) {

//"C:\\sally\\bootstrap栅格"是根据自己随意找的一个图片形式，"png"是图片的格式  

snapFree cam=new snapFree("C:\\bootstrap栅格","jpg");  

cam.snapshot(200,200,300,400);  

}  

}

package com.auto.Test12306;

import org.junit.Test;

public class wirtecontext {

private String contxt = "写入内容";

private String path = "C:\\output.txt";

@Test

public void wirteContext() {

TestLogin testLogin = new TestLogin();

//传入 书写内容和路径

testLogin.outPut(contxt,path);

}

}

package com.auto.Test12306;

import java.io.File;

import java.io.BufferedWriter;

import java.io.FileWriter;

import org.junit.Test;

public class TestLogin {

@Test

public void outPut(String wirteContext,String path) {

try {

/* 写入Txt文件 */

File writename = new File(path); // 相对路径，如果没有则要建立一个新的output。txt文件

writename.createnewfile(); // 创建新文件

BufferedWriter out = new BufferedWriter(new FileWriter(writename));

out.write(wirteContext);

out.flush(); // 把缓存区内容压入文件

out.close(); // 最后记得关闭文件

} catch (Exception e) {

e.printStackTrace();

}

}

}

package com.auto.lyzb.utils;

import java.nio.charset.Charset;

import java.util.ArrayList;

import com.csvreader.CsvReader;

public class ReadCSV {

public static void main(String args[])

{

ArrayList<String []> list = readCsv("C:\\Users\\AdMinistrator\\Desktop\\test.csv");

for(int i=0;i<list.size();i++)

{

System.out.println(list.get(i)[0]);//name

System.out.println(list.get(i)[1]);//age

System.out.println(list.get(i)[2]);//sex

System.out.println("-------------------");

}

}

public static ArrayList<String []> readCsv(String filePath)

{

ArrayList<String[]> list = new ArrayList<String []>();//创建保存数据集合

//创建csvreader读取

CsvReader cReader = null;

try {

cReader = new CsvReader(filePath,',',Charset.forName("GBK"));

//是否跳过表头

cReader.readheaders();

//录入数据

while(cReader.readRecord()){

   list.add(cReader.getValues());

}

} catch (Exception e) {

e.printStackTrace();

}finally{

cReader.close();

}

//如果使用testng的DataProvider,可以返回一个二维数组

Object data[][] = new Object[list.size()][];

for(int i=0;i<list.size();i++)

{

data[i]=list.get(i);

}        

return list;

}

}

<dependency>

<groupId>com.hynnet</groupId>

<artifactId>jxl</artifactId>

<version>2.6.12.1</version>

</dependency>

package Auto.StreamaxTest.Case;  

import java.io.File;  

import jxl.*;   

public class ReadExcel{  

public static void main(String[] args) {  

Sheet sheet = null;  

Workbook book = null;  

Cell cell1 = null;  

try {   

//hello.xls为要读取的excel文件名  

book= Workbook.getWorkbook(new File("C:/Users/Administrator/Desktop/Auto_CSV File/AddCarGroup.xls"));   

//获得第一个工作表对象(ecxel中sheet的编号从0开始,0,1,2,3,....)  

sheet=book.getSheet(0);   

//获取左上角的单元格  

cell1=sheet.getCell(1,1);  

System.out.println(cell1.getcontents());   

}  

catch(Exception e)  { }   

}  

} 

package CommUtils;

import java.io.File;

import java.util.ArrayList;

import jxl.Workbook;

import jxl.write.Label;

import jxl.write.WritableSheet;

import jxl.write.WritableWorkbook;

public class WriteExcel {

private static ArrayList<String> arrayList;

public static void main(String[] args) {

arrayList = new ArrayList<String>();

arrayList.add("Streamax1");

arrayList.add("Streamax2");

arrayList.add("Streamax3");

WriteExcel.createExcel("c:/test.xls","第一层", 0);

}

public static void createExcel(String path,String sheetNmae,int sheetNumber) {

try {

// 在path路径下建立一个excel文件

WritableWorkbook wbook = Workbook.createWorkbook(new File(path));

// 创建一个工作表 第一个工作区

WritableSheet wsheet = wbook.createSheet(sheetNmae, sheetNumber);

//WritableSheet wsheet1 = wbook.createSheet("数据清单2", 1);

//WritableSheet wsheet2 = wbook.createSheet("数据清单3", 2);

/**

// 设置excel里的字体

WritableFont wf = new WritableFont(WritableFont.ARIAL, 12, WritableFont.NO_BOLD, false);

// 给标题规定字体的格式

WritableCellformat titleFormat = new WritableCellFormat(wf);

String[] title = { "账号", "密码" };

// 设置表头

for (int i = 0; i < title.length; i++) {

// 一列列的打印表头 按照我们规定的格式

Label excelTitle = new Label(i, 0, title[i], titleFormat);

// 把标头加到我们的工作区

wsheet.addCell(excelTitle);

}

*/

for(int i = 0;i<arrayList.size(); i++){

//向第i列 第二行写入内容

Label lable = new Label(i, 1, arrayList.get(i));

// 把值加到工作表中

wsheet.addCell(lable);

}

// 写入文件

wbook.write();

wbook.close();

System.out.println("创建成功!");

} catch (Exception e) {

// TODO: handle exception

}

}

}

package streamax.test;    

import java.io.BufferedWriter;    

import java.io.File;    

import java.io.filenotfoundException;    

import java.io.FileWriter;    

import java.io.IOException;    

public class WriteCSV {    

 public static void Wirtecsv(String filePath,String WirteContext) {    

try {    

 File csv = new File(filePath); // CSV数据文件  例如：C:/Users/Administrator/Desktop/Test1/Test.csv

 BufferedWriter bw = new BufferedWriter(new FileWriter(csv, true)); // 附加   

 // 添加新的数据行   

 bw.write(WirteContext);    

 bw.newLine();    

 bw.close();    

} catch (FileNotFoundException e) {    

 // File对象的创建过程中的异常捕获   

 e.printStackTrace();    

} catch (IOException e) {    

 // BufferedWriter在关闭对象捕捉异常   

 e.printStackTrace();    

}    

 }    

}  

package com.auto.lyzb.user;

import org.openqa.selenium.WebDriver;

import org.openqa.selenium.WebElement;

import org.openqa.selenium.support.FindBy;

public class RegisterBean {

private WebDriver driver;

@FindBy(linkText = "用户名")

private WebElement ele1;

@FindBy(linkText = "密码")

private WebElement ele2;

@FindBy(linkText = "登录")

private WebElement ele3;

//构造函数  创建对象时传入driver

public RegisterBean(WebDriver driver) {

super();

this.driver = driver;

}

public void login(String username,String password) {

ele1.sendKeys(username);

ele2.sendKeys(password);

ele3.click();

}

}

package com.auto.lyzb.user;

import static org.testng.Assert.assertequals;

import static org.testng.Assert.assertTrue;

import org.openqa.selenium.WebDriver;

import org.openqa.selenium.firefox.FirefoxDriver;

import org.testng.annotations.AfterMethod;

import org.testng.annotations.Test;

public class RegisterNew {

private static WebDriver driver = new FirefoxDriver();

// RegisterBean有有参构造函数

private RegisterBean rs = new RegisterBean(driver);

@Test

public void registrnew() {

driver.get("http://xx.xx.128.xx:xx端口");

rs.login("123", "qwe");

//断言

assertEquals(true, true);

}

@AfterMethod

public void closedIE() {

driver.quit();

}

}

package Utils;

import java.io.File;

import java.io.fileinputstream;

import java.io.Fileoutputstream;

import java.io.InputStream;

import org.junit.Test;

public class RW {

private RW rw = new RW();

/**

*

*/

@Test

public void Test1() {

//将照片复制1000张到D:/123这个文件夹下

for (int i = 0; i < 1000; i++) {

rw.copyFile("c:/1.JPG", "D:/123/"+i+".JPG");

}

}

@Test

public void Test2() {

rw.copyfolder("D:/123", "C:/1234");

}

/***

* 复制单个文件*

*

* @param oldPath

*            String 原文件路径 如：c:/fqf.txt*

* @param newPath

*            String 复制后路径 如：f:/fqf.txt*@return boolean

*/

public void copyFile(String oldPath, String newPath) {

try {

int bytesum = 0;

int byteread = 0;

File oldfile = new File(oldPath);

if (oldfile.exists()) { // 文件存在时

InputStream inStream = new FileInputStream(oldPath); // 读入原文件

FileOutputStream fs = new FileOutputStream(newPath);

byte[] buffer = new byte[1444];

int length;

while ((byteread = inStream.read(buffer)) != -1) {

bytesum += byteread; // 字节数 文件大小

// System.out.println(bytesum);

fs.write(buffer, 0, byteread);

}

inStream.close();

}

} catch (Exception e) {

System.out.println("复制单个文件操作出错");

e.printStackTrace();

}

}

/**

* 复制整个文件夹内容

*

* @param oldPath

*            String 原文件路径 如：c:/fqf

* @param newPath

*            String 复制后路径 如：f:/fqf/ff

* @return boolean

*/

public void copyFolder(String oldPath, String newPath) {

try {

(new File(newPath)).mkdirs(); // 如果文件夹不存在 则建立新文件夹

File a = new File(oldPath);

String[] file = a.list();

File temp = null;

for (int i = 0; i < file.length; i++) {

if (oldPath.endsWith(File.separator)) {

temp = new File(oldPath + file[i]);

} else {

temp = new File(oldPath + File.separator + file[i]);

}

if (temp.isFile()) {

FileInputStream input = new FileInputStream(temp);

FileOutputStream output = new FileOutputStream(newPath + "/" + (temp.getName()).toString());

byte[] b = new byte[1024 * 5];

int len;

while ((len = input.read(b)) != -1) {

output.write(b, 0, len);

}

output.flush();

output.close();

input.close();

}

if (temp.isDirectory()) {// 如果是子文件夹

copyFolder(oldPath + "/" + file[i], newPath + "/" + file[i]);

}

}

} catch (Exception e) {

System.out.println("复制整个文件夹内容操作出错");

e.printStackTrace();

}

}

}

package Test1;

import java.io.File;

import java.io.FileInputStream;

import org.apache.commons.net.ftp.FTPClient;

import org.apache.commons.net.ftp.FTPClientConfig;

import org.testng.annotations.Test;

public class ftpUpload {

@Test

public void testFtpClient(){  

//创建一个FTPClient对象  

FTPClient ftpClient = new FTPClient();  

//创建FTP连接,端口号可以设定，我的是22,默认是21  

try {

ftpClient.connect("xxx.xxx.151.47",21);  

//登录FTP服务器，使用用户名和密码   下列写法是没有密码的写法

ftpClient.login("anonymous", "");  

//ftpClient.enterRemotePassiveMode();  

//ftpClient.setControlEncoding("gb18030");  

//上传文件,读取本地文件  

String path = "C:/1.jpg";  

FileInputStream inputStream = new FileInputStream(new File(path));  

//设置上传的路径  

String pathname = "/abc";//这个路径就是FTP服务端存储的路径，先要确保abc这个文件目录存在

ftpClient.changeWorkingDirectory(pathname); //保证上传到指定的目录下 如果没有这一句 将会上传到ftp根目录下

//参数一：服务器端文档名；参数二：上传文档的inputStream  

String remote = "1.jpg";  

ftpClient.storeFile(remote, inputStream);  

//关闭连接  

ftpClient.logout();

} catch (Exception e) {

e.printStackTrace();

}  

}

}

package streamax.test;

import java.io.File;

import java.util.ArrayList;

import java.util.List;

public class ReadFileNanes {

//文件名列表

private static List<String> fileList = new ArrayList<String>();

/**

* @param args

*/

public static void main(String[] args) {

//该目录下有123.jpg 和 20180515153634_10.jpg两文件

printAllFileNames("C:/Users/Administrator/Desktop/Test1");

}

public static void printAllFileNames(String path) {

if (null != path) {

File file = new File(path);

if (file.exists()) {

   File[] list = file.listFiles();

   if(null != list){

       for (File child : list) {

           if (child.isFile()) {

               fileList.add(child.getabsolutePath());

               String filePath = child.getAbsolutePath();

               String jieguo = filePath.substring(filePath.indexof("Test1")+6,filePath.indexOf(".jpg")+4);

               System.out.println(jieguo);

               //输出内容为

//                            123.jpg

//                            20180515153634_10.jpg

//                            System.out.println("==============");

           } else if (child.isDirectory()) {

               printAllFileNames(child.getAbsolutePath());

           }

       }

   }

}

}

}

}

package CommUtils;

import java.awt.Rectangle;

import java.awt.image.BufferedImage;

import java.io.File;

import java.io.FileInputStream;

import java.io.IOException;

import java.util.Iterator;

import javax.imageio.ImageIO;

import javax.imageio.ImageReadParam;

import javax.imageio.ImageReader;

import javax.imageio.stream.ImageInputStream;

public class snapFree {

// ===源图片路径名称如:c:\1.jpg

private String srcpath;

// ===剪切图片存放路径名称.如:c:\2.jpg

private String subpath;

// ===剪切点x坐标

private int x;

private int y;

// ===剪切点宽度

private int width;

private int height;

public snapFree() {

}

public snapFree(int x, int y, int width, int height) {

this.x = x;

this.y = y;

this.width = width;

this.height = height;

}

/**

* 对图片裁剪，并把裁剪完蛋新图片保存 。

*/

public void cut() throws IOException {

FileInputStream is = null;

ImageInputStream iis = null;

try {

// 读取图片文件

is = new FileInputStream(srcpath);

/*

* 返回包含所有当前已注册 ImageReader 的 Iterator，这些 ImageReader 声称能够解码指定格式。

* 参数：formatName - 包含非正式格式名称 . （例如 "jpeg" 或 "tiff"）等 。

*/

Iterator<ImageReader> it = ImageIO.getImageReadersByFormatName("jpg");

ImageReader reader = it.next();

// 获取图片流

iis = ImageIO.createImageInputStream(is);

/*

* <p>iis:读取源.true:只向前搜索 </p>.将它标记为 ‘只向前搜索’。

* 此设置意味着包含在输入源中的图像将只按顺序读取，可能允许 reader 避免缓存包含与以前已经读取的图像关联的数据的那些输入部分。

*/

reader.setInput(iis, true);

/*

* <p>描述如何对流进行解码的类<p>.用于指定如何在输入时从 Java Image I/O

* 框架的上下文中的流转换一幅图像或一组图像。用于特定图像格式的插件 将从其 ImageReader 实现的

* getDefaultReadParam 方法中返回 ImageReadParam 的实例。

*/

ImageReadParam param = reader.getDefaultReadParam();

/*

* 图片裁剪区域。Rectangle 指定了坐标空间中的一个区域，通过 Rectangle 对象

* 的左上顶点的坐标（x，y）、宽度和高度可以定义这个区域。

*/

Rectangle rect = new Rectangle(x, y, width, height);

// 提供一个 BufferedImage，将其用作解码像素数据的目标。

param.setSourceRegion(rect);

/*

* 使用所提供的 ImageReadParam 读取通过索引 imageIndex 指定的对象，并将 它作为一个完整的

* BufferedImage 返回。

*/

BufferedImage bi = reader.read(0, param);

// 保存新图片

ImageIO.write(bi, "jpg", new File(subpath));

} finally {

if (is != null)

is.close();

if (iis != null)

iis.close();

}

}

public int getHeight() {

return height;

}

public void setHeight(int height) {

this.height = height;

}

public String getSrcpath() {

return srcpath;

}

public void setSrcpath(String srcpath) {

this.srcpath = srcpath;

}

public String getSubpath() {

return subpath;

}

public void setSubpath(String subpath) {

this.subpath = subpath;

}

public int getWidth() {

return width;

}

public void setWidth(int width) {

this.width = width;

}

public int getX() {

return x;

}

public void setX(int x) {

this.x = x;

}

public int getY() {

return y;

}

public void setY(int y) {

this.y = y;

}

public static void main(String[] args) {

snapFree operateImage = new snapFree(469, 222, 100, 100);//开始的宽度，开始的高度，截图宽度，截图高度

operateImage.srcpath = "C:/Users/Administrator/Desktop/1.jpg";

operateImage.subpath = "C:/Users/Administrator/Desktop/2.jpg";

try {

operateImage.cut();

} catch (IOException e) {

e.printStackTrace();

}

}

}

package enter.entrys;

import javax.swing.JFrame;  

import javax.swing.JOptionPane;  

import org.junit.Test;  

public class demo {

@Test

public void test(){

int n = JOptionPane.showConfirmDialog(null, "确认删除吗?", "确认对话框", JOptionPane.YES_NO_OPTION);

if (n == JOptionPane.YES_OPTION) {   

//    JOptionPane.showmessageDialog(new JFrame(),"已删除");

System.out.println("执行确定的操作");

} else if (n == JOptionPane.NO_OPTION) {   

//    JOptionPane.showMessageDialog(new JFrame(),"已取消");  

System.out.println("执行取消的操作");

}

}

}