Create pcep topology parent aggregator
[bgpcep.git] / pcep / topology / topology-provider / src / test / java / org / opendaylight / bgpcep / pcep / topology / provider / IncrementalSynchronizationProcedureTest.java
diff --git a/pcep/topology/topology-provider/src/test/java/org/opendaylight/bgpcep/pcep/topology/provider/IncrementalSynchronizationProcedureTest.java b/pcep/topology/topology-provider/src/test/java/org/opendaylight/bgpcep/pcep/topology/provider/IncrementalSynchronizationProcedureTest.java
new file mode 100644 (file)
index 0000000..8a43de6
--- /dev/null
@@ -0,0 +1,155 @@
+/*
+ * Copyright (c) 2015 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.bgpcep.pcep.topology.provider;
+
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertFalse;
+import static org.opendaylight.protocol.pcep.pcc.mock.spi.MsgBuilderUtil.createLsp;
+import static org.opendaylight.protocol.pcep.pcc.mock.spi.MsgBuilderUtil.createPath;
+import static org.opendaylight.protocol.util.CheckUtil.readDataOperational;
+
+import com.google.common.base.Optional;
+import java.math.BigInteger;
+import java.util.Collections;
+import org.junit.Before;
+import org.junit.Test;
+import org.opendaylight.protocol.pcep.PCEPSession;
+import org.opendaylight.protocol.pcep.pcc.mock.spi.MsgBuilderUtil;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev150714.Tlvs3;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev150714.Tlvs3Builder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev150714.lsp.db.version.tlv.LspDbVersion;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev150714.lsp.db.version.tlv.LspDbVersionBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated.rev131126.Stateful1;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated.rev131126.Stateful1Builder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.OperationalStatus;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.Pcrpt;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.PlspId;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.SymbolicPathName;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.Tlvs1;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.Tlvs1Builder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.identifiers.tlv.LspIdentifiersBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.object.LspBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.stateful.capability.tlv.StatefulBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.symbolic.path.name.tlv.SymbolicPathNameBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.Open;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.OpenBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.open.TlvsBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev150820.LspId;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.PccSyncState;
+
+public class IncrementalSynchronizationProcedureTest extends AbstractPCEPSessionTest<Stateful07TopologySessionListenerFactory> {
+
+    private Stateful07TopologySessionListener listener;
+
+    @Override
+    @Before
+    public void setUp() throws Exception {
+        super.setUp();
+        this.listener = (Stateful07TopologySessionListener) getSessionListener();
+    }
+
+    @Test
+    public void testStateSynchronizationPerformed() throws Exception {
+        PCEPSession session = getPCEPSession(getOpen(null), getOpen(null));
+        this.listener.onSessionUp(session);
+        //report LSP + LSP-DB version number
+        final Pcrpt pcRpt = getPcrpt(1L, "test");
+        this.listener.onMessage(session, pcRpt);
+        readDataOperational(getDataBroker(), this.pathComputationClientIId, pcc -> {
+            assertFalse(pcc.getReportedLsp().isEmpty());
+            return pcc;
+        });
+
+        this.listener.onSessionDown(session, new IllegalArgumentException());
+        this.listener = (Stateful07TopologySessionListener) getSessionListener();
+
+        //session up - expect sync (LSP-DBs do not match)
+        final LspDbVersion localDbVersion = new LspDbVersionBuilder().setLspDbVersionValue(BigInteger.valueOf(2L)).build();
+        session = getPCEPSession(getOpen(localDbVersion), getOpen(null));
+        this.listener.onSessionUp(session);
+        readDataOperational(getDataBroker(), this.pathComputationClientIId, pcc -> {
+            //check node - IncrementalSync state
+            assertEquals(PccSyncState.IncrementalSync, pcc.getStateSync());
+            //check reported LSP - persisted from previous session
+            assertFalse(pcc.getReportedLsp().isEmpty());
+            return pcc;
+        });
+
+        //report LSP2 + LSP-DB version number 2
+        final Pcrpt pcRpt2 = getPcrpt(2L,"testsecond");
+        this.listener.onMessage(session, pcRpt2);
+        readDataOperational(getDataBroker(), this.pathComputationClientIId, pcc -> {
+            //check node - synchronized
+            assertEquals(PccSyncState.IncrementalSync, pcc.getStateSync());
+            //check reported LSP is not empty
+            assertEquals(2, pcc.getReportedLsp().size());
+            return pcc;
+        });
+
+        //sync rpt + LSP-DB
+        final Pcrpt syncMsg = getSyncPcrt();
+        this.listener.onMessage(session, syncMsg);
+        readDataOperational(getDataBroker(), this.pathComputationClientIId, pcc -> {
+            //check node - synchronized
+            assertEquals(PccSyncState.Synchronized, pcc.getStateSync());
+            //check reported LSP is empty, LSP state from previous session was purged
+            assertEquals(2, pcc.getReportedLsp().size());
+            return pcc;
+        });
+
+        //report LSP3 + LSP-DB version number 4
+        final Pcrpt pcRpt3 = getPcrpt(3L,"testthird");
+        this.listener.onMessage(session, pcRpt3);
+        readDataOperational(getDataBroker(), this.pathComputationClientIId, pcc -> {
+            //check node - synchronized
+            assertEquals(PccSyncState.Synchronized, pcc.getStateSync());
+            assertEquals(3,pcc.getReportedLsp().size());
+            return pcc;
+        });
+    }
+
+    private Open getOpen(final LspDbVersion dbVersion) {
+        return new OpenBuilder(super.getLocalPref()).setTlvs(new TlvsBuilder().addAugmentation(Tlvs1.class,
+            new Tlvs1Builder().setStateful(new StatefulBuilder().addAugmentation(Stateful1.class,
+                new Stateful1Builder().setInitiation(Boolean.TRUE).build())
+            .addAugmentation(org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.
+                optimizations.rev150714.Stateful1.class, new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml
+                .ns.yang.controller.pcep.sync.optimizations.rev150714.Stateful1Builder()
+                .setIncludeDbVersion(Boolean.TRUE).setDeltaLspSyncCapability(Boolean.TRUE).build())
+            .build()).build()).addAugmentation(Tlvs3.class, new Tlvs3Builder().setLspDbVersion(dbVersion).build())
+            .build()).build();
+    }
+
+    private Pcrpt getPcrpt(final Long val, final String pathname) {
+        return MsgBuilderUtil.createPcRtpMessage(new LspBuilder().setPlspId(new PlspId(val)).setTlvs(
+            new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.
+                object.lsp.TlvsBuilder().setLspIdentifiers(new LspIdentifiersBuilder().setLspId(new LspId(val)).build())
+                .setSymbolicPathName(new SymbolicPathNameBuilder().setPathName(new SymbolicPathName(
+                    pathname.getBytes())).build()).addAugmentation(org.opendaylight.yang.gen.v1.urn.opendaylight.params
+                .xml.ns.yang.controller.pcep.sync.optimizations.rev150714.Tlvs1.class, new org.opendaylight.yang.gen.v1.
+                urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev150714.Tlvs1Builder()
+                .setLspDbVersion(new LspDbVersionBuilder().setLspDbVersionValue(BigInteger.valueOf(val)).build())
+                .build()).build()).setPlspId(new PlspId(val)
+        ).setSync(true).setRemove(false).setOperational(OperationalStatus.Active).build(), Optional.absent(),
+            createPath(Collections.emptyList()));
+    }
+
+    private Pcrpt getSyncPcrt() {
+        return MsgBuilderUtil.createPcRtpMessage(createLsp(0, false, Optional.of(
+                new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.
+                    lsp.object.lsp.TlvsBuilder().addAugmentation(org.opendaylight.yang.gen.v1.urn.opendaylight.params.
+                        xml.ns.yang.controller.pcep.sync.optimizations.rev150714.Tlvs1.class,
+                    new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync
+                        .optimizations.rev150714.Tlvs1Builder().setLspDbVersion(new LspDbVersionBuilder()
+                        .setLspDbVersionValue(BigInteger.valueOf(3L)).build()).build()).build()),
+            true, false), Optional.absent(),
+            createPath(Collections.emptyList()));
+    }
+}