wait for number of blocks instead of seconds and check successful votes

gov/vote.go

@@ -30,10 +30,16 @@ func SubmitAllVotesForProposal(bin *utils.Binary, proposalID int) error {
 		return errors.New("no accounts with delegations found")
 	}
 
-	utils.Wait(1)
+	if err := utils.WaitNBlocks(bin, 1); err != nil {
+		return err
+	}
+
 	bin.Logger.Info().Msgf("voting for proposal %d", proposalID)
 
-	var out string
+	var (
+		out             string
+		successfulVotes int
+	)
 
 	for _, acc := range accsWithDelegations {
 		out, err = VoteForProposal(bin, proposalID, acc.Name)
@@ -49,9 +55,14 @@ func SubmitAllVotesForProposal(bin *utils.Binary, proposalID int) error {
 			bin.Logger.Error().Msgf("could not vote using key %s: %v", acc.Name, err)
 		} else {
 			bin.Logger.Info().Msgf("voted using key %s", acc.Name)
+			successfulVotes += 1
 		}
 	}
 
+	if successfulVotes == 0 {
+		return errors.New("there were no successful votes for the proposal, please check logs")
+	}
+
 	return nil
 }
 

utils/utils.go

@@ -187,7 +187,23 @@ func GetTxHashFromTxResponse(cdc *codec.ProtoCodec, out string) (string, error)
 	return txHash.TxHash, nil
 }
 
-// Wait waits for the specified amount of seconds.
-func Wait(seconds int) {
-	time.Sleep(time.Duration(seconds) * time.Second)
+// WaitNBlocks waits for the specified amount of blocks being produced
+// on the connected network.
+func WaitNBlocks(bin *Binary, n int) error {
+	currentBlockRes, err := ExecuteQuery(bin, QueryArgs{
+		Subcommand: []string{"q", "block"},
+	})
+	if err != nil {
+		return err
+	}
+
+	fmt.Printf("block res: \n%s\n", currentBlockRes)
+	panic("")
+
+	for {
+		bin.Logger.Debug().Msgf("waiting for %d blocks\n", n)
+		break
+	}
+
+	return nil
 }