Fix TxStateTest after removing equals() from diffsets

neo4j · Aug 29, 2018 · 91f5a3c · 91f5a3c
1 parent 89b9a93
commit 91f5a3c
Showing 1 changed file with 14 additions and 4 deletions.
diff --git a/community/kernel/src/test/java/org/neo4j/kernel/impl/api/state/TxStateTest.java b/community/kernel/src/test/java/org/neo4j/kernel/impl/api/state/TxStateTest.java
@@ -300,9 +300,9 @@ public void shouldComputeIndexUpdatesOnTxStateWithAddedNodes()
         UnmodifiableMap<ValueTuple,? extends LongDiffSets> diffSets = state.getIndexUpdates( indexOn_1_1.schema() );
 
         // THEN
-        assertEquals( addedNodes( 42L ), diffSets.get( ValueTuple.of( Values.stringValue( "value42" ) ) ) );
-        assertEquals( addedNodes( 43L ), diffSets.get( ValueTuple.of( Values.stringValue( "value43" ) ) ) );
-        assertEquals( addedNodes( 41L ), diffSets.get( ValueTuple.of( Values.stringValue( "value41" ) ) ) );
+        assertEqualDiffSets( addedNodes( 42L ), diffSets.get( ValueTuple.of( Values.stringValue( "value42" ) ) ) );
+        assertEqualDiffSets( addedNodes( 43L ), diffSets.get( ValueTuple.of( Values.stringValue( "value43" ) ) ) );
+        assertEqualDiffSets( addedNodes( 41L ), diffSets.get( ValueTuple.of( Values.stringValue( "value41" ) ) ) );
     }
 
     @Test
@@ -318,7 +318,11 @@ public void shouldComputeSortedIndexUpdatesOnTxStateWithAddedNodes()
 
         TreeMap<ValueTuple,LongDiffSets> expected = sortedAddedNodesDiffSets( 42, 41, 43 );
         // THEN
-        assertEquals( expected, diffSets );
+        assertEquals( expected.keySet(), diffSets.keySet() );
+        for ( final ValueTuple key : expected.keySet() )
+        {
+            assertEqualDiffSets( expected.get( key ), diffSets.get( key ) );
+        }
     }
 
     // endregion
@@ -959,4 +963,10 @@ private <T> void withProperties( Collection<Pair<Long,T>> nodesWithValues )
             }
         };
     }
+
+    private static void assertEqualDiffSets( LongDiffSets expected, LongDiffSets actual )
+    {
+        assertEquals( expected.getRemoved(), actual.getRemoved() );
+        assertEquals( expected.getAdded(), actual.getAdded() );
+    }
 }