diff --git a/src/policy/src/test/java/eu/teraflow/policy/MutinyContextServiceWrapper.java b/src/policy/src/test/java/eu/teraflow/policy/MutinyContextServiceWrapper.java
new file mode 100644
index 0000000000000000000000000000000000000000..eff6bcf6219e2c771ee875ad5e1fa56e8deadaa1
--- /dev/null
+++ b/src/policy/src/test/java/eu/teraflow/policy/MutinyContextServiceWrapper.java
@@ -0,0 +1,18 @@
+package eu.teraflow.policy;
+
+import context.ContextOuterClass.Device;
+import context.ContextOuterClass.DeviceId;
+import context.MutinyContextServiceGrpc;
+import io.smallrye.mutiny.Uni;
+
+public class MutinyContextServiceWrapper {
+    private final MutinyContextServiceGrpc.MutinyContextServiceStub stub;
+
+    public MutinyContextServiceWrapper(MutinyContextServiceGrpc.MutinyContextServiceStub stub) {
+        this.stub = stub;
+    }
+
+    public Uni<Device> getDevice(DeviceId deviceId) {
+        return stub.getDevice(deviceId);
+    }
+}