test/minitest_helper.rb in powertrack-1.0.3 vs test/minitest_helper.rb in powertrack-1.1.0

@@ -28,14 +28,15 @@
 
     @pwtk_config
   end
 
   # Returns a brand-new stream based on the config found in test/powertrack.yml.
-  def new_stream
+  def new_stream(replay=false)
     PowerTrack::Stream.new(
       powertrack_config[:username],
       powertrack_config[:password],
       powertrack_config[:account_name],
       powertrack_config[:data_source],
-      powertrack_config[:stream_label])
+      replay ? 'prod' : powertrack_config[:stream_label],
+      replay: replay)
   end
 end