From b7d3bc46f895d82446b3e987ede7afecf537faaf Mon Sep 17 00:00:00 2001
From: Boris Nagaev <bnagaev@gmail.com>
Date: Mon, 21 Apr 2025 00:57:17 -0300
Subject: [PATCH] sweepbatcher: auto-increment sweep.ID in StoreMock

---
 sweepbatcher/store_mock.go | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/sweepbatcher/store_mock.go b/sweepbatcher/store_mock.go
index 47b27ee8..90896aac 100644
--- a/sweepbatcher/store_mock.go
+++ b/sweepbatcher/store_mock.go
@@ -15,6 +15,7 @@ type StoreMock struct {
 	batches map[int32]dbBatch
 	sweeps  map[wire.OutPoint]dbSweep
 	mu      sync.Mutex
+	sweepID int32
 }
 
 // NewStoreMock instantiates a new mock store.
@@ -122,7 +123,18 @@ func (s *StoreMock) UpsertSweep(ctx context.Context, sweep *dbSweep) error {
 	s.mu.Lock()
 	defer s.mu.Unlock()
 
-	s.sweeps[sweep.Outpoint] = *sweep
+	sweepCopy := *sweep
+
+	if old, exists := s.sweeps[sweep.Outpoint]; exists {
+		// Preserve existing sweep ID.
+		sweepCopy.ID = old.ID
+	} else {
+		// Assign fresh sweep ID.
+		sweepCopy.ID = s.sweepID
+		s.sweepID++
+	}
+
+	s.sweeps[sweep.Outpoint] = sweepCopy
 
 	return nil
 }