Account for stake held by the current node while waiting for the supe…

…rmajority to join gossip
solana-labs · Jan 8, 2020 · 43897de · 43897de
1 parent 4b577aa
commit 43897de
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/core/src/validator.rs b/core/src/validator.rs
@@ -299,7 +299,7 @@ impl Validator {
 
         if config.wait_for_supermajority {
             info!(
-                "Waiting more than 66% of activated stake at slot {} to be in gossip...",
+                "Waiting for more than 66% of activated stake at slot {} to be in gossip...",
                 bank.slot()
             );
             loop {
@@ -630,6 +630,8 @@ fn get_stake_percent_in_gossip(
                 activated_stake
             );
             gossip_stake += activated_stake;
+        } else if vote_state.node_pubkey == cluster_info.read().unwrap().id() {
+            gossip_stake += activated_stake;
         }
     }
 

diff --git a/local-cluster/tests/local_cluster.rs b/local-cluster/tests/local_cluster.rs
@@ -159,6 +159,8 @@ fn test_validator_exit_2() {
     let num_nodes = 2;
     let mut validator_config = ValidatorConfig::default();
     validator_config.rpc_config.enable_validator_exit = true;
+    validator_config.wait_for_supermajority = true;
+
     let config = ClusterConfig {
         cluster_lamports: 10_000,
         node_stakes: vec![100; num_nodes],