From 29737ab5e4ff5044d5d060aac913d17b6b164a46 Mon Sep 17 00:00:00 2001
From: Andrew Fitzgerald <apfitzge@gmail.com>
Date: Thu, 25 Jan 2024 10:22:27 -0800
Subject: [PATCH] Use ThreadLocalMultiIterator for tests (#34947)

* Use ThreadLocalMultiIterator for tests

* some validator config was not using default_for_test
---
 core/src/validator.rs                | 1 +
 local-cluster/tests/local_cluster.rs | 4 ++--
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/core/src/validator.rs b/core/src/validator.rs
index 23045aa35..f624dae1e 100644
--- a/core/src/validator.rs
+++ b/core/src/validator.rs
@@ -343,6 +343,7 @@ impl ValidatorConfig {
         Self {
             enforce_ulimit_nofile: false,
             rpc_config: JsonRpcConfig::default_for_test(),
+            block_production_method: BlockProductionMethod::ThreadLocalMultiIterator,
             ..Self::default()
         }
     }
diff --git a/local-cluster/tests/local_cluster.rs b/local-cluster/tests/local_cluster.rs
index 5a4ca6e4c..b79a1c4e3 100644
--- a/local-cluster/tests/local_cluster.rs
+++ b/local-cluster/tests/local_cluster.rs
@@ -2268,7 +2268,7 @@ fn test_hard_fork_with_gap_in_roots() {
 
     let validator_config = ValidatorConfig {
         snapshot_config: LocalCluster::create_dummy_load_only_snapshot_config(),
-        ..ValidatorConfig::default()
+        ..ValidatorConfig::default_for_test()
     };
     let mut config = ClusterConfig {
         cluster_lamports: 100_000,
@@ -5236,7 +5236,7 @@ fn test_duplicate_shreds_switch_failure() {
                 validator_keypair,
                 validator_config: ValidatorConfig {
                     voting_disabled,
-                    ..ValidatorConfig::default()
+                    ..ValidatorConfig::default_for_test()
                 },
                 in_genesis,
             }