diff --git a/src/main/Main.java b/src/main/Main.java index 0cf764d..325dd80 100644 --- a/src/main/Main.java +++ b/src/main/Main.java @@ -17,8 +17,8 @@ formParam.add(new RestParam("userPass", "abc")); api.setupRestTest(Method.GET, queryParam, formParam, pathParam); api.doRestTest(); - for (int i = 0; i < RestApi.ACCOUNT_NUM; i++) { - System.out.println(api.getResult(i)); + for (String result : api.getResults()) { + System.out.println(result); } System.out.println(api.getJudgeResult()); } diff --git a/src/main/RestApi.java b/src/main/RestApi.java index ebe01f6..2cd0430 100644 --- a/src/main/RestApi.java +++ b/src/main/RestApi.java @@ -10,19 +10,19 @@ private ArrayList formParam; private ArrayList pathParam; private String judgeResult; - public static final int ACCOUNT_NUM = 2; + public static final int ACCOUNT_NUM = 5; public void setAccount(String url) { for (int i = 0; i < ACCOUNT_NUM; i++) accounts.add(new RestAccount(url)); } - public RestAccount getAccount(int num) { - return accounts.get(num); + public ArrayList getAccounts() { + return accounts; } - public String getResult(int num) { - return results.get(num); + public ArrayList getResults() { + return results; } public String getJudgeResult() { @@ -59,7 +59,9 @@ account.doAnything(); results.add(account.doReceive()); } - if (results.get(0).equals(results.get(1))) { + @SuppressWarnings("unchecked") + ArrayList tmplist = (ArrayList) results.clone(); + if (results.equals(tmplist)) { judgeResult = "GET"; } else {