Test cleanup

This commit is contained in:
Sean Braithwaite
2019-07-19 11:18:21 +03:00
parent 1828f860f7
commit a790f14340

View File

@ -55,15 +55,19 @@ func TestTouchPeer(t *testing.T) {
func TestPeerHeight(t *testing.T) { func TestPeerHeight(t *testing.T) {
sc := newSchedule(initHeight) sc := newSchedule(initHeight)
assert.Nil(t, sc.addPeer(peerID), assert.NoError(t, sc.addPeer(peerID),
"Adding a peer should return no error") "Adding a peer should return no error")
assert.Nil(t, sc.setPeerHeight(peerID, peerHeight)) assert.NoError(t, sc.setPeerHeight(peerID, peerHeight))
for i := initHeight; i <= peerHeight; i++ { for i := initHeight; i <= peerHeight; i++ {
assert.Equal(t, sc.getStateAtHeight(i), blockStateNew, assert.Equal(t, sc.getStateAtHeight(i), blockStateNew,
"Expected all blocks to be in blockStateNew") "Expected all blocks to be in blockStateNew")
assert.Equal(t, len(sc.getPeersAtHeight(i)), 1, peerIDs := []p2p.ID{}
"Expected the block to be registered to the 1 peer") for _, peer := range sc.getPeersAtHeight(i) {
//TODO check peerID peerIDs = append(peerIDs, peer.peerID)
}
assert.Containsf(t, peerIDs, peerID,
"Expected %s to have block %d", peerID, i)
} }
} }