Java tutorial
/* * Copyright (c) 2013 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 */ package org.opendaylight.netconf.it; import static org.junit.Assert.assertEquals; import static org.junit.Assert.assertNotNull; import com.google.common.base.Throwables; import com.google.common.io.ByteStreams; import com.google.common.util.concurrent.CheckedFuture; import io.netty.channel.Channel; import io.netty.channel.ChannelFuture; import io.netty.channel.EventLoopGroup; import io.netty.channel.local.LocalAddress; import io.netty.channel.nio.NioEventLoopGroup; import io.netty.util.HashedWheelTimer; import io.netty.util.concurrent.GlobalEventExecutor; import java.io.ByteArrayInputStream; import java.io.IOException; import java.io.InputStream; import java.net.InetSocketAddress; import java.net.SocketAddress; import java.util.ArrayList; import java.util.Arrays; import java.util.Collection; import java.util.Collections; import java.util.List; import java.util.concurrent.TimeUnit; import org.junit.After; import org.junit.Before; import org.opendaylight.controller.config.facade.xml.ConfigSubsystemFacadeFactory; import org.opendaylight.controller.config.facade.xml.osgi.EnumResolver; import org.opendaylight.controller.config.facade.xml.osgi.YangStoreService; import org.opendaylight.controller.config.manager.impl.AbstractConfigTest; import org.opendaylight.controller.config.manager.impl.factoriesresolver.HardcodedModuleFactoriesResolver; import org.opendaylight.controller.config.spi.ModuleFactory; import org.opendaylight.controller.config.yang.test.impl.DepTestImplModuleFactory; import org.opendaylight.controller.config.yang.test.impl.IdentityTestModuleFactory; import org.opendaylight.controller.config.yang.test.impl.MultipleDependenciesModuleFactory; import org.opendaylight.controller.config.yang.test.impl.NetconfTestImplModuleFactory; import org.opendaylight.controller.config.yang.test.impl.TestImplModuleFactory; import org.opendaylight.netconf.api.NetconfMessage; import org.opendaylight.netconf.api.monitoring.NetconfMonitoringService; import org.opendaylight.netconf.client.NetconfClientDispatcherImpl; import org.opendaylight.netconf.client.SimpleNetconfClientSessionListener; import org.opendaylight.netconf.client.conf.NetconfClientConfiguration; import org.opendaylight.netconf.client.conf.NetconfClientConfigurationBuilder; import org.opendaylight.netconf.confignetconfconnector.osgi.NetconfOperationServiceFactoryImpl; import org.opendaylight.netconf.impl.NetconfServerDispatcherImpl; import org.opendaylight.netconf.impl.NetconfServerSessionNegotiatorFactory; import org.opendaylight.netconf.impl.SessionIdProvider; import org.opendaylight.netconf.impl.osgi.AggregatedNetconfOperationServiceFactory; import org.opendaylight.netconf.impl.osgi.NetconfMonitoringServiceImpl; import org.opendaylight.netconf.mapping.api.NetconfOperationServiceFactory; import org.opendaylight.netconf.monitoring.osgi.NetconfMonitoringActivator; import org.opendaylight.netconf.monitoring.osgi.NetconfMonitoringOperationService; import org.opendaylight.netconf.util.test.XmlFileLoader; import org.opendaylight.protocol.framework.NeverReconnectStrategy; import org.opendaylight.yangtools.sal.binding.generator.util.BindingRuntimeContext; import org.opendaylight.yangtools.yang.binding.BindingMapping; import org.opendaylight.yangtools.yang.model.api.SchemaContext; import org.opendaylight.yangtools.yang.model.api.SchemaContextProvider; import org.opendaylight.yangtools.yang.model.repo.api.SchemaSourceException; import org.opendaylight.yangtools.yang.model.repo.api.SourceIdentifier; import org.opendaylight.yangtools.yang.model.repo.api.YangTextSchemaSource; import org.opendaylight.yangtools.yang.model.repo.spi.SchemaSourceProvider; import org.opendaylight.yangtools.yang.parser.spi.meta.ReactorException; import org.opendaylight.yangtools.yang.parser.spi.source.SourceException; import org.opendaylight.yangtools.yang.parser.stmt.rfc6020.YangInferencePipeline; public abstract class AbstractNetconfConfigTest extends AbstractConfigTest { public static final String LOOPBACK_ADDRESS = "127.0.0.1"; public static final int SERVER_CONNECTION_TIMEOUT_MILLIS = 5000; private static final int RESOURCE_TIMEOUT_MINUTES = 2; static ModuleFactory[] FACTORIES = { new TestImplModuleFactory(), new DepTestImplModuleFactory(), new NetconfTestImplModuleFactory(), new IdentityTestModuleFactory(), new MultipleDependenciesModuleFactory() }; protected ConfigSubsystemFacadeFactory configSubsystemFacadeFactory; private EventLoopGroup nettyThreadgroup; private HashedWheelTimer hashedWheelTimer; private NetconfClientDispatcherImpl clientDispatcher; private Channel serverTcpChannel; private NetconfMessage getConfig; private NetconfMessage get; /** * @Before in subclasses is called after this method. */ @Before public void setUpAbstractNetconfConfigTest() throws Exception { super.initConfigTransactionManagerImpl(new HardcodedModuleFactoriesResolver(mockedContext, FACTORIES)); nettyThreadgroup = new NioEventLoopGroup(); hashedWheelTimer = new HashedWheelTimer(); loadMessages(); setUpTestInitial(); final AggregatedNetconfOperationServiceFactory factoriesListener = new AggregatedNetconfOperationServiceFactory(); final NetconfMonitoringService netconfMonitoringService = getNetconfMonitoringService(factoriesListener); configSubsystemFacadeFactory = new ConfigSubsystemFacadeFactory(configRegistryClient, configRegistryClient, getYangStore()); factoriesListener.onAddNetconfOperationServiceFactory( new NetconfOperationServiceFactoryImpl(configSubsystemFacadeFactory)); factoriesListener.onAddNetconfOperationServiceFactory( new NetconfMonitoringActivator.NetconfMonitoringOperationServiceFactory( new NetconfMonitoringOperationService(netconfMonitoringService))); for (final NetconfOperationServiceFactory netconfOperationServiceFactory : getAdditionalServiceFactories( factoriesListener)) { factoriesListener.onAddNetconfOperationServiceFactory(netconfOperationServiceFactory); } serverTcpChannel = startNetconfTcpServer(factoriesListener, netconfMonitoringService); clientDispatcher = new NetconfClientDispatcherImpl(getNettyThreadgroup(), getNettyThreadgroup(), getHashedWheelTimer()); } /** * Called before setUp method is executed, so test classes can set up resources before * setUpAbstractNetconfConfigTest method is called. */ protected void setUpTestInitial() throws Exception { } private void loadMessages() throws Exception { this.getConfig = XmlFileLoader.xmlFileToNetconfMessage("netconfMessages/getConfig.xml"); this.get = XmlFileLoader.xmlFileToNetconfMessage("netconfMessages/get.xml"); } public NetconfMessage getGetConfig() { return getConfig; } public NetconfMessage getGet() { return get; } private Channel startNetconfTcpServer(final AggregatedNetconfOperationServiceFactory listener, final NetconfMonitoringService monitoring) throws Exception { final NetconfServerDispatcherImpl dispatch = createDispatcher(listener, monitoring); final ChannelFuture s; if (getTcpServerAddress() instanceof LocalAddress) { s = dispatch.createLocalServer(((LocalAddress) getTcpServerAddress())); } else { s = dispatch.createServer(((InetSocketAddress) getTcpServerAddress())); } s.await(RESOURCE_TIMEOUT_MINUTES, TimeUnit.MINUTES); return s.channel(); } protected Iterable<NetconfOperationServiceFactory> getAdditionalServiceFactories( final AggregatedNetconfOperationServiceFactory factoriesListener) throws Exception { return Collections.emptySet(); } protected NetconfMonitoringService getNetconfMonitoringService( final AggregatedNetconfOperationServiceFactory factoriesListener) throws Exception { return new NetconfMonitoringServiceImpl(factoriesListener); } protected abstract SocketAddress getTcpServerAddress(); public NetconfClientDispatcherImpl getClientDispatcher() { return clientDispatcher; } private HardcodedYangStoreService getYangStore() throws IOException { final Collection<InputStream> yangDependencies = getBasicYangs(); return new HardcodedYangStoreService(yangDependencies, getBindingRuntimeContext()); } static Collection<InputStream> getBasicYangs() throws IOException { final List<String> paths = Arrays.asList("/META-INF/yang/config.yang", "/META-INF/yang/rpc-context.yang", "/META-INF/yang/config-test.yang", "/META-INF/yang/config-test-impl.yang", "/META-INF/yang/test-types.yang", "/META-INF/yang/test-groups.yang", "/META-INF/yang/ietf-inet-types.yang"); final Collection<InputStream> yangDependencies = new ArrayList<>(); final List<String> failedToFind = new ArrayList<>(); for (final String path : paths) { final InputStream resourceAsStream = NetconfITTest.class.getResourceAsStream(path); if (resourceAsStream == null) { failedToFind.add(path); } else { yangDependencies.add(resourceAsStream); } } assertEquals("Some yang files were not found", Collections.<String>emptyList(), failedToFind); return yangDependencies; } protected NetconfServerDispatcherImpl createDispatcher( final AggregatedNetconfOperationServiceFactory factoriesListener, final NetconfMonitoringService sessionMonitoringService) { final SessionIdProvider idProvider = new SessionIdProvider(); final NetconfServerSessionNegotiatorFactory serverNegotiatorFactory = new NetconfServerSessionNegotiatorFactory( hashedWheelTimer, factoriesListener, idProvider, SERVER_CONNECTION_TIMEOUT_MILLIS, sessionMonitoringService); final NetconfServerDispatcherImpl.ServerChannelInitializer serverChannelInitializer = new NetconfServerDispatcherImpl.ServerChannelInitializer( serverNegotiatorFactory); return new NetconfServerDispatcherImpl(serverChannelInitializer, nettyThreadgroup, nettyThreadgroup); } protected HashedWheelTimer getHashedWheelTimer() { return hashedWheelTimer; } protected EventLoopGroup getNettyThreadgroup() { return nettyThreadgroup; } /** * @After in subclasses is be called before this. */ @After public void cleanUpNetconf() throws Exception { serverTcpChannel.close().await(RESOURCE_TIMEOUT_MINUTES, TimeUnit.MINUTES); hashedWheelTimer.stop(); nettyThreadgroup.shutdownGracefully().await(RESOURCE_TIMEOUT_MINUTES, TimeUnit.MINUTES); } public NetconfClientConfiguration getClientConfiguration(final InetSocketAddress tcpAddress, final int timeout) { final NetconfClientConfigurationBuilder b = NetconfClientConfigurationBuilder.create(); b.withAddress(tcpAddress); b.withSessionListener(new SimpleNetconfClientSessionListener()); b.withReconnectStrategy(new NeverReconnectStrategy(GlobalEventExecutor.INSTANCE, timeout)); b.withConnectionTimeoutMillis(timeout); return b.build(); } public static final class HardcodedYangStoreService extends YangStoreService { public HardcodedYangStoreService(final Collection<? extends InputStream> inputStreams, final BindingRuntimeContext bindingRuntimeContext) throws IOException { super(new SchemaContextProvider() { @Override public SchemaContext getSchemaContext() { return getSchema(inputStreams); } }, new SchemaSourceProvider<YangTextSchemaSource>() { @Override public CheckedFuture<? extends YangTextSchemaSource, SchemaSourceException> getSource( SourceIdentifier sourceIdentifier) { return null; } }); refresh(bindingRuntimeContext); } private static SchemaContext getSchema(final Collection<? extends InputStream> inputStreams) { final ArrayList<InputStream> byteArrayInputStreams = new ArrayList<>(); for (final InputStream inputStream : inputStreams) { assertNotNull(inputStream); final byte[] content; try { content = ByteStreams.toByteArray(inputStream); } catch (IOException e) { throw new IllegalStateException("Cannot read " + inputStream, e); } final ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(content); byteArrayInputStreams.add(byteArrayInputStream); } for (final InputStream inputStream : byteArrayInputStreams) { try { inputStream.reset(); } catch (final IOException e) { throw new RuntimeException(e); } } try { return YangInferencePipeline.RFC6020_REACTOR.newBuild().buildEffective(byteArrayInputStreams); } catch (SourceException | ReactorException e) { throw Throwables.propagate(e); } } @Override public EnumResolver getEnumResolver() { return new EnumResolver() { @Override public String fromYang(final String enumType, final String enumYangValue) { return BindingMapping.getClassName(enumYangValue); } @Override public String toYang(final String enumType, final String enumJavaValue) { return enumJavaValue.toLowerCase(); } }; } } }