diff options
Diffstat (limited to 'core/test/marshaller.go')
| -rw-r--r-- | core/test/marshaller.go | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/core/test/marshaller.go b/core/test/marshaller.go index fc42639..a1b15b6 100644 --- a/core/test/marshaller.go +++ b/core/test/marshaller.go @@ -95,6 +95,12 @@ func (m *DefaultMarshaller) Unmarshal( break } msg = final + case "packed-state-changes": + packed := &packedStateChanges{} + if err = json.Unmarshal(payload, packed); err != nil { + break + } + msg = *packed default: if m.fallback == nil { err = fmt.Errorf("unknown msg type: %v", msgType) @@ -136,6 +142,9 @@ func (m *DefaultMarshaller) Marshal( case *typesDKG.Finalize: msgType = "dkg-finalize" payload, err = json.Marshal(msg) + case packedStateChanges: + msgType = "packed-state-changes" + payload, err = json.Marshal(msg) default: if m.fallback == nil { err = fmt.Errorf("unknwon message type: %v", msg) |