HttpClient +Testng(3)对返回结果进行处理：json格式化，方便验证结果

上次的运行结果可看出，http返回结果是个字符串类型，没有规律性。而正常情况下，我们是需要提取返回直接中的值和预期值进行比较，看返回结果是否正确。

所以我们对返回结果进行json格式化，这样就方便验证结果。

首先，还是修改FirstTest.java，如下所示：

import net.sf.json.JSONObject;
import org.apache.http.HttpEntity;
import org.apache.http.HttpResponse;
import org.apache.http.NameValuePair;
import org.apache.http.client.HttpClient;
import org.apache.http.client.entity.UrlEncodedFormEntity;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.impl.client.DefaultHttpClient;
import org.apache.http.message.BasicNameValuePair;
import org.apache.http.util.EntityUtils;
import org.testng.Assert;
import org.testng.annotations.DataProvider;
import org.testng.annotations.Test;

import java.io.File;
import java.util.ArrayList;
import java.util.List;

public class FirstTest {

    private HttpClient httpClient = new DefaultHttpClient();
    private HttpPost httppost;
    private HttpResponse response;
    private HttpEntity entity;
    private String postResult;
    private JSONObject jsonResult;

    @DataProvider(name = "LoginData")
    public Object[][] login() {
        // 测试数据准备
        String file = "." + File.separator + "TestData" + File.separator + "LoginTest.xls";
        Object[][] records ;
        records = ExcelReader.getExpectationData(file, "login");
        return records ;
    }

    @Test(dataProvider = "LoginData")
    public void loginJDTest(String caseDescription,String loginURL, String uuid,String eid,String fp,String _t,
                            String loginType,String loginname,String nloginpwd, String chkRememberMe,
                            String authcode,String pubKey,String sa_token,String seqSid) {

        System.out.println("=====" + caseDescription + "=====");
        //创建一个httppost请求
        httppost = new HttpPost(loginURL);

        //创建Post请求参数
        List<NameValuePair> formparams1 = new ArrayList<NameValuePair>();
        formparams1.add(new BasicNameValuePair("uuid",uuid));
        formparams1.add(new BasicNameValuePair("eid",eid));
        formparams1.add(new BasicNameValuePair("fp",fp));
        formparams1.add(new BasicNameValuePair("_t",_t));
        formparams1.add(new BasicNameValuePair("loginType",loginType));
        formparams1.add(new BasicNameValuePair("loginname",loginname));
        formparams1.add(new BasicNameValuePair("nloginpwd",nloginpwd));
        formparams1.add(new BasicNameValuePair("chkRememberMe",chkRememberMe));
        formparams1.add(new BasicNameValuePair("authcode",authcode));
        formparams1.add(new BasicNameValuePair("pubKey",pubKey));
        formparams1.add(new BasicNameValuePair("sa_token",sa_token));
        formparams1.add(new BasicNameValuePair("seqSid",seqSid));

        try {
            httppost.setEntity(new UrlEncodedFormEntity(formparams1,"UTF-8"));
            response = httpClient.execute(httppost);
            entity = response.getEntity();
            // 在这里可以用Jsoup之类的工具对返回结果进行分析，以判断创建是否成功
            postResult = EntityUtils.toString(entity, "UTF-8");

            //System.out.println("查看登录接口请求返回的结果：" + postResult);

            //由于京东这个接口返回的请求头尾是“（）”，所以需要去掉，才可以转化成Json格式
            if (postResult.startsWith("(")){
                postResult = postResult.substring(1,postResult.length());
            }
            if (postResult.endsWith(")")){
                postResult = postResult.substring(0,postResult.length()-1);
            }
            //转换成json格式
            jsonResult = JSONObject.fromObject(postResult); //该结果可用来提取任何想要的值
            //Assert.assertEquals(jsonResult.getString("_t"),"_ntiyYur"); //assert验证结果，注意这个验证结果会失败,
                                                                          // 因为京东对登录结果做了限制，不让我们这样做接口测试，但是我们可以当做例子
            JsonFormatTool tool = new JsonFormatTool();
            String Result = tool.formatJson(jsonResult.toString()); //自定义控制台打印结果的格式
            System.out.println("请求后返回的结果：" + Result);
        } catch (Exception e) {
            e.printStackTrace();
        }

        httppost.releaseConnection();
    }
}

上面我们使用JSONObject来对返回结果进行格式化，但是由于JSONObject本身返回的结果在控制台打印出来的内容页不利于我们肉眼查看，所以这里我们自己写了一个JSON转化格式的工具。

package utl;

public class JsonFormatTool {

    /**
     * 单位缩进字符串。
     */
    private static String SPACE = "   ";

    /**
     * 返回格式化JSON字符串。
     *
     * @param json 未格式化的JSON字符串。
     * @return 格式化的JSON字符串。
     */
    public String formatJson(String json)
    {
        StringBuffer result = new StringBuffer();

        int length = json.length();
        int number = 0;
        char key = 0;

        //遍历输入字符串。
        for (int i = 0; i < length; i++)
        {
            //1、获取当前字符。
            key = json.charAt(i);

            //2、如果当前字符是前方括号、前花括号做如下处理：
            if((key == '[') || (key == '{') )
            {
                //（1）如果前面还有字符，并且字符为“：”，打印：换行和缩进字符字符串。
                if((i - 1 > 0) && (json.charAt(i - 1) == ':'))
                {
                    result.append('\n');
                    result.append(indent(number));
                }

                //（2）打印：当前字符。
                result.append(key);

                //（3）前方括号、前花括号，的后面必须换行。打印：换行。
                result.append('\n');

                //（4）每出现一次前方括号、前花括号；缩进次数增加一次。打印：新行缩进。
                number++;
                result.append(indent(number));

                //（5）进行下一次循环。
                continue;
            }

            //3、如果当前字符是后方括号、后花括号做如下处理：
            if((key == ']') || (key == '}') )
            {
                //（1）后方括号、后花括号，的前面必须换行。打印：换行。
                result.append('\n');

                //（2）每出现一次后方括号、后花括号；缩进次数减少一次。打印：缩进。
                number--;
                result.append(indent(number));

                //（3）打印：当前字符。
                result.append(key);

                //（4）如果当前字符后面还有字符，并且字符不为“，”，打印：换行。
                if(((i + 1) < length) && (json.charAt(i + 1) != ','))
                {
                    result.append('\n');
                }

                //（5）继续下一次循环。
                continue;
            }

            //4、如果当前字符是逗号。逗号后面换行，并缩进，不改变缩进次数。
            if((key == ','))
            {
                result.append(key);
                result.append('\n');
                result.append(indent(number));
                continue;
            }

            //5、打印：当前字符。
            result.append(key);
        }

        return result.toString();
    }

    /**
     * 返回指定次数的缩进字符串。每一次缩进三个空格，即SPACE。
     *
     * @param number 缩进次数。
     * @return 指定缩进次数的字符串。
     */
    private String indent(int number)
    {
        StringBuffer result = new StringBuffer();
        for(int i = 0; i < number; i++)
        {
            result.append(SPACE);
        }
        return result.toString();
    }
}

则HttpClient接口测试一个小框架算完成了。

整个项目的代码也已上传到CSDN：

HTTPClient代码