/*
*Copyright (c) 2005-2010, 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.mediator.test.throttle;
import org.testng.annotations.AfterClass;
import org.testng.annotations.BeforeClass;
import org.testng.annotations.Test;
import org.wso2.esb.integration.common.utils.ESBIntegrationTest;
import org.wso2.carbon.esb.mediator.test.throttle.utils.ConcurrencyAndRequestThrottleTestClient;
import org.wso2.carbon.esb.mediator.test.throttle.utils.ThrottleTestCounter;
import java.util.ArrayList;
import java.util.Calendar;
import java.util.Collections;
import java.util.List;
import static org.testng.Assert.assertEquals;
import static org.testng.Assert.assertTrue;
public class ConcurrencyAndRequestThrottlingTestRequest extends ESBIntegrationTest {
private final int CONCURRENT_CLIENTS = 10;
private final int AS_POLICY_ACCESS_GRANTED = 4;
private final int AS_POLICY_ACCESS_DENIED = 6;
private final int THROTTLE_MAX_MSG_COUNT = 4;
//Since Max Msg Count is 4 even concurrent access is 10 it will throttle after 4
private List list;
private ConcurrencyAndRequestThrottleTestClient[] concurrencyAndRequestThrottleTestClients;
private Thread[] clients;
private ThrottleTestCounter clientsDone;
private int grantedRequests;
private int deniedRequests;
private ThrottleTestCounter requestThrottledClients;
@BeforeClass(alwaysRun = true)
public void setEnvironment() throws Exception {
super.init();
loadESBConfigurationFromClasspath("/artifacts/ESB/synapseconfig/throttle/ConcurrencyAndRequestBasedThrottlingRequest.xml");
list = Collections.synchronizedList(new ArrayList());
concurrencyAndRequestThrottleTestClients = new ConcurrencyAndRequestThrottleTestClient[CONCURRENT_CLIENTS];
clients = new Thread[CONCURRENT_CLIENTS];
clientsDone = new ThrottleTestCounter();
requestThrottledClients = new ThrottleTestCounter();
grantedRequests = 0;
deniedRequests = 0;
initClients(); //initialising Axis2Clients
}
//use file system registry
@Test(groups = "wso2.esb",
description = "Use of both concurrency throttling and request rate based throttling- Throttle Request Massages", enabled = false)
public void testConcurrencyAndRequestBasedPolicyThrottlingRequest()
throws InterruptedException {
startClients();
while (clientsDone.getCount() < CONCURRENT_CLIENTS) {
Thread.sleep(1000);
}
for (Object aList : list) {
if (aList.toString().equals("Access Granted")) {
grantedRequests++;
} else if (aList.toString().equals("Access Denied")) {
deniedRequests++;
}
}
assertEquals(grantedRequests, AS_POLICY_ACCESS_GRANTED, "Fault: Concurrent throttle policy failure");
assertEquals(deniedRequests, AS_POLICY_ACCESS_DENIED, "Fault: Concurrent throttle policy failure");
assertTrue(grantedRequests == requestThrottledClients.getCount(), "Fault: Request throttle policy failure");
}
@AfterClass(alwaysRun = true)
public void destroy() throws Exception {
concurrencyAndRequestThrottleTestClients = null;
list = null;
clients = null;
clientsDone = null;
requestThrottledClients = null;
super.cleanup();
}
private void initClients() {
for (int i = 0; i < CONCURRENT_CLIENTS; i++) {
concurrencyAndRequestThrottleTestClients[i] = new ConcurrencyAndRequestThrottleTestClient(
getProxyServiceURLHttp("StockQuoteProxy"), list, clientsDone, requestThrottledClients, THROTTLE_MAX_MSG_COUNT);
}
for (int i = 0; i < CONCURRENT_CLIENTS; i++) {
clients[i] = new Thread(concurrencyAndRequestThrottleTestClients[i]);
}
}
private void startClients() {
for (int i = 0; i < CONCURRENT_CLIENTS; i++) {
clients[i].start();
}
int aliveCount = 0;
Calendar startTime = Calendar.getInstance();
while (aliveCount < CONCURRENT_CLIENTS) {
if ((Calendar.getInstance().getTimeInMillis() - startTime.getTimeInMillis()) > 60000) {
break;
}
if (clients[aliveCount].isAlive()) {
aliveCount = 0;
continue;
}
aliveCount++;
}
}
}