From 102c2ef9e845b1c76bfb292a8369b5c6ea491d39 Mon Sep 17 00:00:00 2001
From: nprentza <nprentza@redhat.com>
Date: Wed, 2 Aug 2023 08:41:54 +0300
Subject: [PATCH] init uniqueObjectTypesInStore

---
 .../core/SimpleSerializationReliableRefObjectStore.java      | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/drools-reliability/drools-reliability-core/src/main/java/org/drools/reliability/core/SimpleSerializationReliableRefObjectStore.java b/drools-reliability/drools-reliability-core/src/main/java/org/drools/reliability/core/SimpleSerializationReliableRefObjectStore.java
index 2fc9f3b36b1..6ccf4d6e38b 100644
--- a/drools-reliability/drools-reliability-core/src/main/java/org/drools/reliability/core/SimpleSerializationReliableRefObjectStore.java
+++ b/drools-reliability/drools-reliability-core/src/main/java/org/drools/reliability/core/SimpleSerializationReliableRefObjectStore.java
@@ -32,6 +32,9 @@ public class SimpleSerializationReliableRefObjectStore extends SimpleSerializati
     public SimpleSerializationReliableRefObjectStore(Storage<Long, StoredObject> storage) {
         super(storage);
         uniqueObjectTypesInStore = new HashMap<>();
+        if (storage.size()>0){
+            storage.values().forEach(sObject -> {this.putIntoObjectTypesList( sObject.getObject() );});
+        }
         this.storage = storage.size()>0 ? updateObjectReferences(storage) : storage;
     }
 
@@ -100,7 +103,7 @@ private List<Field> getReferencedObjects(Object object){
         Field[] fields = object.getClass().getDeclaredFields();
 
         List<Field> fieldsWithTypeInTheStore = Arrays.stream(fields)
-                .filter(field -> uniqueObjectTypesInStore.keySet().contains(field.getType()))
+                .filter(field -> uniqueObjectTypesInStore.containsKey(field.getType()))
                 .collect(Collectors.toList());
         return fieldsWithTypeInTheStore;
     }