1 files changed, 34 insertions, 0 deletions
diff --git a/spec/support/helpers/redis_commands/recorder.rb b/spec/support/helpers/redis_commands/recorder.rb
new file mode 100644
index 00000000000..05a1aa67853
--- /dev/null
+++ b/spec/support/helpers/redis_commands/recorder.rb
@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+module RedisCommands
+  class Recorder
+    def initialize(pattern: nil, &block)
+      @log = []
+      @pattern = pattern
+
+      record(&block) if block
+    end
+
+    attr_reader :log
+
+    def record(&block)
+      ActiveSupport::Notifications.subscribed(method(:callback), 'redis.process_commands', &block)
+    end
+
+    def by_command(command)
+      @log.select { |record| record.include?(command) }
+    end
+
+    def count
+      @count ||= @log.count
+    end
+
+    private
+
+    def callback(name, start, finish, message_id, values)
+      commands = values[:commands]
+
+      @log << commands.flatten if @pattern.nil? || commands.to_s.include?(@pattern)
+    end
+  end
+end