association_collection.rb in essay-1.1.0

- old
+ new

@@ -3,11 +3,11 @@
 
 require 'essay/abstract_collection'
 
 module Essay
   class AssociationCollection < AbstractCollection
-    def construct_features(assoc_name)
-      Essay::AssociationFeatures.new(@env.merge(association_name: assoc_name))
+    def construct_features(name)
+      Essay::AssociationFeatures.new(env.merge(association: traits.associations.fetch(name)))
     end
 
     protected :construct_features
   end
 end