pcep/topology/topology-provider/src/test/java/org/opendaylight/bgpcep/pcep/topology/provider/IncrementalSynchronizationProcedureTest.java

   1 /*
   2  * Copyright (c) 2015 Cisco Systems, Inc. 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.bgpcep.pcep.topology.provider;
   9
  10 import static org.junit.Assert.assertEquals;
  11 import static org.junit.Assert.assertFalse;
  12 import static org.opendaylight.protocol.pcep.pcc.mock.spi.MsgBuilderUtil.createLsp;
  13 import static org.opendaylight.protocol.pcep.pcc.mock.spi.MsgBuilderUtil.createPath;
  14 import static org.opendaylight.protocol.util.CheckTestUtil.readDataOperational;
  15
  16 import java.util.Collections;
  17 import java.util.Optional;
  18 import org.junit.Before;
  19 import org.junit.Test;
  20 import org.opendaylight.protocol.pcep.PCEPSession;
  21 import org.opendaylight.protocol.pcep.pcc.mock.spi.MsgBuilderUtil;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev181109.Tlvs3Builder;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev181109.lsp.db.version.tlv.LspDbVersion;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev181109.lsp.db.version.tlv.LspDbVersionBuilder;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.initiated.rev181109.Stateful1Builder;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.OperationalStatus;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.Pcrpt;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.PlspId;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.SymbolicPathName;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.Tlvs1Builder;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.lsp.identifiers.tlv.LspIdentifiersBuilder;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.lsp.object.LspBuilder;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.stateful.capability.tlv.StatefulBuilder;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.symbolic.path.name.tlv.SymbolicPathNameBuilder;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.open.object.Open;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.open.object.OpenBuilder;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.open.object.open.TlvsBuilder;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev150820.LspId;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev200120.PccSyncState;
  40 import org.opendaylight.yangtools.yang.common.Uint32;
  41 import org.opendaylight.yangtools.yang.common.Uint64;
  42
  43 public class IncrementalSynchronizationProcedureTest
  44         extends AbstractPCEPSessionTest<PCEPTopologySessionListenerFactory> {
  45
  46     private PCEPTopologySessionListener listener;
  47
  48     @Override
  49     @Before
  50     public void setUp() throws Exception {
  51         super.setUp();
  52         this.listener = (PCEPTopologySessionListener) getSessionListener();
  53     }
  54
  55     @Test
  56     public void testStateSynchronizationPerformed() throws Exception {
  57         PCEPSession session = getPCEPSession(getOpen(null), getOpen(null));
  58         this.listener.onSessionUp(session);
  59         //report LSP + LSP-DB version number
  60         final Pcrpt pcRpt = getPcrpt(Uint32.ONE, "test");
  61         this.listener.onMessage(session, pcRpt);
  62         readDataOperational(getDataBroker(), this.pathComputationClientIId, pcc -> {
  63             assertFalse(pcc.nonnullReportedLsp().isEmpty());
  64             return pcc;
  65         });
  66
  67         this.listener.onSessionDown(session, new IllegalArgumentException());
  68         this.listener = (PCEPTopologySessionListener) getSessionListener();
  69
  70         //session up - expect sync (LSP-DBs do not match)
  71         final LspDbVersion localDbVersion = new LspDbVersionBuilder()
  72                 .setLspDbVersionValue(Uint64.TWO).build();
  73         session = getPCEPSession(getOpen(localDbVersion), getOpen(null));
  74         this.listener.onSessionUp(session);
  75         readDataOperational(getDataBroker(), this.pathComputationClientIId, pcc -> {
  76             //check node - IncrementalSync state
  77             assertEquals(PccSyncState.IncrementalSync, pcc.getStateSync());
  78             //check reported LSP - persisted from previous session
  79             assertFalse(pcc.nonnullReportedLsp().isEmpty());
  80             return pcc;
  81         });
  82
  83         //report LSP2 + LSP-DB version number 2
  84         final Pcrpt pcRpt2 = getPcrpt(Uint32.TWO, "testsecond");
  85         this.listener.onMessage(session, pcRpt2);
  86         readDataOperational(getDataBroker(), this.pathComputationClientIId, pcc -> {
  87             //check node - synchronized
  88             assertEquals(PccSyncState.IncrementalSync, pcc.getStateSync());
  89             //check reported LSP is not empty
  90             assertEquals(2, pcc.nonnullReportedLsp().size());
  91             return pcc;
  92         });
  93
  94         //sync rpt + LSP-DB
  95         final Pcrpt syncMsg = getSyncPcrt();
  96         this.listener.onMessage(session, syncMsg);
  97         readDataOperational(getDataBroker(), this.pathComputationClientIId, pcc -> {
  98             //check node - synchronized
  99             assertEquals(PccSyncState.Synchronized, pcc.getStateSync());
 100             //check reported LSP is empty, LSP state from previous session was purged
 101             assertEquals(2, pcc.nonnullReportedLsp().size());
 102             return pcc;
 103         });
 104
 105         //report LSP3 + LSP-DB version number 4
 106         final Pcrpt pcRpt3 = getPcrpt(Uint32.valueOf(3), "testthird");
 107         this.listener.onMessage(session, pcRpt3);
 108         readDataOperational(getDataBroker(), this.pathComputationClientIId, pcc -> {
 109             //check node - synchronized
 110             assertEquals(PccSyncState.Synchronized, pcc.getStateSync());
 111             assertEquals(3,pcc.nonnullReportedLsp().size());
 112             return pcc;
 113         });
 114     }
 115
 116     private Open getOpen(final LspDbVersion dbVersion) {
 117         return new OpenBuilder(super.getLocalPref())
 118                 .setTlvs(new TlvsBuilder()
 119                     .addAugmentation(new Tlvs1Builder()
 120                         .setStateful(new StatefulBuilder()
 121                             .addAugmentation(new Stateful1Builder().setInitiation(Boolean.TRUE).build())
 122                             .addAugmentation(new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang
 123                                 .controller.pcep.sync.optimizations.rev181109.Stateful1Builder()
 124                                 .setIncludeDbVersion(Boolean.TRUE).setDeltaLspSyncCapability(Boolean.TRUE)
 125                                 .build())
 126                             .build())
 127                         .build())
 128                     .addAugmentation(new Tlvs3Builder().setLspDbVersion(dbVersion).build())
 129                     .build())
 130                 .build();
 131     }
 132
 133     private static Pcrpt getPcrpt(final Uint32 val, final String pathname) {
 134         return MsgBuilderUtil.createPcRtpMessage(new LspBuilder().setPlspId(new PlspId(val)).setTlvs(
 135             new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.lsp
 136                     .object.lsp.TlvsBuilder().setLspIdentifiers(new LspIdentifiersBuilder()
 137                     .setLspId(new LspId(val)).build())
 138                 .setSymbolicPathName(new SymbolicPathNameBuilder().setPathName(new SymbolicPathName(
 139                     pathname.getBytes())).build()).addAugmentation(new org.opendaylight.yang.gen.v1.urn.opendaylight
 140                         .params.xml.ns.yang.controller.pcep.sync.optimizations.rev181109.Tlvs1Builder()
 141                         .setLspDbVersion(new LspDbVersionBuilder().setLspDbVersionValue(Uint64.valueOf(val)).build())
 142                         .build()).build()).setPlspId(new PlspId(val)
 143         ).setSync(true).setRemove(false).setOperational(OperationalStatus.Active).build(), Optional.empty(),
 144             createPath(Collections.emptyList()));
 145     }
 146
 147     private static Pcrpt getSyncPcrt() {
 148         return MsgBuilderUtil.createPcRtpMessage(createLsp(Uint32.ZERO, false, Optional.of(
 149                 new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109
 150                         .lsp.object.lsp.TlvsBuilder().addAugmentation(new org.opendaylight.yang.gen.v1.urn.opendaylight
 151                             .params.xml.ns.yang.controller.pcep.sync.optimizations.rev181109.Tlvs1Builder()
 152                             .setLspDbVersion(new LspDbVersionBuilder().setLspDbVersionValue(Uint64.valueOf(3L)).build())
 153                             .build()).build()),
 154                 true, false), Optional.empty(),
 155                 createPath(Collections.emptyList()));
 156     }
 157 }