Fixed a test

hypermodeinc · Jan 7, 2025 · b2b64b3 · b2b64b3
1 parent 6163160
commit b2b64b3
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/posting/list.go b/posting/list.go
@@ -873,6 +873,12 @@ func GetConflictKey(pk x.ParsedKey, key []byte, t *pb.DirectedEdge) uint64 {
 	return conflictKey
 }
 
+// SetTs allows us to set the transaction timestamp in mutation map. Should be used before the posting list is passed
+// on to the functions that would read the data.
+func (l *List) SetTs(readTs uint64) {
+	l.mutationMap.setTs(readTs)
+}
+
 func (l *List) addMutationInternal(ctx context.Context, txn *Txn, t *pb.DirectedEdge) error {
 	l.AssertLock()
 

diff --git a/worker/worker_test.go b/worker/worker_test.go
@@ -56,6 +56,7 @@ func commitTransaction(t *testing.T, edge *pb.DirectedEdge, l *posting.List) {
 	startTs := timestamp()
 	txn := posting.Oracle().RegisterStartTs(startTs)
 	l = txn.Store(l)
+	l.SetTs(startTs)
 	require.NoError(t, l.AddMutationWithIndex(context.Background(), edge, txn))
 
 	commit := commitTs(startTs)