sweep: use channeldb.MakeTestDB

This commit is contained in:
Andras Banki-Horvath 2020-06-24 12:57:23 +02:00
parent f2a08e420e
commit 905990eb54
No known key found for this signature in database
GPG Key ID: 80E5375C094198D8

@ -1,8 +1,6 @@
package sweep package sweep
import ( import (
"io/ioutil"
"os"
"testing" "testing"
"github.com/btcsuite/btcd/chaincfg/chainhash" "github.com/btcsuite/btcd/chaincfg/chainhash"
@ -10,38 +8,13 @@ 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
}
// TestStore asserts that the store persists the presented data to disk and is // TestStore asserts that the store persists the presented data to disk and is
// able to retrieve it again. // able to retrieve it again.
func TestStore(t *testing.T) { func TestStore(t *testing.T) {
t.Run("bolt", func(t *testing.T) { t.Run("bolt", func(t *testing.T) {
// Create new store. // Create new store.
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)
} }