replicate/mdsal-replicate-netty/src/test/java/org/opendaylight/mdsal/replicate/netty/IntegrationTest.java

   1 /*
   2  * Copyright (c) 2020 PANTHEON.tech, s.r.o. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.mdsal.replicate.netty;
   9
  10 import static org.mockito.ArgumentMatchers.any;
  11 import static org.mockito.ArgumentMatchers.eq;
  12 import static org.mockito.Mockito.doReturn;
  13 import static org.mockito.Mockito.mock;
  14 import static org.mockito.Mockito.timeout;
  15 import static org.mockito.Mockito.verify;
  16
  17 import java.net.Inet4Address;
  18 import java.time.Duration;
  19 import org.junit.After;
  20 import org.junit.Before;
  21 import org.junit.Test;
  22 import org.junit.runner.RunWith;
  23 import org.mockito.junit.MockitoJUnitRunner;
  24 import org.opendaylight.mdsal.binding.dom.adapter.test.AbstractDataBrokerTest;
  25 import org.opendaylight.mdsal.common.api.CommitInfo;
  26 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
  27 import org.opendaylight.mdsal.dom.api.DOMDataBroker;
  28 import org.opendaylight.mdsal.dom.api.DOMDataTreeWriteTransaction;
  29 import org.opendaylight.mdsal.dom.api.DOMTransactionChain;
  30 import org.opendaylight.mdsal.eos.dom.simple.SimpleDOMEntityOwnershipService;
  31 import org.opendaylight.mdsal.singleton.dom.impl.DOMClusterSingletonServiceProviderImpl;
  32 import org.opendaylight.yangtools.concepts.Registration;
  33 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  34 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
  35
  36 @RunWith(MockitoJUnitRunner.StrictStubs.class)
  37 public class IntegrationTest extends AbstractDataBrokerTest {
  38     private static final int TEST_PORT = 4000;
  39
  40     private AbstractBootstrapSupport support;
  41     private DOMClusterSingletonServiceProviderImpl css;
  42
  43     @Before
  44     public void before() {
  45         support = AbstractBootstrapSupport.create();
  46         css = new DOMClusterSingletonServiceProviderImpl(new SimpleDOMEntityOwnershipService());
  47         css.initializeProvider();
  48     }
  49
  50     @After
  51     public void after() throws InterruptedException {
  52         support.close();
  53         css.close();
  54     }
  55
  56     @Test
  57     public void testSourceToSink() throws InterruptedException {
  58         // Make sure to start source...
  59         final Registration source = NettyReplication.createSource(support, getDomBroker(), css, true, TEST_PORT);
  60         // ... and give it some time start up and open up the port
  61         Thread.sleep(1000);
  62
  63         // Mocking for sink...
  64         final DOMTransactionChain sinkChain = mock(DOMTransactionChain.class);
  65         final DOMDataTreeWriteTransaction sinkTx = mock(DOMDataTreeWriteTransaction.class);
  66         doReturn(CommitInfo.emptyFluentFuture()).when(sinkTx).commit();
  67         doReturn(sinkTx).when(sinkChain).newWriteOnlyTransaction();
  68         final DOMDataBroker sinkBroker = mock(DOMDataBroker.class);
  69         doReturn(sinkChain).when(sinkBroker).createMergingTransactionChain(any());
  70
  71         // Kick of the sink ...
  72         final Registration sink = NettyReplication.createSink(support, sinkBroker, css, true,
  73             Inet4Address.getLoopbackAddress(), TEST_PORT, Duration.ZERO);
  74         // ... and sync on it starting up
  75         verify(sinkBroker, timeout(1000)).createMergingTransactionChain(any());
  76
  77         // FIXME: add a few writes to the broker so we have multiple transactions and verify deltas
  78
  79         verify(sinkChain, timeout(2000)).newWriteOnlyTransaction();
  80         verify(sinkTx, timeout(1000)).put(eq(LogicalDatastoreType.CONFIGURATION), eq(YangInstanceIdentifier.empty()),
  81             any(ContainerNode.class));
  82         verify(sinkTx, timeout(1000)).commit();
  83
  84         sink.close();
  85         source.close();
  86     }
  87 }