diff options
Diffstat (limited to 'core/test')
| -rw-r--r-- | core/test/state-change-request.go | 8 | ||||
| -rw-r--r-- | core/test/state.go | 5 |
2 files changed, 11 insertions, 2 deletions
diff --git a/core/test/state-change-request.go b/core/test/state-change-request.go index 84d4d3f..1515fd2 100644 --- a/core/test/state-change-request.go +++ b/core/test/state-change-request.go @@ -185,7 +185,11 @@ func (req *StateChangeRequest) String() (ret string) { ret += fmt.Sprintf( "{Type:AddNode %s", types.NewNodeID(req.Payload.(crypto.PublicKey)).String()[:6]) + default: + panic(fmt.Errorf( + "attempting to dump unknown type of state change request: %v", + req.Type)) } - panic(fmt.Errorf( - "attempting to dump unknown type of state change request: %v", req.Type)) + ret += "}" + return } diff --git a/core/test/state.go b/core/test/state.go index b360fa2..30ed8af 100644 --- a/core/test/state.go +++ b/core/test/state.go @@ -177,6 +177,11 @@ func (s *State) Snapshot() (*types.Config, []crypto.PublicKey) { }, nodes } +// AttachLogger allows to attach custom logger. +func (s *State) AttachLogger(logger common.Logger) { + s.logger = logger +} + func (s *State) unpackPayload( raw *rawStateChangeRequest) (v interface{}, err error) { switch raw.Type { |