Migrate away from legacy setters
[bgpcep.git] / bgp / parser-api / src / test / java / org / opendaylight / protocol / bgp / parser / APITest.java
index daaafa85f9bbe376ad942ae58f436580fd226fa8..4f53fab5d946238a1fbf87c4c0c8f8592aea9c74 100644 (file)
@@ -8,59 +8,85 @@
 package org.opendaylight.protocol.bgp.parser;
 
 import static org.junit.Assert.assertEquals;
-import static org.junit.Assert.assertNull;
+import static org.junit.Assert.assertTrue;
 
 import java.util.ArrayList;
 import java.util.List;
-
 import org.junit.Test;
 import org.opendaylight.protocol.util.ByteArray;
-import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.AsNumber;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.Open;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.OpenBuilder;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.BgpParameters;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.BgpParametersBuilder;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.bgp.parameters.c.parameters.As4BytesCaseBuilder;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.bgp.parameters.c.parameters.as4.bytes._case.As4BytesCapabilityBuilder;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.AsNumber;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.Open;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.OpenBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.open.message.BgpParameters;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.open.message.BgpParametersBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.open.message.bgp.parameters.OptionalCapabilities;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.open.message.bgp.parameters.OptionalCapabilitiesBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.open.message.bgp.parameters.optional.capabilities.CParametersBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.open.message.bgp.parameters.optional.capabilities.c.parameters.As4BytesCapabilityBuilder;
+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.yangtools.yang.common.Uint16;
+import org.opendaylight.yangtools.yang.common.Uint32;
+import org.opendaylight.yangtools.yang.common.Uint8;
 
 public class APITest {
 
-       @Test
-       public void testDocumentedException() {
-               final BGPDocumentedException de = new BGPDocumentedException("Some message", BGPError.BAD_BGP_ID);
-               assertEquals("Some message", de.getMessage());
-               assertEquals(BGPError.BAD_BGP_ID, de.getError());
-               assertNull(de.getData());
+    @Test
+    public void testDocumentedException() {
+        final BGPDocumentedException de = new BGPDocumentedException("Some message", BGPError.BAD_BGP_ID);
+        assertEquals("Some message", de.getMessage());
+        assertEquals(BGPError.BAD_BGP_ID, de.getError());
+        assertEquals(0, de.getData().length);
+
+        final BGPDocumentedException doc = BGPDocumentedException.badMessageLength("Wrong length", 5000);
+        assertEquals(5000, ByteArray.bytesToInt(doc.getData()));
+    }
 
-               final BGPDocumentedException doc = BGPDocumentedException.badMessageLength("Wrong length", 5000);
-               assertEquals(5000, ByteArray.bytesToInt(doc.getData()));
-       }
+    @Test
+    public void testParsingException() {
+        final BGPParsingException de = new BGPParsingException("Some message");
+        assertEquals("Some message", de.getMessage());
 
-       @Test
-       public void testParsingException() {
-               final BGPParsingException de = new BGPParsingException("Some message");
-               assertEquals("Some message", de.getMessage());
-       }
+        final BGPParsingException de1 = new BGPParsingException("Some message", new IllegalArgumentException("text"));
+        assertEquals("text", de1.getCause().getMessage());
+    }
 
-       @Test
-       public void testBGPError() {
-               assertEquals(BGPError.BAD_MSG_TYPE, BGPError.forValue(1, 3));
-       }
+    @Test
+    public void testBGPError() {
+        assertEquals(BGPError.BAD_MSG_TYPE, BGPError.forValue(Uint8.ONE, Uint8.valueOf(3)));
+    }
 
-       @Test
-       public void testTerminationReason() {
-               assertEquals(BGPError.BAD_PEER_AS.toString(), new BGPTerminationReason(BGPError.BAD_PEER_AS).getErrorMessage());
-       }
+    @Test
+    public void testAsNumberUtil() {
+        final List<BgpParameters> params = new ArrayList<>();
+        final List<OptionalCapabilities> capas = new ArrayList<>();
+        capas.add(new OptionalCapabilitiesBuilder().setCParameters(new CParametersBuilder().addAugmentation(
+                CParameters1.class, new CParameters1Builder().setMultiprotocolCapability(
+                        new MultiprotocolCapabilityBuilder().build()).build()).build()).build());
+        capas.add(new OptionalCapabilitiesBuilder().setCParameters(
+                new CParametersBuilder().setAs4BytesCapability(new As4BytesCapabilityBuilder().setAsNumber(
+                        new AsNumber(Uint32.valueOf(35))).build()).build()).build());
+        params.add(new BgpParametersBuilder().setOptionalCapabilities(capas).build());
+        final Open open1 = new OpenBuilder().setBgpParameters(params).build();
+        assertEquals(35L, AsNumberUtil.advertizedAsNumber(open1).getValue().longValue());
 
-       @Test
-       public void testAsNumberUtil() {
-               final List<BgpParameters> params = new ArrayList<>();
-               params.add(new BgpParametersBuilder().setCParameters(
-                               new As4BytesCaseBuilder().setAs4BytesCapability(new As4BytesCapabilityBuilder().setAsNumber(new AsNumber(35L)).build()).build()).build());
-               final Open open1 = new OpenBuilder().setBgpParameters(params).build();
-               assertEquals(35L, AsNumberUtil.advertizedAsNumber(open1).getValue().longValue());
+        final Open open2 = new OpenBuilder().setMyAsNumber(Uint16.valueOf(10)).build();
+        assertEquals(10, AsNumberUtil.advertizedAsNumber(open2).getValue().intValue());
+    }
 
-               final Open open2 = new OpenBuilder().setMyAsNumber(10).build();
-               assertEquals(10, AsNumberUtil.advertizedAsNumber(open2).getValue().intValue());
-       }
+    @Test
+    public void testBgpExtendedMessageUtil() {
+        final List<BgpParameters> params = new ArrayList<>();
+        final List<OptionalCapabilities> capas = new ArrayList<>();
+        capas.add(new OptionalCapabilitiesBuilder().setCParameters(new CParametersBuilder().addAugmentation(
+                CParameters1.class, new CParameters1Builder()
+                        .setMultiprotocolCapability(new MultiprotocolCapabilityBuilder()
+                                .build()).build()).build()).build());
+        capas.add(new OptionalCapabilitiesBuilder().setCParameters(
+                BgpExtendedMessageUtil.EXTENDED_MESSAGE_CAPABILITY).build());
+        params.add(new BgpParametersBuilder().setOptionalCapabilities(capas).build());
+        final Open open1 = new OpenBuilder().setBgpParameters(params).build();
+        assertTrue(BgpExtendedMessageUtil.advertizedBgpExtendedMessageCapability(open1));
+    }
 }