Code Review / bgpcep.git / blobdiff
? search:
summary | shortlog | log | commit | commitdiff | review | tree
raw | inline | side by side
Bump upstreams to 2022.09
[bgpcep.git] / bgp / parser-impl / src / test / java / org / opendaylight / protocol / bgp / parser / impl / MultiProtocolCapabilityHandlerTest.java
diff --git a/bgp/parser-impl/src/test/java/org/opendaylight/protocol/bgp/parser/impl/MultiProtocolCapabilityHandlerTest.java b/bgp/parser-impl/src/test/java/org/opendaylight/protocol/bgp/parser/impl/MultiProtocolCapabilityHandlerTest.java
index c17a60384baa141665e76628d75eeabc45e2b88b..d094a7db5491798f38896603e030949f297a2394 100644 (file)
--- a/bgp/parser-impl/src/test/java/org/opendaylight/protocol/bgp/parser/impl/MultiProtocolCapabilityHandlerTest.java
+++ b/bgp/parser-impl/src/test/java/org/opendaylight/protocol/bgp/parser/impl/MultiProtocolCapabilityHandlerTest.java
@@ -5,60 +5,68 @@
  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
  * and is available at http://www.eclipse.org/legal/epl-v10.html
  */
-
 package org.opendaylight.protocol.bgp.parser.impl;
 
 import static org.junit.Assert.assertEquals;
+import static org.mockito.Mockito.doReturn;
 
 import io.netty.buffer.ByteBuf;
 import io.netty.buffer.Unpooled;
 import org.junit.Before;
 import org.junit.Test;
+import org.junit.runner.RunWith;
 import org.mockito.Mock;
-import org.mockito.Mockito;
-import org.mockito.MockitoAnnotations;
+import org.mockito.junit.MockitoJUnitRunner;
 import org.opendaylight.protocol.bgp.parser.BGPDocumentedException;
 import org.opendaylight.protocol.bgp.parser.BGPParsingException;
 import org.opendaylight.protocol.bgp.parser.impl.message.open.MultiProtocolCapabilityHandler;
 import org.opendaylight.protocol.bgp.parser.spi.AddressFamilyRegistry;
 import org.opendaylight.protocol.bgp.parser.spi.SubsequentAddressFamilyRegistry;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.message.bgp.parameters.optional.capabilities.CParameters;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.message.bgp.parameters.optional.capabilities.CParametersBuilder;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.CParameters1;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.CParameters1Builder;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.mp.capabilities.MultiprotocolCapabilityBuilder;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.Ipv6AddressFamily;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.UnicastSubsequentAddressFamily;
-
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.open.message.bgp.parameters.optional.capabilities.CParameters;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.open.message.bgp.parameters.optional.capabilities.CParametersBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.CParameters1;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.CParameters1Builder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.mp.capabilities.MultiprotocolCapabilityBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.Ipv6AddressFamily;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.UnicastSubsequentAddressFamily;
+
+@RunWith(MockitoJUnitRunner.StrictStubs.class)
 public class MultiProtocolCapabilityHandlerTest {
-
-    private final static Class afi = Ipv6AddressFamily.class;
-    @Mock private AddressFamilyRegistry afir;
-    @Mock private AddressFamilyRegistry afirExpection;
-    private final static Class safi = UnicastSubsequentAddressFamily.class;
-    @Mock private SubsequentAddressFamilyRegistry safir;
-    @Mock private SubsequentAddressFamilyRegistry safirException;
+    private static final Ipv6AddressFamily AFI = Ipv6AddressFamily.VALUE;
+    private static final UnicastSubsequentAddressFamily SAFI = UnicastSubsequentAddressFamily.VALUE;
+
+    @Mock
+    private AddressFamilyRegistry afir;
+    @Mock
+    private AddressFamilyRegistry afirExpection;
+    @Mock
+    private SubsequentAddressFamilyRegistry safir;
+    @Mock
+    private SubsequentAddressFamilyRegistry safirException;
 
     @Before
     public void setUp() {
-        MockitoAnnotations.initMocks(this);
-        Mockito.doReturn(260).when(this.afir).numberForClass(afi);
-        Mockito.doReturn(this.afi).when(this.afir).classForFamily(260);
+        doReturn(260).when(this.afir).numberForClass(AFI);
+        doReturn(MultiProtocolCapabilityHandlerTest.AFI).when(this.afir).classForFamily(260);
 
-        Mockito.doReturn(null).when(this.afirExpection).numberForClass(afi);
-        Mockito.doReturn(null).when(this.afirExpection).classForFamily(260);
+        doReturn(null).when(this.afirExpection).numberForClass(AFI);
 
-        Mockito.doReturn(4).when(this.safir).numberForClass(safi);
-        Mockito.doReturn(this.safi).when(this.safir).classForFamily(4);
+        doReturn(4).when(this.safir).numberForClass(SAFI);
+        doReturn(MultiProtocolCapabilityHandlerTest.SAFI).when(this.safir).classForFamily(4);
 
-        Mockito.doReturn(null).when(this.safirException).numberForClass(safi);
-        Mockito.doReturn(null).when(this.safirException).classForFamily(4);
+        doReturn(null).when(this.safirException).numberForClass(SAFI);
     }
 
     @Test
     public void testCapabilityHandler() throws BGPDocumentedException, BGPParsingException {
-        final CParameters capabilityToSerialize = new CParametersBuilder().addAugmentation(CParameters1.class, new CParameters1Builder().setMultiprotocolCapability(
-            new MultiprotocolCapabilityBuilder().setAfi(this.afi).setSafi(this.safi).build()).build()).build();
+        final CParameters capabilityToSerialize = new CParametersBuilder()
+                .addAugmentation(new CParameters1Builder()
+                    .setMultiprotocolCapability(new MultiprotocolCapabilityBuilder()
+                        .setAfi(MultiProtocolCapabilityHandlerTest.AFI)
+                        .setSafi(MultiProtocolCapabilityHandlerTest.SAFI)
+                        .build())
+                    .build())
+                .build();
 
         final ByteBuf bytes = Unpooled.buffer();
         final MultiProtocolCapabilityHandler handler = new MultiProtocolCapabilityHandler(this.afir, this.safir);
@@ -68,33 +76,49 @@ public class MultiProtocolCapabilityHandlerTest {
         assertEquals(capabilityToSerialize.hashCode(), newCaps.hashCode());
     }
 
-    @Test(expected=IllegalArgumentException.class)
+    @Test(expected = IllegalArgumentException.class)
     public void testUnhandledAfi() {
-        final CParameters capabilityToSerialize = new CParametersBuilder().addAugmentation(CParameters1.class, new CParameters1Builder().setMultiprotocolCapability(
-            new MultiprotocolCapabilityBuilder().setAfi(this.afi).setSafi(this.safi).build()).build()).build();
+        final CParameters capabilityToSerialize = new CParametersBuilder()
+                .addAugmentation(new CParameters1Builder()
+                    .setMultiprotocolCapability(new MultiprotocolCapabilityBuilder()
+                        .setAfi(MultiProtocolCapabilityHandlerTest.AFI)
+                        .setSafi(MultiProtocolCapabilityHandlerTest.SAFI).build())
+                    .build())
+                .build();
 
         final ByteBuf bytes = Unpooled.buffer();
-        final MultiProtocolCapabilityHandler handler = new MultiProtocolCapabilityHandler(this.afirExpection, this.safir);
+        final MultiProtocolCapabilityHandler handler = new MultiProtocolCapabilityHandler(this.afirExpection,
+            this.safir);
         handler.serializeCapability(capabilityToSerialize, bytes);
     }
 
-    @Test(expected=IllegalArgumentException.class)
+    @Test(expected = IllegalArgumentException.class)
     public void testUnhandledSafi() {
-        final CParameters capabilityToSerialize = new CParametersBuilder().addAugmentation(CParameters1.class, new CParameters1Builder().setMultiprotocolCapability(
-            new MultiprotocolCapabilityBuilder().setAfi(this.afi).setSafi(this.safi).build()).build()).build();
+        final CParameters capabilityToSerialize = new CParametersBuilder()
+                .addAugmentation(new CParameters1Builder()
+                    .setMultiprotocolCapability(new MultiprotocolCapabilityBuilder()
+                        .setAfi(MultiProtocolCapabilityHandlerTest.AFI)
+                        .setSafi(MultiProtocolCapabilityHandlerTest.SAFI)
+                        .build())
+                    .build())
+                .build();
 
         final ByteBuf bytes = Unpooled.buffer();
-        final MultiProtocolCapabilityHandler handler = new MultiProtocolCapabilityHandler(this.afir, this.safirException);
+        final MultiProtocolCapabilityHandler handler = new MultiProtocolCapabilityHandler(this.afir,
+            this.safirException);
         handler.serializeCapability(capabilityToSerialize, bytes);
     }
 
     @Test
     public void noSerializationTest() {
-        final CParameters capabilityNoAugmentation = new CParametersBuilder().addAugmentation(CParameters1.class, null).build();
-        final CParameters capabilityNoMP = new CParametersBuilder().addAugmentation(CParameters1.class, new CParameters1Builder().build()).build();
+        final CParameters capabilityNoAugmentation = new CParametersBuilder().removeAugmentation(CParameters1.class)
+                .build();
+        final CParameters capabilityNoMP = new CParametersBuilder().addAugmentation(new CParameters1Builder().build())
+                .build();
 
         final ByteBuf bytes = Unpooled.buffer();
-        final MultiProtocolCapabilityHandler handler = new MultiProtocolCapabilityHandler(this.afir, this.safirException);
+        final MultiProtocolCapabilityHandler handler = new MultiProtocolCapabilityHandler(this.afir,
+            this.safirException);
         handler.serializeCapability(capabilityNoAugmentation, bytes);
         assertEquals(0, bytes.readableBytes());
         handler.serializeCapability(capabilityNoMP, bytes);
BGP/PCEP protocol library