OXM is nearing completion

org.springframework.web.servlet/src/test/java/org/springframework/web/servlet/view/xml/MarshallingViewTest.java

@@ -0,0 +1,138 @@
+/*
+ * Copyright 2007 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.web.servlet.view.xml;
+
+import java.util.HashMap;
+import java.util.Map;
+import javax.servlet.ServletException;
+import javax.xml.transform.stream.StreamResult;
+
+import static org.easymock.EasyMock.*;
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.fail;
+import org.junit.Before;
+import org.junit.Test;
+
+import org.springframework.mock.web.MockHttpServletRequest;
+import org.springframework.mock.web.MockHttpServletResponse;
+import org.springframework.oxm.Marshaller;
+
+public class MarshallingViewTest {
+
+	private MarshallingView view;
+
+	private Marshaller marshallerMock;
+
+	@Before
+	public void createView() throws Exception {
+		marshallerMock = createMock(Marshaller.class);
+		view = new MarshallingView(marshallerMock);
+	}
+
+	@Test
+	public void testGetContentType() {
+		assertEquals("Invalid content type", "application/xml", view.getContentType());
+	}
+
+	@Test
+	public void renderModelKey() throws Exception {
+		Object toBeMarshalled = new Object();
+		String modelKey = "key";
+		view.setModelKey(modelKey);
+		Map model = new HashMap();
+		model.put(modelKey, toBeMarshalled);
+
+		MockHttpServletRequest request = new MockHttpServletRequest();
+		MockHttpServletResponse response = new MockHttpServletResponse();
+
+		expect(marshallerMock.supports(Object.class)).andReturn(true);
+		marshallerMock.marshal(eq(toBeMarshalled), isA(StreamResult.class));
+
+		replay(marshallerMock);
+		view.render(model, request, response);
+		assertEquals("Invalid content type", "application/xml", response.getContentType());
+		assertEquals("Invalid content length", 0, response.getContentLength());
+		verify(marshallerMock);
+	}
+
+	@Test
+	public void renderModelKeyUnsupported() throws Exception {
+		Object toBeMarshalled = new Object();
+		String modelKey = "key";
+		view.setModelKey(modelKey);
+		Map model = new HashMap();
+		model.put(modelKey, toBeMarshalled);
+
+		MockHttpServletRequest request = new MockHttpServletRequest();
+		MockHttpServletResponse response = new MockHttpServletResponse();
+
+		expect(marshallerMock.supports(Object.class)).andReturn(false);
+
+		replay(marshallerMock);
+		try {
+			view.render(model, request, response);
+			fail("ServletException expected");
+		}
+		catch (ServletException ex) {
+			// expected
+		}
+		verify(marshallerMock);
+	}
+
+	@Test
+	public void renderNoModelKey() throws Exception {
+		Object toBeMarshalled = new Object();
+		String modelKey = "key";
+		Map model = new HashMap();
+		model.put(modelKey, toBeMarshalled);
+
+		MockHttpServletRequest request = new MockHttpServletRequest();
+		MockHttpServletResponse response = new MockHttpServletResponse();
+
+		expect(marshallerMock.supports(Object.class)).andReturn(true);
+		marshallerMock.marshal(eq(toBeMarshalled), isA(StreamResult.class));
+
+		replay(marshallerMock);
+		view.render(model, request, response);
+		assertEquals("Invalid content type", "application/xml", response.getContentType());
+		assertEquals("Invalid content length", 0, response.getContentLength());
+		verify(marshallerMock);
+	}
+
+	@Test
+	public void testRenderUnsupportedModel() throws Exception {
+		Object toBeMarshalled = new Object();
+		String modelKey = "key";
+		Map model = new HashMap();
+		model.put(modelKey, toBeMarshalled);
+
+		MockHttpServletRequest request = new MockHttpServletRequest();
+		MockHttpServletResponse response = new MockHttpServletResponse();
+
+		expect(marshallerMock.supports(Object.class)).andReturn(false);
+
+		replay(marshallerMock);
+		try {
+			view.render(model, request, response);
+			fail("ServletException expected");
+		}
+		catch (ServletException ex) {
+			// expected
+		}
+		verify(marshallerMock);
+	}
+}