channeldb: mark ApplyMigration as a test helper

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

View File

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