AddressApiControllerQueryTest.java

package org.xmx0632.deliciousfruit.api.v1;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;

import java.util.ArrayList;
import java.util.List;

import org.junit.BeforeClass;
import org.junit.Test;
import org.springframework.http.HttpEntity;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpMethod;
import org.springframework.http.ResponseEntity;
import org.springframework.web.client.RestTemplate;
import org.xmx0632.deliciousfruit.api.v1.bo.AllReceiversResponse;
import org.xmx0632.deliciousfruit.api.v1.bo.AllReceiversResponse.ReceiverBo;
import org.xmx0632.deliciousfruit.api.v1.bo.RequestBo;
import org.xmx0632.deliciousfruit.functional.BaseControllerTestCase;

public class AddressApiControllerQueryTest extends BaseControllerTestCase {

	private final RestTemplate restTemplate = new RestTemplate();

	private static String url;

	@BeforeClass
	public static void initUrl() {
		// baseUrl = "http://116.228.195.82:18080/deliciousfruit/api/v1";
		url = baseUrl + "/address/queryaddress";
	}

	@Test
	public void testQuerySuccess() throws Exception {

		HttpHeaders requestHeaders = createHttpHeader("user2", "password");

		HttpEntity<RequestBo> requestEntity = new HttpEntity<RequestBo>(null,
				requestHeaders);

		ResponseEntity<AllReceiversResponse> result = restTemplate.exchange(
				url, HttpMethod.GET, requestEntity, AllReceiversResponse.class);

		List<ReceiverBo> receivers = result.getBody().getReceivers();
		assertTrue(receivers.size() >= 2);
		List<ReceiverBo> newReceivers = new ArrayList<ReceiverBo>();
		for (ReceiverBo receiverBo : receivers) {
			if (receiverBo.getId() == 1 || receiverBo.getId() == 2) {
				receiverBo.setIsDefault(0);
				newReceivers.add(receiverBo);
			}
		}
		assertEquals(genAll().toString(), newReceivers.toString());

		formatHttpInfoPrint(HttpMethod.GET, url, requestHeaders,
				"测试查询收货地址信息, 成功", null, jsonMapper.toJson(result.getBody()));

	}

	private List<ReceiverBo> genAll() {
		List<ReceiverBo> receivers = new ArrayList<ReceiverBo>();

		ReceiverBo receiverInfo = new ReceiverBo();
		receiverInfo.setId(1l);
		receiverInfo.setIsDefault(0);
		receiverInfo.setAddress("云桥路133");
		receiverInfo.setCity("浦东");
		receiverInfo.setDistrict("金桥");
		receiverInfo.setName("张三");
		receiverInfo.setPhoneNumber("13811118888");
		receiverInfo.setProvince("上海");
		receiverInfo.setIsDefault(1);

		ReceiverBo receiverInfo2 = new ReceiverBo();
		receiverInfo2.setId(2l);
		receiverInfo.setIsDefault(0);
		receiverInfo2.setAddress("云桥路111");
		receiverInfo2.setCity("浦东");
		receiverInfo2.setDistrict("金桥");
		receiverInfo2.setName("李四");
		receiverInfo2.setPhoneNumber("13811118888");
		receiverInfo2.setProvince("上海");
		receivers.add(receiverInfo);
		receivers.add(receiverInfo2);
		return receivers;
	}
}