features/step_definitions/cvg_steps.rb in cvg-0.1.0 vs features/step_definitions/cvg_steps.rb in cvg-0.1.1

@@ -1,5 +1,18 @@
+def cvgme(args, input_csv_paths)
+  bin_path = File.expand_path '../../../bin/cvg', __FILE__
+  cmd = "#{bin_path} #{args.join(' ')} #{input_csv_paths.join(' ')}"
+  child = POSIX::Spawn::Child.new cmd
+  if child.err.present?
+    $stderr.puts
+    $stderr.puts cmd
+    $stderr.puts child.err
+    $stderr.puts
+  end
+  child.out.strip
+end
+
 Before do
   @input_csv_paths = []
   @args = []
 end
 
@@ -12,19 +25,16 @@
 When(/^you pass arguments (.+)$/) do |args|
   @args << args.strip
 end
 
 Then(/^you get output$/) do |expected_output_csv|
-  bin_path = File.expand_path '../../../bin/cvg', __FILE__
-  cmd = "#{bin_path} #{@args.join(' ')} #{@input_csv_paths.join(' ')}"
-  child = POSIX::Spawn::Child.new cmd
-  if child.err.present?
-    $stderr.puts
-    $stderr.puts cmd
-    $stderr.puts child.err
-    $stderr.puts
-  end
-  expect(child.out.strip).to eq(expected_output_csv.strip)
+  got_csv = cvgme(@args, @input_csv_paths)
+  expect(got_csv).to eq(expected_output_csv.strip)
+
+  expected_count = CSV.parse(expected_output_csv.strip, headers: :first_row).length
+  got_count = cvgme((@args + ['--count']), @input_csv_paths)
+  expect(got_count).to match(/Count: #{expected_count}/)
+
   @input_csv_paths.each do |path|
     if File.dirname(File.expand_path(path)).start_with?(Dir.tmpdir)
       File.unlink path
     end
   end