2 * Copyright (c) 2015 Cisco Systems, Inc. and others. All rights reserved.
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
9 package org.opendaylight.bgpcep.pcep.topology.provider;
11 import static org.junit.Assert.assertEquals;
12 import static org.junit.Assert.assertFalse;
13 import static org.junit.Assert.assertNull;
14 import static org.opendaylight.protocol.pcep.pcc.mock.spi.MsgBuilderUtil.createLsp;
15 import static org.opendaylight.protocol.pcep.pcc.mock.spi.MsgBuilderUtil.createPath;
16 import static org.opendaylight.protocol.util.CheckTestUtil.readDataOperational;
18 import java.math.BigInteger;
19 import java.util.Collections;
20 import java.util.Optional;
21 import java.util.concurrent.ExecutionException;
22 import org.junit.Before;
23 import org.junit.Test;
24 import org.opendaylight.protocol.pcep.PCEPSession;
25 import org.opendaylight.protocol.pcep.pcc.mock.spi.MsgBuilderUtil;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev181109.PathComputationClient1;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev181109.Tlvs3;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev181109.Tlvs3Builder;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev181109.lsp.db.version.tlv.LspDbVersion;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev181109.lsp.db.version.tlv.LspDbVersionBuilder;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated.rev181109.Stateful1;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated.rev181109.Stateful1Builder;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.OperationalStatus;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.Pcrpt;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.PlspId;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.SymbolicPathName;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.Tlvs1;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.Tlvs1Builder;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.lsp.identifiers.tlv.LspIdentifiersBuilder;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.lsp.object.LspBuilder;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.stateful.capability.tlv.StatefulBuilder;
42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.symbolic.path.name.tlv.SymbolicPathNameBuilder;
43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.open.object.Open;
44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.open.object.OpenBuilder;
45 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.open.object.open.TlvsBuilder;
46 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev150820.LspId;
47 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev181109.PccSyncState;
49 public class StateSynchronizationAvoidanceProcedureTest extends
50 AbstractPCEPSessionTest<Stateful07TopologySessionListenerFactory> {
52 private Stateful07TopologySessionListener listener;
56 public void setUp() throws Exception {
58 this.listener = (Stateful07TopologySessionListener) getSessionListener();
62 public void testNodePersisted() throws ExecutionException, InterruptedException {
63 final PCEPSession session = getPCEPSession(getOpen(null), getOpen(null));
64 this.listener.onSessionUp(session);
65 //report LSP + LSP-DB version number
66 final Pcrpt pcRpt = MsgBuilderUtil.createPcRtpMessage(new LspBuilder().setTlvs(
67 new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.lsp
68 .object.lsp.TlvsBuilder().addAugmentation(org.opendaylight.yang.gen.v1.urn.opendaylight.params
69 .xml.ns.yang.controller.pcep.sync.optimizations.rev181109.Tlvs1.class,
70 new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync
71 .optimizations.rev181109.Tlvs1Builder().setLspDbVersion(new LspDbVersionBuilder()
72 .setLspDbVersionValue(BigInteger.ONE).build()).build()).build())
73 .setPlspId(new PlspId(1L)).setSync(false).setRemove(false)
74 .setOperational(OperationalStatus.Active).build(),
75 Optional.of(MsgBuilderUtil.createSrp(1L)), null);
76 this.listener.onMessage(session, pcRpt);
78 readDataOperational(getDataBroker(), this.pathComputationClientIId.builder()
79 .augmentation(PathComputationClient1.class)
80 .child(LspDbVersion.class).build(), dbVersion -> {
81 assertEquals(1L, dbVersion.getLspDbVersionValue().longValue());
86 this.listener.onSessionDown(session, new IllegalStateException());
87 readDataOperational(getDataBroker(), TOPO_IID, topology -> {
88 assertFalse(topology.getNode().isEmpty());
91 //check topology - node is persisted
95 public void testStateSynchronizationSkipped() throws Exception {
96 //session up - sync skipped (LSP-DBs match)
97 final LspDbVersion lspDbVersion = new LspDbVersionBuilder().setLspDbVersionValue(BigInteger.ONE).build();
98 final PCEPSession session = getPCEPSession(getOpen(lspDbVersion), getOpen(lspDbVersion));
99 this.listener.onSessionUp(session);
100 //check node - synchronized
101 readDataOperational(getDataBroker(), this.pathComputationClientIId, pcc -> {
102 assertEquals(PccSyncState.Synchronized, pcc.getStateSync());
108 public void testStateSynchronizationPerformed() throws Exception {
109 PCEPSession session = getPCEPSession(getOpen(null), getOpen(null));
110 this.listener.onSessionUp(session);
111 //report LSP + LSP-DB version number
112 final Pcrpt pcRpt = MsgBuilderUtil.createPcRtpMessage(new LspBuilder().setPlspId(new PlspId(1L))
113 .setTlvs(new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful
114 .rev181109.lsp.object.lsp.TlvsBuilder().setLspIdentifiers(new LspIdentifiersBuilder()
115 .setLspId(new LspId(1L)).build()).setSymbolicPathName(new SymbolicPathNameBuilder()
116 .setPathName(new SymbolicPathName("test".getBytes())).build()).addAugmentation(org.opendaylight
117 .yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev181109
118 .Tlvs1.class, new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller
119 .pcep.sync.optimizations.rev181109.Tlvs1Builder().setLspDbVersion(new LspDbVersionBuilder()
120 .setLspDbVersionValue(BigInteger.ONE).build()).build()).build())
121 .setPlspId(new PlspId(1L)).setSync(true).setRemove(false).setOperational(OperationalStatus.Active)
122 .build(), Optional.empty(), createPath(Collections.emptyList()));
123 this.listener.onMessage(session, pcRpt);
124 readDataOperational(getDataBroker(), this.pathComputationClientIId, pcc -> {
125 assertFalse(pcc.getReportedLsp().isEmpty());
129 this.listener.onSessionDown(session, new IllegalArgumentException("Simulate Exception"));
130 this.listener = (Stateful07TopologySessionListener) getSessionListener();
132 //session up - expect sync (LSP-DBs do not match)
133 final LspDbVersion localDbVersion = new LspDbVersionBuilder()
134 .setLspDbVersionValue(BigInteger.valueOf(2L)).build();
135 session = getPCEPSession(getOpen(localDbVersion), getOpen(null));
136 this.listener.onSessionUp(session);
138 readDataOperational(getDataBroker(), this.pathComputationClientIId, pcc -> {
139 //check node - not synchronized
140 assertEquals(PccSyncState.InitialResync, pcc.getStateSync());
141 //check reported LSP - persisted from previous session
142 assertFalse(pcc.getReportedLsp().isEmpty());
147 final Pcrpt syncMsg = MsgBuilderUtil.createPcRtpMessage(createLsp(0, false, Optional.of(
148 new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109
149 .lsp.object.lsp.TlvsBuilder().addAugmentation(org.opendaylight.yang.gen.v1.urn.opendaylight
150 .params.xml.ns.yang.controller.pcep.sync.optimizations.rev181109.Tlvs1.class,
151 new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync
152 .optimizations.rev181109.Tlvs1Builder().setLspDbVersion(new LspDbVersionBuilder()
153 .setLspDbVersionValue(BigInteger.valueOf(2L)).build()).build()).build()),
154 true, false), Optional.empty(),
155 createPath(Collections.emptyList()));
156 this.listener.onMessage(session, syncMsg);
157 readDataOperational(getDataBroker(), this.pathComputationClientIId, pcc -> {
158 //check node - synchronized
159 assertEquals(PccSyncState.Synchronized, pcc.getStateSync());
160 //check reported LSP is empty, LSP state from previous session was purged
161 assertNull(pcc.getReportedLsp());
166 private Open getOpen(final LspDbVersion dbVersion) {
167 return new OpenBuilder(super.getLocalPref()).setTlvs(new TlvsBuilder().addAugmentation(Tlvs1.class,
168 new Tlvs1Builder().setStateful(new StatefulBuilder()
169 .addAugmentation(Stateful1.class, new Stateful1Builder().setInitiation(Boolean.TRUE).build())
170 .addAugmentation(org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync
171 .optimizations.rev181109.Stateful1.class, new org.opendaylight.yang.gen.v1.urn.opendaylight.params
172 .xml.ns.yang.controller.pcep.sync.optimizations.rev181109.Stateful1Builder()
173 .setIncludeDbVersion(Boolean.TRUE).build()).build()).build()).addAugmentation(Tlvs3.class,
174 new Tlvs3Builder().setLspDbVersion(dbVersion).build()).build()).build();