[LoadTest] Add missing block query client to load test (#642)

pokt-network · Jun 30, 2024 · 9de7577 · 9de7577
1 parent 0ed3518
commit 9de7577
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/load-testing/tests/relays_stress_helpers_test.go b/load-testing/tests/relays_stress_helpers_test.go
@@ -1361,10 +1361,15 @@ func (s *relaysSuite) countClaimAndProofs() {
 
 // querySharedParams queries the current on-chain shared module parameters for use
 // over the duration of the test.
-func (s *relaysSuite) querySharedParams() {
+func (s *relaysSuite) querySharedParams(queryNodeRPCURL string) {
 	s.Helper()
 
 	deps := depinject.Supply(s.txContext.GetClientCtx())
+
+	blockQueryClient, err := sdkclient.NewClientFromNode(queryNodeRPCURL)
+	require.NoError(s, err)
+	deps = depinject.Configs(deps, depinject.Supply(blockQueryClient))
+
 	sharedQueryClient, err := query.NewSharedQuerier(deps)
 	require.NoError(s, err)
 

diff --git a/load-testing/tests/relays_stress_test.go b/load-testing/tests/relays_stress_test.go
@@ -358,7 +358,7 @@ func (s *relaysSuite) LocalnetIsRunning() {
 	s.countClaimAndProofs()
 
 	// Query for the current shared on-chain params.
-	s.querySharedParams()
+	s.querySharedParams(loadTestParams.TestNetNode)
 
 	// Some suppliers may already be staked at genesis, ensure that staking during
 	// this test succeeds by increasing the sake amount.