--- /dev/null
+/*
+ * Copyright (c) 2014 Cisco Systems, Inc. and others. All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html.html
+ */
+
+package org.opendaylight.controller.netconf.it;
+
+import static java.lang.Thread.sleep;
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.fail;
+import static org.opendaylight.controller.netconf.it.NetconfITSecureTest.getSessionListener;
+
+import com.google.common.base.Throwables;
+import com.google.common.collect.Lists;
+import com.google.common.util.concurrent.FutureCallback;
+import com.google.common.util.concurrent.Futures;
+import com.google.common.util.concurrent.ListenableFuture;
+import io.netty.channel.nio.NioEventLoopGroup;
+import io.netty.util.HashedWheelTimer;
+import java.net.InetSocketAddress;
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.LinkedList;
+import java.util.List;
+import java.util.concurrent.ExecutorService;
+import java.util.concurrent.Executors;
+import java.util.concurrent.Future;
+import java.util.concurrent.TimeUnit;
+import java.util.concurrent.TimeoutException;
+import java.util.concurrent.atomic.AtomicInteger;
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Test;
+import org.opendaylight.controller.netconf.api.NetconfMessage;
+import org.opendaylight.controller.netconf.client.NetconfClientDispatcher;
+import org.opendaylight.controller.netconf.client.NetconfClientDispatcherImpl;
+import org.opendaylight.controller.netconf.client.TestingNetconfClient;
+import org.opendaylight.controller.netconf.test.tool.Main.Params;
+import org.opendaylight.controller.netconf.test.tool.NetconfDeviceSimulator;
+import org.opendaylight.controller.netconf.util.test.XmlFileLoader;
+import org.opendaylight.controller.sal.connect.netconf.listener.NetconfDeviceCommunicator;
+import org.opendaylight.yangtools.yang.common.QName;
+import org.opendaylight.yangtools.yang.common.RpcResult;
+
+public class NetconfITSecureTestTool
+{
+
+ //set up port both for testool device and test
+ public static final int PORT = 17833;
+ private static final InetSocketAddress TLS_ADDRESS = new InetSocketAddress("127.0.0.1", PORT);
+
+ private String xmlFile = "netconfMessages/editConfig.xml";
+
+ private ExecutorService msgExec = Executors.newFixedThreadPool(8);
+
+ Collection<Future<?>> tasks = new LinkedList<Future<?>>();
+
+ final NetconfDeviceSimulator netconfDeviceSimulator = new NetconfDeviceSimulator();
+
+ @Before
+ public void setUp() throws Exception {
+
+ //Set up parameters for testtool device
+ Params params = new Params();
+ params.debug = true;
+ params.deviceCount = 1;
+ params.startingPort = PORT;
+ params.ssh = true;
+ params.exi = true;
+
+ final List<Integer> openDevices = netconfDeviceSimulator.start(params);
+ }
+
+ @After
+ public void tearDown() throws Exception {
+
+ }
+
+ /**
+ * Test all requests are handled properly and no mismatch occurs in listener
+ */
+ @Test(timeout = 6*60*1000)
+ public void testSecureStress() throws Exception {
+
+ final int requests = 4000;
+
+ List<Future<?>> tasks = new ArrayList<>();
+
+ final NetconfClientDispatcher dispatch = new NetconfClientDispatcherImpl(new NioEventLoopGroup(), new NioEventLoopGroup(), new HashedWheelTimer());
+
+ final NetconfDeviceCommunicator sessionListener = getSessionListener();
+
+ try (TestingNetconfClient netconfClient = new TestingNetconfClient("testing-ssh-client", dispatch, NetconfITSecureTest.getClientConfiguration(sessionListener, TLS_ADDRESS));)
+ {
+
+ final AtomicInteger responseCounter = new AtomicInteger(0);
+ final List<ListenableFuture<RpcResult<NetconfMessage>>> futures = Lists.newArrayList();
+
+ for (int i = 0; i < requests; i++) {
+
+ NetconfMessage getConfig = XmlFileLoader.xmlFileToNetconfMessage(xmlFile);
+
+ getConfig = NetconfITSecureTest.changeMessageId(getConfig,i);
+
+ Runnable worker = new NetconfITSecureTestToolRunnable(getConfig,i, sessionListener, futures, responseCounter);
+
+ tasks.add(msgExec.submit(worker));
+
+ }
+
+ msgExec.shutdown();
+
+ // Wait for every future
+ for (final Future<?> task : tasks){
+ try
+ {
+
+ task.get(3, TimeUnit.MINUTES);
+ } catch (final TimeoutException e) {
+ fail(String.format("Request %d is not responding", tasks.indexOf(task)));
+ }
+ }
+
+ for (final ListenableFuture<RpcResult<NetconfMessage>> future : futures) {
+ try {
+
+ future.get(3, TimeUnit.MINUTES);
+ } catch (final TimeoutException e) {
+ fail(String.format("Reply %d is not responding", futures.indexOf(future)));
+ }
+ }
+
+ sleep(5000);
+
+ assertEquals(requests, responseCounter.get());
+
+ }
+ }
+
+ class NetconfITSecureTestToolRunnable implements Runnable {
+
+ private NetconfMessage getConfig;
+ private int it;
+ private NetconfDeviceCommunicator sessionListener;
+ private List<ListenableFuture<RpcResult<NetconfMessage>>> futures;
+ private AtomicInteger responseCounter;
+
+ public NetconfITSecureTestToolRunnable(NetconfMessage getConfig, int it, NetconfDeviceCommunicator sessionListener, List<ListenableFuture<RpcResult<NetconfMessage>>> futures, AtomicInteger responseCounter){
+ this.getConfig = getConfig;
+ this.it = it;
+ this.sessionListener = sessionListener;
+ this.futures = futures;
+ this.responseCounter = responseCounter;
+ }
+
+ @Override
+ public void run(){
+
+ ListenableFuture<RpcResult<NetconfMessage>> netconfMessageFuture;
+
+ netconfMessageFuture = sessionListener.sendRequest(getConfig, QName.create("namespace", "2012-12-12", "get"));
+
+ futures.add(netconfMessageFuture);
+ Futures.addCallback(netconfMessageFuture, new FutureCallback<RpcResult<NetconfMessage>>() {
+
+ @Override
+ public void onSuccess(final RpcResult<NetconfMessage> result) {
+
+ if(result.isSuccessful()&result.getErrors().isEmpty()) {
+ responseCounter.incrementAndGet();
+ } else {
+
+ fail(String.format("Message result not ok %s", result.getErrors().toString()));
+
+ }
+ }
+
+ @Override
+ public void onFailure(final Throwable t) {
+
+ fail(String.format("Message failed %s", Throwables.getStackTraceAsString(t)));
+
+ }
+ }
+ );
+ }
+ }
+
+}
Code Review / controller.git / blobdiff