/*
    * Copyright 2002-2013 the original author or authors.
    *
    * Licensed under the Apache License, Version 2.0 (the "License");
    * you may not use this file except in compliance with the License.
    * You may obtain a copy of the License at
    *
    *      http://www.apache.org/licenses/LICENSE-2.0
    *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   */
  
  package org.springframework.test.web.servlet.samples.standalone.resultmatchers;
  
  import java.nio.charset.Charset;
  
  import org.junit.Before;
  import org.junit.Test;
  
  import org.springframework.http.MediaType;
  import org.springframework.stereotype.Controller;
  import org.springframework.test.web.servlet.MockMvc;
  import org.springframework.web.bind.annotation.RequestMapping;
  import org.springframework.web.bind.annotation.ResponseBody;
  
  import static org.hamcrest.Matchers.*;
  import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.*;
  import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.*;
  import static org.springframework.test.web.servlet.setup.MockMvcBuilders.*;
  
  /**
   * Examples of defining expectations on the response content, content type, and
   * the character encoding.
   *
   * @author Rossen Stoyanchev
   *
   * @see JsonPathAssertionTests
   * @see XmlContentAssertionTests
   * @see XpathAssertionTests
   */
  public class ContentAssertionTests {
  
  	public static final MediaType TEXT_PLAIN_UTF8 = new MediaType("text", "plain", Charset.forName("UTF-8"));
  
  	private MockMvc mockMvc;
  
  	@Before
  	public void setup() {
  		this.mockMvc = standaloneSetup(new SimpleController()).alwaysExpect(status().isOk()).build();
  	}
  
  	@Test
  	public void testContentType() throws Exception {
  		this.mockMvc.perform(get("/handle").accept(MediaType.TEXT_PLAIN))
  			.andExpect(content().contentType(MediaType.TEXT_PLAIN))
  			.andExpect(content().contentType("text/plain"));
  
  		this.mockMvc.perform(get("/handleUtf8"))
  			.andExpect(content().contentType(MediaType.valueOf("text/plain;charset=UTF-8")))
  			.andExpect(content().contentType("text/plain;charset=UTF-8"))
  			.andExpect(content().contentTypeCompatibleWith("text/plain"))
  			.andExpect(content().contentTypeCompatibleWith(MediaType.TEXT_PLAIN));
  	}
  
  	@Test
  	public void testContentAsString() throws Exception {
  
  		this.mockMvc.perform(get("/handle").accept(MediaType.TEXT_PLAIN))
  			.andExpect(content().string("Hello world!"));
  
  		this.mockMvc.perform(get("/handleUtf8"))
  			.andExpect(content().string("\u3053\u3093\u306b\u3061\u306f\u4e16\u754c\uff01"));
  
  		// Hamcrest matchers...
  		this.mockMvc.perform(get("/handle").accept(MediaType.TEXT_PLAIN)).andExpect(content().string(equalTo("Hello world!")));
  		this.mockMvc.perform(get("/handleUtf8")).andExpect(content().string(equalTo("\u3053\u3093\u306b\u3061\u306f\u4e16\u754c\uff01")));
  	}
  
  	@Test
  	public void testContentAsBytes() throws Exception {
  
  		this.mockMvc.perform(get("/handle").accept(MediaType.TEXT_PLAIN))
  			.andExpect(content().bytes("Hello world!".getBytes("ISO-8859-1")));
  
  		this.mockMvc.perform(get("/handleUtf8"))
  			.andExpect(content().bytes("\u3053\u3093\u306b\u3061\u306f\u4e16\u754c\uff01".getBytes("UTF-8")));
  	}
  
  	@Test
  	public void testContentStringMatcher() throws Exception {
  		this.mockMvc.perform(get("/handle").accept(MediaType.TEXT_PLAIN))
  			.andExpect(content().string(containsString("world")));
  	}
  
  	@Test
 	public void testCharacterEncoding() throws Exception {
 
 		this.mockMvc.perform(get("/handle").accept(MediaType.TEXT_PLAIN))
 			.andExpect(content().encoding("ISO-8859-1"))
 			.andExpect(content().string(containsString("world")));
 
 		this.mockMvc.perform(get("/handleUtf8"))
 			.andExpect(content().encoding("UTF-8"))
 			.andExpect(content().bytes("\u3053\u3093\u306b\u3061\u306f\u4e16\u754c\uff01".getBytes("UTF-8")));
 	}
 
 
 	@Controller
 	private static class SimpleController {
 
 		@RequestMapping(value="/handle", produces="text/plain")
 		@ResponseBody
 		public String handle() {
 			return "Hello world!";
 		}
 
 		@RequestMapping(value="/handleUtf8", produces="text/plain;charset=UTF-8")
 		@ResponseBody
 		public String handleWithCharset() {
 			return "\u3053\u3093\u306b\u3061\u306f\u4e16\u754c\uff01";	// "Hello world! (Japanese)
 		}
 	}
 
 }