java 自动化测试小功能集锦

1. java 调用执行shell 命令

package com.mkyong.shell;
import java.io.BufferedReader;
import java.io.InputStreamReader;

public class ExecuteShellComand {
public static void main(String[] args) {
ExecuteShellComand obj = new ExecuteShellComand();
String domainName = "google.com";
//in mac oxs
String command = "ping -c 3 " + domainName;

//in windows
//String command = "ping -n 3 " + domainName;
String output = obj.executeCommand(command);
System.out.println(output);
}

private String executeCommand(String command) {
StringBuffer output = new StringBuffer();
Process p;
try {
p = Runtime.getRuntime().exec(command);
p.waitFor();
BufferedReader reader =
new BufferedReader(new InputStreamReader(p.getInputStream()));
String line = "";
while ((line = reader.readLine())!= null) {
output.append(line + "\n");
}
} catch (Exception e) {
e.printStackTrace();
}
return output.toString();
}
}

转自：http://www.mkyong.com/java/how-to-execute-shell-command-from-java/

2. java 按时间创建执行日志目录

public static String mkReportDir(){

// 创建报告存放目录，以日期为标记

String reportRootDir="/root/log/"; // windows平台 D:\\reports

String timeDir = new SimpleDateFormat("yyyy_MM_dd_HH_mm_ss").format(new Date());

String reportDir = reportRootDir + timeDir; // linux平台

// String reportDir = reportRootDir + "\\\\" + timeDir; // windows 平台

File file = new File(reportDir);

if(!file.getParentFile().exists()){

file.getParentFile().mkdir();

}

if(!file.exists() && !file.isDirectory()){

file.mkdir();

}

System.out.println(file.getAbsolutePath());

return file.getAbsolutePath();

}

3. java 将log写入文件

FileWriter fileWriter = new FileWriter(reportDir+"/1.log", true);

BufferedWriter bWriter = new BufferedWriter(fileWriter);

bWriter.write("123");

bWriter.newLine();

bWriter.flush();

bWriter.close();

4. java连接mysql数据库

String driver = "com.mysql.jdbc.Driver";

String url = "jdbc:mysql://12345678xxx.sh.cdb.myqcloud.com:3306/xxxdb";

String user = "xxxuser";

String password = "xxxpassword";

int result = 0;

try {

Class.forName(driver);

Connection conn = DriverManager.getConnection(url, user, password);

if(!conn.isClosed()){

}else{

System.out.println("数据库连接失败");

}

Statement statement = conn.createStatement();

ResultSet rs = statement.executeQuery(sql);

while(rs.next()){

result = rs.getInt("dbResult"); // 如 String sql = "select count(*) dbResult from orders where userid = 888";

}

/* while(rs.next()){

dbResult = rs.getString(dbResult);

System.out.println(dbResult);

}*/

rs.close();

System.out.println("connection closed...");

conn.close();

return result;

} catch (ClassNotFoundException | SQLException e) {

// TODO Auto-generated catch block

e.printStackTrace();

System.out.println("没有返回");

return 0;

}

5. 创建并写入excel表

String logDir = "c:\\log\\wxq\\";

String fileName = logDir + new SimpleDateFormat("yyyy_MM_dd").format(new Date()) + ".xls";

File file = new File(fileName);

WritableWorkbook book;

if (file.exists()) {

file.delete();

Thread.sleep(100);

}

book = Workbook.createWorkbook(file);

WritableSheet sheet = book.createSheet("群聊人数统计", 0);

Label xiaoqu_title = new Label(0, 0, "小区名");

Label renshu_title = new Label(1, 0, "人数");

sheet.addCell(xiaoqu_title);

sheet.addCell(renshu_title);

book.write();

book.close();

6. java 调用执行js 并返回结果：

JavascriptExecutor js = (JavascriptExecutor) wd; // WebDriver

title_count = (String) js.executeScript("return document.getElementsByClassName(\"title_count ng-binding ng-scope\")[0].innerText;");

7. map有序化，按key值：

Demo.java 文件

package com.yang.selenium;

import java.util.HashMap;

import java.util.Map;

import java.util.TreeMap;

public class Demo {

public static void main(String[] args) {

HashMap<String, String> map = new HashMap<String, String>();

map.put("重庆一号", "2");

map.put("重庆二号", "2");

map.put("重庆三号", "2");

map.put("重庆四号", "2");

map.put("重庆五号", "2");

map.put("合肥一号", "3");

map.put("合肥二号", "3");

map.put("合肥三号", "3");

map.put("合肥四号", "3");

map.put("合肥五号", "3");

map.put("南京一号", "5");

map.put("南京二号", "5");

map.put("南京三号", "5");

map.put("南京四号", "5");

map.put("南京五号", "5");

int sum=0;

for(String key: map.keySet()){

sum+=Integer.parseInt(map.get(key));

System.out.println(key + " : " + map.get(key));

}

System.out.println("sum : " + sum);

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

Map<String, String> resultMap = sortMapByKey(map);
//按Key进行排序

for (Map.Entry<String, String> entry : resultMap.entrySet())

{

System.out.println(entry.getKey() + " " + entry.getValue());

}

}

public static Map<String, String> sortMapByKey(Map<String, String> map){

if(map == null || map.isEmpty()){

return null;

}

Map<String, String> sortMap = new TreeMap<String, String>(new MapKeyComparator());

sortMap.putAll(map);

return sortMap;

}

}

MapKeyComparator.java 文件

package com.yang.selenium;

import java.util.Comparator;

public class MapKeyComparator implements Comparator<String>{

public int compare(String str1, String str2) {

return str1.compareTo(str2);

}

}