Bump upstreams to 2022.09
[bgpcep.git] / bgp / parser-impl / src / test / java / org / opendaylight / protocol / bgp / parser / impl / BGPParserTest.java
index 77bb869561066e5aacf17a6d300904054bc9de1a..21f61cfe8978c4ecdf26ab182fd92263331304f1 100644 (file)
@@ -535,13 +535,13 @@ public class BGPParserTest {
         final Update message = BGPParserTest.updateParser.parseMessageBody(Unpooled.copiedBuffer(body), messageLength,
             null);
 
-        final Class<? extends AddressFamily> afi = message.getAttributes()
-            .augmentation(AttributesUnreach.class).getMpUnreachNlri().getAfi();
-        final Class<? extends SubsequentAddressFamily> safi = message.getAttributes()
-            .augmentation(AttributesUnreach.class).getMpUnreachNlri().getSafi();
+        final AddressFamily afi = message.getAttributes().augmentation(AttributesUnreach.class)
+            .getMpUnreachNlri().getAfi();
+        final SubsequentAddressFamily safi = message.getAttributes().augmentation(AttributesUnreach.class)
+            .getMpUnreachNlri().getSafi();
 
-        assertEquals(Ipv6AddressFamily.class, afi);
-        assertEquals(UnicastSubsequentAddressFamily.class, safi);
+        assertEquals(Ipv6AddressFamily.VALUE, afi);
+        assertEquals(UnicastSubsequentAddressFamily.VALUE, safi);
 
         final ByteBuf buffer = Unpooled.buffer();
         BGPParserTest.updateParser.serializeMessage(message, buffer);
@@ -570,13 +570,13 @@ public class BGPParserTest {
         final Update message = BGPParserTest.updateParser.parseMessageBody(Unpooled.copiedBuffer(body), messageLength,
             null);
 
-        final Class<? extends AddressFamily> afi = message.getAttributes()
-            .augmentation(AttributesUnreach.class).getMpUnreachNlri().getAfi();
-        final Class<? extends SubsequentAddressFamily> safi = message.getAttributes()
-            .augmentation(AttributesUnreach.class).getMpUnreachNlri().getSafi();
+        final AddressFamily afi = message.getAttributes().augmentation(AttributesUnreach.class).getMpUnreachNlri()
+            .getAfi();
+        final SubsequentAddressFamily safi = message.getAttributes().augmentation(AttributesUnreach.class)
+            .getMpUnreachNlri().getSafi();
 
-        assertEquals(Ipv6AddressFamily.class, afi);
-        assertEquals(UnicastSubsequentAddressFamily.class, safi);
+        assertEquals(Ipv6AddressFamily.VALUE, afi);
+        assertEquals(UnicastSubsequentAddressFamily.VALUE, safi);
 
         final ByteBuf buffer = Unpooled.buffer();
         BGPParserTest.updateParser.serializeMessage(message, buffer);