Merge "Return response codes from policy controller"

[nonrtric.git] / policy-agent / src / test / java / org / oransc / policyagent / utils / MockA1Client.java

diff --git a/policy-agent/src/test/java/org/oransc/policyagent/utils/MockA1Client.java b/policy-agent/src/test/java/org/oransc/policyagent/utils/MockA1Client.java
index 1a93b4d..aca29f5 100644 (file)
--- a/policy-agent/src/test/java/org/oransc/policyagent/utils/MockA1Client.java
+++ b/policy-agent/src/test/java/org/oransc/policyagent/utils/MockA1Client.java
@@ -42,23 +42,27 @@ public class MockA1Client implements A1Client {
 
     @Override
     public Mono<Collection<String>> getPolicyTypeIdentities(String nearRtRicUrl) {
-        Vector<String> result = new Vector<>();
-        for (PolicyType p : this.policyTypes.getAll()) {
-            result.add(p.name());
+        synchronized (this.policyTypes) {
+            Vector<String> result = new Vector<>();
+            for (PolicyType p : this.policyTypes.getAll()) {
+                result.add(p.name());
+            }
+            return Mono.just(result);
         }
-        return Mono.just(result);
     }
 
     @Override
     public Mono<Collection<String>> getPolicyIdentities(String nearRtRicUrl) {
-        Vector<String> result = new Vector<>();
-        for (Policy policy : getPolicies(nearRtRicUrl).getAll()) {
-            if (policy.ric().getConfig().baseUrl().equals(nearRtRicUrl)) {
-                result.add(policy.id());
+        synchronized (this.policies) {
+            Vector<String> result = new Vector<>();
+            for (Policy policy : getPolicies(nearRtRicUrl).getAll()) {
+                if (policy.ric().getConfig().baseUrl().equals(nearRtRicUrl)) {
+                    result.add(policy.id());
+                }
             }
-        }
 
-        return Mono.just(result);
+            return Mono.just(result);
+        }
     }
 
     @Override