From 492417dd55f6aaf0436c316634b66bf4882420f8 Mon Sep 17 00:00:00 2001
From: Jimmy Hu <jimmy.hu@dexon.org>
Date: Mon, 7 Jan 2019 12:04:43 +0800
Subject: simulation: Fix simulation stuffs (#405)

* test: fix marshal randomness pullrequest

* Add result for witness latency
---
 core/test/network.go | 8 ++++++++
 1 file changed, 8 insertions(+)

(limited to 'core/test')

diff --git a/core/test/network.go b/core/test/network.go
index 066d36c..49342f0 100644
--- a/core/test/network.go
+++ b/core/test/network.go
@@ -75,6 +75,8 @@ func (req *PullRequest) MarshalJSON() (b []byte, err error) {
 		idAsBytes, err = json.Marshal(req.Identity.(common.Hashes))
 	case "vote":
 		idAsBytes, err = json.Marshal(req.Identity.(types.Position))
+	case "randomness":
+		idAsBytes, err = json.Marshal(req.Identity.(common.Hashes))
 	default:
 		err = fmt.Errorf("unknown ID type for pull request: %v", req.Type)
 	}
@@ -113,6 +115,12 @@ func (req *PullRequest) UnmarshalJSON(data []byte) (err error) {
 			break
 		}
 		ID = pos
+	case "randomness":
+		hashes := common.Hashes{}
+		if err = json.Unmarshal(rawReq.Identity, &hashes); err != nil {
+			break
+		}
+		ID = hashes
 	default:
 		err = fmt.Errorf("unknown pull request type: %v", rawReq.Type)
 	}
-- 
cgit v1.2.3