Merge pull request datamapper#21 from jwkoelewijn/relationships

Applied dkubbs patch from http://pastie.org/1625163 to fix relationships

lib/dm-core/model.rb

@@ -194,10 +194,12 @@ def self.extra_extensions
     def self.extended(descendant)
       descendants << descendant
 
-      descendant.instance_variable_set(:@valid,         false)
-      descendant.instance_variable_set(:@base_model,    descendant)
-      descendant.instance_variable_set(:@storage_names, {})
-      descendant.instance_variable_set(:@default_order, {})
+      descendant.instance_eval do
+        @valid           = false unless instance_variable_defined?(:@valid)
+        @base_model    ||= descendant
+        @storage_names ||= {}
+        @default_order ||= {}
+      end
 
       descendant.extend(Chainable)
 
@@ -210,6 +212,11 @@ def self.extended(descendant)
       def inherited(descendant)
         descendants << descendant
 
+        descendant.instance_eval do
+          @valid        = false unless instance_variable_defined?(:@valid)
+          @base_model ||= base_model
+        end
+
         descendant.instance_variable_set(:@valid,         false)
         descendant.instance_variable_set(:@base_model,    base_model)
         descendant.instance_variable_set(:@storage_names, @storage_names.dup)

lib/dm-core/model/property.rb

@@ -9,13 +9,18 @@ module Property
       extend Chainable
 
       def self.extended(model)
-        model.instance_variable_set(:@properties,               {})
-        model.instance_variable_set(:@field_naming_conventions, {})
+        model.instance_eval do
+          @properties               ||= {}
+          @field_naming_conventions ||= {}
+        end
       end
 
       chainable do
         def inherited(model)
-          model.instance_variable_set(:@properties,               {})
+          model.instance_eval do
+            @properties ||= {}
+          end
+
           model.instance_variable_set(:@field_naming_conventions, @field_naming_conventions.dup)
 
           @properties.each do |repository_name, properties|

lib/dm-core/model/relationship.rb

@@ -18,7 +18,9 @@ module Relationship
       #
       # @api private
       def self.extended(model)
-        model.instance_variable_set(:@relationships, {})
+        model.instance_eval do
+          @relationships ||= {}
+        end
       end
 
       chainable do
@@ -27,7 +29,9 @@ def self.extended(model)
         #
         # @api private
         def inherited(model)
-          model.instance_variable_set(:@relationships, {})
+          model.instance_eval do
+            @relationships ||= {}
+          end
 
           @relationships.each do |repository_name, relationships|
             model_relationships = model.relationships(repository_name)

spec/public/model/relationship_spec.rb

@@ -1038,3 +1038,64 @@ class ::Company
     end
   end
 end
+
+describe DataMapper::Associations do
+  before :all do
+    module ::Vehicle
+      def self.included(base)
+        base.extend(ClassMethods)
+      end
+
+      module ClassMethods
+
+        def vehicle_with_wheels(*args)
+          include DataMapper::Resource
+
+          property :id, DataMapper::Property::Serial
+          property :license_plate, String
+        end
+      end
+    end
+
+    class Exhaust
+      include DataMapper::Resource
+
+      property :id, Serial
+      belongs_to :car
+    end
+
+    class Suspension
+      include DataMapper::Resource
+
+      property :id, Serial
+      property :brand, String
+    end
+
+    class Car
+      include DataMapper::Resource
+      has n, :suspensions
+    end
+
+    class Car
+      include DataMapper::Resource
+      include Vehicle
+
+      vehicle_with_wheels
+
+      property :number_of_seats, Integer
+      has n, :exhausts
+
+    end
+
+    DataMapper.finalize
+  end
+
+  it "should have all relationships when Resource is included once" do
+    Exhaust.relationships.any? {|r| r.name == :car }.should be_true
+  end
+
+  it "should not forget about relationships when Resource is included more than once" do
+    Car.relationships.any? {|r| r.name == :exhausts }.should be_true
+    Car.relationships.any? {|r| r.name == :suspensions }.should be_true
+  end
+end