80 lines
2.6 KiB
Go
80 lines
2.6 KiB
Go
// Copyright (c) 2014 The btcsuite developers
|
|
// Use of this source code is governed by an ISC
|
|
// license that can be found in the LICENSE file.
|
|
|
|
/*
|
|
This test file is part of the waddrmgr package rather than than the
|
|
waddrmgr_test package so it can bridge access to the internals to properly test
|
|
cases which are either not possible or can't reliably be tested via the public
|
|
interface. The functions are only exported while the tests are being run.
|
|
*/
|
|
|
|
package waddrmgr
|
|
|
|
import (
|
|
"errors"
|
|
|
|
"github.com/roasbeef/btcwallet/snacl"
|
|
)
|
|
|
|
// TstLatestMgrVersion makes the unexported latestMgrVersion variable available
|
|
// for change when the tests are run.
|
|
var TstLatestMgrVersion = &latestMgrVersion
|
|
|
|
// Replace the Manager.newSecretKey function with the given one and calls
|
|
// the callback function. Afterwards the original newSecretKey
|
|
// function will be restored.
|
|
func TstRunWithReplacedNewSecretKey(callback func()) {
|
|
orig := newSecretKey
|
|
defer func() {
|
|
newSecretKey = orig
|
|
}()
|
|
newSecretKey = func(passphrase *[]byte, config *ScryptOptions) (*snacl.SecretKey, error) {
|
|
return nil, snacl.ErrDecryptFailed
|
|
}
|
|
callback()
|
|
}
|
|
|
|
// TstCheckPublicPassphrase returns true if the provided public passphrase is
|
|
// correct for the manager.
|
|
func (m *Manager) TstCheckPublicPassphrase(pubPassphrase []byte) bool {
|
|
secretKey := snacl.SecretKey{Key: &snacl.CryptoKey{}}
|
|
secretKey.Parameters = m.masterKeyPub.Parameters
|
|
err := secretKey.DeriveKey(&pubPassphrase)
|
|
return err == nil
|
|
}
|
|
|
|
// failingCryptoKey is an implementation of the EncryptorDecryptor interface
|
|
// with intentionally fails when attempting to encrypt or decrypt with it.
|
|
type failingCryptoKey struct {
|
|
cryptoKey
|
|
}
|
|
|
|
// Encrypt intenionally returns a failure when invoked to test error paths.
|
|
//
|
|
// This is part of the EncryptorDecryptor interface implementation.
|
|
func (c *failingCryptoKey) Encrypt(in []byte) ([]byte, error) {
|
|
return nil, errors.New("failed to encrypt")
|
|
}
|
|
|
|
// Decrypt intenionally returns a failure when invoked to test error paths.
|
|
//
|
|
// This is part of the EncryptorDecryptor interface implementation.
|
|
func (c *failingCryptoKey) Decrypt(in []byte) ([]byte, error) {
|
|
return nil, errors.New("failed to decrypt")
|
|
}
|
|
|
|
// TstRunWithFailingCryptoKeyPriv runs the provided callback with the
|
|
// private crypto key replaced with a version that fails to help test error
|
|
// paths.
|
|
func TstRunWithFailingCryptoKeyPriv(m *Manager, callback func()) {
|
|
orig := m.cryptoKeyPriv
|
|
defer func() {
|
|
m.cryptoKeyPriv = orig
|
|
}()
|
|
m.cryptoKeyPriv = &failingCryptoKey{}
|
|
callback()
|
|
}
|
|
|
|
// TstDefaultAccountName is the constant defaultAccountName exported for tests.
|
|
const TstDefaultAccountName = defaultAccountName
|