mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-05-22 23:07:04 +00:00
25 lines
623 B
Go
25 lines
623 B
Go
package testing
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/kaspanet/kaspad/app/protocol/protocolerrors"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
func checkFlowError(t *testing.T, err error, isProtocolError bool, shouldBan bool, contains string) {
|
|
pErr := protocolerrors.ProtocolError{}
|
|
if errors.As(err, &pErr) != isProtocolError {
|
|
t.Fatalf("Unexepcted error %+v", err)
|
|
}
|
|
|
|
if pErr.ShouldBan != shouldBan {
|
|
t.Fatalf("Exepcted shouldBan %t but got %t", shouldBan, pErr.ShouldBan)
|
|
}
|
|
|
|
if !strings.Contains(err.Error(), contains) {
|
|
t.Fatalf("Unexpected error. Expected error to contain '%s' but got: %+v", contains, err)
|
|
}
|
|
}
|