Java Code Examples for org.springframework.http.MockHttpOutputMessage#getBodyAsString()

@Test  // SPR-13318
public void writeSubTypeList() throws Exception {
	MockHttpOutputMessage outputMessage = new MockHttpOutputMessage();
	List<MyBean> beans = new ArrayList<>();
	MyBean foo = new MyBean();
	foo.setString("Foo");
	foo.setNumber(42);
	beans.add(foo);
	MyBean bar = new MyBean();
	bar.setString("Bar");
	bar.setNumber(123);
	beans.add(bar);
	ParameterizedTypeReference<List<MyInterface>> typeReference =
			new ParameterizedTypeReference<List<MyInterface>>() {};

	this.converter.writeInternal(beans, typeReference.getType(), outputMessage);

	String result = outputMessage.getBodyAsString(StandardCharsets.UTF_8);
	assertTrue(result.contains("\"string\":\"Foo\""));
	assertTrue(result.contains("\"number\":42"));
	assertTrue(result.contains("\"string\":\"Bar\""));
	assertTrue(result.contains("\"number\":123"));
}
 

@Test
public void jsonView() throws Exception {
	MockHttpOutputMessage outputMessage = new MockHttpOutputMessage();
	JacksonViewBean bean = new JacksonViewBean();
	bean.setWithView1("with");
	bean.setWithView2("with");
	bean.setWithoutView("without");

	MappingJacksonValue jacksonValue = new MappingJacksonValue(bean);
	jacksonValue.setSerializationView(MyJacksonView1.class);
	this.converter.write(jacksonValue, null, outputMessage);

	String result = outputMessage.getBodyAsString(StandardCharsets.UTF_8);
	assertThat(result, containsString("<withView1>with</withView1>"));
	assertThat(result, not(containsString("<withView2>with</withView2>")));
	assertThat(result, not(containsString("<withoutView>without</withoutView>")));
}
 

@Test
public void write() throws IOException {
	MockHttpOutputMessage outputMessage = new MockHttpOutputMessage();
	MyBean body = new MyBean();
	body.setString("Foo");
	body.setNumber(42);
	body.setFraction(42F);
	body.setArray(new String[] {"Foo", "Bar"});
	body.setBool(true);
	body.setBytes(new byte[] {0x1, 0x2});
	converter.write(body, null, outputMessage);
	String result = outputMessage.getBodyAsString(StandardCharsets.UTF_8);
	assertTrue(result.contains("\"string\":\"Foo\""));
	assertTrue(result.contains("\"number\":42"));
	assertTrue(result.contains("fraction\":42.0"));
	assertTrue(result.contains("\"array\":[\"Foo\",\"Bar\"]"));
	assertTrue(result.contains("\"bool\":true"));
	assertTrue(result.contains("\"bytes\":\"AQI=\""));
	assertEquals("Invalid content-type", new MediaType("application", "json", StandardCharsets.UTF_8),
			outputMessage.getHeaders().getContentType());
}
 

@Test
public void classLevelJsonView() throws Exception {
	MockHttpOutputMessage outputMessage = new MockHttpOutputMessage();
	JacksonViewBean bean = new JacksonViewBean();
	bean.setWithView1("with");
	bean.setWithView2("with");
	bean.setWithoutView("without");

	MappingJacksonValue jacksonValue = new MappingJacksonValue(bean);
	jacksonValue.setSerializationView(MyJacksonView3.class);
	this.converter.writeInternal(jacksonValue, null, outputMessage);

	String result = outputMessage.getBodyAsString(StandardCharsets.UTF_8);
	assertThat(result, not(containsString("\"withView1\":\"with\"")));
	assertThat(result, not(containsString("\"withView2\":\"with\"")));
	assertThat(result, containsString("\"withoutView\":\"without\""));
}
 

@Test  // SPR-13318
public void writeSubTypeList() throws Exception {
	MockHttpOutputMessage outputMessage = new MockHttpOutputMessage();
	List<MyBean> beans = new ArrayList<>();
	MyBean foo = new MyBean();
	foo.setString("Foo");
	foo.setNumber(42);
	beans.add(foo);
	MyBean bar = new MyBean();
	bar.setString("Bar");
	bar.setNumber(123);
	beans.add(bar);
	ParameterizedTypeReference<List<MyInterface>> typeReference =
			new ParameterizedTypeReference<List<MyInterface>>() {};

	this.converter.writeInternal(beans, typeReference.getType(), outputMessage);

	String result = outputMessage.getBodyAsString(StandardCharsets.UTF_8);
	assertTrue(result.contains("\"string\":\"Foo\""));
	assertTrue(result.contains("\"number\":42"));
	assertTrue(result.contains("\"string\":\"Bar\""));
	assertTrue(result.contains("\"number\":123"));
}
 

@Test
public void writeWithBaseType() throws IOException {
	MockHttpOutputMessage outputMessage = new MockHttpOutputMessage();
	MyBean body = new MyBean();
	body.setString("Foo");
	body.setNumber(42);
	body.setFraction(42F);
	body.setArray(new String[] {"Foo", "Bar"});
	body.setBool(true);
	body.setBytes(new byte[] {0x1, 0x2});
	this.converter.write(body, MyBase.class, null, outputMessage);
	Charset utf8 = StandardCharsets.UTF_8;
	String result = outputMessage.getBodyAsString(utf8);
	assertTrue(result.contains("\"string\":\"Foo\""));
	assertTrue(result.contains("\"number\":42"));
	assertTrue(result.contains("fraction\":42.0"));
	assertTrue(result.contains("\"array\":[\"Foo\",\"Bar\"]"));
	assertTrue(result.contains("\"bool\":true"));
	assertTrue(result.contains("\"bytes\":[1,2]"));
	assertEquals("Invalid content-type", new MediaType("application", "json", utf8),
			outputMessage.getHeaders().getContentType());
}
 

@Test
public void jsonpAndJsonView() throws Exception {
	MockHttpOutputMessage outputMessage = new MockHttpOutputMessage();
	JacksonViewBean bean = new JacksonViewBean();
	bean.setWithView1("with");
	bean.setWithView2("with");
	bean.setWithoutView("without");

	MappingJacksonValue jacksonValue = new MappingJacksonValue(bean);
	jacksonValue.setSerializationView(MyJacksonView1.class);
	jacksonValue.setJsonpFunction("callback");
	this.converter.writeInternal(jacksonValue, null, outputMessage);

	String result = outputMessage.getBodyAsString(Charset.forName("UTF-8"));
	assertThat(result, startsWith("/**/callback("));
	assertThat(result, endsWith(");"));
	assertThat(result, containsString("\"withView1\":\"with\""));
	assertThat(result, not(containsString("\"withView2\":\"with\"")));
	assertThat(result, not(containsString("\"withoutView\":\"without\"")));
}
 

@Test
public void classLevelJsonView() throws Exception {
	MockHttpOutputMessage outputMessage = new MockHttpOutputMessage();
	JacksonViewBean bean = new JacksonViewBean();
	bean.setWithView1("with");
	bean.setWithView2("with");
	bean.setWithoutView("without");

	MappingJacksonValue jacksonValue = new MappingJacksonValue(bean);
	jacksonValue.setSerializationView(MyJacksonView3.class);
	this.converter.writeInternal(jacksonValue, null, outputMessage);

	String result = outputMessage.getBodyAsString(StandardCharsets.UTF_8);
	assertThat(result, not(containsString("\"withView1\":\"with\"")));
	assertThat(result, not(containsString("\"withView2\":\"with\"")));
	assertThat(result, containsString("\"withoutView\":\"without\""));
}
 

@Test
public void fieldLevelJsonView() throws Exception {
	MockHttpOutputMessage outputMessage = new MockHttpOutputMessage();
	JacksonViewBean bean = new JacksonViewBean();
	bean.setWithView1("with");
	bean.setWithView2("with");
	bean.setWithoutView("without");

	MappingJacksonValue jacksonValue = new MappingJacksonValue(bean);
	jacksonValue.setSerializationView(MyJacksonView1.class);
	this.converter.writeInternal(jacksonValue, null, outputMessage);

	String result = outputMessage.getBodyAsString(StandardCharsets.UTF_8);
	assertThat(result, containsString("\"withView1\":\"with\""));
	assertThat(result, not(containsString("\"withView2\":\"with\"")));
	assertThat(result, not(containsString("\"withoutView\":\"without\"")));
}
 

@Test
public void writeWithBaseType() throws IOException {
	MockHttpOutputMessage outputMessage = new MockHttpOutputMessage();
	MyBean body = new MyBean();
	body.setString("Foo");
	body.setNumber(42);
	body.setFraction(42F);
	body.setArray(new String[] {"Foo", "Bar"});
	body.setBool(true);
	body.setBytes(new byte[] {0x1, 0x2});
	this.converter.write(body, MyBase.class, null, outputMessage);
	Charset utf8 = StandardCharsets.UTF_8;
	String result = outputMessage.getBodyAsString(utf8);
	assertTrue(result.contains("\"string\":\"Foo\""));
	assertTrue(result.contains("\"number\":42"));
	assertTrue(result.contains("fraction\":42.0"));
	assertTrue(result.contains("\"array\":[\"Foo\",\"Bar\"]"));
	assertTrue(result.contains("\"bool\":true"));
	assertTrue(result.contains("\"bytes\":[1,2]"));
	assertEquals("Invalid content-type", new MediaType("application", "json", utf8),
			outputMessage.getHeaders().getContentType());
}
 

@Test
public void write() throws IOException {
	MockHttpOutputMessage outputMessage = new MockHttpOutputMessage();
	MyBean body = new MyBean();
	body.setString("Foo");
	body.setNumber(42);
	body.setFraction(42F);
	body.setArray(new String[] {"Foo", "Bar"});
	body.setBool(true);
	body.setBytes(new byte[] {0x1, 0x2});
	this.converter.write(body, null, outputMessage);
	Charset utf8 = StandardCharsets.UTF_8;
	String result = outputMessage.getBodyAsString(utf8);
	assertTrue(result.contains("\"string\":\"Foo\""));
	assertTrue(result.contains("\"number\":42"));
	assertTrue(result.contains("fraction\":42.0"));
	assertTrue(result.contains("\"array\":[\"Foo\",\"Bar\"]"));
	assertTrue(result.contains("\"bool\":true"));
	assertTrue(result.contains("\"bytes\":[1,2]"));
	assertEquals("Invalid content-type", new MediaType("application", "json", utf8),
			outputMessage.getHeaders().getContentType());
}
 

@Test
public void fieldLevelJsonView() throws Exception {
	MockHttpOutputMessage outputMessage = new MockHttpOutputMessage();
	JacksonViewBean bean = new JacksonViewBean();
	bean.setWithView1("with");
	bean.setWithView2("with");
	bean.setWithoutView("without");

	MappingJacksonValue jacksonValue = new MappingJacksonValue(bean);
	jacksonValue.setSerializationView(MyJacksonView1.class);
	this.converter.writeInternal(jacksonValue, null, outputMessage);

	String result = outputMessage.getBodyAsString(StandardCharsets.UTF_8);
	assertThat(result, containsString("\"withView1\":\"with\""));
	assertThat(result, not(containsString("\"withView2\":\"with\"")));
	assertThat(result, not(containsString("\"withoutView\":\"without\"")));
}
 

@Test
public void write() throws IOException {
	MockHttpOutputMessage outputMessage = new MockHttpOutputMessage();
	MyBean body = new MyBean();
	body.setString("Foo");
	body.setNumber(42);
	body.setFraction(42F);
	body.setArray(new String[]{"Foo", "Bar"});
	body.setBool(true);
	body.setBytes(new byte[]{0x1, 0x2});
	this.converter.write(body, null, outputMessage);
	Charset utf8 = Charset.forName("UTF-8");
	String result = outputMessage.getBodyAsString(utf8);
	assertTrue(result.contains("\"string\":\"Foo\""));
	assertTrue(result.contains("\"number\":42"));
	assertTrue(result.contains("fraction\":42.0"));
	assertTrue(result.contains("\"array\":[\"Foo\",\"Bar\"]"));
	assertTrue(result.contains("\"bool\":true"));
	assertTrue(result.contains("\"bytes\":[1,2]"));
	assertEquals("Invalid content-type", new MediaType("application", "json", utf8),
			outputMessage.getHeaders().getContentType());
}
 

@Test
public void writeJsonWithJavaFormat() throws IOException {
	this.converter = new ProtobufHttpMessageConverter(
			new ProtobufHttpMessageConverter.ProtobufJavaFormatSupport(),
			this.extensionRegistry);
	MockHttpOutputMessage outputMessage = new MockHttpOutputMessage();
	MediaType contentType = MediaType.APPLICATION_JSON_UTF8;
	this.converter.write(this.testMsg, contentType, outputMessage);

	assertEquals(contentType, outputMessage.getHeaders().getContentType());

	final String body = outputMessage.getBodyAsString(Charset.forName("UTF-8"));
	assertFalse("body is empty", body.isEmpty());

	Msg.Builder builder = Msg.newBuilder();
	JsonFormat.parser().merge(body, builder);
	assertEquals(this.testMsg, builder.build());

	assertNull(outputMessage.getHeaders().getFirst(
			ProtobufHttpMessageConverter.X_PROTOBUF_MESSAGE_HEADER));
	assertNull(outputMessage.getHeaders().getFirst(
			ProtobufHttpMessageConverter.X_PROTOBUF_SCHEMA_HEADER));
}
 

@Test
public void writeJsonWithGoogleProtobuf() throws IOException {
	this.converter = new ProtobufHttpMessageConverter(
			new ProtobufHttpMessageConverter.ProtobufJavaUtilSupport(null, null),
			this.extensionRegistry);
	MockHttpOutputMessage outputMessage = new MockHttpOutputMessage();
	MediaType contentType = MediaType.APPLICATION_JSON;
	this.converter.write(this.testMsg, contentType, outputMessage);

	assertEquals(contentType, outputMessage.getHeaders().getContentType());

	final String body = outputMessage.getBodyAsString(Charset.forName("UTF-8"));
	assertFalse("body is empty", body.isEmpty());

	Msg.Builder builder = Msg.newBuilder();
	JsonFormat.parser().merge(body, builder);
	assertEquals(this.testMsg, builder.build());

	assertNull(outputMessage.getHeaders().getFirst(
			ProtobufHttpMessageConverter.X_PROTOBUF_MESSAGE_HEADER));
	assertNull(outputMessage.getHeaders().getFirst(
			ProtobufHttpMessageConverter.X_PROTOBUF_SCHEMA_HEADER));
}
 

@Test
public void prettyPrint() throws Exception {
	MockHttpOutputMessage outputMessage = new MockHttpOutputMessage();
	PrettyPrintBean bean = new PrettyPrintBean();
	bean.setName("Jason");

	this.converter.setPrettyPrint(true);
	this.converter.writeInternal(bean, null, outputMessage);
	String result = outputMessage.getBodyAsString(StandardCharsets.UTF_8);

	assertEquals("{" + NEWLINE_SYSTEM_PROPERTY +
			"  \"name\" : \"Jason\"" + NEWLINE_SYSTEM_PROPERTY + "}", result);
}
 

@Test
public void partialContentMultipleByteRanges() throws Exception {
	MockHttpOutputMessage outputMessage = new MockHttpOutputMessage();
	Resource body = new ClassPathResource("byterangeresource.txt", getClass());
	List<HttpRange> rangeList = HttpRange.parseRanges("bytes=0-5,7-15,17-20,22-38");
	List<ResourceRegion> regions = new ArrayList<>();
	for(HttpRange range : rangeList) {
		regions.add(range.toResourceRegion(body));
	}

	converter.write(regions, MediaType.TEXT_PLAIN, outputMessage);

	HttpHeaders headers = outputMessage.getHeaders();
	assertThat(headers.getContentType().toString(), Matchers.startsWith("multipart/byteranges;boundary="));
	String boundary = "--" + headers.getContentType().toString().substring(30);
	String content = outputMessage.getBodyAsString(StandardCharsets.UTF_8);
	String[] ranges = StringUtils.tokenizeToStringArray(content, "\r\n", false, true);

	assertThat(ranges[0], is(boundary));
	assertThat(ranges[1], is("Content-Type: text/plain"));
	assertThat(ranges[2], is("Content-Range: bytes 0-5/39"));
	assertThat(ranges[3], is("Spring"));

	assertThat(ranges[4], is(boundary));
	assertThat(ranges[5], is("Content-Type: text/plain"));
	assertThat(ranges[6], is("Content-Range: bytes 7-15/39"));
	assertThat(ranges[7], is("Framework"));

	assertThat(ranges[8], is(boundary));
	assertThat(ranges[9], is("Content-Type: text/plain"));
	assertThat(ranges[10], is("Content-Range: bytes 17-20/39"));
	assertThat(ranges[11], is("test"));

	assertThat(ranges[12], is(boundary));
	assertThat(ranges[13], is("Content-Type: text/plain"));
	assertThat(ranges[14], is("Content-Range: bytes 22-38/39"));
	assertThat(ranges[15], is("resource content."));
}
 

@Test  // SPR-13318
public void writeSubType() throws Exception {
	MockHttpOutputMessage outputMessage = new MockHttpOutputMessage();
	MyBean bean = new MyBean();
	bean.setString("Foo");
	bean.setNumber(42);

	this.converter.writeInternal(bean, MyInterface.class, outputMessage);

	String result = outputMessage.getBodyAsString(StandardCharsets.UTF_8);
	assertTrue(result.contains("\"string\":\"Foo\""));
	assertTrue(result.contains("\"number\":42"));
}
 

@Test  // SPR-13318
public void writeSubType() throws Exception {
	MockHttpOutputMessage outputMessage = new MockHttpOutputMessage();
	MyBean bean = new MyBean();
	bean.setString("Foo");
	bean.setNumber(42);

	this.converter.writeInternal(bean, MyInterface.class, outputMessage);

	String result = outputMessage.getBodyAsString(Charset.forName("UTF-8"));
	assertTrue(result.contains("\"string\":\"Foo\""));
	assertTrue(result.contains("\"number\":42"));
}
 

@Test
public void prettyPrint() throws Exception {
	MockHttpOutputMessage outputMessage = new MockHttpOutputMessage();
	PrettyPrintBean bean = new PrettyPrintBean();
	bean.setName("Jason");

	this.converter.setPrettyPrint(true);
	this.converter.writeInternal(bean, null, outputMessage);
	String result = outputMessage.getBodyAsString(Charset.forName("UTF-8"));

	assertEquals("{" + NEWLINE_SYSTEM_PROPERTY + "  \"name\" : \"Jason\"" + NEWLINE_SYSTEM_PROPERTY + "}", result);
}