Add reset method to clean kafka memory between runs

catawiki · Mar 7, 2019 · cea79cc · cea79cc
1 parent ab4f07b
commit cea79cc
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/lib/fake/kafka.rb b/lib/fake/kafka.rb
@@ -33,5 +33,11 @@ def consumer(*options)
     def producer(*)
       Producer.new(self)
     end
+
+    # Used to clean test between runs
+    def reset
+      @messages = []
+      @paused_partitions = {}
+    end
   end
 end