fe0f60991a
Doing that may cause erratic test failures when we run them in parallel, so move the functions the tests need to mock as struct fields that are not shared across tests.
101 lines
3.4 KiB
Go
101 lines
3.4 KiB
Go
/*
|
|
* Copyright (c) 2014 The btcsuite developers
|
|
*
|
|
* 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 votingpool
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"reflect"
|
|
"runtime"
|
|
"testing"
|
|
|
|
"github.com/btcsuite/btclog"
|
|
"github.com/btcsuite/btcwallet/waddrmgr"
|
|
)
|
|
|
|
func init() {
|
|
runtime.GOMAXPROCS(runtime.NumCPU())
|
|
|
|
// Enable logging (Debug level) to aid debugging failing tests.
|
|
logger, err := btclog.NewLoggerFromWriter(os.Stdout, btclog.DebugLvl)
|
|
if err != nil {
|
|
fmt.Printf("Failed to initialize stdout logger: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
UseLogger(logger)
|
|
}
|
|
|
|
// TstCheckError ensures the passed error is a votingpool.Error with an error
|
|
// code that matches the passed error code.
|
|
func TstCheckError(t *testing.T, testName string, gotErr error, wantErrCode ErrorCode) {
|
|
vpErr, ok := gotErr.(Error)
|
|
if !ok {
|
|
t.Errorf("%s: unexpected error type - got %T (%s), want %T",
|
|
testName, gotErr, gotErr, Error{})
|
|
}
|
|
if vpErr.ErrorCode != wantErrCode {
|
|
t.Errorf("%s: unexpected error code - got %s (%s), want %s",
|
|
testName, vpErr.ErrorCode, vpErr, wantErrCode)
|
|
}
|
|
}
|
|
|
|
// TstRunWithManagerUnlocked calls the given callback with the manager unlocked,
|
|
// and locks it again before returning.
|
|
func TstRunWithManagerUnlocked(t *testing.T, mgr *waddrmgr.Manager, callback func()) {
|
|
if err := mgr.Unlock(privPassphrase); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer mgr.Lock()
|
|
callback()
|
|
}
|
|
|
|
// TstCheckWithdrawalStatusMatches compares s1 and s2 using reflect.DeepEqual
|
|
// and calls t.Fatal() if they're not identical.
|
|
func TstCheckWithdrawalStatusMatches(t *testing.T, s1, s2 WithdrawalStatus) {
|
|
if s1.Fees() != s2.Fees() {
|
|
t.Fatalf("Wrong amount of network fees; want %d, got %d", s1.Fees(), s2.Fees())
|
|
}
|
|
|
|
if !reflect.DeepEqual(s1.Sigs(), s2.Sigs()) {
|
|
t.Fatalf("Wrong tx signatures; got %x, want %x", s1.Sigs(), s2.Sigs())
|
|
}
|
|
|
|
if !reflect.DeepEqual(s1.NextInputAddr(), s2.NextInputAddr()) {
|
|
t.Fatalf("Wrong NextInputAddr; got %v, want %v", s1.NextInputAddr(), s2.NextInputAddr())
|
|
}
|
|
|
|
if !reflect.DeepEqual(s1.NextChangeAddr(), s2.NextChangeAddr()) {
|
|
t.Fatalf("Wrong NextChangeAddr; got %v, want %v", s1.NextChangeAddr(), s2.NextChangeAddr())
|
|
}
|
|
|
|
if !reflect.DeepEqual(s1.Outputs(), s2.Outputs()) {
|
|
t.Fatalf("Wrong WithdrawalOutputs; got %v, want %v", s1.Outputs(), s2.Outputs())
|
|
}
|
|
|
|
if !reflect.DeepEqual(s1.transactions, s2.transactions) {
|
|
t.Fatalf("Wrong transactions; got %v, want %v", s1.transactions, s2.transactions)
|
|
}
|
|
|
|
// The above checks could be replaced by this one, but when they fail the
|
|
// failure msg wouldn't give us much clue as to what is not equal, so we do
|
|
// the individual checks above and use this one as a catch-all check in case
|
|
// we forget to check any of the individual fields.
|
|
if !reflect.DeepEqual(s1, s2) {
|
|
t.Fatalf("Wrong WithdrawalStatus; got %v, want %v", s1, s2)
|
|
}
|
|
}
|