1 files changed, 178 insertions, 0 deletions
diff --git a/src/test/java/org/onap/dmaap/mr/cambria/utils/DMaaPResponseBuilderTest.java b/src/test/java/org/onap/dmaap/mr/cambria/utils/DMaaPResponseBuilderTest.java
new file mode 100644
index 0000000..326e399
--- /dev/null
+++ b/src/test/java/org/onap/dmaap/mr/cambria/utils/DMaaPResponseBuilderTest.java
@@ -0,0 +1,178 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * ONAP Policy Engine
+ * ================================================================================
+ * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
+ * ================================================================================
+ * 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.
+ * ============LICENSE_END=========================================================
+ */
+
+ package org.onap.dmaap.mr.cambria.utils;
+
+import org.json.JSONException;
+import org.json.JSONObject;
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Test;
+import org.onap.dmaap.dmf.mr.beans.DMaaPContext;
+import org.onap.dmaap.dmf.mr.utils.DMaaPResponseBuilder;
+import org.springframework.mock.web.MockHttpServletRequest;
+import org.springframework.mock.web.MockHttpServletResponse;
+
+import java.io.IOException;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNull;
+
+public class DMaaPResponseBuilderTest {
+	
+	DMaaPContext dMaapContext;
+	MockHttpServletRequest request;
+	MockHttpServletResponse response;
+
+	@Before
+	public void setUp() throws Exception {
+		
+		dMaapContext = new DMaaPContext();
+		request = new MockHttpServletRequest();
+		response = new MockHttpServletResponse();
+		dMaapContext.setRequest(request);
+		dMaapContext.setResponse(response);
+	}
+
+	@After
+	public void tearDown() throws Exception {
+	}
+
+	@Test
+	public void testsetNoCacheHeadings(){		
+		DMaaPResponseBuilder.setNoCacheHeadings(dMaapContext);		
+		assertEquals("no-cache", response.getHeader("Pragma"));
+	}
+	
+	@Test
+	public void testrespondOk() throws JSONException, IOException{
+		JSONObject jsonObject = new JSONObject();
+		jsonObject.put("Name", "Test");
+		
+		DMaaPResponseBuilder.respondOk(dMaapContext, jsonObject);
+		assertEquals("application/json", response.getContentType());
+		assertEquals(200, response.getStatus());
+		
+		request.setMethod("HEAD");
+		
+		DMaaPResponseBuilder.respondOk(dMaapContext, jsonObject);
+		assertEquals("application/json", response.getContentType());
+		assertEquals(200, response.getStatus());
+	}
+	
+	@Test
+	public void testrespondOkNoContent(){
+		DMaaPResponseBuilder.respondOkNoContent(dMaapContext);
+		assertEquals(204, response.getStatus());
+	}
+	
+	@Test
+	public void testrespondOkNoContentError(){
+		dMaapContext.setResponse(null);
+		DMaaPResponseBuilder.respondOkNoContent(dMaapContext);
+		assertNull(dMaapContext.getResponse());
+	}
+	
+	@Test
+	public void testrespondOkWithHtml(){
+		DMaaPResponseBuilder.respondOkWithHtml(dMaapContext, "<head></head>");
+		
+		assertEquals("text/html", response.getContentType());
+		DMaaPResponseBuilder.respondOkWithHtml(dMaapContext, "<head></head>");
+		assertEquals(200, response.getStatus());
+	}
+	
+	@Test
+	public void testrespondOkWithHtmlError(){
+		dMaapContext.setResponse(null);
+		DMaaPResponseBuilder.respondOkWithHtml(dMaapContext, "<head></head>");
+		assertNull(dMaapContext.getResponse());
+	}
+	
+	@Test
+	public void testrespondWithError(){
+		DMaaPResponseBuilder.respondWithError(dMaapContext, 500, "InternalServerError");
+		assertEquals(500, response.getStatus());
+	}
+	
+	@Test(expected=NullPointerException.class)
+	public void testInvalidrespondWithError(){
+		dMaapContext.setResponse(null);
+		DMaaPResponseBuilder.respondWithError(dMaapContext, 500, "InternalServerError");
+	}
+	
+	@Test
+	public void testrespondWithJsonError(){
+		JSONObject o = new JSONObject();
+		o.put("status", 500);
+		o.put("message", "InternalServerError");
+		DMaaPResponseBuilder.respondWithError(dMaapContext, 500, o);
+		assertEquals(500, response.getStatus());
+	}
+	
+	@Test
+	public void testInvalidrespondWithJsonError(){
+		JSONObject o = new JSONObject();
+		o.put("status", 500);
+		o.put("message", "InternalServerError");
+		dMaapContext.setResponse(null);
+		DMaaPResponseBuilder.respondWithError(dMaapContext, 500, o);
+		assertNull(dMaapContext.getResponse());
+	}
+	
+	@Test
+	public void testrespondWithErrorInJson(){
+		DMaaPResponseBuilder.respondWithErrorInJson(dMaapContext, 500, "InternalServerError");
+		
+		assertEquals("application/json", response.getContentType());
+		assertEquals(500, response.getStatus());
+	}
+	
+	@Test
+	public void testsendErrorAndBody(){
+		DMaaPResponseBuilder.sendErrorAndBody(dMaapContext, 500, "InternalServerError", "text/html");
+		
+		assertEquals("text/html", response.getContentType());
+		assertEquals(500, response.getStatus());
+		
+		request.setMethod("HEAD");
+		
+		DMaaPResponseBuilder.sendErrorAndBody(dMaapContext, 500, "InternalServerError", "text/html");
+		
+		assertEquals("text/html", response.getContentType());
+		assertEquals(500, response.getStatus());
+		
+	}
+	
+	@Test
+	public void testgetStreamForBinaryResponse() throws IOException{
+		DMaaPResponseBuilder.getStreamForBinaryResponse(dMaapContext);
+		
+		assertEquals("application/octet-stream", response.getContentType());
+		assertEquals(200, response.getStatus());
+	}
+	
+	@Test(expected=NullPointerException.class)
+	public void testgetStreamForBinaryResponseError() throws IOException{
+		dMaapContext.setResponse(null);
+		DMaaPResponseBuilder.getStreamForBinaryResponse(dMaapContext);
+	}
+
+}