添加随机词语返回方法
This commit is contained in:
彭宇琦
parent
61b8752b29
commit
932fd306e2
|
|
@ -0,0 +1,19 @@
|
|||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<project name="工资单管理">
|
||||
<templet>
|
||||
<xpath id='1'>//iframe[contains(@src,'${src}')]</xpath>
|
||||
<xpath id='2'>//li[text()='${mon}']</xpath>
|
||||
</templet>
|
||||
|
||||
<element name='日历'>
|
||||
<xpath is_use='true' temp_id='2'/>
|
||||
</element>
|
||||
|
||||
<iframe name='选择用户窗体'>
|
||||
<xpath is_use='true' temp_id='1' src='/project/salary/choiceUser.jsp'/>
|
||||
|
||||
<element name='选择所有人员'>
|
||||
<xpath is_use='true'>//a[text()='选择所有人员']</xpath>
|
||||
</element>
|
||||
</iframe>
|
||||
</project>
|
||||
|
|
@ -0,0 +1,43 @@
|
|||
package pres.auxiliary.tool.randomstring;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.Arrays;
|
||||
|
||||
public enum MobleNumberType {
|
||||
/**
|
||||
* 中国电信运营商号码段
|
||||
*/
|
||||
CHINA_TELECOM(133, 1349, 149, 153, 162, 1700, 1701, 1702, 173, 17400, 17401, 17402, 17403, 17404, 17405, 177, 180, 181, 189, 191, 193, 199),
|
||||
/**
|
||||
* 中国移动运营商号码段
|
||||
*/
|
||||
CHINA_MOBILE(1340, 1341, 1342, 1343, 1344, 1345, 1346, 135, 136, 137, 138, 139, 147, 148, 150, 151, 152, 1570, 1571, 1572, 1573, 1574, 1575, 1576, 1577, 1578, 1579, 158, 159, 165, 1703, 1705, 1706, 172, 178, 182, 183, 184, 187, 188, 198),
|
||||
/**
|
||||
* 中国联通运营商号码段
|
||||
*/
|
||||
CHINA_UNICOM(130, 131, 132, 145, 146, 155, 156, 166, 167, 1704, 1707, 1708, 1709, 171, 175, 176, 185, 186);
|
||||
|
||||
/**
|
||||
* 用于表示当前运营商号码段正则表达式
|
||||
*/
|
||||
ArrayList<String> numberList = new ArrayList<>();
|
||||
|
||||
/**
|
||||
* 用于初始化数据
|
||||
* @param nums 号码段
|
||||
*/
|
||||
private MobleNumberType(int...nums) {
|
||||
//添加数据
|
||||
Arrays.stream(nums).forEach(num -> {
|
||||
numberList.add(String.valueOf(num));
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* 用于返回运营商号码段
|
||||
* @return 号码段
|
||||
*/
|
||||
public ArrayList<String> getRegex() {
|
||||
return numberList;
|
||||
}
|
||||
}
|
||||
|
|
@ -110,33 +110,47 @@ public class PresetString {
|
|||
* @return 生成的身份证号码
|
||||
*/
|
||||
public static String IdentityCard() {
|
||||
String s = "45030119";
|
||||
|
||||
rs.clear();
|
||||
rs.addMode(StringMode.NUM);
|
||||
s += rs.toString(2);
|
||||
|
||||
Random r = new Random();
|
||||
|
||||
// 生成月
|
||||
int i = (1 + r.nextInt(11));
|
||||
// 判断生成的随机字符串是否小于10，小于10则向s中补全一个0
|
||||
if (i < 10) {
|
||||
s += "0";
|
||||
StringBuilder sb = new StringBuilder();
|
||||
RandomString rs = new RandomString("123456789");
|
||||
|
||||
//省份
|
||||
int[] province = {13,14,15,21,22,23,31,32,33,34,35,36,37,41,42,43,44,45,46,50,51,52,53,54,61,62,63,64,65,71,81,82,91};
|
||||
//城市
|
||||
String[] city = {"01", "02", "03"};
|
||||
//区县
|
||||
String[] county = {"01", "02", "03", "04", "05"};
|
||||
//生日年份前两位
|
||||
int[] year = {19, 18};
|
||||
|
||||
//加权数
|
||||
int[] factors = {7, 9, 10, 5, 8, 4, 2, 1, 6, 3, 7, 9, 10, 5, 8, 4, 2};
|
||||
//校验位
|
||||
String[] parity = {"1", "0", "X", "9", "8", "7", "6", "5", "4", "3", "2"};
|
||||
|
||||
//随机添加一个省份
|
||||
sb.append(province[new Random().nextInt(province.length)]);
|
||||
//随机添加一个城市
|
||||
sb.append(city[new Random().nextInt(city.length)]);
|
||||
//随机添加一个区县
|
||||
sb.append(county[new Random().nextInt(county.length)]);
|
||||
//随机生日年份
|
||||
sb.append(year[new Random().nextInt(year.length)]).append(rs.toString(2));
|
||||
//随机生日月份及日子
|
||||
sb.append("0" + rs.toString(1)).append("0" + rs.toString(1));
|
||||
//随机三位数
|
||||
sb.append(rs.toString(3));
|
||||
|
||||
//计算加权数
|
||||
int sum = 0;
|
||||
for (int i = 0; i < 17; i++) {
|
||||
int code = Integer.valueOf(sb.substring(i, i + 1));
|
||||
int factor = Integer.valueOf(factors[i]);
|
||||
sum += (code * factor);
|
||||
}
|
||||
s += i;
|
||||
|
||||
// 生成日
|
||||
i = (1 + r.nextInt(27));
|
||||
// 判断生成的随机字符串是否小于10，小于10则向s中补全一个0
|
||||
if (i < 10) {
|
||||
s += "0";
|
||||
}
|
||||
s += i;
|
||||
|
||||
s += rs.toString(4);
|
||||
|
||||
return s;
|
||||
//根据加权数添加校验位
|
||||
sb.append(parity[(sum % 11)]);
|
||||
|
||||
return sb.toString();
|
||||
}
|
||||
|
||||
/**
|
||||
|
|
@ -176,6 +190,9 @@ public class PresetString {
|
|||
|
||||
return s;
|
||||
}
|
||||
}
|
||||
|
||||
private static String getMobleNumber(MobleNumberType mobleNumberType) {
|
||||
|
||||
}
|
||||
|
||||
|
|
|
|||
|
|
@ -0,0 +1,127 @@
|
|||
package pres.auxiliary.tool.randomstring;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.Arrays;
|
||||
import java.util.Random;
|
||||
|
||||
public class RandomWord {
|
||||
/**
|
||||
* 用于存储需要随机返回的词语
|
||||
*/
|
||||
private ArrayList<String> returnWordList = new ArrayList<>();
|
||||
/**
|
||||
* 用于存储默认返回的词语
|
||||
*/
|
||||
private ArrayList<String> conditionWordList = new ArrayList<>();
|
||||
|
||||
/**
|
||||
* 指向生成的词语是否能重复
|
||||
*/
|
||||
private boolean isRepeat = true;
|
||||
|
||||
/**
|
||||
* 用于向无需条件亦可直接返回的词语组中添加词语
|
||||
* @param words 词语组
|
||||
*/
|
||||
public void addRetuenWord(String...words) {
|
||||
returnWordList.addAll(Arrays.asList(words));
|
||||
}
|
||||
|
||||
/**
|
||||
* 用于向无需条件亦可直接返回的词语组中添加词语
|
||||
* @param wordLst 词语集合
|
||||
*/
|
||||
public void addRetuenWord(ArrayList<String> wordLst) {
|
||||
conditionWordList.addAll(wordLst);
|
||||
}
|
||||
|
||||
/**
|
||||
* 用于向符合条件后返回的词语组中添加词语
|
||||
* @param words 词语组
|
||||
*/
|
||||
public void addConditionWord(String...words) {
|
||||
conditionWordList.addAll(Arrays.asList(words));
|
||||
}
|
||||
|
||||
/**
|
||||
* 用于向符合条件后返回的词语组中添加词语
|
||||
* @param wordLst 词语集合
|
||||
*/
|
||||
public void addConditionWord(ArrayList<String> wordLst) {
|
||||
returnWordList.addAll(wordLst);
|
||||
}
|
||||
|
||||
/**
|
||||
* 用于设置词语是否可以重复生成
|
||||
* @param isRepeat 词语是否可重复
|
||||
*/
|
||||
public void setRepeat(boolean isRepeat) {
|
||||
this.isRepeat = isRepeat;
|
||||
}
|
||||
|
||||
/**
|
||||
* 用于随机生成指定随机个数的词语
|
||||
* @param minLength 最少词语个数
|
||||
* @param maxLength 最多词语个数
|
||||
* @return 生成的词语集合
|
||||
*/
|
||||
public ArrayList<String> toWord(int minLength, int maxLength) {
|
||||
return toWord(getLength(minLength, maxLength));
|
||||
}
|
||||
|
||||
/**
|
||||
* 用于随机生成指定个数的词语
|
||||
* @param length 输出的词语个数
|
||||
* @return 生成的词语集合
|
||||
*/
|
||||
public ArrayList<String> toWord(int length) {
|
||||
//克隆returnWordList，便于对不能生成重复词语的情况下，移除集合中的元素，达到快速返回的目的
|
||||
@SuppressWarnings("unchecked")
|
||||
ArrayList<String> wordList = (ArrayList<String>) returnWordList.clone();
|
||||
|
||||
//存储生成的词语
|
||||
ArrayList<String> randomWordList = new ArrayList<>();
|
||||
//循环，生成相应的词语
|
||||
for (int i = 0; i < length; i++) {
|
||||
//获取随机下标
|
||||
int index = getRandomIndex(wordList);
|
||||
//存储下标对应的词语
|
||||
randomWordList.add(wordList.get(index));
|
||||
//若生成的词语不可重复，则直接移除wordList中对应的词语
|
||||
if (!isRepeat) {
|
||||
wordList.remove(index);
|
||||
}
|
||||
}
|
||||
|
||||
return wordList;
|
||||
}
|
||||
|
||||
/**
|
||||
* 用于根据词语是否允许重复，返回实际的最大生成词语个数
|
||||
* @param maxLength 设置的最大词语生成个数
|
||||
* @return 实际最大词语生成个数
|
||||
*/
|
||||
private int getLength(int minLength, int maxLength) {
|
||||
//若词语不允许重复生成且maxLength大于returnWordList.size()，则将maxLength改为maxLength
|
||||
maxLength = (isRepeat && maxLength > returnWordList.size()) ? returnWordList.size() : maxLength;
|
||||
//为避免最小长度与最大长度传参相反，故此处进行一个调换
|
||||
int max = Math.max(minLength, maxLength);
|
||||
int min = Math.min(minLength, maxLength);
|
||||
|
||||
//返回指定长度下的一个随机长度
|
||||
return (new Random().nextInt(max - min + 1) + min);
|
||||
}
|
||||
|
||||
/**
|
||||
* 根据指定的集合，返回随机的一个下标
|
||||
* @param wordList 词语集合
|
||||
* @return 集合元素的随机下标
|
||||
*/
|
||||
private int getRandomIndex(ArrayList<String> wordList) {
|
||||
//根据条件获取集合的长度
|
||||
int maxLength = wordList.size();
|
||||
//生成最大长度范围内的随机数
|
||||
return new Random().nextInt(maxLength);
|
||||
|
||||
}
|
||||
}
|
||||
|
|
@ -137,9 +137,9 @@ public class ListFileRead {
|
|||
columnIndex = columnIndex < 0 ? 0 : columnIndex;
|
||||
|
||||
//若传值大于最大maxRowNum时，则直接赋予maxRowNum
|
||||
startRowIndex = (startRowIndex >= maxRowNum) ? maxRowNum - 1 : startRowIndex;
|
||||
startRowIndex = (startRowIndex >= maxRowNum) ? maxRowNum : startRowIndex;
|
||||
startRowIndex = startRowIndex < 0 ? 0 : startRowIndex;
|
||||
endRowIndex = (endRowIndex >= maxRowNum) ? maxRowNum - 1 : endRowIndex;
|
||||
endRowIndex = (endRowIndex >= maxRowNum) ? maxRowNum : endRowIndex;
|
||||
endRowIndex = endRowIndex < 0 ? 0 : endRowIndex;
|
||||
|
||||
//若两个传值相同，则endRowIndex+1
|
||||
|
|
@ -185,9 +185,9 @@ public class ListFileRead {
|
|||
endColumnIndex = endColumnIndex < 0 ? 0 : endColumnIndex;
|
||||
|
||||
//若传值大于最大maxRowNum时，则直接赋予maxRowNum，小于0，则直接赋予0
|
||||
startRowIndex = (startRowIndex >= maxRowNum) ? maxRowNum - 1 : startRowIndex;
|
||||
startRowIndex = (startRowIndex >= maxRowNum) ? maxRowNum : startRowIndex;
|
||||
startRowIndex = startRowIndex < 0 ? 0 : startRowIndex;
|
||||
endRowIndex = (endRowIndex >= maxRowNum) ? maxRowNum - 1 : endRowIndex;
|
||||
endRowIndex = (endRowIndex >= maxRowNum) ? maxRowNum : endRowIndex;
|
||||
endRowIndex = endRowIndex < 0 ? 0 : endRowIndex;
|
||||
|
||||
//若两个传值相同，则endXXXIndex+1
|
||||
|
|
|
|||
|
|
@ -1,169 +0,0 @@
|
|||
package pres.auxiliary.work.selenium.datadriven;
|
||||
|
||||
public class NumberDataDriven extends DataDriven {
|
||||
// /**
|
||||
// * 无参构造，不做任何操作，请使用{@link #setDataFile(File)}方法定义数据驱动文件
|
||||
// */
|
||||
// public NumberDataDriven() {
|
||||
// super();
|
||||
// }
|
||||
//
|
||||
// /**
|
||||
// * 用于指定数据驱动文件
|
||||
// *
|
||||
// * @param dataFile
|
||||
// * 数据驱动文件对象
|
||||
// */
|
||||
// public NumberDataDriven(File dataFile) {
|
||||
// super(dataFile);
|
||||
// }
|
||||
//
|
||||
// @Override
|
||||
// public File createDataFile(String saveFileName, int minSize, int maxSize, int length) throws IOException {
|
||||
// //用于存储生成的字符
|
||||
// StringBuilder sb = new StringBuilder("");
|
||||
// // 循环，向文件中添加随机数字
|
||||
// for (int i = 0; i < length; i++) {
|
||||
// // 添加随机数字
|
||||
// sb.append(new RandomString(StringMode.NUM).toString(minSize, maxSize));
|
||||
// sb.append("\r\n");
|
||||
// }
|
||||
// //删除最后一个换行符
|
||||
// sb.delete(sb.lastIndexOf("\r\n"), sb.lastIndexOf("\r\n") + "\r\n".length());
|
||||
//
|
||||
// // 返回数据驱动文件
|
||||
// return createDataFile(saveFileName, sb);
|
||||
// }
|
||||
//
|
||||
// @Override
|
||||
// public File createDataFile(String saveFileName, int size, int length) throws IOException {
|
||||
// //用于存储生成的字符
|
||||
// StringBuilder sb = new StringBuilder("");
|
||||
// // 循环，向文件中添加随机数字
|
||||
// for (int i = 0; i < length; i++) {
|
||||
// // 添加随机数字
|
||||
// sb.append(new RandomString(StringMode.NUM).toString(size));
|
||||
// sb.append("\r\n");
|
||||
// }
|
||||
// //删除最后一个换行符
|
||||
// sb.delete(sb.lastIndexOf("\r\n"), sb.lastIndexOf("\r\n") + "\r\n".length());
|
||||
//
|
||||
// // 返回数据驱动文件
|
||||
// return createDataFile(saveFileName, sb);
|
||||
// }
|
||||
//
|
||||
// /**
|
||||
// * 该方法用于从数字1开始，每次增加1步长来生成自然顺序的数字，并写入文件中
|
||||
// *
|
||||
// * @param saveFileName
|
||||
// * 数据驱动文件的文件名，若为空或为null，则默认为NewDataDriven（文件的后缀为.txt）
|
||||
// * @param length
|
||||
// * 数据生成个数
|
||||
// * @return 生成的数据驱动文件对象
|
||||
// * @throws IOException
|
||||
// *
|
||||
// * @see #createDataFile(String, int, int)
|
||||
// * @see #createDataFile(String, int, int, int)
|
||||
// */
|
||||
// public File createContinuesNumberDataFile(String saveFileName, int length) throws IOException {
|
||||
// //用于存储生成的字符
|
||||
// StringBuilder sb = new StringBuilder("");
|
||||
// // 循环，向文件中添加随机数字
|
||||
// for (int i = 1; i < length + 1; i++) {
|
||||
// // 添加随机数字
|
||||
// sb.append("" + i);
|
||||
// sb.append("\r\n");
|
||||
// }
|
||||
// //删除最后一个换行符
|
||||
// sb.delete(sb.lastIndexOf("\r\n"), sb.lastIndexOf("\r\n") + "\r\n".length());
|
||||
//
|
||||
// // 返回数据驱动文件
|
||||
// return createDataFile(saveFileName, sb);
|
||||
// }
|
||||
//
|
||||
// /**
|
||||
// * 该方法用于根据起始数字，通过增加一定的步长来生成自然顺序的数字，并写入文件中
|
||||
// *
|
||||
// * @param saveFileName
|
||||
// * 数据驱动的文件名
|
||||
// * @param startNum
|
||||
// * 起始数字
|
||||
// * @param step
|
||||
// * 步长
|
||||
// * @param length
|
||||
// * 生成数字的个数
|
||||
// * @return 数据驱动文件对象
|
||||
// * @throws IOException
|
||||
// */
|
||||
// public File createContinuesNumberDataFile(String saveFileName, int startNum, int step, int length)
|
||||
// throws IOException {
|
||||
// //用于存储生成的字符
|
||||
// StringBuilder sb = new StringBuilder("");
|
||||
// // 循环，向文件中添加随机数字
|
||||
// for (int i = 0; i < length; i++) {
|
||||
// // 添加随机数字
|
||||
// sb.append("" + startNum);
|
||||
// sb.append("\r\n");
|
||||
//
|
||||
// // 数字增加一步
|
||||
// startNum += step;
|
||||
// }
|
||||
// //删除最后一个换行符
|
||||
// sb.delete(sb.lastIndexOf("\r\n"), sb.lastIndexOf("\r\n") + "\r\n".length());
|
||||
//
|
||||
// // 返回数据驱动文件
|
||||
// return createDataFile(saveFileName, sb);
|
||||
// }
|
||||
//
|
||||
// /**
|
||||
// * 该方法用于向文件中添加身份证信息
|
||||
// *
|
||||
// * @param saveFileName
|
||||
// * 数据驱动的文件名
|
||||
// * @param length
|
||||
// * 生成身份证的个数
|
||||
// * @return 数据驱动文件对象
|
||||
// * @throws IOException
|
||||
// */
|
||||
// public File createIDCardDataFile(String saveFileName, int length) throws IOException {
|
||||
// //用于存储生成的字符
|
||||
// StringBuilder sb = new StringBuilder("");
|
||||
// // 循环，向文件中添加随机数字
|
||||
// for (int i = 0; i < length; i++) {
|
||||
// // 添加随机数字
|
||||
// sb.append(PresetString.IdentityCard());
|
||||
// sb.append("\r\n");
|
||||
// }
|
||||
// //删除最后一个换行符
|
||||
// sb.delete(sb.lastIndexOf("\r\n"), sb.lastIndexOf("\r\n") + "\r\n".length());
|
||||
//
|
||||
// // 返回数据驱动文件
|
||||
// return createDataFile(saveFileName, sb);
|
||||
// }
|
||||
//
|
||||
// /**
|
||||
// * 该方法用于向文件中添加手机号码信息
|
||||
// *
|
||||
// * @param saveFileName
|
||||
// * 数据驱动的文件名
|
||||
// * @param length
|
||||
// * 生成手机号码的个数
|
||||
// * @return 数据驱动文件对象
|
||||
// * @throws IOException
|
||||
// */
|
||||
// public File createPhoneDataFile(String saveFileName, int length) throws IOException {
|
||||
// //用于存储生成的字符
|
||||
// StringBuilder sb = new StringBuilder("");
|
||||
// // 循环，向文件中添加随机数字
|
||||
// for (int i = 0; i < length; i++) {
|
||||
// // 添加随机数字
|
||||
// sb.append("139" + new RandomString(StringMode.NUM).toString(8));
|
||||
// sb.append("\r\n");
|
||||
// }
|
||||
// //删除最后一个换行符
|
||||
// sb.delete(sb.lastIndexOf("\r\n"), sb.lastIndexOf("\r\n") + "\r\n".length());
|
||||
//
|
||||
// // 返回数据驱动文件
|
||||
// return createDataFile(saveFileName, sb);
|
||||
// }
|
||||
}
|
||||
|
|
@ -1,41 +0,0 @@
|
|||
package pres.auxiliary.work.selenium.datadriven;
|
||||
|
||||
public class StringDataDriven extends DataDriven {
|
||||
|
||||
/*
|
||||
@Override
|
||||
public File createDataFile(String saveFileName, int minSize, int maxSize, int length) throws IOException, DataNotFoundException {
|
||||
// 用于存储生成的字符
|
||||
StringBuilder sb = new StringBuilder("");
|
||||
// 循环，向文件中添加随机数字
|
||||
for (int i = 0; i < length; i++) {
|
||||
// 添加随机字母
|
||||
sb.append(new RandomString(StringMode.LOW, StringMode.CAP).toString(minSize, maxSize));
|
||||
sb.append("\r\n");
|
||||
}
|
||||
// 删除最后一个换行符
|
||||
sb.delete(sb.lastIndexOf("\r\n"), sb.lastIndexOf("\r\n") + "\r\n".length());
|
||||
|
||||
// 返回数据驱动文件
|
||||
return createDataFile(saveFileName, sb);
|
||||
}
|
||||
|
||||
@Override
|
||||
public File createDataFile(String saveFileName, int size, int length) throws IOException {
|
||||
// 用于存储生成的字符
|
||||
StringBuilder sb = new StringBuilder("");
|
||||
// 循环，向文件中添加随机数字
|
||||
for (int i = 0; i < length; i++) {
|
||||
// 添加随机数字
|
||||
sb.append(new RandomString(StringMode.LOW, StringMode.CAP).toString(size));
|
||||
sb.append("\r\n");
|
||||
}
|
||||
// 删除最后一个换行符
|
||||
sb.delete(sb.lastIndexOf("\r\n"), sb.lastIndexOf("\r\n") + "\r\n".length());
|
||||
|
||||
// 返回数据驱动文件
|
||||
return createDataFile(saveFileName, sb);
|
||||
}
|
||||
*/
|
||||
|
||||
}
|
||||
|
|
@ -55,7 +55,8 @@ public class TestNGDataDriver {
|
|||
/**
|
||||
* 用于存储读取到的数据
|
||||
*/
|
||||
private ArrayList<Data> dataList = new ArrayList<>();
|
||||
// private ArrayList<Data> dataList = new ArrayList<>();
|
||||
private HashMap<String, List<Object>> dataMap = new HashMap<>(16);
|
||||
|
||||
/**
|
||||
* 用于存储自定义的公式
|
||||
|
|
@ -67,6 +68,11 @@ public class TestNGDataDriver {
|
|||
*/
|
||||
private Time time = new Time();
|
||||
|
||||
/**
|
||||
* 用于存储最大列的元素个数
|
||||
*/
|
||||
private int maxColumnSize = -1;
|
||||
|
||||
/**
|
||||
* 构造类，并将默认的数据驱动函数（{@link Functions}）加载至类中
|
||||
*/
|
||||
|
|
@ -124,11 +130,14 @@ public class TestNGDataDriver {
|
|||
public void addDataDriver(File dataFile, String pattern, boolean isFirstTitle) throws IOException {
|
||||
//读取文件内的数据
|
||||
ListFileRead data = new ListFileRead(dataFile, pattern);
|
||||
//由于数据按照列的形式读取，故此处将数据转置
|
||||
data.tableTransposition();
|
||||
//获取当前标题行的总数
|
||||
int titleIndex = titleList.size();
|
||||
|
||||
//若设定首行为标题行，则根据首行元素读取标题
|
||||
//若设定首行为标题行，则根据首行元素读取标题；若首行非标题行，则以列数进行命名
|
||||
if (isFirstTitle) {
|
||||
//将读取到的数据转置，得到以行的形式存储数据
|
||||
data.tableTransposition();
|
||||
//此时，返回data中第一列元素即为标题行的元素
|
||||
//定位到首行
|
||||
List<String> dataTitleList = data.getColumn(0);
|
||||
|
||||
|
|
@ -148,22 +157,44 @@ public class TestNGDataDriver {
|
|||
titleList.add(title);
|
||||
}
|
||||
}
|
||||
//存储标题后再重新转置，使data以列的形式进行返回
|
||||
data.tableTransposition();
|
||||
} else {
|
||||
//若首行不是标题行，则获取data的最大列，按照“元素列 + 行号”的形式进行命名
|
||||
for (int index = 0; index < data.getMaxColumnNumber(); index++) {
|
||||
titleList.add("元素列" + (titleIndex + index));
|
||||
}
|
||||
}
|
||||
|
||||
//遍历所有行，读取所有的数据
|
||||
for (int rowIndex = isFirstTitle ? 1 : 0; rowIndex < data.getMaxColumnNumber(); rowIndex++) {
|
||||
addFileRowData(isFirstTitle ? rowIndex - 1 : rowIndex, data.getColumn(rowIndex));
|
||||
for (int columnIndex = 0; columnIndex < data.getMaxColumnNumber(); columnIndex++) {
|
||||
//将List<String>转换为List<Object>
|
||||
List<Object> objectList = new ArrayList<>();
|
||||
objectList.addAll(data.getColumn(columnIndex, isFirstTitle ? 1 : 0, data.getCoulumnSize(columnIndex)));
|
||||
System.out.println(objectList);
|
||||
//存储数据
|
||||
addData(titleList.get(columnIndex + titleIndex), objectList);
|
||||
}
|
||||
}
|
||||
|
||||
public void addData(String title, Object...objects) {
|
||||
//根据列名是否存在，调用相应的添加元素的方法
|
||||
if (titleList.indexOf(title) > -1) {
|
||||
addOriginalColumn(titleList.indexOf(title), objects);
|
||||
} else {
|
||||
titleList.add(title);
|
||||
addNewColumn(objects);
|
||||
/**
|
||||
* 用于在非文件中添加数据驱动
|
||||
* @param title 数据列标题
|
||||
* @param objects 相应的元素
|
||||
*/
|
||||
public void addDataDriver(String title, Object...objects) {
|
||||
//判断title是否为空，若为空，则将其改为默认命名
|
||||
if (title == null || title.isEmpty()) {
|
||||
title = "数据" + (1 + columnSize());
|
||||
}
|
||||
|
||||
//判断列表是否存在，若不存在，则重新添加标题
|
||||
if (titleList.indexOf(title) < 0) {
|
||||
titleList.add(title);
|
||||
}
|
||||
|
||||
//添加数据
|
||||
addData(title, Arrays.asList(objects));
|
||||
}
|
||||
|
||||
/**
|
||||
|
|
@ -179,13 +210,28 @@ public class TestNGDataDriver {
|
|||
* 一个{@link Data}类对象，通过该对象，对获取到的所有数据进行返回，以简化在编写脚本时
|
||||
* 的传参
|
||||
*
|
||||
* @return TestNG数据驱动
|
||||
* @return TestNG形式的数据驱动
|
||||
*/
|
||||
public Object[][] getDataDriver() {
|
||||
Object[][] objs = new Object[dataList.size()][1];
|
||||
//构造Object[][]
|
||||
Object[][] objs = new Object[maxColumnSize][1];
|
||||
|
||||
for (int i = 0; i < dataList.size(); i++) {
|
||||
objs[i][0] = dataList.get(i);
|
||||
//遍历所有的行，按照最大行添加数据
|
||||
for (int i = 0; i < maxColumnSize; i++) {
|
||||
//由于在lambda中不能直接引用变量i，故需要编写一个封装类
|
||||
AtomicInteger index = new AtomicInteger(i);
|
||||
//遍历标题列表，按照标题列表的顺序，写入到data中
|
||||
Data data = new Data();
|
||||
titleList.forEach(title -> {
|
||||
//由于列表长度不一致，导致存储时超出数组的元素无法读取，若抛出异常，则存储空串
|
||||
try {
|
||||
data.addData(dataMap.get(title).get(index.get()));
|
||||
} catch (Exception e) {
|
||||
data.addData("");
|
||||
}
|
||||
});
|
||||
//存储数据
|
||||
objs[i][0] = data;
|
||||
}
|
||||
|
||||
return objs;
|
||||
|
|
@ -196,117 +242,51 @@ public class TestNGDataDriver {
|
|||
* @return 数据驱动的总行数
|
||||
*/
|
||||
public int getRowSize() {
|
||||
return dataList.size();
|
||||
//最大列的元素个数即为最大的行数
|
||||
return maxColumnSize;
|
||||
}
|
||||
|
||||
/**
|
||||
* 用于根据列名称返回指定列的元素个数
|
||||
* 用于根据列名称返回指定列的元素个数，若标题不存在，则返回0
|
||||
* @param title 列名称
|
||||
* @return 列元素个数
|
||||
*/
|
||||
public int getColumnSize(String title) {
|
||||
int index = titleList.indexOf(title);
|
||||
if (index > -1) {
|
||||
return getColumnSize(index);
|
||||
if (dataMap.containsKey(title)) {
|
||||
return dataMap.get(title).size();
|
||||
} else {
|
||||
return 0;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 用于根据列下标返回指定列的元素个数
|
||||
* 用于根据列下标返回指定列的元素个数，若输入的参数不存在，则返回0
|
||||
* @param index 列下标
|
||||
* @return 列元素个数
|
||||
*/
|
||||
public int getColumnSize(int index) {
|
||||
AtomicInteger rowIndex = new AtomicInteger(0);
|
||||
//遍历所有行的指定列元素
|
||||
dataList.forEach(data -> {
|
||||
//判断当前元素是否为空，若为空，则结束循环，否则，其rowIndex+1
|
||||
if (data.getOriginalString(index).isEmpty()) {
|
||||
return;
|
||||
} else {
|
||||
rowIndex.addAndGet(1);
|
||||
}
|
||||
});
|
||||
|
||||
return rowIndex.get();
|
||||
try {
|
||||
return getColumnSize(titleList.get(index));
|
||||
} catch (Exception e) {
|
||||
return 0;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 用于将一行的数据转为Data类对象
|
||||
* 将一行数据添加至相应的dataMap中
|
||||
* @param rowDataList 一行数据
|
||||
* @return 返回一行数据的Data类对象
|
||||
*/
|
||||
private void addFileRowData(int rowInde, List<String> rowDataList) {
|
||||
//将所有的空元素替换为“${ }”，以表示该行存在元素，只是为空
|
||||
rowDataList.replaceAll(text -> {
|
||||
return text.isEmpty() ? "${ }" : text;
|
||||
});
|
||||
|
||||
//当rowInde大于dataList的总列数时，则表示未对该行进行存储，此时存在两种情况：
|
||||
//1.第一次文件，此时dataList并无数据，调用方法时为初次添加数据，则按照正常方式添加
|
||||
//2.非第一次读取文件，此时dataList已存在上次读取的数据，当写入的数据行数大于存储的行数
|
||||
//时，直接添加数据会造成数据向左移，则需要读取原始列表的内容，将原来没有的数据补全后再
|
||||
//添加数据
|
||||
if (rowInde > dataList.size() - 1) {
|
||||
Data data = new Data();
|
||||
//判断rowDataList元素与上一行元素个数一致，若一致，则表示当前为第一次读取
|
||||
if (dataList.size() != 0 && dataList.get(rowInde - 1).dataList.size() != rowDataList.size()) {
|
||||
//若上一行的元素个数与rowDataList的元素个数不一致，则此时是非第一次读取文件，rowDataList存在数据，则
|
||||
//将rowDataList之前的列都置为空串
|
||||
int oldListSize = dataList.get(rowInde - 1).dataList.size();
|
||||
for (int index = 0; index < oldListSize - rowDataList.size(); index++) {
|
||||
data.addData("${ }");
|
||||
}
|
||||
}
|
||||
|
||||
//存储数据
|
||||
rowDataList.forEach(data :: addData);
|
||||
dataList.add(data);
|
||||
} else {
|
||||
rowDataList.forEach(dataList.get(rowInde) :: addData);
|
||||
private void addData(String title, List<Object> columnDataList) {
|
||||
//判断标题是否存在，若不存在，则进行构造
|
||||
if (!dataMap.containsKey(title)) {
|
||||
dataMap.put(title, new ArrayList<Object>());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 存储新的列元素
|
||||
* @param objects 元素组
|
||||
*/
|
||||
private void addNewColumn(Object...objects) {
|
||||
//获取当前数据驱动中的行数
|
||||
int maxRowSize = getRowSize();
|
||||
//获取当前行元素与需要添加的元素个数之间的差值
|
||||
int diff = objects.length - maxRowSize;
|
||||
//对元素进行存储
|
||||
dataMap.get(title).addAll(columnDataList);
|
||||
|
||||
//为保证数据统一，按照当前元素最大行进行存储，若objects中的元素个数不足时，使用空函数对其占位，保证列表的统一
|
||||
for (int index = 0; index < maxRowSize; index++) {
|
||||
try {
|
||||
dataList.get(index).addData(objects[index].toString());
|
||||
}catch (Exception e) {
|
||||
dataList.get(index).addData("${ }");
|
||||
}
|
||||
}
|
||||
|
||||
//判断差值是否大于0，若大于0，则按照已存在的行进行存储
|
||||
if (diff > 0) {
|
||||
//存储剩余的元素
|
||||
Object[] obj = new Object[diff];
|
||||
for (int index = 0; index < diff; index++) {
|
||||
obj[index] = objects[maxRowSize + index];
|
||||
}
|
||||
|
||||
addOriginalColumn(titleList.size() - 1, obj);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 在已存在的列中添加元素
|
||||
* @param index 列下标
|
||||
* @param objects 元素组
|
||||
*/
|
||||
private void addOriginalColumn(int index, Object...objects) {
|
||||
if ()
|
||||
//判断该列的元素个数是否大于最大列元素个数，若大于，则将其个数替换为最大元素个数
|
||||
int size = dataMap.get(title).size();
|
||||
maxColumnSize = size > maxColumnSize ? size : maxColumnSize;
|
||||
}
|
||||
|
||||
/**
|
||||
|
|
@ -339,7 +319,7 @@ public class TestNGDataDriver {
|
|||
/**
|
||||
* 用于存储数据
|
||||
*/
|
||||
private ArrayList<String> dataList = new ArrayList<>();
|
||||
private ArrayList<Object> dataList = new ArrayList<>();
|
||||
|
||||
/**
|
||||
* 用于根据列名称，以字符串的形式返回数据
|
||||
|
|
@ -357,7 +337,7 @@ public class TestNGDataDriver {
|
|||
* @return 字符串形式的数据
|
||||
*/
|
||||
public String getString(int index) {
|
||||
return index < dataList.size() ? disposeContent(dataList.get(index)) : "";
|
||||
return index < dataList.size() ? disposeContent(dataList.get(index).toString()) : "";
|
||||
// return index < dataList.size() ? dataList.get(index) : ""; //调试使用，输出存储的原始内容
|
||||
}
|
||||
|
||||
|
|
@ -475,11 +455,29 @@ public class TestNGDataDriver {
|
|||
return new Time(getString(listName));
|
||||
}
|
||||
|
||||
/**
|
||||
* 用于根据列名称，以{@link Object}的形式对数据进行返回
|
||||
* @param listName 列表名称
|
||||
* @return {@link Object}类型的数据
|
||||
*/
|
||||
public Object getObject(String listName) {
|
||||
return dataList.get(titleList.indexOf(listName));
|
||||
}
|
||||
|
||||
/**
|
||||
* 用于根据列名称，以{@link Object}的形式对数据进行返回
|
||||
* @param listName 列表名称
|
||||
* @return {@link Object}类型的数据
|
||||
*/
|
||||
public Object getObject(int index) {
|
||||
return dataList.get(index);
|
||||
}
|
||||
|
||||
/**
|
||||
* 用于添加元素
|
||||
* @param content 元素内容
|
||||
*/
|
||||
private void addData(String content) {
|
||||
private void addData(Object content) {
|
||||
dataList.add(content);
|
||||
}
|
||||
|
||||
|
|
@ -489,6 +487,7 @@ public class TestNGDataDriver {
|
|||
* @return 处理后的内容
|
||||
*/
|
||||
private String disposeContent(String content) {
|
||||
//TODO 想办法在处理时输出Object类
|
||||
//判断传入的内容是否包含公式起始符号，若不包含，则返回原串
|
||||
if (content.indexOf(FUNCTION_SIGN_START) < 0) {
|
||||
return content;
|
||||
|
|
@ -543,14 +542,5 @@ public class TestNGDataDriver {
|
|||
//若内容不与任何正则匹配，则返回原始内容
|
||||
return null;
|
||||
}
|
||||
|
||||
/**
|
||||
* 用于返回index对应列中存储的元素的原始内容，若列不存在，或无元素，则返回空
|
||||
* @param index 列下标
|
||||
* @return 列中的内容
|
||||
*/
|
||||
private String getOriginalString(int index) {
|
||||
return index < dataList.size() ? dataList.get(index) : "";
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
|
|||
|
|
@ -1,4 +1,4 @@
|
|||
package pres.auxiliary.tool.readfile;
|
||||
package pres.auxiliary.work.selenium.datadriven;
|
||||
|
||||
import java.io.File;
|
||||
import java.io.IOException;
|
||||
|
|
|
|||
|
|
@ -79,4 +79,13 @@ public class TestNGDataDriverTest {
|
|||
testNGData.addFunction(Functions.getTime());
|
||||
testNGData.addDataDriver(functionFile, "Sheet1", true);
|
||||
}
|
||||
|
||||
/**
|
||||
* 测试{@link TestNGDataDriver#addDataDriver(String, Object...)}方法
|
||||
*/
|
||||
@Test
|
||||
public void addDataDriverTest_String_Objects() {
|
||||
testNGData.addDataDriver("123", "1", "2", "3");
|
||||
testNGData.addDataDriver("docx3", "1", "2", "3");
|
||||
}
|
||||
}
|
||||
|
|
|
|||
|
|
@ -3,6 +3,7 @@
|
|||
<templet>
|
||||
<!-- 窗体模板 -->
|
||||
<xpath id='1'>//iframe[contains(@src,'${src}')]</xpath>
|
||||
<xpath id='3'>//lable[contains(text(), '${condition}')]/../span/input[1]</xpath>
|
||||
</templet>
|
||||
|
||||
<iframe name='主窗体'>
|
||||
|
|
@ -16,6 +17,10 @@
|
|||
<element name='单位名称'>
|
||||
<xpath is_use='true'>//*[@id="listBox"]/li[1]/div[1]/p/span[1]</xpath>
|
||||
</element>
|
||||
|
||||
<element name='搜索条件'>
|
||||
<xpath is_use='true' temp_id='3' />
|
||||
</element>
|
||||
</iframe>
|
||||
</iframe>
|
||||
</project>
|
||||
|
|
@ -3,6 +3,9 @@
|
|||
<templet>
|
||||
<xpath id='1'>//XXX模板控件1[@X='${name}']/div/div[@${att}='${id}']/input</xpath>
|
||||
<css id='2'>http body ${tagName}</css>
|
||||
<xpath id='3'>//XXX模板控件1[@X='${src}']/div[@name='${name}']</xpath>
|
||||
<xpath id='4'>//XXX模板控件1[@X='${src}']/div[@name='${name}']/div[@is='${str1}' and text()='${str1}']</xpath>
|
||||
<xpath id='5'>//XXX模板控件1[@X='${src}']/div[@name='${name}']/div[@is='${str1}' and text()='${src}']/span[text()='${str2}']/span[id='${aaaa}']</xpath>
|
||||
</templet>
|
||||
|
||||
<element name='XX控件1'>
|
||||
|
|
@ -62,6 +65,25 @@
|
|||
<element name='XX控件11'>
|
||||
<xpath is_use='true' temp_id='1' id='Test' att='src'/>
|
||||
<css is_use='true' temp_id='2' tagName='div'/>
|
||||
<xpath is_use='true' temp_id='3' />
|
||||
</element>
|
||||
|
||||
<element name='XX控件12'>
|
||||
<xpath is_use='true' temp_id='3' />
|
||||
</element>
|
||||
</iframe>
|
||||
|
||||
<iframe name='窗体3' >
|
||||
<xpath is_use='true'>//窗体3[@X='${ccc}']</xpath>
|
||||
|
||||
<element name='XX控件13'>
|
||||
<xpath is_use='true' temp_id='3' />
|
||||
</element>
|
||||
<element name='XX控件14'>
|
||||
<xpath is_use='true' temp_id='4' />
|
||||
</element>
|
||||
<element name='XX控件15'>
|
||||
<xpath is_use='true' temp_id='5' />
|
||||
</element>
|
||||
</iframe>
|
||||
</project>
|
||||
Loading…
Reference in New Issue