pcep/spi/src/test/java/org/opendaylight/protocol/pcep/spi/RegistryTest.java

   1 /*
   2  * Copyright (c) 2014 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.spi;
   9
  10 import static org.mockito.ArgumentMatchers.any;
  11 import static org.mockito.ArgumentMatchers.anyBoolean;
  12 import static org.mockito.Mockito.doNothing;
  13 import static org.mockito.Mockito.doReturn;
  14
  15 import io.netty.buffer.ByteBuf;
  16 import io.netty.buffer.Unpooled;
  17 import java.util.ArrayList;
  18 import java.util.Collections;
  19 import java.util.List;
  20 import org.junit.Before;
  21 import org.junit.Test;
  22 import org.junit.runner.RunWith;
  23 import org.mockito.Mock;
  24 import org.mockito.junit.MockitoJUnitRunner;
  25 import org.opendaylight.protocol.pcep.spi.pojo.AbstractPCEPExtensionProviderActivator;
  26 import org.opendaylight.protocol.pcep.spi.pojo.ServiceLoaderPCEPExtensionProviderContext;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.iana.rev130816.EnterpriseNumber;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.message.rev181109.Keepalive;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.message.rev181109.KeepaliveBuilder;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.Message;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.Object;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.ObjectHeader;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.Tlv;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.exclude.route.object.xro.Subobject;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.exclude.route.object.xro.SubobjectBuilder;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.of.list.tlv.OfList;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.of.list.tlv.OfListBuilder;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.open.object.OpenBuilder;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.rp.object.Rp;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.vendor.information.EnterpriseSpecificInformation;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.vendor.information.objects.VendorInformationObject;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.vendor.information.tlvs.VendorInformationTlv;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev150820.basic.explicit.route.subobjects.subobject.type.AsNumberCase;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev150820.basic.explicit.route.subobjects.subobject.type.AsNumberCaseBuilder;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev150820.basic.explicit.route.subobjects.subobject.type.IpPrefixCase;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev150820.label.subobject.LabelType;
  47 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev150820.label.subobject.label.type.Type1LabelCase;
  48 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev150820.label.subobject.label.type.Type1LabelCaseBuilder;
  49 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev150820.record.route.subobjects.subobject.type.IpPrefixCaseBuilder;
  50 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev150820.record.route.subobjects.subobject.type.LabelCase;
  51 import org.opendaylight.yangtools.concepts.Registration;
  52 import org.opendaylight.yangtools.yang.common.Uint32;
  53
  54 @RunWith(MockitoJUnitRunner.StrictStubs.class)
  55 public class RegistryTest {
  56
  57     @Mock
  58     XROSubobjectParser xroParser;
  59     @Mock
  60     XROSubobjectSerializer xroSerializer;
  61
  62     @Mock
  63     RROSubobjectParser rroParser;
  64     @Mock
  65     RROSubobjectSerializer rroSerializer;
  66
  67     @Mock
  68     EROSubobjectParser eroParser;
  69     @Mock
  70     EROSubobjectSerializer eroSerializer;
  71
  72     @Mock
  73     TlvParser tlvParser;
  74     @Mock
  75     TlvSerializer tlvSerializer;
  76
  77     @Mock
  78     ObjectParser objectParser;
  79
  80     @Mock
  81     ObjectSerializer objectSerializer;
  82
  83     @Mock
  84     MessageParser msgParser;
  85     @Mock
  86     MessageSerializer msgSerializer;
  87
  88     @Mock
  89     LabelParser labelParser;
  90     @Mock
  91     LabelSerializer labelSerializer;
  92
  93     @Mock
  94     VendorInformationObject esi;
  95
  96     @Mock
  97     EnterpriseSpecificInformation es;
  98
  99     @Mock
 100     VendorInformationTlv viTlv;
 101
 102     public final List<Registration> regs = new ArrayList<>();
 103
 104     final AbstractPCEPExtensionProviderActivator activator = new AbstractPCEPExtensionProviderActivator() {
 105
 106         @Override
 107         protected List<? extends Registration> startImpl(final PCEPExtensionProviderContext context) {
 108             return RegistryTest.this.regs;
 109         }
 110     };
 111
 112     final PCEPExtensionProviderContext ctx = ServiceLoaderPCEPExtensionProviderContext.getSingletonInstance();
 113
 114     @SuppressWarnings("unchecked")
 115     @Before
 116     public void setUp() throws PCEPDeserializerException {
 117         doReturn(null).when(this.xroParser).parseSubobject(any(ByteBuf.class), anyBoolean());
 118         doNothing().when(this.xroSerializer).serializeSubobject(any(Subobject.class), any(ByteBuf.class));
 119
 120         doReturn(this.viTlv).when(this.tlvParser).parseTlv(any(ByteBuf.class));
 121         doNothing().when(this.tlvSerializer).serializeTlv(any(Tlv.class), any(ByteBuf.class));
 122
 123         doReturn(5).when(this.objectParser).getObjectClass();
 124         doReturn(1).when(this.objectParser).getObjectType();
 125         doReturn(new OpenBuilder().build()).when(this.objectParser).parseObject(any(ObjectHeader.class),
 126             any(ByteBuf.class));
 127         doNothing().when(this.objectSerializer).serializeObject(any(Object.class), any(ByteBuf.class));
 128
 129         doReturn(null).when(this.msgParser).parseMessage(any(ByteBuf.class), any(List.class));
 130         doNothing().when(this.msgSerializer).serializeMessage(any(Message.class), any(ByteBuf.class));
 131
 132         doNothing().when(this.labelSerializer).serializeLabel(anyBoolean(), anyBoolean(), any(LabelType.class),
 133             any(ByteBuf.class));
 134
 135         doReturn(this.es).when(this.esi).getEnterpriseSpecificInformation();
 136
 137         doReturn(this.es).when(this.viTlv).getEnterpriseSpecificInformation();
 138
 139         doReturn(EnterpriseSpecificInformation.class).when(this.es).implementedInterface();
 140     }
 141
 142     @Test
 143     public void testRegistry() throws PCEPDeserializerException {
 144         this.regs.add(this.ctx.registerXROSubobjectParser(2, this.xroParser));
 145         this.regs.add(this.ctx.registerXROSubobjectSerializer(AsNumberCase.class, this.xroSerializer));
 146
 147         this.regs.add(this.ctx.registerRROSubobjectParser(3, this.rroParser));
 148         this.regs.add(this.ctx.registerRROSubobjectSerializer(LabelCase.class, this.rroSerializer));
 149
 150         this.regs.add(this.ctx.registerEROSubobjectParser(4, this.eroParser));
 151         this.regs.add(this.ctx.registerEROSubobjectSerializer(IpPrefixCase.class, this.eroSerializer));
 152
 153         this.regs.add(this.ctx.registerTlvParser(1, this.tlvParser));
 154         this.regs.add(this.ctx.registerTlvSerializer(OfList.class, this.tlvSerializer));
 155
 156         this.regs.add(this.ctx.registerObjectParser(this.objectParser));
 157         this.regs.add(this.ctx.registerObjectSerializer(Rp.class, this.objectSerializer));
 158
 159         this.regs.add(this.ctx.registerMessageParser(6, this.msgParser));
 160         this.regs.add(this.ctx.registerMessageSerializer(Keepalive.class, this.msgSerializer));
 161
 162         this.regs.add(this.ctx.registerLabelParser(7, this.labelParser));
 163         this.regs.add(this.ctx.registerLabelSerializer(Type1LabelCase.class, this.labelSerializer));
 164
 165         this.regs.add(this.ctx.registerVendorInformationObjectParser(new EnterpriseNumber(Uint32.valueOf(10)),
 166             this.objectParser));
 167         this.regs.add(this.ctx.registerVendorInformationObjectSerializer(EnterpriseSpecificInformation.class,
 168             this.objectSerializer));
 169
 170         this.regs.add(this.ctx.registerVendorInformationTlvParser(new EnterpriseNumber(Uint32.valueOf(12)),
 171             this.tlvParser));
 172         this.regs.add(this.ctx.registerVendorInformationTlvSerializer(EnterpriseSpecificInformation.class,
 173             this.tlvSerializer));
 174
 175         final ByteBuf buffer = Unpooled.buffer();
 176         this.ctx.getXROSubobjectHandlerRegistry().parseSubobject(2, buffer, false);
 177         this.ctx.getXROSubobjectHandlerRegistry().serializeSubobject(new SubobjectBuilder()
 178             .setSubobjectType(new AsNumberCaseBuilder().build()).build(), buffer);
 179
 180         this.ctx.getEROSubobjectHandlerRegistry().parseSubobject(3, buffer, true);
 181         this.ctx.getEROSubobjectHandlerRegistry().serializeSubobject(new org.opendaylight.yang.gen.v1.urn.opendaylight
 182             .params.xml.ns.yang.pcep.types.rev181109.explicit.route.object.ero.SubobjectBuilder()
 183             .setSubobjectType(new AsNumberCaseBuilder().build()).build(), buffer);
 184
 185         this.ctx.getRROSubobjectHandlerRegistry().parseSubobject(4, buffer);
 186         this.ctx.getRROSubobjectHandlerRegistry().serializeSubobject(new org.opendaylight.yang.gen.v1.urn.opendaylight
 187             .params.xml.ns.yang.pcep.types.rev181109.reported.route.object.rro.SubobjectBuilder()
 188             .setSubobjectType(new IpPrefixCaseBuilder().build()).build(), buffer);
 189
 190         this.ctx.getLabelHandlerRegistry().parseLabel(1, buffer);
 191         this.ctx.getLabelHandlerRegistry().serializeLabel(true, false, new Type1LabelCaseBuilder().build(), buffer);
 192
 193         this.ctx.getTlvHandlerRegistry().parseTlv(2, buffer);
 194         this.ctx.getTlvHandlerRegistry().serializeTlv(new OfListBuilder().build(), buffer);
 195
 196         this.ctx.getObjectHandlerRegistry().parseObject(4, 1, new ObjectHeaderImpl(true, false), buffer);
 197         this.ctx.getObjectHandlerRegistry().serializeObject(new OpenBuilder().build(), buffer);
 198
 199         this.ctx.getMessageHandlerRegistry().parseMessage(6, buffer, Collections.emptyList());
 200         this.ctx.getMessageHandlerRegistry().serializeMessage(new KeepaliveBuilder().build(), buffer);
 201
 202         this.ctx.getVendorInformationObjectRegistry().parseVendorInformationObject(
 203             new EnterpriseNumber(Uint32.valueOf(10)), new ObjectHeaderImpl(true, false), buffer);
 204         this.ctx.getVendorInformationObjectRegistry().serializeVendorInformationObject(this.esi, buffer);
 205
 206         this.ctx.getVendorInformationTlvRegistry().parseVendorInformationTlv(
 207             new EnterpriseNumber(Uint32.valueOf(12)), buffer);
 208         this.ctx.getVendorInformationTlvRegistry().serializeVendorInformationTlv(this.viTlv, buffer);
 209
 210         this.activator.start(this.ctx);
 211         this.activator.close();
 212     }
 213 }