test/minitest_helper.rb in powertrack-1.1.1 vs test/minitest_helper.rb in powertrack-1.2.0

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