Commit 340b0718 authored by mdj33's avatar mdj33 Committed by vipwzw

fix ut

parent eaf1c68c
...@@ -250,7 +250,7 @@ func (client *client) SetQueueClient(c queue.Client) { ...@@ -250,7 +250,7 @@ func (client *client) SetQueueClient(c queue.Client) {
}) })
go client.EventLoop() go client.EventLoop()
client.wg.Add(1) client.wg.Add(1)
go client.commitMsgClient.process() go client.commitMsgClient.handler()
client.wg.Add(1) client.wg.Add(1)
go client.CreateBlock() go client.CreateBlock()
client.wg.Add(1) client.wg.Add(1)
...@@ -268,6 +268,11 @@ func (client *client) InitBlock() { ...@@ -268,6 +268,11 @@ func (client *client) InitBlock() {
} }
client.grpcClient = grpcCli client.grpcClient = grpcCli
err = client.commitMsgClient.setSelfConsEnable()
if err != nil {
panic(err)
}
block, err := client.RequestLastBlock() block, err := client.RequestLastBlock()
if err != nil { if err != nil {
panic(err) panic(err)
......
...@@ -930,15 +930,3 @@ func (client *commitMsgClient) isSelfConsEnable(height int64) bool { ...@@ -930,15 +930,3 @@ func (client *commitMsgClient) isSelfConsEnable(height int64) bool {
} }
return false return false
} }
func (client *commitMsgClient) initConfig() {
err := client.setSelfConsEnable()
if err != nil {
panic(err)
}
}
func (client *commitMsgClient) process() {
client.initConfig()
client.handler()
}
...@@ -469,32 +469,32 @@ type VoteTestSuite struct { ...@@ -469,32 +469,32 @@ type VoteTestSuite struct {
exec *Paracross exec *Paracross
} }
func (suite *VoteTestSuite) SetupSuite() { func (s *VoteTestSuite) SetupSuite() {
//para_init(Title) //para_init(Title)
suite.exec = newParacross().(*Paracross) s.exec = newParacross().(*Paracross)
api := new(apimock.QueueProtocolAPI) api := new(apimock.QueueProtocolAPI)
api.On("GetConfig", mock.Anything).Return(chain33TestCfg, nil) api.On("GetConfig", mock.Anything).Return(chain33TestCfg, nil)
suite.exec.SetAPI(api) s.exec.SetAPI(api)
suite.stateDB, _ = dbm.NewGoMemDB("state", "state", 1024) s.stateDB, _ = dbm.NewGoMemDB("state", "state", 1024)
// memdb 不支持KVDB接口, 等测试完Exec , 再扩展 memdb // memdb 不支持KVDB接口, 等测试完Exec , 再扩展 memdb
//suite.localDB, _ = dbm.NewGoMemDB("local", "local", 1024) //s.localDB, _ = dbm.NewGoMemDB("local", "local", 1024)
suite.localDB = new(dbmock.KVDB) s.localDB = new(dbmock.KVDB)
suite.exec.SetLocalDB(suite.localDB) s.exec.SetLocalDB(s.localDB)
suite.exec.SetStateDB(suite.stateDB) s.exec.SetStateDB(s.stateDB)
suite.exec.SetEnv(0, 0, 0) s.exec.SetEnv(0, 0, 0)
stageKey := calcParaSelfConsStagesKey() stageKey := calcParaSelfConsStagesKey()
stage := &pt.SelfConsensStage{BlockHeight: 0, Enable: pt.ParaConfigYes} stage := &pt.SelfConsensStage{BlockHeight: 0, Enable: pt.ParaConfigYes}
stages := &pt.SelfConsensStages{Items: []*pt.SelfConsensStage{stage}} stages := &pt.SelfConsensStages{Items: []*pt.SelfConsensStage{stage}}
suite.stateDB.Set(stageKey, types.Encode(stages)) s.stateDB.Set(stageKey, types.Encode(stages))
value, err := suite.stateDB.Get(stageKey) value, err := s.stateDB.Get(stageKey)
if err != nil { if err != nil {
suite.T().Error("get setup stages failed", err) s.T().Error("get setup stages failed", err)
return return
} }
assert.Equal(suite.T(), value, types.Encode(stages)) assert.Equal(s.T(), value, types.Encode(stages))
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment