features/steps/junit_steps.rb in xcpretty-0.0.6 vs features/steps/junit_steps.rb in xcpretty-0.0.7

@@ -1,21 +1,33 @@
+Given(/^I have tests in my suite from 2 classes$/) do
+  add_run_input SAMPLE_OCUNIT_TEST
+  add_run_input SAMPLE_KIWI_TEST
+end
+
 Then(/^I should see a failed test node in my report$/) do
-  junit_report.root.elements.to_a.detect do |node|
+  junit_report_root.elements.to_a.detect do |node|
     element = node.elements.to_a.first
     element && element.name == "failure"
   end.should_not be_nil
 end
 
 Then(/^I should see a passing test node in my report$/) do
-  junit_report.root.elements.to_a.detect do |node|
+  junit_report_root.elements.to_a.detect do |node|
     node.attributes["time"] != nil
   end.should_not be_nil
 end
 
 Then(/^I should see a test suite node$/) do
-  junit_report.root.should_not be_nil
+  junit_report_root.elements.to_a.first.should_not be_nil
 end
 
 Then(/^I should see (\d+) tests in my report$/) do |test_count|
-  junit_report.root.attributes["tests"].should == test_count
-  junit_report.root.elements.to_a.count.should == test_count.to_i
-end
\ No newline at end of file
+  junit_report_root.attributes["tests"].should == test_count
+  junit_report_root.elements.to_a.count.should == test_count.to_i
+end
+
+Then(/^I should see (\d+) test suites$/) do |count|
+  suites = junit_report.root.elements.to_a
+  suites.size.should == count.to_i
+  suites.select {|s| s.name == 'testsuite' }.size.should == count.to_i
+end
+