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 authored and Johan T. Halseth committed 2021-02-24 14:34:57 +01:00
1 parent db23e34f74
commit c6a8816043
1 file changed
+4
+4
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)
}