Improve coverage for dmaap messageservice

[dmaap/messagerouter/messageservice.git] / src / test / java / com / att / nsa / dmaap / util / ContentLengthInterceptorTest.java

diff --git a/src/test/java/com/att/nsa/dmaap/util/ContentLengthInterceptorTest.java b/src/test/java/com/att/nsa/dmaap/util/ContentLengthInterceptorTest.java
index 5d9a3a6..36d52e0 100644 (file)
--- a/src/test/java/com/att/nsa/dmaap/util/ContentLengthInterceptorTest.java
+++ b/src/test/java/com/att/nsa/dmaap/util/ContentLengthInterceptorTest.java
@@ -20,19 +20,41 @@
 \r
 package com.att.nsa.dmaap.util;\r
 \r
-import static org.junit.Assert.*;\r
+import static org.junit.Assert.assertTrue;\r
+\r
+import java.util.Map;\r
+\r
+import javax.servlet.http.HttpServletRequest;\r
+import javax.servlet.http.HttpServletResponse;\r
 \r
 import org.junit.After;\r
 import org.junit.Before;\r
 import org.junit.Test;\r
-\r
+import org.junit.runner.RunWith;\r
+import org.mockito.InjectMocks;\r
+import org.mockito.Mock;\r
+import org.powermock.api.mockito.PowerMockito;\r
+import org.powermock.core.classloader.annotations.PrepareForTest;\r
+import org.powermock.modules.junit4.PowerMockRunner;\r
+\r
+@RunWith(PowerMockRunner.class)\r
+@PrepareForTest({ System.class })\r
 public class ContentLengthInterceptorTest {\r
+       @InjectMocks\r
+       ContentLengthInterceptor interceptor = null;\r
+\r
+       @Mock\r
+       Map map;\r
 \r
-       private ContentLengthInterceptor interceptor = null;\r
+       @Mock\r
+       HttpServletRequest req;\r
+\r
+       @Mock\r
+       HttpServletResponse res;\r
 \r
        @Before\r
        public void setUp() throws Exception {\r
-               interceptor = new ContentLengthInterceptor();\r
+               // interceptor = new ContentLengthInterceptor();\r
        }\r
 \r
        @After\r
@@ -40,34 +62,22 @@ public class ContentLengthInterceptorTest {
        }\r
 \r
        @Test\r
-       public void testAllowOrReject() {\r
-\r
-               try {\r
-                       interceptor.allowOrReject(null, null, null);\r
-               } catch (Exception e) {\r
-                       // TODO Auto-generated catch block\r
-                       e.printStackTrace();\r
-               }\r
-\r
+       public void testAllowOrReject() throws Exception {\r
+               PowerMockito.when(req.getHeader("Transfer-Encoding")).thenReturn("UTF-8");\r
+               PowerMockito.when(req.getHeader("Content-Length")).thenReturn("1027");\r
+               interceptor.allowOrReject(req, res, map);\r
                assertTrue(true);\r
-\r
        }\r
 \r
        @Test\r
        public void testGetDefLength() {\r
-\r
                interceptor.getDefLength();\r
-\r
                assertTrue(true);\r
-\r
-               \r
        }\r
-       \r
+\r
        @Test\r
        public void testSetDefLength() {\r
-\r
                interceptor.setDefLength("defLength");\r
-\r
                assertTrue(true);\r
 \r
        }\r