Report bad-request for invalid YAML

Added classes and modified code to report bad-request when a servlet
attempts to read invalid YAML.

Change-Id: Iacddee92a448fb69d5c778a3c3f3f2b5528983f7
Issue-ID: POLICY-2085
Signed-off-by: Jim Hahn <jrh3@att.com>

1 files changed, 91 insertions, 53 deletions

diff --git a/policy-endpoints/src/test/java/org/onap/policy/common/endpoints/http/server/test/RestServerTest.java b/policy-endpoints/src/test/java/org/onap/policy/common/endpoints/http/server/test/RestServerTest.java
index ee28b96d..cd40f012 100644
--- a/policy-endpoints/src/test/java/org/onap/policy/common/endpoints/http/server/test/RestServerTest.java
+++ b/policy-endpoints/src/test/java/org/onap/policy/common/endpoints/http/server/test/RestServerTest.java
@@ -20,6 +20,7 @@
 
 package org.onap.policy.common.endpoints.http.server.test;
 
+import static org.assertj.core.api.Assertions.assertThat;
 import static org.assertj.core.api.Assertions.assertThatIllegalArgumentException;
 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertNotNull;
@@ -34,6 +35,7 @@ import java.io.IOException;
 import java.io.PrintWriter;
 import java.net.HttpURLConnection;
 import java.net.URL;
+import java.nio.charset.StandardCharsets;
 import java.util.Arrays;
 import java.util.Base64;
 import java.util.Properties;
@@ -45,7 +47,7 @@ import javax.ws.rs.Produces;
 import javax.ws.rs.core.MediaType;
 import javax.ws.rs.core.Response;
 import lombok.Getter;
-import org.junit.After;
+import org.apache.commons.io.IOUtils;
 import org.junit.AfterClass;
 import org.junit.Before;
 import org.junit.BeforeClass;
@@ -56,6 +58,7 @@ import org.onap.policy.common.endpoints.http.server.HttpServletServerFactory;
 import org.onap.policy.common.endpoints.http.server.JsonExceptionMapper;
 import org.onap.policy.common.endpoints.http.server.RestServer;
 import org.onap.policy.common.endpoints.http.server.RestServer.Factory;
+import org.onap.policy.common.endpoints.http.server.YamlExceptionMapper;
 import org.onap.policy.common.endpoints.http.server.YamlMessageBodyHandler;
 import org.onap.policy.common.endpoints.http.server.aaf.AafAuthFilter;
 import org.onap.policy.common.endpoints.parameters.RestServerParameters;
@@ -66,6 +69,7 @@ import org.onap.policy.common.utils.network.NetworkUtil;
 import org.powermock.reflect.Whitebox;
 
 public class RestServerTest {
+    private static final String APPLICATION_YAML = "application/yaml";
     private static final String SERVER1 = "my-server-A";
     private static final String SERVER2 = "my-server-B";
     private static final String FACTORY_FIELD = "factory";
@@ -74,24 +78,56 @@ public class RestServerTest {
     private static final String PASS = "my-pass";
     private static final Integer PORT = 9876;
     private static final String USER = "my-user";
+
     private static Factory saveFactory;
+    private static RestServer realRest;
+    private static int realPort;
+    private static RestServerParameters params;
 
-    private RestServer realRest;
     private RestServer rest;
     private HttpServletServer server1;
     private HttpServletServer server2;
     private Factory factory;
     private HttpServletServerFactory serverFactory;
-    private RestServerParameters params;
+    private String errorMsg;
 
+    /**
+     * Starts the REST server.
+     * @throws Exception if an error occurs
+     */
     @BeforeClass
-    public static void setUpBeforeClass() {
+    public static void setUpBeforeClass() throws Exception {
         saveFactory = Whitebox.getInternalState(RestServer.class, FACTORY_FIELD);
+
+        realPort = NetworkUtil.allocPort();
+
+        initRealParams();
+
+        realRest = new RestServer(params, null, RealProvider.class) {
+            @Override
+            protected Properties getServerProperties(RestServerParameters restServerParameters, String names) {
+                Properties props = super.getServerProperties(restServerParameters, names);
+
+                String svcpfx = PolicyEndPointProperties.PROPERTY_HTTP_SERVER_SERVICES + "."
+                                + restServerParameters.getName();
+                props.setProperty(svcpfx + PolicyEndPointProperties.PROPERTY_HTTP_SWAGGER_SUFFIX, "false");
+
+                return props;
+            }
+        };
+
+        realRest.start();
+        assertTrue(NetworkUtil.isTcpPortOpen(params.getHost(), params.getPort(), 100, 100));
     }
 
+    /**
+     * Restores the factory and stops the REST server.
+     */
     @AfterClass
     public static void tearDownAfterClass() {
         Whitebox.setInternalState(RestServer.class, FACTORY_FIELD, saveFactory);
+
+        realRest.stop();
     }
 
     /**
@@ -103,7 +139,8 @@ public class RestServerTest {
         server2 = mock(HttpServletServer.class);
         factory = mock(Factory.class);
         serverFactory = mock(HttpServletServerFactory.class);
-        params = mock(RestServerParameters.class);
+
+        initParams();
 
         when(factory.getServerFactory()).thenReturn(serverFactory);
         when(serverFactory.build(any())).thenReturn(Arrays.asList(server1, server2));
@@ -111,27 +148,7 @@ public class RestServerTest {
         when(server1.getName()).thenReturn(SERVER1);
         when(server2.getName()).thenReturn(SERVER2);
 
-        when(params.getHost()).thenReturn(HOST);
-        when(params.getName()).thenReturn(PARAM_NAME);
-        when(params.getPassword()).thenReturn(PASS);
-        when(params.getPort()).thenReturn(PORT);
-        when(params.getUserName()).thenReturn(USER);
-        when(params.isAaf()).thenReturn(true);
-        when(params.isHttps()).thenReturn(true);
-
         Whitebox.setInternalState(RestServer.class, FACTORY_FIELD, factory);
-
-        realRest = null;
-    }
-
-    /**
-     * Stops the rest server.
-     */
-    @After
-    public void tearDown() {
-        if (realRest != null) {
-            realRest.stop();
-        }
     }
 
     @Test
@@ -213,41 +230,33 @@ public class RestServerTest {
         assertEquals("true", props.getProperty(svcpfx + PolicyEndPointProperties.PROPERTY_HTTP_HTTPS_SUFFIX));
         assertEquals("true", props.getProperty(svcpfx + PolicyEndPointProperties.PROPERTY_AAF_SUFFIX));
         assertEquals(String.join(",", GsonMessageBodyHandler.class.getName(), YamlMessageBodyHandler.class.getName(),
-                        JsonExceptionMapper.class.getName()),
+                        JsonExceptionMapper.class.getName(), YamlExceptionMapper.class.getName()),
                         props.getProperty(svcpfx + PolicyEndPointProperties.PROPERTY_HTTP_SERIALIZATION_PROVIDER));
     }
 
     @Test
     public void testInvalidJson() throws Exception {
-        when(params.getHost()).thenReturn("localhost");
-        when(params.getPort()).thenReturn(NetworkUtil.allocPort());
-        when(params.isHttps()).thenReturn(false);
-        when(params.isAaf()).thenReturn(false);
-
-        // use real factory
-        Whitebox.setInternalState(RestServer.class, FACTORY_FIELD, saveFactory);
-
-        realRest = new RestServer(params, null, RealProvider.class) {
-            @Override
-            protected Properties getServerProperties(RestServerParameters restServerParameters, String names) {
-                Properties props = super.getServerProperties(restServerParameters, names);
-
-                String svcpfx = PolicyEndPointProperties.PROPERTY_HTTP_SERVER_SERVICES + "."
-                                + restServerParameters.getName();
-                props.setProperty(svcpfx + PolicyEndPointProperties.PROPERTY_HTTP_SWAGGER_SUFFIX, "false");
-
-                return props;
-            }
-        };
-
-        realRest.start();
-        assertTrue(NetworkUtil.isTcpPortOpen(params.getHost(), params.getPort(), 100, 100));
+        initRealParams();
 
         assertEquals(200, roundTrip(new StandardCoder().encode(new MyRequest())));
         assertEquals(400, roundTrip("{'bogus-json'"));
+        assertThat(errorMsg).contains("Invalid request");
+    }
+
+    @Test
+    public void testInvalidYaml() throws Exception {
+        initRealParams();
+
+        assertEquals(200, roundTrip(new StandardCoder().encode(new MyRequest()), APPLICATION_YAML));
+        assertEquals(400, roundTrip("<bogus yaml", APPLICATION_YAML));
+        assertThat(errorMsg).contains("Invalid request");
     }
 
     private int roundTrip(String request) throws IOException {
+        return roundTrip(request, MediaType.APPLICATION_JSON);
+    }
+
+    private int roundTrip(String request, String mediaType) throws IOException {
         URL url = new URL("http://" + params.getHost() + ":" + params.getPort() + "/request");
         HttpURLConnection conn = (HttpURLConnection) url.openConnection();
         conn.setDoInput(true);
@@ -255,14 +264,22 @@ public class RestServerTest {
         conn.setRequestMethod("POST");
         String auth = params.getUserName() + ":" + params.getPassword();
         conn.setRequestProperty("Authorization", "Basic " + Base64.getEncoder().encodeToString(auth.getBytes()));
-        conn.setRequestProperty("Content-type", MediaType.APPLICATION_JSON);
+        conn.setRequestProperty("Content-type", mediaType);
         conn.connect();
 
         try (PrintWriter wtr = new PrintWriter(conn.getOutputStream())) {
             wtr.write(request);
         }
 
-        return conn.getResponseCode();
+        int code = conn.getResponseCode();
+
+        if (code == 200) {
+            errorMsg = "";
+        } else {
+            errorMsg = IOUtils.toString(conn.getErrorStream(), StandardCharsets.UTF_8);
+        }
+
+        return code;
     }
 
     @Test
@@ -277,6 +294,27 @@ public class RestServerTest {
         assertNotNull(saveFactory.getServerFactory());
     }
 
+    private static void initRealParams() {
+        initParams();
+
+        when(params.getHost()).thenReturn("localhost");
+        when(params.getPort()).thenReturn(realPort);
+        when(params.isHttps()).thenReturn(false);
+        when(params.isAaf()).thenReturn(false);
+    }
+
+    private static void initParams() {
+        params = mock(RestServerParameters.class);
+
+        when(params.getHost()).thenReturn(HOST);
+        when(params.getName()).thenReturn(PARAM_NAME);
+        when(params.getPassword()).thenReturn(PASS);
+        when(params.getPort()).thenReturn(PORT);
+        when(params.getUserName()).thenReturn(USER);
+        when(params.isAaf()).thenReturn(true);
+        when(params.isHttps()).thenReturn(true);
+    }
+
     private static class Filter extends AafAuthFilter {
         @Override
         protected String getPermissionType(HttpServletRequest request) {
@@ -302,8 +340,8 @@ public class RestServerTest {
     }
 
     @Path("/")
-    @Produces(MediaType.APPLICATION_JSON)
-    @Consumes(MediaType.APPLICATION_JSON)
+    @Produces({MediaType.APPLICATION_JSON, APPLICATION_YAML})
+    @Consumes({MediaType.APPLICATION_JSON, APPLICATION_YAML})
     public static class RealProvider {
         @POST
         @Path("/request")