2 * Copyright (c) 2016 Cisco Systems, Inc. and others. All rights reserved.
4 * This program and the accompanying materials are made available under the
5 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6 * and is available at http://www.eclipse.org/legal/epl-v10.html
9 package org.opendaylight.protocol.bgp.benchmark.app;
11 import java.util.concurrent.ExecutionException;
12 import javax.management.MalformedObjectNameException;
13 import org.junit.After;
14 import org.junit.Assert;
15 import org.junit.Before;
16 import org.junit.Test;
17 import org.mockito.Mock;
18 import org.mockito.Mockito;
19 import org.mockito.MockitoAnnotations;
20 import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
21 import org.opendaylight.controller.md.sal.binding.test.AbstractDataBrokerTest;
22 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
23 import org.opendaylight.controller.md.sal.common.api.data.ReadFailedException;
24 import org.opendaylight.controller.sal.binding.api.BindingAwareBroker.RoutedRpcRegistration;
25 import org.opendaylight.controller.sal.binding.api.RpcProviderRegistry;
26 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
27 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Prefix;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev150305.ipv4.routes.Ipv4Routes;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.ApplicationRib;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.ApplicationRibId;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.ApplicationRibKey;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.rib.Tables;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.rib.TablesKey;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.Ipv4AddressFamily;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.UnicastSubsequentAddressFamily;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.odl.bgp.app.peer.benchmark.rev160309.AddPrefixInputBuilder;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.odl.bgp.app.peer.benchmark.rev160309.AddPrefixOutput;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.odl.bgp.app.peer.benchmark.rev160309.DeletePrefixInputBuilder;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.odl.bgp.app.peer.benchmark.rev160309.DeletePrefixOutput;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.odl.bgp.app.peer.benchmark.rev160309.OdlBgpAppPeerBenchmarkService;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.odl.bgp.app.peer.benchmark.rev160309.output.Result;
42 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
43 import org.opendaylight.yangtools.yang.binding.KeyedInstanceIdentifier;
44 import org.opendaylight.yangtools.yang.common.RpcResult;
46 public class AppPeerBenchmarkTest extends AbstractDataBrokerTest {
48 private static final String PREFIX = "1.1.1.1/32";
49 private static final String NH = "127.0.0.1";
50 private static final String PEER_RIB_ID = "app-peer";
52 private static final InstanceIdentifier<ApplicationRib> BASE_IID = KeyedInstanceIdentifier.builder(ApplicationRib.class,
53 new ApplicationRibKey(new ApplicationRibId(PEER_RIB_ID))).build();
54 private static final InstanceIdentifier tablesIId = BASE_IID
55 .child(Tables.class, new TablesKey(Ipv4AddressFamily.class, UnicastSubsequentAddressFamily.class));
56 private static final InstanceIdentifier<Ipv4Routes> ROUTES = tablesIId.child(Ipv4Routes.class);
59 private RpcProviderRegistry rpcRegistry;
61 private RoutedRpcRegistration<OdlBgpAppPeerBenchmarkService> registration;
63 private AppPeerBenchmark appPeerBenchmark;
66 public void setUp() throws MalformedObjectNameException {
67 MockitoAnnotations.initMocks(this);
68 Mockito.doReturn(this.registration).when(this.rpcRegistry).addRpcImplementation(Mockito.<Class<OdlBgpAppPeerBenchmarkService>>any(),
69 Mockito.any(OdlBgpAppPeerBenchmarkService.class));
70 Mockito.doNothing().when(this.registration).close();
71 this.appPeerBenchmark = new AppPeerBenchmark(getDataBroker(), this.rpcRegistry, PEER_RIB_ID);
75 public void testRpcs() throws InterruptedException, ExecutionException, ReadFailedException {
76 final RpcResult<AddPrefixOutput> addRpcResult = this.appPeerBenchmark.addPrefix(new AddPrefixInputBuilder()
79 .setNexthop(new Ipv4Address(NH))
80 .setPrefix(new Ipv4Prefix(PREFIX)).build()).get();
81 final Result addResult = addRpcResult.getResult().getResult();
82 Assert.assertEquals(1, addResult.getCount().intValue());
83 Assert.assertEquals(1, addResult.getRate().intValue());
84 final Ipv4Routes routesAfterAdd = getRoutes();
85 Assert.assertEquals(1, routesAfterAdd.getIpv4Route().size());
87 final RpcResult<DeletePrefixOutput> deleteRpcResult = this.appPeerBenchmark.deletePrefix(new DeletePrefixInputBuilder()
90 .setPrefix(new Ipv4Prefix(PREFIX)).build()).get();
91 final Result deleteResult = deleteRpcResult.getResult().getResult();
92 Assert.assertEquals(1, deleteResult.getCount().intValue());
93 Assert.assertEquals(1, deleteResult.getRate().intValue());
94 final Ipv4Routes routesAfterDelete = getRoutes();
95 Assert.assertTrue(routesAfterDelete.getIpv4Route().isEmpty());
99 public void tearDown() {
100 this.appPeerBenchmark.close();
103 private Ipv4Routes getRoutes() throws ReadFailedException {
104 final ReadOnlyTransaction rTx = getDataBroker().newReadOnlyTransaction();
105 return rTx.read(LogicalDatastoreType.CONFIGURATION, ROUTES).checkedGet().get();