Ticket #2917: habtm-singular-id-get-method.patch

activerecord/test/associations_test.rb

@@ -1372 +1372 @@
-    assert !developer.special_projects.include?(other_project)
-  end
-
+  def test_returns_collection_of_ids
+    francois = Developer.new("name" => "Francois")
+    francois.projects.concat([Project.find(1), Project.find(2)])
+    francois.save
+
+    assert_equal 2, francois.project_ids.size
+    assert francois.project_ids.include?(Project.find(1).id)
+    assert francois.project_ids.include?(Project.find(2).id)
+  end
+
+  def test_assign_through_collection_of_ids
+    francois = Developer.new("name" => "Francois")
+    francois.project_ids = [Project.find(1).id, Project.find(2).id]
+    francois.save
+    francois.reload
+
+    assert_equal 2, francois.projects.size
+    assert francois.projects.include?(Project.find(1))
+    assert francois.projects.include?(Project.find(2))
+  end
-end

activerecord/lib/active_record/associations.rb

@@ -753 +753 @@
-            association
-          end
-
+          define_method("#{Inflector.singularize(association_name)}_ids") do
+            send("#{association_name}").map {|entry| entry.id}
+          end
+
-          define_method("#{Inflector.singularize(association_name)}_ids=") do |new_value|
-            send("#{association_name}=", association_class_name.constantize.find(new_value))
-          end