X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?a=blobdiff_plain;f=bgp%2Frib-impl%2Fsrc%2Ftest%2Fjava%2Forg%2Fopendaylight%2Fprotocol%2Fbgp%2Frib%2Fimpl%2FParserToSalTest.java;h=ce0ae9ac25d1228850a5eeeb376b175cb3a87cad;hb=2072c47aca26c89985fac7a5d09937d93a9fb14b;hp=8c04a1d542286e213f8a8ede8e963c39c344997d;hpb=f00300b6aa21e28947cb6bab56e5ee8a4d3e46f8;p=bgpcep.git diff --git a/bgp/rib-impl/src/test/java/org/opendaylight/protocol/bgp/rib/impl/ParserToSalTest.java b/bgp/rib-impl/src/test/java/org/opendaylight/protocol/bgp/rib/impl/ParserToSalTest.java old mode 100644 new mode 100755 index 8c04a1d542..ce0ae9ac25 --- a/bgp/rib-impl/src/test/java/org/opendaylight/protocol/bgp/rib/impl/ParserToSalTest.java +++ b/bgp/rib-impl/src/test/java/org/opendaylight/protocol/bgp/rib/impl/ParserToSalTest.java @@ -9,123 +9,125 @@ package org.opendaylight.protocol.bgp.rib.impl; import static org.junit.Assert.assertFalse; import static org.junit.Assert.assertTrue; +import static org.mockito.Matchers.any; +import static org.mockito.Mockito.doReturn; +import static org.opendaylight.protocol.bgp.rib.impl.AbstractAddPathTest.AS_NUMBER; +import static org.opendaylight.protocol.bgp.rib.impl.AbstractAddPathTest.BGP_ID; +import static org.opendaylight.protocol.util.CheckUtil.readDataOperational; -import com.google.common.base.Function; -import com.google.common.base.Optional; -import com.google.common.base.Throwables; import com.google.common.collect.Collections2; import com.google.common.collect.ImmutableList; import com.google.common.collect.Lists; import com.google.common.eventbus.EventBus; import io.netty.util.concurrent.GlobalEventExecutor; -import java.io.IOException; import java.net.InetSocketAddress; import java.util.Collection; +import java.util.Collections; import java.util.List; import java.util.concurrent.ExecutionException; -import javax.annotation.Nullable; import org.junit.After; +import org.junit.Before; import org.junit.Test; import org.mockito.Mock; import org.mockito.Mockito; import org.mockito.MockitoAnnotations; -import org.opendaylight.controller.md.sal.binding.api.DataBroker; -import org.opendaylight.controller.md.sal.binding.test.AbstractDataBrokerTest; -import org.opendaylight.controller.md.sal.binding.test.DataBrokerTestCustomizer; -import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType; -import org.opendaylight.controller.sal.core.api.model.SchemaService; +import org.opendaylight.controller.md.sal.binding.test.AbstractConcurrentDataBrokerTest; +import org.opendaylight.controller.md.sal.binding.test.AbstractDataBrokerTestCustomizer; +import org.opendaylight.controller.md.sal.common.api.data.ReadFailedException; +import org.opendaylight.mdsal.binding.dom.codec.api.BindingCodecTreeFactory; +import org.opendaylight.mdsal.binding.generator.impl.GeneratedClassLoadingStrategy; +import org.opendaylight.mdsal.dom.api.DOMSchemaService; +import org.opendaylight.mdsal.singleton.common.api.ClusterSingletonService; +import org.opendaylight.mdsal.singleton.common.api.ClusterSingletonServiceProvider; +import org.opendaylight.mdsal.singleton.common.api.ClusterSingletonServiceRegistration; +import org.opendaylight.protocol.bgp.inet.RIBActivator; +import org.opendaylight.protocol.bgp.mode.impl.base.BasePathSelectionModeFactory; import org.opendaylight.protocol.bgp.parser.BgpTableTypeImpl; import org.opendaylight.protocol.bgp.parser.spi.pojo.ServiceLoaderBGPExtensionProviderContext; import org.opendaylight.protocol.bgp.rib.impl.spi.BGPDispatcher; -import org.opendaylight.protocol.bgp.rib.impl.spi.BGPPeerRegistry; import org.opendaylight.protocol.bgp.rib.mock.BGPMock; import org.opendaylight.protocol.bgp.rib.spi.AbstractRIBExtensionProviderActivator; import org.opendaylight.protocol.bgp.rib.spi.RIBExtensionProviderContext; import org.opendaylight.protocol.bgp.rib.spi.SimpleRIBExtensionProviderContext; import org.opendaylight.protocol.bgp.util.HexDumpBGPFileParser; -import org.opendaylight.protocol.framework.ReconnectStrategyFactory; -import org.opendaylight.tcpmd5.api.KeyMapping; -import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.AsNumber; -import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.Ipv4Address; -import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev150305.ipv4.routes.ipv4.routes.Ipv4Route; -import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev150305.ipv6.routes.ipv6.routes.Ipv6Route; -import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev150210.LinkstateAddressFamily; -import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev150210.LinkstateSubsequentAddressFamily; -import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev150210.linkstate.routes.linkstate.routes.LinkstateRoute; -import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.BgpTableType; -import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.BgpRib; -import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.RibId; -import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.bgp.rib.Rib; -import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.bgp.rib.RibKey; -import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.bgp.rib.rib.LocRib; -import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.rib.Tables; +import org.opendaylight.protocol.concepts.KeyMapping; +import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress; +import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address; +import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev171207.ipv4.routes.ipv4.routes.Ipv4Route; +import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev171207.ipv6.routes.ipv6.routes.Ipv6Route; +import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev171207.LinkstateAddressFamily; +import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev171207.LinkstateSubsequentAddressFamily; +import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev171207.linkstate.routes.linkstate.routes.LinkstateRoute; +import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev171207.BgpTableType; +import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev171207.BgpRib; +import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev171207.PeerRole; +import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev171207.RibId; +import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev171207.rib.Tables; +import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev171207.rib.TablesKey; +import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.BgpId; import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.Ipv4AddressFamily; import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.UnicastSubsequentAddressFamily; -import org.opendaylight.yangtools.binding.data.codec.api.BindingCodecTreeFactory; import org.opendaylight.yangtools.concepts.ListenerRegistration; -import org.opendaylight.yangtools.sal.binding.generator.impl.GeneratedClassLoadingStrategy; import org.opendaylight.yangtools.yang.binding.InstanceIdentifier; +import org.opendaylight.yangtools.yang.binding.YangModuleInfo; import org.opendaylight.yangtools.yang.binding.util.BindingReflections; -public class ParserToSalTest extends AbstractDataBrokerTest { +public class ParserToSalTest extends AbstractConcurrentDataBrokerTest { private static final String TEST_RIB_ID = "testRib"; - - private final String hex_messages = "/bgp_hex.txt"; - + private static final TablesKey TABLE_KEY = new TablesKey(LinkstateAddressFamily.class, LinkstateSubsequentAddressFamily.class); + private static final InstanceIdentifier BGP_IID = InstanceIdentifier.create(BgpRib.class); private BGPMock mock; private AbstractRIBExtensionProviderActivator baseact, lsact; private RIBExtensionProviderContext ext1, ext2; + private final IpAddress localAddress = new IpAddress(new Ipv4Address("127.0.0.1")); @Mock - BGPDispatcher dispatcher; - - @Mock - ReconnectStrategyFactory tcpStrategyFactory; - + private BGPDispatcher dispatcher; @Mock - ReconnectStrategyFactory sessionStrategy; - - BindingCodecTreeFactory codecFactory; - - private SchemaService schemaService; - - @Override - protected java.lang.Iterable getModuleInfos() throws Exception { - return ImmutableList.of(BindingReflections.getModuleInfo(Ipv4Route.class), BindingReflections.getModuleInfo(Ipv6Route.class), BindingReflections.getModuleInfo(LinkstateRoute.class)); - }; + private ClusterSingletonServiceProvider clusterSingletonServiceProvider; + private BindingCodecTreeFactory codecFactory; - @Override - protected DataBrokerTestCustomizer createDataBrokerTestCustomizer() { - final DataBrokerTestCustomizer customizer = super.createDataBrokerTestCustomizer(); - this.codecFactory = customizer.getBindingToNormalized(); - this.schemaService = customizer.getSchemaService(); - return customizer; - } + private DOMSchemaService schemaService; - @Override - protected void setupWithDataBroker(final DataBroker dataBroker) { + @Before + public void setUp() throws Exception { + super.setup(); MockitoAnnotations.initMocks(this); - final List bgpMessages; - try { - bgpMessages = HexDumpBGPFileParser.parseMessages(ParserToSalTest.class.getResourceAsStream(this.hex_messages)); - } catch (final IOException e) { - throw Throwables.propagate(e); - } - this.mock = new BGPMock(new EventBus("test"), ServiceLoaderBGPExtensionProviderContext.getSingletonInstance().getMessageRegistry(), Lists.newArrayList(fixMessages(bgpMessages))); - - Mockito.doReturn(GlobalEventExecutor.INSTANCE.newSucceededFuture(null)).when(this.dispatcher).createReconnectingClient( - Mockito.any(InetSocketAddress.class), Mockito.any(AsNumber.class), - Mockito.any(BGPPeerRegistry.class), Mockito.eq(this.tcpStrategyFactory), Mockito.eq(this.sessionStrategy), + final String hexMessages = "/bgp_hex.txt"; + final List bgpMessages = HexDumpBGPFileParser.parseMessages(ParserToSalTest.class.getResourceAsStream(hexMessages)); + this.mock = new BGPMock(new EventBus("test"), ServiceLoaderBGPExtensionProviderContext + .getSingletonInstance().getMessageRegistry(), Lists.newArrayList(fixMessages(bgpMessages))); + + Mockito.doReturn(GlobalEventExecutor.INSTANCE.newSucceededFuture(null)).when(this.dispatcher) + .createReconnectingClient(Mockito.any(InetSocketAddress.class), Mockito.anyInt(), Mockito.any(KeyMapping.class)); this.ext1 = new SimpleRIBExtensionProviderContext(); this.ext2 = new SimpleRIBExtensionProviderContext(); this.baseact = new RIBActivator(); - this.lsact = new org.opendaylight.protocol.bgp.linkstate.RIBActivator(); + this.lsact = new org.opendaylight.protocol.bgp.linkstate.impl.RIBActivator(); this.baseact.startRIBExtensionProvider(this.ext1); this.lsact.startRIBExtensionProvider(this.ext2); + doReturn(Mockito.mock(ClusterSingletonServiceRegistration.class)).when(this.clusterSingletonServiceProvider). + registerClusterSingletonService(any(ClusterSingletonService.class)); + } + + @Override + protected Iterable getModuleInfos() throws Exception { + return ImmutableList.of( + BindingReflections.getModuleInfo(Ipv4Route.class), + BindingReflections.getModuleInfo(Ipv6Route.class), + BindingReflections.getModuleInfo(LinkstateRoute.class)); + } + + @Override + protected AbstractDataBrokerTestCustomizer createDataBrokerTestCustomizer() { + final AbstractDataBrokerTestCustomizer customizer = super.createDataBrokerTestCustomizer(); + this.codecFactory = customizer.getBindingToNormalized(); + this.schemaService = customizer.getSchemaService(); + return customizer; } @After @@ -135,64 +137,69 @@ public class ParserToSalTest extends AbstractDataBrokerTest { } @Test - public void testWithLinkstate() throws InterruptedException, ExecutionException { - final List tables = ImmutableList.of( - (BgpTableType) new BgpTableTypeImpl(LinkstateAddressFamily.class, LinkstateSubsequentAddressFamily.class)); - final RIBImpl rib = new RIBImpl(new RibId(TEST_RIB_ID), new AsNumber(72L), new Ipv4Address("127.0.0.1"), null, this.ext2, this.dispatcher, this.tcpStrategyFactory, this.codecFactory, this.sessionStrategy, getDataBroker(), getDomBroker(), tables, GeneratedClassLoadingStrategy.getTCCLClassLoadingStrategy()); + public void testWithLinkstate() throws InterruptedException, ExecutionException, ReadFailedException { + final List tables = ImmutableList.of(new BgpTableTypeImpl(LinkstateAddressFamily.class, + LinkstateSubsequentAddressFamily.class)); + final RIBImpl rib = new RIBImpl(this.clusterSingletonServiceProvider, new RibId(TEST_RIB_ID), + AS_NUMBER, new BgpId("127.0.0.1"), null, this.ext2, this.dispatcher, + this.codecFactory, getDomBroker(), tables, Collections.singletonMap(TABLE_KEY, + BasePathSelectionModeFactory.createBestPathSelectionStrategy()), + GeneratedClassLoadingStrategy.getTCCLClassLoadingStrategy()); + rib.instantiateServiceInstance(); + assertTablesExists(tables); rib.onGlobalContextUpdated(this.schemaService.getGlobalContext()); - final BGPPeer peer = new BGPPeer("peer-" + this.mock.toString(), rib); - + final BGPPeer peer = new BGPPeer(this.localAddress, rib, PeerRole.Ibgp, null, Collections.emptySet(), + Collections.emptySet()); + peer.instantiateServiceInstance(); final ListenerRegistration reg = this.mock.registerUpdateListener(peer); reg.close(); - assertTablesExists(tables); } @Test - public void testWithoutLinkstate() throws InterruptedException, ExecutionException { - final List tables = ImmutableList.of((BgpTableType) new BgpTableTypeImpl(Ipv4AddressFamily.class, UnicastSubsequentAddressFamily.class)); - final RIBImpl rib = new RIBImpl(new RibId(TEST_RIB_ID), new AsNumber(72L), new Ipv4Address("127.0.0.1"), null, this.ext1, this.dispatcher, this.tcpStrategyFactory, this.codecFactory, this.sessionStrategy, getDataBroker(), getDomBroker(), tables, GeneratedClassLoadingStrategy.getTCCLClassLoadingStrategy()); + public void testWithoutLinkstate() throws InterruptedException, ExecutionException, ReadFailedException { + final List tables = ImmutableList.of(new BgpTableTypeImpl(Ipv4AddressFamily.class, + UnicastSubsequentAddressFamily.class)); + final RIBImpl rib = new RIBImpl(this.clusterSingletonServiceProvider, new RibId(TEST_RIB_ID), AS_NUMBER, BGP_ID, + null, this.ext1, this.dispatcher, this.codecFactory, getDomBroker(), tables, + Collections.singletonMap(TABLE_KEY, BasePathSelectionModeFactory.createBestPathSelectionStrategy()), + GeneratedClassLoadingStrategy.getTCCLClassLoadingStrategy()); + rib.instantiateServiceInstance(); rib.onGlobalContextUpdated(this.schemaService.getGlobalContext()); - final BGPPeer peer = new BGPPeer("peer-" + this.mock.toString(), rib); - + assertTablesExists(tables); + final BGPPeer peer = new BGPPeer(this.localAddress, rib, PeerRole.Ibgp, null, Collections.emptySet(), + Collections.emptySet()); + peer.instantiateServiceInstance(); final ListenerRegistration reg = this.mock.registerUpdateListener(peer); reg.close(); - assertTablesExists(tables); } - private Collection fixMessages(final Collection bgpMessages) { - return Collections2.transform(bgpMessages, new Function() { - - @Nullable - @Override - public byte[] apply(@Nullable final byte[] input) { - final byte[] ret = new byte[input.length + 1]; - // ff - ret[0] = -1; - for (int i = 0; i < input.length; i++) { - ret[i + 1] = input[i]; - } - return ret; - } + private static Collection fixMessages(final Collection bgpMessages) { + return Collections2.transform(bgpMessages, input -> { + final byte[] ret = new byte[input.length + 1]; + // ff + ret[0] = -1; + System.arraycopy(input, 0, ret, 1, input.length); + return ret; }); } - private void assertTablesExists(final List expectedTables) throws InterruptedException, ExecutionException { - final Optional lockRib = getLocRibTable(); - assertTrue(lockRib.isPresent()); - final List tables = lockRib.get().getTables(); - assertFalse(tables.isEmpty()); - for (final BgpTableType tableType : expectedTables) { - boolean found = false; - for (final Tables table : tables) { - if(table.getAfi().equals(tableType.getAfi()) && table.getSafi().equals(tableType.getSafi())) { - found = true; + private void assertTablesExists(final List expectedTables) + throws InterruptedException, ExecutionException, ReadFailedException { + readDataOperational(getDataBroker(), BGP_IID, bgpRib -> { + final List tables = bgpRib.getRib().get(0).getLocRib().getTables(); + assertFalse(tables.isEmpty()); + + for (final BgpTableType tableType : expectedTables) { + boolean found = false; + for (final Tables table : tables) { + if (table.getAfi().equals(tableType.getAfi()) && table.getSafi().equals(tableType.getSafi())) { + found = true; + assertTrue(Boolean.valueOf(true).equals(table.getAttributes().isUptodate())); + } } + assertTrue(found); } - assertTrue(found); - } - } - - private Optional getLocRibTable() throws InterruptedException, ExecutionException { - return getDataBroker().newReadOnlyTransaction().read(LogicalDatastoreType.OPERATIONAL, InstanceIdentifier.builder(BgpRib.class).child(Rib.class, new RibKey(new RibId(TEST_RIB_ID))).child(LocRib.class).build()).get(); + return bgpRib; + }); } }