AccountTest.java 3.6 KB
package com.xkl.uspih;

import com.xkl.Constants;
import com.xkl.EncodeTools;
import com.xkl.HttpTools;
import org.json.JSONObject;
import org.junit.Test;

import java.util.HashMap;
import java.util.Map;

/**
 * USPIH login and logout
 * zhaoyue
 * 2017-1-14
 */

public class AccountTest {
    private static final String URL_LOGIN = Constants.URL_PREFIX + "/uspih/account";

    /**
     * @param user user account
     * @param pass user password
     * @return
     */
    public static String loginAndGetToken(String user, String pass) {
        Map<String, String> params = new HashMap<String, String>();
        params.put("account", user);
        params.put("password", EncodeTools.encode("MD5", pass)); // encode password with MD5 algorithm
        params.put("t", HttpTools.getNow());
        String response = HttpTools.requestByMap(URL_LOGIN, "POST", params);
        System.out.println("login success:" + response);
        JSONObject jsonObject = new JSONObject(response);
        JSONObject content = jsonObject.getJSONObject("content");
        return content.getString("userId") + "_" + content.getString("token");
    }

    /**
     * @param user   user account
     * @param pass   user password
     * @param newpwd new user password
     * @return
     */
    public static String loginModPwdAndGetToken(String user, String pass, String newpwd) {
        Map<String, String> params = new HashMap<String, String>();
        params.put("account", user);
        params.put("password", EncodeTools.encode("MD5", pass)); // encode password with MD5 algorithm
        params.put("newpwd", EncodeTools.encode("MD5", newpwd)); // encode password with MD5 algorithm
        params.put("t", HttpTools.getNow());
        String response = HttpTools.requestByMap(URL_LOGIN, "PUT", params);
        System.out.println("login success:" + response);
        JSONObject jsonObject = new JSONObject(response);
        JSONObject content = jsonObject.getJSONObject("content");
        return content.getString("userId") + "_" + content.getString("token");
    }

    /**
     * login success test
     */
    @Test
    public void testLoginSuccess() {
        System.out.println(loginAndGetToken(Data.ADMIN_ACCOUNT, Data.ADMIN_PWD));
    }

    /**
     * login fail test
     */
    @Test
    public void testLoginFail() {
        System.out.println(loginAndGetToken(Data.ADMIN_ACCOUNT + "ppppp", Data.ADMIN_PWD));
    }

    /**
     * login and modify password success test
     */
    @Test
    public void testLoginModPwdSuccess() {
        System.out.println(loginModPwdAndGetToken(Data.ADMIN_ACCOUNT, Data.ADMIN_PWD, Data.ADMIN_PWD));
    }

    /**
     * login and modify password fail test
     */
    @Test
    public void testLoginModPwdFail() {
        System.out.println(loginModPwdAndGetToken(Data.ADMIN_ACCOUNT, Data.ADMIN_PWD + "pppppp", Data.ADMIN_PWD));
    }


    @Test
    public void testLogout() {
        String token = loginAndGetToken(Data.ADMIN_ACCOUNT, Data.ADMIN_PWD);
        Map<String, String> params = new HashMap<String, String>();
        params.put("t", HttpTools.getNow());
        String response = HttpTools.requestByMapWithToken(URL_LOGIN, "DELETE", params, token);
        System.out.println("logout success:" + response);
    }

    @Test
    public void testLogoutFail() {
        String token = loginAndGetToken(Data.ADMIN_ACCOUNT, Data.ADMIN_PWD);
        Map<String, String> params = new HashMap<String, String>();
        params.put("t", HttpTools.getNow());
        String response = HttpTools.requestByMapWithToken(URL_LOGIN, "DELETE", params, token + "PPPPPP");
        System.out.println("logout success:" + response);
    }
}