associations_keys should inherit, fixes #763

neo4jrb · Apr 20, 2015 · 8e0153e · 8e0153e
1 parent ab2c261
commit 8e0153e
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/lib/neo4j/active_node/has_n.rb b/lib/neo4j/active_node/has_n.rb
@@ -96,12 +96,13 @@ def associations
       end
 
       def associations_keys
-        @associations_keys ||= []
+        @associations_keys ||= associations.keys
       end
 
       # make sure the inherited classes inherit the <tt>_decl_rels</tt> hash
       def inherited(klass)
         klass.instance_variable_set(:@associations, associations.clone)
+        @associations_keys = klass.associations_keys.clone
         super
       end
 

diff --git a/spec/e2e/inheritance_spec.rb b/spec/e2e/inheritance_spec.rb
@@ -16,6 +16,7 @@
       property :specs # Hash
       index :name
       serialize :specs
+      has_many :out, :models, model_class: false
     end
 
     stub_named_class('Car', Vehicle) do
@@ -71,4 +72,12 @@
       expect(toyota.specs.class).to eq Hash
     end
   end
+
+  describe 'associations' do
+    it 'are inherited' do
+      expect(Car.new).to respond_to(:models)
+      expect(Vehicle.associations_keys).to include(:models)
+      expect(Car.associations_keys).to include(:models)
+    end
+  end
 end