android UiAutomator如何獲取短信驗證碼

本人在測試註冊等功能的時候需要自動把驗證碼填寫到註冊框裏，雖然現在很多手機已經提供了直接複製驗證碼的功能，但我還是想了一個辦法從一段文本中獲取驗證碼。

public int findCode(String message) {
Pattern r = Pattern.compile("[0-9]{6}");
Matcher m = r.matcher(message);
if (m.find()) {
outputNotable("匹配成功");
for(int i=0;i<=m.groupCount();i++){
System.out.println(m.group(i));
}
}
return changeStringToInt(m.group(0));
}
//把string類型轉化爲int
public int changeStringToInt (String text) {
return Integer.parseInt(text);
}

有些驗證碼不是6位數字的，或者短信裏有兩個或者更多連續6爲數字的，估計還得再改改。