From 1b89d475a409eaedf31be859ee9ff2591849bd4d Mon Sep 17 00:00:00 2001
From: Katopodisv <vkatopodis@ubitech.eu>
Date: Wed, 28 Jun 2023 18:26:30 +0300
Subject: [PATCH] test: add PolicyRuleServiceValidation tests

---
 .../policy/MutinyContextServiceWrapper.java    | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)
 create mode 100644 src/policy/src/test/java/eu/teraflow/policy/MutinyContextServiceWrapper.java

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 000000000..eff6bcf62
--- /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);
+    }
+}
-- 
GitLab