Update Statefuful & Initiated yang models
[bgpcep.git] / pcep / topology / 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.junit.Assert.assertNotNull;
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;
17
18 import java.util.Collections;
19 import java.util.Map;
20 import java.util.Optional;
21 import org.junit.Before;
22 import org.junit.Test;
23 import org.opendaylight.protocol.pcep.PCEPSession;
24 import org.opendaylight.protocol.pcep.pcc.mock.spi.MsgBuilderUtil;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev200720.Tlvs3Builder;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev200720.lsp.db.version.tlv.LspDbVersion;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev200720.lsp.db.version.tlv.LspDbVersionBuilder;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.initiated.rev200720.Stateful1Builder;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev200720.OperationalStatus;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev200720.Pcrpt;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev200720.PlspId;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev200720.SrpIdNumber;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev200720.SymbolicPathName;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev200720.Tlvs1Builder;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev200720.lsp.identifiers.tlv.LspIdentifiersBuilder;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev200720.lsp.object.LspBuilder;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev200720.srp.object.SrpBuilder;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev200720.stateful.capability.tlv.StatefulBuilder;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev200720.symbolic.path.name.tlv.SymbolicPathNameBuilder;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.open.object.Open;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.open.object.OpenBuilder;
42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.open.object.open.TlvsBuilder;
43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev150820.LspId;
44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev200120.PccSyncState;
45 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev200120.TriggerSyncInputBuilder;
46 import org.opendaylight.yangtools.yang.common.Uint32;
47 import org.opendaylight.yangtools.yang.common.Uint64;
48
49 public class PCEPTriggeredReSynchronizationProcedureTest
50         extends AbstractPCEPSessionTest<PCEPTopologySessionListenerFactory> {
51     private PCEPTopologySessionListener listener;
52
53     private PCEPSession session;
54     private final LspDbVersion lspDbVersion = new LspDbVersionBuilder().setLspDbVersionValue(Uint64.ONE).build();
55
56     @Override
57     @Before
58     public void setUp() throws Exception {
59         super.setUp();
60         this.listener = (PCEPTopologySessionListener) 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.nonnullReportedLsp().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             final Map<?, ?> lsp = pcc.getReportedLsp();
98             assertNotNull(lsp);
99             assertEquals(1, lsp.size());
100             return pcc;
101         });
102
103         //Trigger Full Resync
104         this.listener.triggerSync(new TriggerSyncInputBuilder().setNode(this.nodeId).build());
105         this.listener.onMessage(this.session, pcRpt);
106         //end of sync
107         this.listener.onMessage(this.session, syncMsg);
108         readDataOperational(getDataBroker(), this.pathComputationClientIId, pcc -> {
109             //check node - synchronized
110             assertEquals(PccSyncState.Synchronized, pcc.getStateSync());
111             //check reported LSP is not empty, Stale LSP state were purged
112             assertEquals(1, pcc.getReportedLsp().size());
113             return pcc;
114         });
115     }
116
117     @Test
118     public void testTriggeredResynchronizationLsp() throws Exception {
119         //session up - sync skipped (LSP-DBs match)
120
121         this.session = getPCEPSession(getOpen(), getOpen());
122         this.listener.onSessionUp(this.session);
123
124         //report LSP + LSP-DB version number
125         final Pcrpt pcRpt = getPcrt();
126         this.listener.onMessage(this.session, pcRpt);
127         readDataOperational(getDataBroker(), this.pathComputationClientIId, pcc -> {
128             assertEquals(PccSyncState.Synchronized, pcc.getStateSync());
129             assertFalse(pcc.nonnullReportedLsp().isEmpty());
130             return pcc;
131         });
132
133         //Trigger Full Resync
134         this.listener.triggerSync(new TriggerSyncInputBuilder().setNode(this.nodeId).setName("test").build());
135         readDataOperational(getDataBroker(), this.pathComputationClientIId, pcc -> {
136             assertEquals(PccSyncState.PcepTriggeredResync, pcc.getStateSync());
137             assertFalse(pcc.nonnullReportedLsp().isEmpty());
138             return pcc;
139         });
140
141         this.listener.onMessage(this.session, pcRpt);
142         readDataOperational(getDataBroker(), this.pathComputationClientIId, pcc -> {
143             assertFalse(pcc.nonnullReportedLsp().isEmpty());
144             return pcc;
145         });
146
147         //sync rpt + LSP-DB
148         final Pcrpt syncMsg = getSyncMsg();
149         this.listener.onMessage(this.session, syncMsg);
150         readDataOperational(getDataBroker(), this.pathComputationClientIId, pcc -> {
151             //check node - synchronized
152             assertEquals(PccSyncState.Synchronized, pcc.getStateSync());
153             //check reported LSP
154             assertEquals(1, pcc.nonnullReportedLsp().size());
155             return pcc;
156         });
157
158         //Trigger Full Resync
159         this.listener.triggerSync(new TriggerSyncInputBuilder().setNode(this.nodeId).setName("test").build());
160         this.listener.onMessage(this.session, syncMsg);
161         readDataOperational(getDataBroker(), this.pathComputationClientIId, pcc -> {
162             //check node - synchronized
163             assertEquals(PccSyncState.Synchronized, pcc.getStateSync());
164             //check reported LSP
165             assertNull(pcc.getReportedLsp());
166             return pcc;
167         });
168     }
169
170     private Open getOpen() {
171         return new OpenBuilder(super.getLocalPref()).setTlvs(new TlvsBuilder().addAugmentation(new Tlvs1Builder()
172             .setStateful(new StatefulBuilder()
173                 .addAugmentation(new Stateful1Builder().setInitiation(Boolean.TRUE).build())
174                 .addAugmentation(new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep
175                     .sync.optimizations.rev200720.Stateful1Builder()
176                         .setIncludeDbVersion(Boolean.TRUE).setTriggeredResync(Boolean.TRUE)
177                         .build())
178                 .build())
179             .build()).addAugmentation(new Tlvs3Builder().setLspDbVersion(this.lspDbVersion).build()).build()).build();
180     }
181
182     private static Pcrpt getSyncMsg() {
183         final SrpBuilder srpBuilder = new SrpBuilder();
184         // not sue whether use 0 instead of nextRequest() or do not insert srp == SRP-ID-number = 0
185         srpBuilder.setOperationId(new SrpIdNumber(Uint32.ONE));
186         return MsgBuilderUtil.createPcRtpMessage(createLsp(Uint32.ZERO, false, Optional.of(
187                 new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev200720.lsp
188                         .object.lsp.TlvsBuilder().addAugmentation(new org.opendaylight.yang.gen.v1.urn.opendaylight
189                             .params.xml.ns.yang.controller.pcep.sync.optimizations.rev200720.Tlvs1Builder()
190                             .setLspDbVersion(new LspDbVersionBuilder().setLspDbVersionValue(Uint64.valueOf(3L))
191                                 .build()).build()).build()), true, false),
192             Optional.of(srpBuilder.build()), createPath(Collections.emptyList()));
193     }
194
195     private static Pcrpt getPcrt() {
196         return MsgBuilderUtil.createPcRtpMessage(new LspBuilder().setPlspId(new PlspId(Uint32.ONE)).setTlvs(
197                 new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev200720.lsp
198                         .object.lsp.TlvsBuilder().setLspIdentifiers(new LspIdentifiersBuilder()
199                         .setLspId(new LspId(Uint32.ONE)).build()).setSymbolicPathName(
200                     new SymbolicPathNameBuilder().setPathName(new SymbolicPathName("test".getBytes())).build())
201                     .addAugmentation(new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller
202                         .pcep.sync.optimizations.rev200720.Tlvs1Builder().setLspDbVersion(new LspDbVersionBuilder()
203                                 .setLspDbVersionValue(Uint64.ONE).build()).build()).build())
204                 .setPlspId(new PlspId(Uint32.ONE)).setSync(true).setRemove(false)
205                         .setOperational(OperationalStatus.Active).build(), Optional.empty(),
206             createPath(Collections.emptyList()));
207     }
208 }