akovalenko/lnd.xprv
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

lnd: use channeldb.MakeTestDB in nursery store tests

Browse Source
This commit is contained in:
Andras Banki-Horvath 2020-06-24 12:55:05 +02:00
parent cbce8e8872
commit f2a08e420e
No known key found for this signature in database
GPG Key ID: 80E5375C094198D8
1 changed files with 3 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
nursery_store_test.go
View File

@ -3,8 +3,6 @@
package lnd package lnd
import ( import (
"io/ioutil"
"os"
"reflect" "reflect"
"testing" "testing"
@ -12,31 +10,6 @@ import (
"github.com/lightningnetwork/lnd/channeldb" "github.com/lightningnetwork/lnd/channeldb"
) )
// makeTestDB creates a new instance of the ChannelDB for testing purposes. A
// callback which cleans up the created temporary directories is also returned
// and intended to be executed after the test completes.
func makeTestDB() (*channeldb.DB, func(), error) {
// First, create a temporary directory to be used for the duration of
// this test.
tempDirName, err := ioutil.TempDir("", "channeldb")
if err != nil {
return nil, nil, err
}
// Next, create channeldb for the first time.
cdb, err := channeldb.Open(tempDirName)
if err != nil {
return nil, nil, err
}
cleanUp := func() {
cdb.Close()
os.RemoveAll(tempDirName)
}
return cdb, cleanUp, nil
}
type incubateTest struct { type incubateTest struct {
nOutputs int nOutputs int
chanPoint *wire.OutPoint chanPoint *wire.OutPoint
@ -75,7 +48,7 @@ func initIncubateTests() {
// TestNurseryStoreInit verifies basic properties of the nursery store before // TestNurseryStoreInit verifies basic properties of the nursery store before
// any modifying calls are made. // any modifying calls are made.
func TestNurseryStoreInit(t *testing.T) { func TestNurseryStoreInit(t *testing.T) {
cdb, cleanUp, err := makeTestDB() cdb, cleanUp, err := channeldb.MakeTestDB()
if err != nil { if err != nil {
t.Fatalf("unable to open channel db: %v", err) t.Fatalf("unable to open channel db: %v", err)
} }
@ -95,7 +68,7 @@ func TestNurseryStoreInit(t *testing.T) {
// outputs through the nursery store, verifying the properties of the // outputs through the nursery store, verifying the properties of the
// intermediate states. // intermediate states.
func TestNurseryStoreIncubate(t *testing.T) { func TestNurseryStoreIncubate(t *testing.T) {
cdb, cleanUp, err := makeTestDB() cdb, cleanUp, err := channeldb.MakeTestDB()
if err != nil { if err != nil {
t.Fatalf("unable to open channel db: %v", err) t.Fatalf("unable to open channel db: %v", err)
} }
@ -336,7 +309,7 @@ func TestNurseryStoreIncubate(t *testing.T) {
// populated entries from the height index as it is purged, and that the last // populated entries from the height index as it is purged, and that the last
// purged height is set appropriately. // purged height is set appropriately.
func TestNurseryStoreGraduate(t *testing.T) { func TestNurseryStoreGraduate(t *testing.T) {
cdb, cleanUp, err := makeTestDB() cdb, cleanUp, err := channeldb.MakeTestDB()
if err != nil { if err != nil {
t.Fatalf("unable to open channel db: %v", err) t.Fatalf("unable to open channel db: %v", err)
} }