Properly handle notifications from the database about a Reference's backing PersistentValue

Author: Sammster10

src/main/java/net/staticstudios/data/DataManager.java

@@ -9,6 +9,7 @@
 import com.zaxxer.hikari.pool.HikariPool;
 import net.staticstudios.data.data.Data;
 import net.staticstudios.data.data.InitialValue;
+import net.staticstudios.data.data.Reference;
 import net.staticstudios.data.data.UniqueData;
 import net.staticstudios.data.data.collection.PersistentManyToManyCollection;
 import net.staticstudios.data.data.collection.PersistentUniqueDataCollection;
@@ -202,6 +203,11 @@ public <T extends UniqueData> List<T> loadAll(Class<T> clazz) {
                         dummyValueMap.put(value.getSchema() + "." + value.getTable(), value);
                     }
 
+                    if (data instanceof Reference<?> reference) {
+                        PersistentValue<?> value = reference.getBackingValue();
+                        dummyValueMap.put(value.getSchema() + "." + value.getTable(), value);
+                    }
+
                     if (data instanceof SimplePersistentCollection<?> collection) {
                         dummySimplePersistentCollectionMap.put(collection.getSchema() + "." + collection.getTable(), collection);
                         dummySimplePersistentCollections.put(data.getHolder().getRootHolder(), collection);

src/main/java/net/staticstudios/data/data/Reference.java

@@ -89,4 +89,8 @@ public DataHolder getHolder() {
     public UniqueData getRootHolder() {
         return holder.getRootHolder();
     }
+
+    public PersistentValue<UUID> getBackingValue() {
+        return id;
+    }
 }

src/test/java/net/staticstudios/data/ReferenceTest.java

@@ -5,6 +5,7 @@
 import net.staticstudios.data.mock.reference.SnapchatUser;
 import net.staticstudios.data.mock.reference.SnapchatUserSettings;
 import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
 import org.junitpioneer.jupiter.RetryingTest;
 
 import java.sql.SQLException;
@@ -74,7 +75,7 @@ public void testLoadingReference() {
         assertFalse(user.getSettings().getEnableFriendRequests());
     }
 
-    @RetryingTest(5)
+    @Test
     public void testSetReference() {
         MockEnvironment environment = getMockEnvironments().getFirst();
         DataManager dataManager = environment.dataManager();
@@ -88,5 +89,17 @@ public void testSetReference() {
 
         user.setFavoriteUser(null);
         assertNull(user.getFavoriteUser());
+
+        waitForDataPropagation();
+
+        try (Statement statement = getConnection().createStatement()) {
+            statement.executeUpdate("update snapchat.users set favorite_user_id = '" + favoriteUser.getId() + "' where id = '" + user.getId() + "'");
+        } catch (SQLException e) {
+            throw new RuntimeException(e);
+        }
+
+        waitForDataPropagation();
+
+        assertEquals(favoriteUser, user.getFavoriteUser());
     }
 }