lib/cucumber/blanket/coverage_data.rb in cucumber-blanket-0.0.2 vs lib/cucumber/blanket/coverage_data.rb in cucumber-blanket-0.0.3

@@ -6,48 +6,49 @@
     # the structure against multiple runs of fresh data
     class CoverageData
       attr_reader :data
 
       def initialize
-        @data = nil
+        @data = {'files'=>{}}
       end
 
       def method_missing *args
-        @data[0].send(*args)
+        @data.send(*args)
       end
 
+      def files
+        self.data['files']
+      end
+
       def accrue! page_data
-        # go through every line of every file and OR it all together
-        # e.g. line 1 is 1 and 0, so it is 1
-        #binding.pry
         if @data.nil?
           @data = page_data
         else
           # for files in page_data ...
-          page_data[0]['files'].each do |filename, linedata|
+          page_data['files'].each do |filename, linedata|
             # that exist in @data
-            if @data[0]['files'].has_key? page_data[0]['files'].first[0]
+            if @data['files'][filename]
               # accrue coverage data, meaning:
               # get a handle on existing linedata and iterate
-              @data[0]['files'][filename].each_with_index do |cov_stat, line_no|
-                new_cov_stat = page_data[0]['files'][filename][line_no]
+              @data['files'][filename].each_with_index do |cov_stat, line_no|
+                new_cov_stat = page_data['files'][filename][line_no]
                 # first we need to deal with nils, as we cannot add them
                 # either side can be nil -- and we want to be strictly additive
                 next if new_cov_stat.nil? # this is not additive, next line
                 # So now we know the new data is definitely not nil
                 # but the existing data could be, so we'll handle that now
                 if cov_stat.nil?
-                  @data[0]['files'][filename][line_no] = new_cov_stat
+                  @data['files'][filename][line_no] = new_cov_stat
                   # We replaced it with the new data, next line please
                   next
                 end
                 # if we ever get here, we're dealing strictly with integers
                 # as a result we just need to sum the two stats
-                @data[0]['files'][filename][line_no] = cov_stat + new_cov_stat
+                @data['files'][filename][line_no] = cov_stat + new_cov_stat
               end
             else # if it does not exist
               # add it to 'files' as is
-              @data[0]['files'][filename] = linedata
+              @data['files'][filename] = linedata
             end
           end
         end
       end
     end