lbcwallet/waddrmgr/error_test.go
Dave Collins c8bdd71074 waddrmgr: Make create/update logic more explicit.
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.
2015-03-04 14:12:47 -06:00

138 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
}
}
}