SpringBoot-RestTemplate测试Controller

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/zhangminemail/article/details/83189106

1、功能测试类

package com.imooc.controller;

import java.io.IOException;
import java.math.BigDecimal;
import java.util.ArrayList;
import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.junit.Before;
import org.junit.FixMethodOrder;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.MethodSorters;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.boot.test.context.SpringBootTest.WebEnvironment;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageImpl;
import org.springframework.http.client.ClientHttpResponse;
import org.springframework.test.context.junit4.SpringRunner;
import org.springframework.util.Assert;
import org.springframework.web.client.ResponseErrorHandler;
import org.springframework.web.client.RestTemplate;

import com.imooc.entity.Product;
import com.imooc.entity.enums.ProductStatus;
import com.imooc.util.RestUtil;

@RunWith(SpringRunner.class)
@SpringBootTest(webEnvironment=WebEnvironment.RANDOM_PORT)
@FixMethodOrder(MethodSorters.NAME_ASCENDING) // case执行顺序
public class ProductControllerTest {

//	@Autowired
//	private TestRestTemplate rest;
	
	private static RestTemplate rest = new RestTemplate();
	
	@Value("http://localhost:${local.server.port}/products")
	private String baseUrl;
	
	// 正常数据
	private static List<Product> normals = new ArrayList<>();
	
	private static List<Product> exceptions = new ArrayList<>();
	
	@Before
	public void init(){
		
		Product p1 = new Product("T0001", "零活宝1号", ProductStatus.AUDITING.getCode(),
				BigDecimal.valueOf(10), BigDecimal.valueOf(1), 7, 
				BigDecimal.valueOf(3.42), "memo", new Date(), new Date(), "zemel", "zemel");
		Product p2 = new Product("T0002", "零活宝2号", ProductStatus.AUDITING.getCode(),
				BigDecimal.valueOf(10), BigDecimal.valueOf(0), 6, 
				BigDecimal.valueOf(3.42), "memo", new Date(), new Date(), "zemel", "zemel");
		Product p3 = new Product("T0003", "零活宝3号", ProductStatus.AUDITING.getCode(),
				BigDecimal.valueOf(100), BigDecimal.valueOf(10),3, 
				BigDecimal.valueOf(3.42), "memo", new Date(), new Date(), "zemel", "zemel");
		normals.add(p1);
		normals.add(p2);
		normals.add(p3);
		
		Product e1 = new Product(null, "零活宝1号", ProductStatus.AUDITING.getCode(),
				BigDecimal.valueOf(10), BigDecimal.valueOf(1), 7, 
				BigDecimal.valueOf(3.42), "memo", new Date(), new Date(), "zemel", "zemel");
		exceptions.add(e1);
		
		// 异常处理对象
		ResponseErrorHandler errorHandler = new ResponseErrorHandler() {
			
			@Override
			public boolean hasError(ClientHttpResponse response) throws IOException {
				
				return true;
			}
			
			@Override
			public void handleError(ClientHttpResponse response) throws IOException {
				// TODO Auto-generated method stub
				
			}
		};
		rest.setErrorHandler(errorHandler);
	}
	
	@Test
	public void testAddProduct() {
		normals.forEach(product -> {
			Product result = RestUtil.postJSON(rest, baseUrl, product, Product.class);
			Assert.notNull(result.getCreateAt(), "插入失败");
		});
	}
	
	@Test
	public void testAddProductException() {
		exceptions.forEach(product -> {
			Map<String, String> result = RestUtil.postJSON(rest, baseUrl, product, HashMap.class);
//			Assert.notNull(result.getCreateAt(), "插入失败");
			System.out.println(result);
			Assert.notNull(result.get("message").equals(product.getName()), "插入成功");
		});
	}
	

	@Test
	public void testFindOne() {
		normals.forEach(p->{
			
			Product result = rest.getForObject(baseUrl+"/"+p.getId(), Product.class);
			Assert.isTrue(result.getId().equals(p.getId()));
		});
		
		exceptions.forEach(p->{
			Product result = rest.getForObject(baseUrl+"/"+p.getId(), Product.class);
			Assert.isNull(result, "查询失败");
		
			
		});
	}
	
	@Test
	public void testQuery() {
		
//		Page<Product> page = rest.getForObject(baseUrl, "", Page.class);
		
		Map<String, Object> params = new HashMap<>();
		params.put("ids", "T0001,T0002");
//		Page<Product> page = RestUtil.postJSON(rest, baseUrl, params, Page.class);
		
		Map page = rest.getForObject(baseUrl, HashMap.class, params);
		System.out.println(page);
		System.out.println(page.get("pageable"));
		System.out.println(page.get("content"));
		Assert.notNull(page);
	}

}

2、工具类

package com.imooc.util;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.http.HttpEntity;
import org.springframework.http.HttpHeaders;
import org.springframework.http.MediaType;
import org.springframework.web.client.RestTemplate;

import java.util.Arrays;
import java.util.List;
import java.util.Map;


public class RestUtil {

    static Logger log = LoggerFactory.getLogger(RestUtil.class);

    /**
     * 发送post 请求
     *
     * @param restTemplate
     * @param url
     * @param param
     * @param responseType
     * @param <T>
     * @return
     */
    public static <T> T postJSON(RestTemplate restTemplate, String url, Object param, Class<T> responseType) {
        HttpEntity<String> formEntity = makePostJSONEntiry(param);
        T result = restTemplate.postForObject(url, formEntity, responseType);
        log.info("rest-post-json 响应信息:{}", JsonUtil.toJson(result));
        return result;
    }

    /**
     * 生成json形式的请求头
     *
     * @param param
     * @return
     */
    public static HttpEntity<String> makePostJSONEntiry(Object param) {
        HttpHeaders headers = new HttpHeaders();
        headers.setContentType(MediaType.APPLICATION_JSON_UTF8);
        headers.add("Accept", MediaType.APPLICATION_JSON_VALUE);
        HttpEntity<String> formEntity = new HttpEntity<String>(
                JsonUtil.toJson(param), headers);
        log.info("rest-post-json-请求参数:{}", formEntity.toString());
        return formEntity;
    }


    public static HttpEntity<String> makePostTextEntiry(Map<String, ? extends Object> param) {
        HttpHeaders headers = new HttpHeaders();
        headers.setContentType(MediaType.APPLICATION_FORM_URLENCODED);
        headers.add("Accept", MediaType.APPLICATION_JSON_VALUE);
        HttpEntity<String> formEntity = new HttpEntity<String>(
                makeGetParamContent(param), headers);
        log.info("rest-post-text-请求参数:{}", formEntity.toString());
        return formEntity;
    }


    /**
     * 生成Get请求内容
     *
     * @param param
     * @param excluedes
     * @return
     */
    public static String makeGetParamContent(Map<String, ? extends Object> param, String... excluedes) {
        StringBuilder content = new StringBuilder();
        List<String> excludeKeys = Arrays.asList(excluedes);
        param.forEach((key, v) -> {
            content.append(key).append("=").append(v).append("&");
        });
        if (content.length() > 0) {
            content.deleteCharAt(content.length() - 1);
        }
        return content.toString();
    }
}