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

channeldb: mark ApplyMigration as a test helper

Browse Source 
With this change, errors from migrations will have the proper local line
number.
This commit is contained in:
Olaoluwa Osuntokun 2020-03-18 19:14:35 -07:00 committed by Johan T. Halseth
parent db23e34f74
commit c6a8816043
No known key found for this signature in database
GPG Key ID: 15BAADA29DA20D26
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
channeldb/migtest/migtest.go
View File

@ -41,6 +41,8 @@ func ApplyMigration(t *testing.T,
beforeMigration, afterMigration, migrationFunc func(tx kvdb.RwTx) error, beforeMigration, afterMigration, migrationFunc func(tx kvdb.RwTx) error,
shouldFail bool) { shouldFail bool) {
t.Helper()
cdb, cleanUp, err := MakeDB() cdb, cleanUp, err := MakeDB()
defer cleanUp() defer cleanUp()
if err != nil { if err != nil {
@ -55,6 +57,8 @@ func ApplyMigration(t *testing.T,
} }
defer func() { defer func() {
t.Helper()
if r := recover(); r != nil { if r := recover(); r != nil {
err = newError(r) err = newError(r)
} }