lib/test_queue/iterator.rb in test-queue-split-0.3.0 vs lib/test_queue/iterator.rb in test-queue-split-0.3.1

@@ -35,10 +35,11 @@
           if @filter
             @filter.call(suite){ yield suite }
           else
             yield suite
           end
-          @stats[suite.to_s] = Time.now - start
+          key = suite.respond_to?(:id) ? suite.id : suite.to_s
+          @stats[key] = Time.now - start
         else
           break
         end
       end
     rescue Errno::ENOENT, Errno::ECONNRESET, Errno::ECONNREFUSED