swarm/storage/localstore: fix synchronization in TestDB_gcSize (#19235)

2 files changed, 12 insertions, 7 deletions

diff --git a/swarm/storage/localstore/gc_test.go b/swarm/storage/localstore/gc_test.go
index c2ad774d3..8ed34384d 100644
--- a/swarm/storage/localstore/gc_test.go
+++ b/swarm/storage/localstore/gc_test.go
@@ -289,12 +289,9 @@ func TestDB_gcSize(t *testing.T) {
 	}
 
 	// DB.Close writes gc size to disk, so
-	// Instead calling Close, simulate database shutdown
+	// Instead calling Close, close the database
 	// without it.
-	close(db.close)
-	db.updateGCWG.Wait()
-	err = db.shed.Close()
-	if err != nil {
+	if err := db.closeWithOptions(false); err != nil {
 		t.Fatal(err)
 	}
 
diff --git a/swarm/storage/localstore/localstore.go b/swarm/storage/localstore/localstore.go
index a66130fd3..35044115b 100644
--- a/swarm/storage/localstore/localstore.go
+++ b/swarm/storage/localstore/localstore.go
@@ -367,6 +367,12 @@ func New(path string, baseKey []byte, o *Options) (db *DB, err error) {
 
 // Close closes the underlying database.
 func (db *DB) Close() (err error) {
+	return db.closeWithOptions(true)
+}
+
+// closeWithOptions provides a more control which part of closing
+// is done for tests.
+func (db *DB) closeWithOptions(writeGCSize bool) (err error) {
 	close(db.close)
 	db.updateGCWG.Wait()
 
@@ -384,8 +390,10 @@ func (db *DB) Close() (err error) {
 		log.Error("localstore: write gc size worker did not return after db close")
 	}
 
-	if err := db.writeGCSize(db.getGCSize()); err != nil {
-		log.Error("localstore: write gc size", "err", err)
+	if writeGCSize {
+		if err := db.writeGCSize(db.getGCSize()); err != nil {
+			log.Error("localstore: write gc size", "err", err)
+		}
 	}
 	return db.shed.Close()
 }