// postEncryptionTesting runs a series of checks on the wallet after it has
// been encrypted, to make sure that locking, unlocking, and spending after
// unlocking are all happening in the correct order and returning the correct
// errors.
func postEncryptionTesting(m modules.Miner, w *Wallet, masterKey crypto.TwofishKey) {
if !w.Encrypted() {
panic("wallet is not encrypted when starting postEncryptionTesting")
}
if w.Unlocked() {
panic("wallet is unlocked when starting postEncryptionTesting")
}
if len(w.seeds) != 0 {
panic("wallet has seeds in it when startin postEncryptionTesting")
}
// Try unlocking and using the wallet.
err := w.Unlock(masterKey)
if err != nil {
panic(err)
}
err = w.Unlock(masterKey)
if err != errAlreadyUnlocked {
panic(err)
}
// Mine enough coins so that a balance appears (and some buffer for the
// send later).
for i := types.BlockHeight(0); i <= types.MaturityDelay+1; i++ {
_, err := m.AddBlock()
if err != nil {
panic(err)
}
}
siacoinBal, _, _ := w.ConfirmedBalance()
if siacoinBal.Cmp(types.NewCurrency64(0)) <= 0 {
panic("wallet balance reported as 0 after maturing some mined blocks")
}
err = w.Unlock(masterKey)
if err != errAlreadyUnlocked {
panic(err)
}
// Lock, unlock, and trying using the wallet some more.
err = w.Lock()
if err != nil {
panic(err)
}
err = w.Lock()
if err != modules.ErrLockedWallet {
panic(err)
}
err = w.Unlock(crypto.TwofishKey{})
if err != modules.ErrBadEncryptionKey {
panic(err)
}
err = w.Unlock(masterKey)
if err != nil {
panic(err)
}
// Verify that the secret keys have been restored by sending coins to the
// void. Send more coins than are received by mining a block.
_, err = w.SendSiacoins(types.CalculateCoinbase(0), types.UnlockHash{})
if err != nil {
panic(err)
}
_, err = m.AddBlock()
if err != nil {
panic(err)
}
siacoinBal2, _, _ := w.ConfirmedBalance()
if siacoinBal2.Cmp(siacoinBal) >= 0 {
panic("balance did not increase")
}
}
