adjust for requestrand msg

irisnet · Sep 3, 2020 · d1af2d8 · d1af2d8
1 parent c1a48f1
commit d1af2d8
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/msg/random/rand.go b/msg/random/rand.go
@@ -1,13 +1,16 @@
 package random
 
 import (
-	types "github.com/irisnet/irishub-sync/types"
+	"github.com/irisnet/irishub-sync/types"
 	"github.com/irisnet/irishub-sync/util/constant"
+	"github.com/irisnet/irishub-sync/store"
 )
 
 type DocTxMsgRequestRand struct {
-	Consumer      string `bson:"consumer"`       // request address
-	BlockInterval uint64 `bson:"block_interval"` // block interval after which the requested random number will be generated
+	Consumer      string      `bson:"consumer"`       // request address
+	BlockInterval uint64      `bson:"block_interval"` // block interval after which the requested random number will be generated
+	Oracle        bool        `bson:"oracle"`
+	ServiceFeeCap store.Coins `bson:"service_fee_cap"`
 }
 
 func (doctx *DocTxMsgRequestRand) Type() string {
@@ -18,4 +21,6 @@ func (doctx *DocTxMsgRequestRand) BuildMsg(txMsg interface{}) {
 	msg := txMsg.(types.MsgRequestRandom)
 	doctx.Consumer = msg.Consumer.String()
 	doctx.BlockInterval = msg.BlockInterval
+	doctx.Oracle = msg.Oracle
+	doctx.ServiceFeeCap = types.ParseCoins(msg.ServiceFeeCap.String())
 }
diff --git a/service/task/task_execute_test.go b/service/task/task_execute_test.go
@@ -108,7 +108,7 @@ func Test_parseBlock(t *testing.T) {
 			name: "test parse block",
 			args: args{
 				client: client,
-				b:      1001,
+				b:      601,
 			},
 		},
 	}