9a85f22a12238356c7893745490b8b5528c32261
[bgpcep.git] / pcep / pcc-mock / src / test / java / org / opendaylight / protocol / pcep / pcc / mock / PCCTriggeredSyncTest.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.protocol.pcep.pcc.mock;
9
10 import static org.junit.Assert.assertNotNull;
11
12 import io.netty.channel.Channel;
13 import java.util.ArrayList;
14 import java.util.Collections;
15 import java.util.List;
16 import org.junit.Test;
17 import org.opendaylight.protocol.pcep.PCEPCapability;
18 import org.opendaylight.protocol.pcep.PCEPSession;
19 import org.opendaylight.protocol.pcep.ietf.stateful07.PCEPStatefulCapability;
20 import org.opendaylight.protocol.pcep.pcc.mock.protocol.PCCPeerProposal;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.PcupdBuilder;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.PlspId;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.SrpIdNumber;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.lsp.object.Lsp;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.lsp.object.LspBuilder;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.pcupd.message.PcupdMessageBuilder;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.pcupd.message.pcupd.message.UpdatesBuilder;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.pcupd.message.pcupd.message.updates.PathBuilder;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.srp.object.Srp;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.srp.object.SrpBuilder;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.Message;
32 import org.opendaylight.yangtools.yang.common.Uint64;
33
34 public class PCCTriggeredSyncTest extends PCCMockCommon {
35     @Test
36     public void testSessionTriggeredSync() throws Exception {
37         final TestingSessionListenerFactory factory = new TestingSessionListenerFactory();
38         final Channel channel = createServer(factory, this.remoteAddress, new PCCPeerProposal());
39         final Uint64 numberOflspAndDBv = Uint64.valueOf(3);
40         final PCEPSession session = createPCCSession(numberOflspAndDBv).get();
41         assertNotNull(session);
42         final TestingSessionListener pceSessionListener = getListener(factory);
43         assertNotNull(pceSessionListener);
44         checkSynchronizedSession(0, pceSessionListener, Uint64.ZERO);
45         this.pccSessionListener.onMessage(session, createTriggerMsg());
46         checkSynchronizedSession(3, pceSessionListener, numberOflspAndDBv);
47         channel.close().get();
48     }
49
50     private static Message createTriggerMsg() {
51         final UpdatesBuilder rb = new UpdatesBuilder();
52         // create PCUpd with mandatory objects and LSP object set to 1
53         final SrpBuilder srpBuilder = new SrpBuilder();
54         srpBuilder.setIgnore(false);
55         srpBuilder.setProcessingRule(false);
56         srpBuilder.setOperationId(new SrpIdNumber(1L));
57         final Srp srp = srpBuilder.build();
58         rb.setSrp(srp);
59
60         final Lsp lsp = new LspBuilder().setPlspId(new PlspId(0L)).setSync(Boolean.TRUE).build();
61         rb.setLsp(lsp);
62
63         final PathBuilder pb = new PathBuilder();
64         rb.setPath(pb.build());
65         final PcupdMessageBuilder ub = new PcupdMessageBuilder();
66         ub.setUpdates(Collections.singletonList(rb.build()));
67         return new PcupdBuilder().setPcupdMessage(ub.build()).build();
68     }
69
70     @Override
71     protected List<PCEPCapability> getCapabilities() {
72         final List<PCEPCapability> caps = new ArrayList<>();
73         caps.add(new PCEPStatefulCapability(true, true, true, true, false, false, true));
74         return caps;
75     }
76 }
77