Cover non-default primary key

Author: nsauk

lib/active_graph/node/query/query_proxy_methods.rb

@@ -272,7 +272,7 @@ def first_and_last(func, target)
 
         # @return [Array] An array of primary key values
         def ids
-          pluck(association_id_key)
+          query.pluck(identity => association_id_key)
         end
 
         # @return [String] The primary key of a the current QueryProxy's model or target class

spec/e2e/query_proxy_methods_spec.rb

@@ -470,6 +470,34 @@ def destroy_called
     it 'returns an empty array when there are no matches' do
       expect(Lesson.where(name: 'nope').ids).to eq([])
     end
+
+    context 'when the association target uses a custom id_property' do
+      before(:each) do
+        stub_node_class('Book') do
+          id_property :book_id, on: :generate_book_id
+          property :title
+          has_many :in, :owners, model_class: 'Reader', origin: :books
+          def generate_book_id
+            "book-#{SecureRandom.hex(4)}"
+          end
+        end
+
+        stub_node_class('Reader') do
+          property :name
+          has_many :out, :books, model_class: 'Book', type: 'OWNS'
+        end
+
+        @reader = Reader.create(name: 'Ada')
+        @b1 = Book.create(title: 'A')
+        @b2 = Book.create(title: 'B')
+        @reader.books << @b1
+        @reader.books << @b2
+      end
+
+      it 'uses the target class primary key when traversing an association' do
+        expect(@reader.books.ids).to match_array([@b1.book_id, @b2.book_id])
+      end
+    end
   end
 
   describe 'distinct' do