test environment to evaluate the return value of put

1 files changed, 10 insertions, 3 deletions

diff --git a/model-data/src/test/java/org/eclipse/viatra/solver/data/map/tests/utils/MapTestEnvironment.java b/model-data/src/test/java/org/eclipse/viatra/solver/data/map/tests/utils/MapTestEnvironment.java
index aaf9f50c..4c043350 100644
--- a/model-data/src/test/java/org/eclipse/viatra/solver/data/map/tests/utils/MapTestEnvironment.java
+++ b/model-data/src/test/java/org/eclipse/viatra/solver/data/map/tests/utils/MapTestEnvironment.java
@@ -103,11 +103,18 @@ public class MapTestEnvironment<K, V> {
         }
 
         public void put(K key, V value) {
-                sut.put(key, value);
+                V oldSutValue = sut.put(key, value);
+                V oldOracleValue;
                 if (value != sut.getDefaultValue()) {
-                        oracle.put(key, value);
+                        oldOracleValue = oracle.put(key, value);
                 } else {
-                        oracle.remove(key);
+                        oldOracleValue = oracle.remove(key);
+                }
+                if(oldSutValue == sut.getDefaultValue() && oldOracleValue != null) {
+                        fail("After put, SUT old value was default, but oracle old walue was " + oldOracleValue);
+                }
+                if(oldSutValue != sut.getDefaultValue()) {
+                        assertEquals(oldOracleValue, oldSutValue);
                 }
         }