lib/dread/graph.rb in dread-0.0.4 vs lib/dread/graph.rb in dread-0.0.5

@@ -10,19 +10,20 @@
       @pluralized = pluralized
     end
 
     # { user: { tweets: { comments: {} }, comments: {}, account_setting: {} } }
     def dependable_collection
-      @dependable_collection ||= Hash.new.tap do |relation_hash|
+      @dependable_collection ||= { relation_name.to_sym => Hash.new.tap do |relation_hash|
         @clazz.reflections.each do |assoc_name, assoc_data|
           if assoc_data.options[:dependent] == :delete
             relation_hash[assoc_name] = {}
           elsif assoc_data.options[:dependent] == :destroy
-            relation_hash[assoc_name] =
-              Graph.new(assoc_data, assoc_data.macro == :has_many).dependable_collection
+            relation_hash.merge!(
+              Graph.new(assoc_data, assoc_data.macro == :has_many).dependable_collection)
           end
         end
       end
+      }
     end
 
     def draw(output='console_output')
       case output
       when 'console_output'
\ No newline at end of file