mirror of
https://github.com/lightningnetwork/lnd.git
synced 2025-08-03 02:02:17 +02:00
walletunlocker: assert new password can be used to open the store
This commit is contained in:
@@ -527,6 +527,12 @@ func TestChangeWalletPasswordNewRootKey(t *testing.T) {
|
||||
t.Fatal("file exists but it shouldn't")
|
||||
}
|
||||
}
|
||||
|
||||
// Close the old db first.
|
||||
require.NoError(t, store.Backend.Close())
|
||||
|
||||
// Check that the new password can be used to open the db.
|
||||
assertPasswordChanged(t, testDir, req.NewPassword)
|
||||
}
|
||||
|
||||
// TestChangeWalletPasswordStateless checks that trying to change the password
|
||||
@@ -630,6 +636,12 @@ func TestChangeWalletPasswordStateless(t *testing.T) {
|
||||
case <-time.After(defaultTestTimeout):
|
||||
t.Fatalf("ChangePassword timed out")
|
||||
}
|
||||
|
||||
// Close the old db first.
|
||||
require.NoError(t, store.Backend.Close())
|
||||
|
||||
// Check that the new password can be used to open the db.
|
||||
assertPasswordChanged(t, testDir, req.NewPassword)
|
||||
}
|
||||
|
||||
func doChangePassword(service *walletunlocker.UnlockerService,
|
||||
@@ -653,3 +665,21 @@ func doChangePassword(service *walletunlocker.UnlockerService,
|
||||
|
||||
close(errChan)
|
||||
}
|
||||
|
||||
// assertPasswordChanged asserts that the new password can be used to open the
|
||||
// store.
|
||||
func assertPasswordChanged(t *testing.T, testDir string, newPassword []byte) {
|
||||
// Open it and read the root key with the new password.
|
||||
store, err := openOrCreateTestMacStore(
|
||||
testDir, &newPassword, testNetParams,
|
||||
)
|
||||
require.NoError(t, err)
|
||||
|
||||
// Assert that we can read the root key.
|
||||
_, _, err = store.RootKey(defaultRootKeyIDContext)
|
||||
require.NoError(t, err)
|
||||
|
||||
// Close the db once done.
|
||||
require.NoError(t, store.Close())
|
||||
require.NoError(t, store.Backend.Close())
|
||||
}
|
||||
|
Reference in New Issue
Block a user