Java tutorial
/* * Copyright 2015 Cask Data, Inc. * * 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 co.cask.cdap.internal.app.services.http.handlers; import co.cask.cdap.gateway.handlers.DashboardHttpHandler; import co.cask.cdap.internal.app.services.http.AppFabricTestBase; import com.google.common.collect.Maps; import com.google.gson.Gson; import com.google.gson.JsonArray; import com.google.gson.JsonElement; import com.google.gson.JsonObject; import com.google.gson.JsonParser; import com.google.gson.reflect.TypeToken; import org.apache.http.HttpResponse; import org.apache.http.util.EntityUtils; import org.jboss.netty.handler.codec.http.HttpResponseStatus; import org.junit.Assert; import org.junit.Test; import java.lang.reflect.Type; import java.util.HashMap; import java.util.Map; /** * Tests for {@link DashboardHttpHandler} */ public class DashboardHttpHandlerTest extends AppFabricTestBase { private static final Gson GSON = new Gson(); private static final Type MAP_STRING_STRING_TYPE = new TypeToken<Map<String, String>>() { }.getType(); @Test public void testCleanSlate() throws Exception { JsonElement dash = getDashboards("mynamespace"); Assert.assertTrue(dash.isJsonArray()); Assert.assertEquals(0, dash.getAsJsonArray().size()); String s = createDashboard("mynamespace", 200); dash = getDashboards("mynamespace"); Assert.assertTrue(dash.isJsonArray()); Assert.assertEquals(1, dash.getAsJsonArray().size()); String id = dash.getAsJsonArray().get(0).getAsJsonObject().get("id").getAsString(); Assert.assertEquals(id, s); Assert.assertEquals(new JsonParser().parse("{}"), dash.getAsJsonArray().get(0).getAsJsonObject().get("config")); deleteDashboard("mynamespace", s, 200); deleteDashboard("mynamespace", s, 404); dash = getDashboards("mynamespace"); Assert.assertTrue(dash.isJsonArray()); Assert.assertEquals(0, dash.getAsJsonArray().size()); } @Test public void testMultiNamespace() throws Exception { int maxRuns = 5; for (int run = 0; run < maxRuns; run++) { int maxNamespace = 10; Map<Integer, String> dashboardIds = Maps.newHashMap(); for (int nsId = 0; nsId < maxNamespace; nsId++) { dashboardIds.put(nsId, createDashboard("myspace" + nsId, 200)); } JsonArray dashboards; for (int nsId = 0; nsId < maxNamespace; nsId++) { dashboards = getDashboards("myspace" + nsId).getAsJsonArray(); Assert.assertEquals(1, dashboards.size()); Assert.assertEquals(dashboardIds.get(nsId), dashboards.get(0).getAsJsonObject().get("id").getAsString()); } for (int nsId = 0; nsId < maxNamespace; nsId++) { deleteDashboard("myspace" + nsId, dashboardIds.get(nsId), 200); deleteDashboard("myspace" + nsId, dashboardIds.get(nsId), 404); Assert.assertEquals(0, getDashboards("myspace" + nsId).getAsJsonArray().size()); } } } @Test public void testProperties() throws Exception { Map<String, Integer> intMap = Maps.newHashMap(); intMap.put("k1", 123); intMap.put("k2", 324); String dash = createDashboard("newspace", GSON.toJson(intMap), 200); JsonObject jsonObject = getContents("newspace", dash, 200).getAsJsonObject().get("config") .getAsJsonObject(); Assert.assertEquals(2, jsonObject.entrySet().size()); Assert.assertEquals(123, jsonObject.get("k1").getAsInt()); Assert.assertEquals(324, jsonObject.get("k2").getAsInt()); Map<String, String> propMap = Maps.newHashMap(); propMap.put("k2", "value2"); propMap.put("k1", "value1"); addProperty("newspace", dash, propMap, 200); jsonObject = getContents("newspace", dash, 200).getAsJsonObject().get("config").getAsJsonObject(); Assert.assertEquals(2, jsonObject.entrySet().size()); Assert.assertEquals("value1", jsonObject.get("k1").getAsString()); Assert.assertEquals("value2", jsonObject.get("k2").getAsString()); propMap.clear(); propMap.put("m1", "n1"); String anotherDash = createDashboard("newspace", GSON.toJson(propMap), 200); jsonObject = getContents("newspace", anotherDash, 200).getAsJsonObject().get("config").getAsJsonObject(); Assert.assertEquals(1, jsonObject.entrySet().size()); Assert.assertEquals("n1", jsonObject.get("m1").getAsString()); addProperty("newspace", anotherDash, new HashMap<String, String>(), 200); jsonObject = getContents("newspace", anotherDash, 200).getAsJsonObject().get("config").getAsJsonObject(); Assert.assertEquals(0, jsonObject.entrySet().size()); String str = "some123 random string!@#"; createDashboard("space", str, 400); deleteDashboard("newspace", dash, 200); deleteDashboard("newspace", dash, 404); deleteDashboard("newspace", anotherDash, 200); } @Test public void testGetDashboards() throws Exception { String dash1 = createDashboard("space1", 200); String dash2 = createDashboard("space2", 200); JsonArray dashList = getDashboards("space1").getAsJsonArray(); Assert.assertEquals(1, dashList.size()); deleteDashboard("space1", dash1, 200); deleteDashboard("space2", dash2, 200); } private void addProperty(String namespace, String name, Map<String, String> props, int expectedStatus) throws Exception { HttpResponse response = doPut( String.format("/v3/namespaces/%s/configuration/dashboards/%s", namespace, name), GSON.toJson(props)); Assert.assertEquals(expectedStatus, response.getStatusLine().getStatusCode()); } private JsonElement getContents(String namespace, String name, int expectedStatus) throws Exception { HttpResponse response = doGet( String.format("/v3/namespaces/%s/configuration/dashboards/%s", namespace, name)); Assert.assertEquals(expectedStatus, response.getStatusLine().getStatusCode()); String s = EntityUtils.toString(response.getEntity()); return new JsonParser().parse(s); } private String createDashboard(String namespace, int expectedStatus) throws Exception { return createDashboard(namespace, null, expectedStatus); } private String createDashboard(String namespace, String contents, int expectedStatus) throws Exception { HttpResponse response = doPost(String.format("/v3/namespaces/%s/configuration/dashboards", namespace), contents); Assert.assertEquals(expectedStatus, response.getStatusLine().getStatusCode()); if (expectedStatus == HttpResponseStatus.OK.getCode()) { String jsonData = EntityUtils.toString(response.getEntity()); Map<String, String> idMap = GSON.fromJson(jsonData, MAP_STRING_STRING_TYPE); Assert.assertEquals(1, idMap.size()); Assert.assertEquals(true, idMap.containsKey("id")); return idMap.get("id"); } return null; } private JsonElement getDashboards(String namespace) throws Exception { String req = String.format("/v3/namespaces/%s/configuration/dashboards", namespace); HttpResponse response = doGet(req); Assert.assertEquals(200, response.getStatusLine().getStatusCode()); String s = EntityUtils.toString(response.getEntity()); return new JsonParser().parse(s); } private void deleteDashboard(String namespace, String name, int expectedStatus) throws Exception { HttpResponse response = doDelete( String.format("/v3/namespaces/%s/configuration/dashboards/%s", namespace, name)); Assert.assertEquals(expectedStatus, response.getStatusLine().getStatusCode()); } }