mirror of
https://github.com/lightningnetwork/lnd.git
synced 2025-06-06 21:20:32 +02:00
channeldb/db_test: move migration helper back to meta_test
This commit is contained in:
parent
79a4203346
commit
090e97cd3b
@ -5,8 +5,6 @@ import (
|
|||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"github.com/go-errors/errors"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestOpenWithCreate(t *testing.T) {
|
func TestOpenWithCreate(t *testing.T) {
|
||||||
@ -35,56 +33,3 @@ func TestOpenWithCreate(t *testing.T) {
|
|||||||
t.Fatalf("channeldb failed to create data directory")
|
t.Fatalf("channeldb failed to create data directory")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// applyMigration is a helper test function that encapsulates the general steps
|
|
||||||
// which are needed to properly check the result of applying migration function.
|
|
||||||
func applyMigration(t *testing.T, beforeMigration, afterMigration func(d *DB),
|
|
||||||
migrationFunc migration, shouldFail bool) {
|
|
||||||
|
|
||||||
cdb, cleanUp, err := makeTestDB()
|
|
||||||
defer cleanUp()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// beforeMigration usually used for populating the database
|
|
||||||
// with test data.
|
|
||||||
beforeMigration(cdb)
|
|
||||||
|
|
||||||
// Create test meta info with zero database version and put it on disk.
|
|
||||||
// Than creating the version list pretending that new version was added.
|
|
||||||
meta := &Meta{DbVersionNumber: 0}
|
|
||||||
if err := cdb.PutMeta(meta); err != nil {
|
|
||||||
t.Fatalf("unable to store meta data: %v", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
versions := []version{
|
|
||||||
{
|
|
||||||
number: 0,
|
|
||||||
migration: nil,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
number: 1,
|
|
||||||
migration: migrationFunc,
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
defer func() {
|
|
||||||
if r := recover(); r != nil {
|
|
||||||
err = errors.New(r)
|
|
||||||
}
|
|
||||||
|
|
||||||
if err == nil && shouldFail {
|
|
||||||
t.Fatal("error wasn't received on migration stage")
|
|
||||||
} else if err != nil && !shouldFail {
|
|
||||||
t.Fatal("error was received on migration stage")
|
|
||||||
}
|
|
||||||
|
|
||||||
// afterMigration usually used for checking the database state and
|
|
||||||
// throwing the error if something went wrong.
|
|
||||||
afterMigration(cdb)
|
|
||||||
}()
|
|
||||||
|
|
||||||
// Sync with the latest version - applying migration function.
|
|
||||||
err = cdb.syncVersions(versions)
|
|
||||||
}
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user