Skip to content

Commit 77a200e

Browse files
committed
test: fix memory profiling settings
1 parent 0ce31d8 commit 77a200e

File tree

1 file changed

+8
-5
lines changed

1 file changed

+8
-5
lines changed

test/prof_mem.rb

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -12,11 +12,13 @@ module ProfMem
1212
SLICED_NUMBERS = Array.new(ATTEMPT_COUNT) { |i| i }.each_slice(MAX_PIPELINE_SIZE).freeze
1313
CLI_TYPES = %w[primary_only scale_read_random scale_read_latency pooled].freeze
1414
MODES = {
15-
single: lambda do |cli|
15+
single: lambda do |client_builder_method|
16+
cli = send(client_builder_method)
1617
ATTEMPT_COUNT.times { |i| cli.call('SET', i, i) }
1718
ATTEMPT_COUNT.times { |i| cli.call('GET', i) }
1819
end,
19-
excessive_pipelining: lambda do |cli|
20+
excessive_pipelining: lambda do |client_builder_method|
21+
cli = send(client_builder_method)
2022
cli.pipelined do |pi|
2123
ATTEMPT_COUNT.times { |i| pi.call('SET', i, i) }
2224
end
@@ -25,7 +27,8 @@ module ProfMem
2527
ATTEMPT_COUNT.times { |i| pi.call('GET', i) }
2628
end
2729
end,
28-
pipelining_in_moderation: lambda do |cli|
30+
pipelining_in_moderation: lambda do |client_builder_method|
31+
cli = send(client_builder_method)
2932
SLICED_NUMBERS.each do |numbers|
3033
cli.pipelined do |pi|
3134
numbers.each { |i| pi.call('SET', i, i) }
@@ -45,8 +48,8 @@ def run
4548
CLI_TYPES.each do |cli_type|
4649
prepare
4750
print_letter(mode, cli_type)
48-
client = send("new_#{cli_type}_client".to_sym)
49-
profile { subject.call(client) }
51+
client_builder_method = "new_#{cli_type}_client".to_sym
52+
profile { subject.call(client_builder_method) }
5053
end
5154
end
5255

0 commit comments

Comments
 (0)