X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?p=controller.git;a=blobdiff_plain;f=opendaylight%2Fmd-sal%2Fstatistics-manager%2Fsrc%2Ftest%2Fjava%2Ftest%2Fmock%2Futil%2FProviderContextMock.java;fp=opendaylight%2Fmd-sal%2Fstatistics-manager%2Fsrc%2Ftest%2Fjava%2Ftest%2Fmock%2Futil%2FProviderContextMock.java;h=9b40a7684fb0d868b5362a3fba4e0b396e6f2cf8;hp=0000000000000000000000000000000000000000;hb=b495d88f63ba0b5b3959d7742ae683c869cf3ccc;hpb=b3e553ce5b3d3e972cbe19465ab7af2fcb39934c

diff --git a/opendaylight/md-sal/statistics-manager/src/test/java/test/mock/util/ProviderContextMock.java b/opendaylight/md-sal/statistics-manager/src/test/java/test/mock/util/ProviderContextMock.java
new file mode 100644
index 0000000000..9b40a7684f
--- /dev/null
+++ b/opendaylight/md-sal/statistics-manager/src/test/java/test/mock/util/ProviderContextMock.java
@@ -0,0 +1,69 @@
+package test.mock.util;
+
+import org.opendaylight.controller.md.sal.binding.api.DataBroker;
+import org.opendaylight.controller.md.sal.common.api.routing.RouteChangeListener;
+import org.opendaylight.controller.sal.binding.api.BindingAwareBroker;
+import org.opendaylight.controller.sal.binding.api.BindingAwareProvider;
+import org.opendaylight.controller.sal.binding.api.BindingAwareService;
+import org.opendaylight.controller.sal.binding.api.NotificationProviderService;
+import org.opendaylight.controller.sal.binding.api.RpcProviderRegistry;
+import org.opendaylight.controller.sal.binding.api.rpc.RpcContextIdentifier;
+import org.opendaylight.yangtools.concepts.ListenerRegistration;
+import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
+import org.opendaylight.yangtools.yang.binding.RpcService;
+
+public class ProviderContextMock implements BindingAwareBroker.ProviderContext {
+
+    RpcProviderRegistry rpcProviderMock;
+    NotificationProviderService notificationProviderService;
+    DataBroker dataBroker;
+
+    public ProviderContextMock(RpcProviderRegistry rpcProviderMock, DataBroker dataBroker,
+                               NotificationProviderService notificationProviderServiceMock) {
+        this.rpcProviderMock = rpcProviderMock;
+        this.dataBroker = dataBroker;
+        this.notificationProviderService = notificationProviderServiceMock;
+    }
+
+    @Override
+    public void registerFunctionality(BindingAwareProvider.ProviderFunctionality functionality) {
+
+    }
+
+    @Override
+    public void unregisterFunctionality(BindingAwareProvider.ProviderFunctionality functionality) {
+
+    }
+
+    @SuppressWarnings("unchecked")
+    @Override
+    public <T extends BindingAwareService> T getSALService(Class<T> service) {
+        if (service.equals(DataBroker.class)) {
+            return (T) dataBroker;
+        }
+        else if (service.equals(NotificationProviderService.class)) {
+            return (T) notificationProviderService;
+        }
+        return null;
+    }
+
+    @Override
+    public <T extends RpcService> BindingAwareBroker.RpcRegistration<T> addRpcImplementation(Class<T> serviceInterface, T implementation) throws IllegalStateException {
+        return null;
+    }
+
+    @Override
+    public <T extends RpcService> BindingAwareBroker.RoutedRpcRegistration<T> addRoutedRpcImplementation(Class<T> serviceInterface, T implementation) throws IllegalStateException {
+        return null;
+    }
+
+    @Override
+    public <L extends RouteChangeListener<RpcContextIdentifier, InstanceIdentifier<?>>> ListenerRegistration<L> registerRouteChangeListener(L listener) {
+        return null;
+    }
+
+    @Override
+    public <T extends RpcService> T getRpcService(Class<T> serviceInterface) {
+        return rpcProviderMock.getRpcService(serviceInterface);
+    }
+}