From f086dd0d57cf3ad55b950b98b37e62750ac9994e Mon Sep 17 00:00:00 2001
From: Jeffrey Picard <jeff@jeffreypicard.com>
Date: Mon, 3 Oct 2022 10:16:23 +0000
Subject: [PATCH] fix testing

---
 db/db.go      | 4 +++-
 db/db_test.go | 5 +++++
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/db/db.go b/db/db.go
index 4ec12e3..f78bcdd 100644
--- a/db/db.go
+++ b/db/db.go
@@ -606,6 +606,8 @@ func GetDBColumnFamilies(name string, secondayPath string, cfNames []string) (*R
 		LastState:        nil,
 		Height:           0,
 		Headers:          nil,
+		OpenIterators:    make(map[string][]chan struct{}),
+		ItMut:            sync.RWMutex{},
 		ShutdownChan:     make(chan struct{}, 1),
 		DoneChan:         make(chan struct{}, 1),
 	}
@@ -1080,7 +1082,7 @@ func GenerateTestData(prefix byte, fileName string) {
 		log.Fatalln(err)
 	}
 
-	options := NewIterateOptions().WithDB(db)
+	options := NewIterateOptions()
 	options.WithRawKey(true).WithRawValue(true).WithIncludeValue(true)
 	options.WithPrefix([]byte{prefix})
 
diff --git a/db/db_test.go b/db/db_test.go
index 85ed7ec..d6ebc9f 100644
--- a/db/db_test.go
+++ b/db/db_test.go
@@ -7,6 +7,7 @@ import (
 	"log"
 	"os"
 	"strings"
+	"sync"
 	"testing"
 
 	dbpkg "github.com/lbryio/herald.go/db"
@@ -93,6 +94,10 @@ func OpenAndFillTmpDBColumnFamlies(filePath string) (*dbpkg.ReadOnlyDBColumnFami
 		LastState:        nil,
 		Height:           0,
 		Headers:          nil,
+		OpenIterators:    make(map[string][]chan struct{}),
+		ItMut:            sync.RWMutex{},
+		ShutdownChan:     make(chan struct{}, 1),
+		DoneChan:         make(chan struct{}, 1),
 	}
 
 	// err = dbpkg.ReadDBState(myDB) //TODO: Figure out right place for this