Cleanup deprecation warnings in bgp/cli

[bgpcep.git] / bgp / cli / src / test / java / org / opendaylight / protocol / bgp / cli / utils / PeerGroupStateCliUtilsTest.java

diff --git a/bgp/cli/src/test/java/org/opendaylight/protocol/bgp/cli/utils/PeerGroupStateCliUtilsTest.java b/bgp/cli/src/test/java/org/opendaylight/protocol/bgp/cli/utils/PeerGroupStateCliUtilsTest.java
index 6d37d05bd94b70dd762687f3470c8b07ab8868c3..af1c10df10ce1ed19e6838e3936539241d907c92 100644 (file)
--- a/bgp/cli/src/test/java/org/opendaylight/protocol/bgp/cli/utils/PeerGroupStateCliUtilsTest.java
+++ b/bgp/cli/src/test/java/org/opendaylight/protocol/bgp/cli/utils/PeerGroupStateCliUtilsTest.java
@@ -17,7 +17,6 @@ import java.nio.charset.StandardCharsets;
 import java.util.Collections;
 import org.junit.Test;
 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.peer.group.PeerGroupBuilder;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180329.PeerGroupStateAugmentation;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180329.PeerGroupStateAugmentationBuilder;
 import org.opendaylight.yangtools.yang.common.Uint32;
 
@@ -39,16 +38,16 @@ public final class PeerGroupStateCliUtilsTest {
 
     @Test
     public void testPeerGroupStateCli() throws IOException {
-        final PeerGroupBuilder peerGroup = new PeerGroupBuilder().setPeerGroupName(TEST_GROUP);
-
-        final PeerGroupStateAugmentation groupState = new PeerGroupStateAugmentationBuilder()
-                .setTotalPrefixes(Uint32.ONE)
-                .setTotalPaths(Uint32.TWO)
-                .build();
-
-        peerGroup.setState(new org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group
-                .StateBuilder().addAugmentation(PeerGroupStateAugmentation.class, groupState).build());
-        PeerGroupStateCliUtils.displayPeerOperationalState(Collections.singletonList(peerGroup.build()), this.stream);
+        PeerGroupStateCliUtils.displayPeerOperationalState(Collections.singletonList(new PeerGroupBuilder()
+            .setPeerGroupName(TEST_GROUP)
+            .setState(new org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group
+                .StateBuilder()
+                    .addAugmentation(new PeerGroupStateAugmentationBuilder()
+                        .setTotalPrefixes(Uint32.ONE)
+                        .setTotalPaths(Uint32.TWO)
+                        .build())
+                    .build())
+            .build()), this.stream);
 
         final String expected = Resources.toString(getClass().getClassLoader().getResource("peer-group.txt"),
             StandardCharsets.UTF_8);