reflection.rb in activefedora-aggregation-0.3.0

- old
+ new

@@ -1,8 +1,25 @@
 module ActiveFedora::Aggregation
   class Reflection < ActiveFedora::Reflection::AssociationReflection
     def association_class
-      ThroughAssociation
+      Association
+    end
+
+    def klass
+      @klass ||= begin
+        klass = class_name.constantize
+        klass.respond_to?(:uri_to_id) ? klass : ActiveFedora::Base
+      rescue NameError => e
+        # If the NameError is a result of the class having a
+        # NameError (e.g. NoMethodError) within it then raise the error.
+        raise e if Object.const_defined? class_name
+        # Otherwise the NameError was a result of not being able to find the class
+        ActiveFedora::Base
+      end
+    end
+
+    def predicate
+      @options[:predicate] || ::RDF::Vocab::ORE.aggregates
     end
 
     def collection?
       true
     end