c8bdd71074
This commit makes the creation and updating of the address manager more explicit so it's easier to upgrade in the future. In particular, rather than treating the initial creation as an upgrade by relying on creating the initial buckets on the fly on each load, the code now explicitly provides distinct create and upgrade paths that are invoked from the Create and Open functions, respectively. It also adds some commented out sample code to illustrate how upgrades should be done and a check to ensure bumping the version number without writing upgrade code results in a new error, ErrUpgrade, being returned. Finally, a test has been added for the new functionality.
137 lines
4.3 KiB
Go
137 lines
4.3 KiB
Go
/*
|
|
* Copyright (c) 2014 Conformal Systems LLC <info@conformal.com>
|
|
*
|
|
* Permission to use, copy, modify, and distribute this software for any
|
|
* purpose with or without fee is hereby granted, provided that the above
|
|
* copyright notice and this permission notice appear in all copies.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
|
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
|
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
|
|
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
|
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
|
|
* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
|
|
* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
|
*/
|
|
|
|
package waddrmgr_test
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/btcsuite/btcwallet/waddrmgr"
|
|
)
|
|
|
|
// TestErrorCodeStringer tests the stringized output for the ErrorCode type.
|
|
func TestErrorCodeStringer(t *testing.T) {
|
|
tests := []struct {
|
|
in waddrmgr.ErrorCode
|
|
want string
|
|
}{
|
|
{waddrmgr.ErrDatabase, "ErrDatabase"},
|
|
{waddrmgr.ErrUpgrade, "ErrUpgrade"},
|
|
{waddrmgr.ErrKeyChain, "ErrKeyChain"},
|
|
{waddrmgr.ErrCrypto, "ErrCrypto"},
|
|
{waddrmgr.ErrInvalidKeyType, "ErrInvalidKeyType"},
|
|
{waddrmgr.ErrNoExist, "ErrNoExist"},
|
|
{waddrmgr.ErrAlreadyExists, "ErrAlreadyExists"},
|
|
{waddrmgr.ErrCoinTypeTooHigh, "ErrCoinTypeTooHigh"},
|
|
{waddrmgr.ErrAccountNumTooHigh, "ErrAccountNumTooHigh"},
|
|
{waddrmgr.ErrLocked, "ErrLocked"},
|
|
{waddrmgr.ErrWatchingOnly, "ErrWatchingOnly"},
|
|
{waddrmgr.ErrInvalidAccount, "ErrInvalidAccount"},
|
|
{waddrmgr.ErrAddressNotFound, "ErrAddressNotFound"},
|
|
{waddrmgr.ErrAccountNotFound, "ErrAccountNotFound"},
|
|
{waddrmgr.ErrDuplicate, "ErrDuplicate"},
|
|
{waddrmgr.ErrTooManyAddresses, "ErrTooManyAddresses"},
|
|
{waddrmgr.ErrWrongPassphrase, "ErrWrongPassphrase"},
|
|
{waddrmgr.ErrWrongNet, "ErrWrongNet"},
|
|
|
|
// The following error codes are defined in pool_error.go.
|
|
{waddrmgr.ErrSeriesStorage, "ErrSeriesStorage"},
|
|
{waddrmgr.ErrSeriesNotExists, "ErrSeriesNotExists"},
|
|
{waddrmgr.ErrSeriesAlreadyExists, "ErrSeriesAlreadyExists"},
|
|
{waddrmgr.ErrSeriesAlreadyEmpowered, "ErrSeriesAlreadyEmpowered"},
|
|
{waddrmgr.ErrKeyIsPrivate, "ErrKeyIsPrivate"},
|
|
{waddrmgr.ErrKeyNeuter, "ErrKeyNeuter"},
|
|
{waddrmgr.ErrKeyMismatch, "ErrKeyMismatch"},
|
|
{waddrmgr.ErrKeysPrivatePublicMismatch, "ErrKeysPrivatePublicMismatch"},
|
|
{waddrmgr.ErrKeyDuplicate, "ErrKeyDuplicate"},
|
|
{waddrmgr.ErrTooFewPublicKeys, "ErrTooFewPublicKeys"},
|
|
{waddrmgr.ErrVotingPoolNotExists, "ErrVotingPoolNotExists"},
|
|
{waddrmgr.ErrScriptCreation, "ErrScriptCreation"},
|
|
{waddrmgr.ErrTooManyReqSignatures, "ErrTooManyReqSignatures"},
|
|
|
|
{0xffff, "Unknown ErrorCode (65535)"},
|
|
}
|
|
t.Logf("Running %d tests", len(tests))
|
|
for i, test := range tests {
|
|
result := test.in.String()
|
|
if result != test.want {
|
|
t.Errorf("String #%d\ngot: %s\nwant: %s", i, result,
|
|
test.want)
|
|
continue
|
|
}
|
|
}
|
|
}
|
|
|
|
// TestManagerError tests the error output for the ManagerError type.
|
|
func TestManagerError(t *testing.T) {
|
|
tests := []struct {
|
|
in waddrmgr.ManagerError
|
|
want string
|
|
}{
|
|
// Manager level error.
|
|
{
|
|
waddrmgr.ManagerError{Description: "human-readable error"},
|
|
"human-readable error",
|
|
},
|
|
|
|
// Encapsulated database error.
|
|
{
|
|
waddrmgr.ManagerError{
|
|
Description: "failed to store master private " +
|
|
"key parameters",
|
|
ErrorCode: waddrmgr.ErrDatabase,
|
|
Err: fmt.Errorf("underlying db error"),
|
|
},
|
|
"failed to store master private key parameters: " +
|
|
"underlying db error",
|
|
},
|
|
|
|
// Encapsulated key chain error.
|
|
{
|
|
waddrmgr.ManagerError{
|
|
Description: "failed to derive extended key " +
|
|
"branch 0",
|
|
ErrorCode: waddrmgr.ErrKeyChain,
|
|
Err: fmt.Errorf("underlying error"),
|
|
},
|
|
"failed to derive extended key branch 0: underlying " +
|
|
"error",
|
|
},
|
|
|
|
// Encapsulated crypto error.
|
|
{
|
|
waddrmgr.ManagerError{
|
|
Description: "failed to decrypt account 0 " +
|
|
"private key",
|
|
ErrorCode: waddrmgr.ErrCrypto,
|
|
Err: fmt.Errorf("underlying error"),
|
|
},
|
|
"failed to decrypt account 0 private key: underlying " +
|
|
"error",
|
|
},
|
|
}
|
|
|
|
t.Logf("Running %d tests", len(tests))
|
|
for i, test := range tests {
|
|
result := test.in.Error()
|
|
if result != test.want {
|
|
t.Errorf("Error #%d\ngot: %s\nwant: %s", i, result,
|
|
test.want)
|
|
continue
|
|
}
|
|
}
|
|
}
|