package com.sandu.ximon.admin.base;

import org.springframework.http.MediaType;
import org.springframework.test.web.servlet.request.MockHttpServletRequestBuilder;
import org.springframework.test.web.servlet.request.MockMvcRequestBuilders;

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

public class BaseMockMvcRequestBuilders {
    //全局的请求头数据
    public final static Map<String, Object> headers = new HashMap<>();

    //配置请求头的数据（包含token之类）
    static {
        headers.put("Authorization", "Bearer eyJhbGciOiJIUzUxMiJ9.eyJzdWIiOiJhZG1pbiIsImF1dGgiOiJhZG1pbiIsImNyZWQiOjAsImV4cCI6MTY0MzIxNTgwN30.b9hHNpLDe3XThtCr8alFl4DZlfWM1Rx2o_AvFnutUQEqnyuh0JDlYigcev8zk_tKuSPDpXtWVsdKKQqlKLRb3g");
    }

    /**
     * post请求
     *
     * @param url
     * @return
     */
    public static MockHttpServletRequestBuilder post(String url) {
        MockHttpServletRequestBuilder temp = MockMvcRequestBuilders.post(url).contentType(MediaType.APPLICATION_JSON)//请求的数据格式
                .characterEncoding("UTF-8");//设置请求的编码
        //设置请求头数据
        for (String key : headers.keySet()) {
            temp.header(key, headers.get(key));
        }
        return temp;
    }

    /**
     * get请求
     *
     * @param url
     * @return
     */
    public static MockHttpServletRequestBuilder get(String url) {
        MockHttpServletRequestBuilder temp = MockMvcRequestBuilders.get(url).contentType(MediaType.APPLICATION_JSON)//请求的数据格式
                .characterEncoding("UTF-8");//设置请求的编码
        //设置请求头数据
        for (String key : headers.keySet()) {
            temp.header(key, headers.get(key));
        }
        return temp;
    }
}