From b61f35e956431336f371dd149eb563b6e778f67b Mon Sep 17 00:00:00 2001
From: Shubham Aggarwal <sbmaggarwal@gmail.com>
Date: Tue, 29 Nov 2016 14:25:32 +0530
Subject: [PATCH] Updated Test

---
 javax-servlets/pom.xml                        |  6 +--
 .../src/main/java/com/root/FormServlet.java   |  5 +-
 .../test/java/com/root/FormServletTest.java   | 53 +++++++------------
 3 files changed, 23 insertions(+), 41 deletions(-)

diff --git a/javax-servlets/pom.xml b/javax-servlets/pom.xml
index 2d2e3a4b8a..76434a8589 100644
--- a/javax-servlets/pom.xml
+++ b/javax-servlets/pom.xml
@@ -22,9 +22,9 @@
             <scope>test</scope>
         </dependency>
         <dependency>
-            <groupId>org.mockito</groupId>
-            <artifactId>mockito-all</artifactId>
-            <version>1.10.19</version>
+            <groupId>org.apache.httpcomponents</groupId>
+            <artifactId>httpclient</artifactId>
+            <version>4.5</version>
             <scope>test</scope>
         </dependency>
 
diff --git a/javax-servlets/src/main/java/com/root/FormServlet.java b/javax-servlets/src/main/java/com/root/FormServlet.java
index 7c345215e1..9c50698710 100644
--- a/javax-servlets/src/main/java/com/root/FormServlet.java
+++ b/javax-servlets/src/main/java/com/root/FormServlet.java
@@ -7,7 +7,6 @@ import javax.servlet.http.HttpServlet;
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
 import java.io.IOException;
-import java.io.PrintWriter;
 
 @WebServlet(name = "FormServlet", urlPatterns = "/calculateServlet")
 public class FormServlet extends HttpServlet {
@@ -26,8 +25,8 @@ public class FormServlet extends HttpServlet {
             response.setHeader("Test", "Success");
             response.setHeader("BMI", String.valueOf(bmi));
 
-            PrintWriter out= response.getWriter();
-            out.write(String.valueOf(bmi));
+            response.setHeader("Test", "Success");
+            response.setHeader("BMI", String.valueOf(bmi));
 
             RequestDispatcher dispatcher = request.getRequestDispatcher("index.jsp");
             dispatcher.forward(request, response);
diff --git a/javax-servlets/src/test/java/com/root/FormServletTest.java b/javax-servlets/src/test/java/com/root/FormServletTest.java
index 2d6ce22db5..437741e12d 100644
--- a/javax-servlets/src/test/java/com/root/FormServletTest.java
+++ b/javax-servlets/src/test/java/com/root/FormServletTest.java
@@ -1,51 +1,34 @@
 package com.root;
 
-import org.junit.Before;
+import org.apache.http.HttpResponse;
+import org.apache.http.client.HttpClient;
+import org.apache.http.client.entity.UrlEncodedFormEntity;
+import org.apache.http.client.methods.HttpPost;
+import org.apache.http.impl.client.DefaultHttpClient;
+import org.apache.http.message.BasicNameValuePair;
 import org.junit.Test;
-import org.mockito.Mock;
-import org.mockito.MockitoAnnotations;
 
-import javax.servlet.RequestDispatcher;
-import javax.servlet.http.HttpServletRequest;
-import javax.servlet.http.HttpServletResponse;
-import java.io.PrintWriter;
-import java.io.StringWriter;
+import java.util.ArrayList;
+import java.util.List;
 
 import static org.junit.Assert.assertEquals;
-import static org.mockito.Mockito.verify;
-import static org.mockito.Mockito.when;
 
 public class FormServletTest {
 
-    @Mock
-    HttpServletRequest request;
-
-    @Mock
-    HttpServletResponse response;
-
-    @Mock
-    RequestDispatcher requestDispatcher;
-
-    @Before
-    public void setUp() throws Exception {
-        MockitoAnnotations.initMocks(this);
-    }
-
     @Test
-    public void testFormServlet() throws Exception {
+    public void whenPostRequestUsingHttpClient_thenCorrect() throws Exception {
 
-        when(request.getParameter("height")).thenReturn("2");
-        when(request.getParameter("weight")).thenReturn("80");
+        HttpClient client = new DefaultHttpClient();
+        HttpPost method = new HttpPost("http://localhost:8080/calculateServlet");
 
-        StringWriter sw = new StringWriter();
-        PrintWriter pw = new PrintWriter(sw);
-        when(response.getWriter()).thenReturn(pw);
+        List<BasicNameValuePair> nvps = new ArrayList<BasicNameValuePair>();
+        nvps.add(new BasicNameValuePair("height", String.valueOf(2)));
+        nvps.add(new BasicNameValuePair("weight", String.valueOf(80)));
 
-        new FormServlet().doPost(request, response);
+        method.setEntity(new UrlEncodedFormEntity(nvps));
+        HttpResponse httpResponse = client.execute(method);
 
-        verify(request).setAttribute("bmi", 20.0);
-
-        String result = sw.getBuffer().toString().trim();
-        assertEquals("20.0", result);
+        assertEquals("Success", httpResponse.getHeaders("Test")[0].getValue());
+        assertEquals("20.0", httpResponse.getHeaders("BMI")[0].getValue());
     }
 }