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);

    }

}

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();

    }

}