@Step("query user")
public void queryUser() {

    String url = "http://119.29.99.89:8080/query";

    String result = sendGetRequest(url);

    //用到了FastJSON来处理返回的Json数据
    JSONArray users = JSON.parseArray(result);
    for (int i = 0; i < users.size(); i++) {
        UserModel user = JSON.parseObject(users.get(i).toString(), UserModel.class);
        Gauge.writeMessage(user.toString());
    }
    assertTrue(users != null && !users.isEmpty());
}

//发送一个GET请求
private String sendGetRequest(String url) {
    String result = "";
    BufferedReader in = null;

    try {
        URL realUrl = new URL(url);
        URLConnection connection = realUrl.openConnection();

        connection.connect();

        in = new BufferedReader(new InputStreamReader(connection.getInputStream()));
        String line;
        while ((line = in.readLine()) != null) {
            result += line;
        }
    } catch (MalformedURLException e) {
        e.printStackTrace();
    } catch (IOException e) {
        e.printStackTrace();
    } finally {
        try {
            if (in != null) {
                in.close();
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
    return result;
}

@Step("add users from table <table>")
public void addUser(Table userTable) {

    boolean result = true;
    String url = "http://119.29.99.89:8080/add";

    for (TableRow row : userTable.getTableRows()) {
        UserModel userModel = new UserModel();
        userModel.setUsername(row.getCell("username"));
        userModel.setSex(row.getCell("sex"));
        userModel.setAge(Integer.valueOf(row.getCell("age")));
        userModel.setSchool(row.getCell("school"));
        userModel.setMajor(row.getCell("major"));
        userModel.setAddress(row.getCell("address"));

        String requestString = JSON.toJSONString(userModel);

        String response = sendPostRequest(requestString, url);
        JSONObject responseMap = JSON.parseObject(response);
        result = result && responseMap.get("STATE").equals("SUCCESSED");
    }
    assertTrue(result);
}

//发送POST请求
 private String sendPostRequest(String requestString, String url) {
    PrintWriter out = null;
    BufferedReader in = null;
    String result = "";
    try {
        URL realUrl = new URL(url);
        URLConnection conn = realUrl.openConnection();

        conn.setRequestProperty("content-type", "application/json");
        conn.setDoOutput(true);
        conn.setDoInput(true);

        out = new PrintWriter(conn.getOutputStream());
        out.print(requestString);
        out.flush();

        in = new BufferedReader(new InputStreamReader(conn.getInputStream()));
        String line;
        while ((line = in.readLine()) != null) {
            result += line;
        }
    } catch (Exception e) {
        e.printStackTrace();
    } finally {
        try {
            if (out != null) {
                out.close();
            }
            if (in != null) {
                in.close();
            }
        } catch (IOException ex) {
            ex.printStackTrace();
        }
    }
    return result;
}

public class UserModel {
    private int id;
    private String username;
    private String sex;
    private int age;
    private String school;
    private String major;
    private String address;

    public UserModel() {
        super();
    }

    public UserModel(int id, String username, String sex, int age, String school, String major, String address) {
        super();
        this.id = id;
        this.username = username;
        this.sex = sex;
        this.age = age;
        this.school = school;
        this.major = major;
        this.address = address;
    }

    public int getId() {
        return id;
    }

    public void setId(int id) {
        this.id = id;
    }

    public String getUsername() {
        return username;
    }

    public void setUsername(String username) {
        this.username = username;
    }

    public String getSex() {
        return sex;
    }

    public void setSex(String sex) {
        this.sex = sex;
    }

    public int getAge() {
        return age;
    }

    public void setAge(int age) {
        this.age = age;
    }

    public String getSchool() {
        return school;
    }

    public void setSchool(String school) {
        this.school = school;
    }

    public String getMajor() {
        return major;
    }

    public void setMajor(String major) {
        this.major = major;
    }

    public String getAddress() {
        return address;
    }

    public void setAddress(String address) {
        this.address = address;
    }

    @Override
    public String toString() {
        return "UserModel{" +
                "id=" + id +
                ", username='" + username + '\'' +
                ", sex='" + sex + '\'' +
                ", age=" + age +
                ", school='" + school + '\'' +
                ", major='" + major + '\'' +
                ", address='" + address + '\'' +
                '}';
    }
}