BUG-9079 Make PCEP session recoverable from exception
[bgpcep.git] / pcep / topology-provider / src / test / java / org / opendaylight / bgpcep / pcep / topology / provider / PCEPTriggeredReSynchronizationProcedureTest.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.CheckUtil.readDataOperational;
15
16 import com.google.common.base.Optional;
17 import java.math.BigInteger;
18 import java.util.Collections;
19 import java.util.List;
20 import org.junit.Before;
21 import org.junit.Test;
22 import org.opendaylight.protocol.pcep.PCEPSession;
23 import org.opendaylight.protocol.pcep.pcc.mock.spi.MsgBuilderUtil;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev150714.Tlvs3;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev150714.Tlvs3Builder;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev150714.lsp.db.version.tlv.LspDbVersion;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev150714.lsp.db.version.tlv.LspDbVersionBuilder;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated.rev131126.Stateful1;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated.rev131126.Stateful1Builder;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.OperationalStatus;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.Pcrpt;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.PlspId;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.SrpIdNumber;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.SymbolicPathName;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.Tlvs1;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.Tlvs1Builder;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.identifiers.tlv.LspIdentifiersBuilder;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.object.LspBuilder;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.srp.object.SrpBuilder;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.stateful.capability.tlv.StatefulBuilder;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.symbolic.path.name.tlv.SymbolicPathNameBuilder;
42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.Open;
43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.OpenBuilder;
44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.open.TlvsBuilder;
45 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev150820.LspId;
46 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.PccSyncState;
47 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.TriggerSyncInputBuilder;
48 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.pcep.client.attributes.path.computation.client.ReportedLsp;
49
50 public class PCEPTriggeredReSynchronizationProcedureTest extends AbstractPCEPSessionTest<Stateful07TopologySessionListenerFactory> {
51     private Stateful07TopologySessionListener listener;
52
53     private PCEPSession session;
54     private final LspDbVersion lspDbVersion = new LspDbVersionBuilder().setLspDbVersionValue(BigInteger.ONE).build();
55
56     @Override
57     @Before
58     public void setUp() throws Exception {
59         super.setUp();
60         this.listener = (Stateful07TopologySessionListener) getSessionListener();
61     }
62
63     @Test
64     public void testTriggeredResynchronization() throws Exception {
65         //session up - sync skipped (LSP-DBs match)
66         this.session = getPCEPSession(getOpen(), getOpen());
67         this.listener.onSessionUp(this.session);
68
69         //report LSP + LSP-DB version number
70         final Pcrpt pcRpt = getPcrt();
71         this.listener.onMessage(this.session, pcRpt);
72
73         readDataOperational(getDataBroker(), this.pathComputationClientIId, pcc -> {
74             assertEquals(PccSyncState.Synchronized, pcc.getStateSync());
75             assertFalse(pcc.getReportedLsp().isEmpty());
76             return pcc;
77         });
78
79         //PCEP Trigger Full Resync
80         this.listener.triggerSync(new TriggerSyncInputBuilder().setNode(this.nodeId).build());
81         readDataOperational(getDataBroker(), this.pathComputationClientIId, pcc -> {
82             assertEquals(PccSyncState.PcepTriggeredResync, pcc.getStateSync());
83             return pcc;
84         });
85
86         //end of sync
87         final Pcrpt syncMsg = getSyncMsg();
88         this.listener.onMessage(this.session, syncMsg);
89         readDataOperational(getDataBroker(), this.pathComputationClientIId, pcc -> {
90             //check node - synchronized
91             assertEquals(PccSyncState.Synchronized, pcc.getStateSync());
92             return pcc;
93         });
94
95         this.listener.onMessage(this.session, pcRpt);
96         readDataOperational(getDataBroker(), this.pathComputationClientIId, pcc -> {
97             assertEquals(1, pcc.getReportedLsp().size());
98             return pcc;
99         });
100
101         //Trigger Full Resync
102         this.listener.triggerSync(new TriggerSyncInputBuilder().setNode(this.nodeId).build());
103         this.listener.onMessage(this.session, pcRpt);
104         //end of sync
105         this.listener.onMessage(this.session, syncMsg);
106         readDataOperational(getDataBroker(), this.pathComputationClientIId, pcc -> {
107             //check node - synchronized
108             assertEquals(PccSyncState.Synchronized, pcc.getStateSync());
109             //check reported LSP is not empty, Stale LSP state were purged
110             assertEquals(1, pcc.getReportedLsp().size());
111             return pcc;
112         });
113     }
114
115     @Test
116     public void testTriggeredResynchronizationLsp() throws Exception {
117         //session up - sync skipped (LSP-DBs match)
118
119         this.session = getPCEPSession(getOpen(), getOpen());
120         this.listener.onSessionUp(this.session);
121
122         //report LSP + LSP-DB version number
123         final Pcrpt pcRpt = getPcrt();
124         this.listener.onMessage(this.session, pcRpt);
125         readDataOperational(getDataBroker(), this.pathComputationClientIId, pcc -> {
126             assertEquals(PccSyncState.Synchronized, pcc.getStateSync());
127             final List<ReportedLsp> reportedLspPcc = pcc.getReportedLsp();
128             assertFalse(reportedLspPcc.isEmpty());
129             return pcc;
130         });
131
132         //Trigger Full Resync
133         this.listener.triggerSync(new TriggerSyncInputBuilder().setNode(this.nodeId).setName("test").build());
134         readDataOperational(getDataBroker(), this.pathComputationClientIId, pcc -> {
135             assertEquals(PccSyncState.PcepTriggeredResync, pcc.getStateSync());
136             assertFalse(pcc.getReportedLsp().isEmpty());
137             return pcc;
138         });
139
140         this.listener.onMessage(this.session, pcRpt);
141         readDataOperational(getDataBroker(), this.pathComputationClientIId, pcc -> {
142             assertFalse(pcc.getReportedLsp().isEmpty());
143             return pcc;
144         });
145
146         //sync rpt + LSP-DB
147         final Pcrpt syncMsg = getSyncMsg();
148         this.listener.onMessage(this.session, syncMsg);
149         readDataOperational(getDataBroker(), this.pathComputationClientIId, pcc -> {
150             //check node - synchronized
151             assertEquals(PccSyncState.Synchronized, pcc.getStateSync());
152             //check reported LSP
153             assertEquals(1, pcc.getReportedLsp().size());
154             return pcc;
155         });
156
157         //Trigger Full Resync
158         this.listener.triggerSync(new TriggerSyncInputBuilder().setNode(this.nodeId).setName("test").build());
159         this.listener.onMessage(this.session, syncMsg);
160         readDataOperational(getDataBroker(), this.pathComputationClientIId, pcc -> {
161             //check node - synchronized
162             assertEquals(PccSyncState.Synchronized, pcc.getStateSync());
163             //check reported LSP
164             assertEquals(0, pcc.getReportedLsp().size());
165             return pcc;
166         });
167     }
168
169     private Open getOpen() {
170         return new OpenBuilder(super.getLocalPref()).setTlvs(new TlvsBuilder().addAugmentation(Tlvs1.class,
171             new Tlvs1Builder().setStateful(new StatefulBuilder().addAugmentation(Stateful1.class, new Stateful1Builder()
172                 .setInitiation(Boolean.TRUE).build()).addAugmentation(org.opendaylight.yang.gen.v1.urn.opendaylight
173                     .params.xml.ns.yang.controller.pcep.sync.optimizations.rev150714.Stateful1.class,
174                 new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations
175                     .rev150714.Stateful1Builder().setIncludeDbVersion(Boolean.TRUE).setTriggeredResync(Boolean.TRUE)
176                     .build()).build()).build()).addAugmentation(Tlvs3.class, new Tlvs3Builder()
177             .setLspDbVersion(this.lspDbVersion).build()).build()).build();
178     }
179
180     private Pcrpt getSyncMsg() {
181         final SrpBuilder srpBuilder = new SrpBuilder();
182         // not sue whether use 0 instead of nextRequest() or do not insert srp == SRP-ID-number = 0
183         srpBuilder.setOperationId(new SrpIdNumber(1L));
184         return MsgBuilderUtil.createPcRtpMessage(createLsp(0, false, Optional.of(
185                 new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.object.lsp.TlvsBuilder().addAugmentation(
186                     org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev150714.Tlvs1.class,
187                     new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev150714.Tlvs1Builder()
188                         .setLspDbVersion(new LspDbVersionBuilder().setLspDbVersionValue(BigInteger.valueOf(3L)).build()).build()).build()), true, false),
189                             Optional.of(srpBuilder.build()), createPath(Collections.emptyList()));
190     }
191
192     private Pcrpt getPcrt() {
193         return MsgBuilderUtil.createPcRtpMessage(new LspBuilder().setPlspId(new PlspId(1L)).setTlvs(
194                 new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.object.lsp.
195                     TlvsBuilder().setLspIdentifiers(new LspIdentifiersBuilder().setLspId(new LspId(1L)).build()).setSymbolicPathName(
196                     new SymbolicPathNameBuilder().setPathName(new SymbolicPathName("test".getBytes())).build())
197                     .addAugmentation(org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev150714.Tlvs1.class,
198                         new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev150714.Tlvs1Builder()
199                             .setLspDbVersion(new LspDbVersionBuilder().setLspDbVersionValue(BigInteger.ONE).build()).build()).build())
200                 .setPlspId(new PlspId(1L)).setSync(true).setRemove(false).setOperational(OperationalStatus.Active).build(), Optional.absent(),
201             createPath(Collections.emptyList()));
202     }
203 }