/* * Copyright (c) 2016, WSO2 Inc. (http://www.wso2.org) All Rights Reserved. * * WSO2 Inc. licenses this file to you 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 org.wso2.carbon.esb.endpoint.test; import java.io.IOException; import java.io.InputStream; import java.io.OutputStream; import java.net.URL; import java.net.URLConnection; import org.testng.Assert; import org.testng.annotations.BeforeClass; import org.testng.annotations.Test; import org.wso2.esb.integration.common.utils.ESBIntegrationTest; public class DynamicTimeoutEndpointTestCase extends ESBIntegrationTest { @BeforeClass(alwaysRun = true) public void init() throws Exception { super.init(); } @Test(groups = {"wso2.esb"}, description = "Test default endpoint with dynamic timeout") public void testDynamicTimeoutEndpoint() throws IOException { String request = "{}"; String response = sendRequest(getProxyServiceURLHttp("MockTimeoutProxy"), request); Assert.assertEquals(response, "{ \"quote\" : \"IBM\"}"); } @Test(groups = {"wso2.esb"}, description = "Test delayed endpoint with dynamic timeout") public void testDynamicDelayedTimeoutEndpoint() throws IOException { String request = "{}"; String response = sendRequest(getProxyServiceURLHttp("MockDelayedTimeoutProxy"), request); Assert.assertEquals(response, "{\"error_code\" : 101504, \"error_msg\" : \"Send timeout\"}"); } @Test(groups = {"wso2.esb"}, description = "Test template endpoint with dynamic timeout") public void testDynamicTimeoutTemplateEndpoint() throws IOException { String request = "{}"; String response = sendRequest(getProxyServiceURLHttp("MockTimeoutTemplateProxy"), request); Assert.assertEquals(response, "{ \"quote\" : \"IBM\"}"); } @Test(groups = {"wso2.esb"}, description = "Test template endpoint with dynamic delayed timeout") public void testDynamicDelayedTimeoutTemplateEndpoint() throws IOException { String request = "{}"; String response = sendRequest(getProxyServiceURLHttp("MockDelayedTimeoutTemplateProxy"), request); Assert.assertEquals(response, "{\"error_code\" : 101504, \"error_msg\" : \"Send timeout\"}"); } private String sendRequest(String addUrl, String request) throws IOException { String charset = "UTF-8"; URLConnection connection = new URL(addUrl).openConnection(); connection.setDoOutput(true); connection.setRequestProperty("Accept-Charset", charset); connection.setRequestProperty("Content-Type", "application/json" + ";charset=" + charset); OutputStream output = null; try { output = connection.getOutputStream(); output.write(request.getBytes(charset)); } finally { if (output != null) { output.close(); } } InputStream response = connection.getInputStream(); String out = "[Fault] No Response."; if (response != null) { StringBuilder sb = new StringBuilder(); byte[] bytes = new byte[1024]; int len; while ((len = response.read(bytes)) != -1) { sb.append(new String(bytes, 0, len)); } out = sb.toString(); } return out; } }