opendaylight/md-sal/statistics-manager/src/test/java/test/mock/util/ProviderContextMock.java

   1 package test.mock.util;
   2
   3 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
   4 import org.opendaylight.controller.md.sal.common.api.routing.RouteChangeListener;
   5 import org.opendaylight.controller.sal.binding.api.BindingAwareBroker;
   6 import org.opendaylight.controller.sal.binding.api.BindingAwareProvider;
   7 import org.opendaylight.controller.sal.binding.api.BindingAwareService;
   8 import org.opendaylight.controller.sal.binding.api.NotificationProviderService;
   9 import org.opendaylight.controller.sal.binding.api.RpcProviderRegistry;
  10 import org.opendaylight.controller.sal.binding.api.rpc.RpcContextIdentifier;
  11 import org.opendaylight.yangtools.concepts.ListenerRegistration;
  12 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  13 import org.opendaylight.yangtools.yang.binding.RpcService;
  14
  15 public class ProviderContextMock implements BindingAwareBroker.ProviderContext {
  16
  17     RpcProviderRegistry rpcProviderMock;
  18     NotificationProviderService notificationProviderService;
  19     DataBroker dataBroker;
  20
  21     public ProviderContextMock(RpcProviderRegistry rpcProviderMock, DataBroker dataBroker,
  22                                NotificationProviderService notificationProviderServiceMock) {
  23         this.rpcProviderMock = rpcProviderMock;
  24         this.dataBroker = dataBroker;
  25         this.notificationProviderService = notificationProviderServiceMock;
  26     }
  27
  28     @Override
  29     public void registerFunctionality(BindingAwareProvider.ProviderFunctionality functionality) {
  30
  31     }
  32
  33     @Override
  34     public void unregisterFunctionality(BindingAwareProvider.ProviderFunctionality functionality) {
  35
  36     }
  37
  38     @SuppressWarnings("unchecked")
  39     @Override
  40     public <T extends BindingAwareService> T getSALService(Class<T> service) {
  41         if (service.equals(DataBroker.class)) {
  42             return (T) dataBroker;
  43         }
  44         else if (service.equals(NotificationProviderService.class)) {
  45             return (T) notificationProviderService;
  46         }
  47         return null;
  48     }
  49
  50     @Override
  51     public <T extends RpcService> BindingAwareBroker.RpcRegistration<T> addRpcImplementation(Class<T> serviceInterface, T implementation) throws IllegalStateException {
  52         return null;
  53     }
  54
  55     @Override
  56     public <T extends RpcService> BindingAwareBroker.RoutedRpcRegistration<T> addRoutedRpcImplementation(Class<T> serviceInterface, T implementation) throws IllegalStateException {
  57         return null;
  58     }
  59
  60     @Override
  61     public <L extends RouteChangeListener<RpcContextIdentifier, InstanceIdentifier<?>>> ListenerRegistration<L> registerRouteChangeListener(L listener) {
  62         return null;
  63     }
  64
  65     @Override
  66     public <T extends RpcService> T getRpcService(Class<T> serviceInterface) {
  67         return rpcProviderMock.getRpcService(serviceInterface);
  68     }
  69 }