import static org.junit.Assert.assertEquals;
import static org.opendaylight.protocol.bgp.cli.utils.BGPOperationalStateUtils.PROTOCOLS_IID;
import static org.opendaylight.protocol.bgp.cli.utils.NeighborStateCliUtilsTest.NEIGHBOR_ADDRESS;
-import static org.opendaylight.protocol.bgp.cli.utils.PeerGroupStateCliUtilsTest.UTF8;
+import com.google.common.io.Resources;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.PrintStream;
-import java.util.Collections;
+import java.nio.charset.StandardCharsets;
import java.util.concurrent.ExecutionException;
-import org.apache.commons.io.IOUtils;
import org.junit.Test;
import org.opendaylight.mdsal.binding.api.WriteTransaction;
import org.opendaylight.mdsal.binding.dom.adapter.test.AbstractConcurrentDataBrokerTest;
import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.policy.types.rev151009.BGP;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180329.NetworkInstanceProtocol;
import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
+import org.opendaylight.yangtools.yang.binding.util.BindingMap;
public class BGPOperationalStateUtilsTest extends AbstractConcurrentDataBrokerTest {
static final String RIB_ID = "test-rib";
private static final String RIB_NOT_FOUND = "RIB not found for [test-rib]\n";
private final ByteArrayOutputStream output = new ByteArrayOutputStream();
- private final PrintStream stream = new PrintStream(this.output);
+ private final PrintStream stream = new PrintStream(output);
@Test
public void testDisplayBgpOperationalStateNotFound() {
- BGPOperationalStateUtils.displayBgpOperationalState(getDataBroker(), this.stream, RIB_ID, null, null);
- assertEquals(RIB_NOT_FOUND, this.output.toString());
+ BGPOperationalStateUtils.displayBgpOperationalState(getDataBroker(), stream, RIB_ID, null, null);
+ assertEquals(RIB_NOT_FOUND, output.toString());
}
@Test
public void testDisplayBgpOperationalStateFound() throws IOException, ExecutionException, InterruptedException {
createDefaultProtocol();
- BGPOperationalStateUtils.displayBgpOperationalState(getDataBroker(), this.stream, RIB_ID, null, null);
- final String expected = IOUtils.toString(
- getClass().getClassLoader().getResourceAsStream("global.txt"), UTF8);
- assertEquals(expected, this.output.toString());
+ BGPOperationalStateUtils.displayBgpOperationalState(getDataBroker(), stream, RIB_ID, null, null);
+ final String expected = Resources.toString(getClass().getClassLoader().getResource("global.txt"),
+ StandardCharsets.UTF_8);
+ assertEquals(expected, output.toString());
}
private void createDefaultProtocol() throws ExecutionException, InterruptedException {
final Bgp bgp = new BgpBuilder()
.setGlobal(GlobalStateCliUtilsTest.buildGlobal(true).build())
.setNeighbors(new NeighborsBuilder().setNeighbor(
- Collections.singletonList(NeighborStateCliUtilsTest.createBasicNeighbor())).build())
+ BindingMap.of(NeighborStateCliUtilsTest.createBasicNeighbor())).build())
.build();
GlobalStateCliUtilsTest.buildGlobal(true);
final InstanceIdentifier<Bgp> bgpIID = PROTOCOLS_IID
- .child(Protocol.class, new ProtocolKey(BGP.class, RIB_ID))
+ .child(Protocol.class, new ProtocolKey(BGP.VALUE, RIB_ID))
.augmentation(NetworkInstanceProtocol.class).child(Bgp.class);
wt.mergeParentStructurePut(LogicalDatastoreType.OPERATIONAL, bgpIID, bgp);
wt.commit().get();
@Test
public void testDisplayNeighborOperationalState() throws IOException, ExecutionException, InterruptedException {
createDefaultProtocol();
- BGPOperationalStateUtils.displayBgpOperationalState(getDataBroker(), this.stream, RIB_ID, null,
+ BGPOperationalStateUtils.displayBgpOperationalState(getDataBroker(), stream, RIB_ID, null,
NEIGHBOR_ADDRESS);
- final String expected = IOUtils.toString(
- getClass().getClassLoader().getResourceAsStream("empty-neighbor.txt"), UTF8);
- assertEquals(expected, this.output.toString());
+ final String expected = Resources.toString(getClass().getClassLoader().getResource("empty-neighbor.txt"),
+ StandardCharsets.UTF_8);
+ assertEquals(expected, output.toString());
}
}
\ No newline at end of file