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

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)
 	}
 

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,9 +390,11 @@ func (db *DB) Close() (err error) {
 		log.Error("localstore: write gc size worker did not return after db close")
 	}
 
+	if writeGCSize {
 		if err := db.writeGCSize(db.getGCSize()); err != nil {
 			log.Error("localstore: write gc size", "err", err)
 		}
+	}
 	return db.shed.Close()
 }