tenacity 0.2.0 → 0.3.0

data/test/associations/belongs_to_test.rb

@@ -2,133 +2,42 @@ require 'test_helper'
 
 class BelongsToTest < Test::Unit::TestCase
 
-  context "A class with a belongs_to association to another class" do
-    setup do
-      setup_fixtures
-      @car = ActiveRecordCar.create
-      @wheel = MongoMapperWheel.create(:active_record_car => @car)
-    end
-
-    should "memoize the association" do
-      assert_equal @car, @wheel.active_record_car
-
-      other_car = ActiveRecordCar.create
-      assert_equal @car, MongoMapperWheel.find(@wheel.id).active_record_car
-      MongoMapperWheel.update(@wheel.id, :active_record_car => other_car)
-      assert_equal other_car, MongoMapperWheel.find(@wheel.id).active_record_car
-
-      assert_equal @car, @wheel.active_record_car
-      assert_equal other_car, @wheel.active_record_car(true)
-    end
-
-    should "be able to specify the class name of the associated class" do
-      dashboard = MongoMapperDashboard.create
-      ash_tray = MongoMapperAshTray.create(:dashboard => dashboard)
-      assert_equal dashboard, ash_tray.dashboard
-    end
-
-    should "be able to specify the foreign key to use for the associated class" do
-      car = ActiveRecordCar.create
-      windshield = CouchRestWindshield.create(:active_record_car => car)
-      assert_equal car.id.to_s, windshield.car_id
-      assert !windshield.respond_to?(:active_record_car_id)
-
-      engine = ActiveRecordEngine.create(:active_record_car => car)
-      assert_equal car.id, engine.car_id
-      assert !engine.respond_to?(:active_record_car_id)
-    end
-  end
-
-  context "A MongoMapper class with belongs_to association to an ActiveRecord class" do
-    setup do
-      setup_fixtures
-      @car = ActiveRecordCar.create
-      @wheel = MongoMapperWheel.create
-    end
-
-    should "be able to fetch the id of the associated object" do
-      @wheel.active_record_car_id = @car.id
-      @wheel.save
-      assert_equal @car.id, MongoMapperWheel.find(@wheel.id).active_record_car_id.to_i
-    end
-
-    should "be able to load the associated object" do
-      @wheel.active_record_car = @car
-      @wheel.save
-      assert_equal @car.id, MongoMapperWheel.find(@wheel.id).active_record_car_id.to_i
-      assert_equal @car, MongoMapperWheel.find(@wheel.id).active_record_car
-    end
-
-    should "be able to load the associated object if all we have is the id" do
-      @wheel.active_record_car_id = @car.id
-      @wheel.save
-      assert_equal @car, MongoMapperWheel.find(@wheel.id).active_record_car
-    end
-
-    should "return nil if no association is set" do
-      assert_nil MongoMapperWheel.find(@wheel.id).active_record_car
-    end
-  end
-
-  context "An ActiveRecord class with belongs_to association to a MongoMapper class" do
-    setup do
-      setup_fixtures
-      @wheel = MongoMapperWheel.create
-      @nut = ActiveRecordNut.create
-    end
-
-    should "be able to fetch the id of the associated object" do
-      @nut.mongo_mapper_wheel_id = @wheel.id
-      @nut.save
-      assert_equal @wheel.id.to_s, ActiveRecordNut.find(@nut.id).mongo_mapper_wheel_id
-    end
-
-    should "be able to load the associated object" do
-      @nut.mongo_mapper_wheel = @wheel
-      @nut.save
-      assert_equal @wheel.id.to_s, ActiveRecordNut.find(@nut.id).mongo_mapper_wheel_id
-      assert_equal @wheel, ActiveRecordNut.find(@nut.id).mongo_mapper_wheel
-    end
-
-    should "be be able to load the associated object if all we have is the id" do
-      @nut.mongo_mapper_wheel_id = @wheel.id
-      @nut.save
-      assert_equal @wheel, ActiveRecordNut.find(@nut.id).mongo_mapper_wheel
-    end
-
-    should "return nil if no association is set" do
-      assert_nil ActiveRecordNut.find(@nut.id).mongo_mapper_wheel
-    end
-  end
-
-  context "A CouchRest class with belongs_to association to a MongoMapper class" do
-    setup do
-      setup_all_fixtures
-      @dashboard = MongoMapperDashboard.create
-      @radio = CouchRestRadio.create({})
-    end
-
-    should "be able to fetch the id of the associated object" do
-      @radio.mongo_mapper_dashboard_id = @dashboard.id
-      @radio.save
-      assert_equal @dashboard.id.to_s, CouchRestRadio.find(@radio.id).mongo_mapper_dashboard_id
-    end
-
-    should "be able to load the associated object" do
-      @radio.mongo_mapper_dashboard = @dashboard
-      @radio.save
-      assert_equal @dashboard.id.to_s, CouchRestRadio.find(@radio.id).mongo_mapper_dashboard_id
-      assert_equal @dashboard, CouchRestRadio.find(@radio.id).mongo_mapper_dashboard
-    end
-
-    should "be be able to load the associated object if all we have is the id" do
-      @radio.mongo_mapper_dashboard_id = @dashboard.id
-      @radio.save
-      assert_equal @dashboard, CouchRestRadio.find(@radio.id).mongo_mapper_dashboard
-    end
-
-    should "return nil if no association is set" do
-      assert_nil CouchRestRadio.find(@radio.id).mongo_mapper_dashboard
+  for_each_orm_extension_combination do |source, target|
+    context "A #{source} class with a belongs_to association to a #{target} class" do
+      setup do
+        setup_fixtures_for(source, target)
+
+        @source_class = class_for_extension(source)
+        @source = @source_class.create({})
+        @target_class = class_for_extension(target, :belongs_to)
+        @target = @target_class.create({})
+
+        @foreign_key = foreign_key_for(source, :belongs_to)
+        @foreign_key_id = foreign_key_id_for(source, :belongs_to)
+      end
+
+      should "be able to fetch the id of the associated object" do
+        @target.send("#{@foreign_key_id}=", @source.id.to_s)
+        @target.save
+        assert_equal @source.id.to_s, @target_class._t_find(@target.id.to_s).send(@foreign_key_id)
+      end
+
+      should "be able to load the associated object" do
+        @target.send("#{@foreign_key}=", @source)
+        @target.save
+        assert_equal @source.id.to_s, @target_class._t_find(@target.id.to_s).send(@foreign_key_id)
+        assert_equal @source, @target_class._t_find(@target.id.to_s).send(@foreign_key)
+      end
+
+      should "be be able to load the associated object if all we have is the id" do
+        @target.send("#{@foreign_key_id}=", @source.id.to_s)
+        @target.save
+        assert_equal @source, @target_class._t_find(@target.id.to_s).send(@foreign_key)
+      end
+
+      should "return nil if no association is set" do
+        assert_nil @target_class._t_find(@target.id.to_s).send(@foreign_key)
+      end
     end
   end
 

data/test/associations/has_many_test.rb

@@ -2,203 +2,84 @@ require 'test_helper'
 
 class HasManyTest < Test::Unit::TestCase
 
-  context "A class with a belongs_to association to another class" do
-    setup do
-      setup_fixtures
-      @car = ActiveRecordCar.create
-      @wheels = [MongoMapperWheel.create, MongoMapperWheel.create, MongoMapperWheel.create]
-
-      @car.mongo_mapper_wheels = @wheels
-      @car.save
-    end
-
-    should "memoize the association" do
-      assert_equal @wheels, @car.mongo_mapper_wheels
-
-      other_wheels = [MongoMapperWheel.create, MongoMapperWheel.create, MongoMapperWheel.create]
-      assert_equal @wheels, ActiveRecordCar.find(@car.id).mongo_mapper_wheels
-      ActiveRecordCar.find(@car.id).update_attribute(:mongo_mapper_wheels, other_wheels)
-      assert_equal other_wheels, ActiveRecordCar.find(@car.id).mongo_mapper_wheels
-
-      assert_equal @wheels, @car.mongo_mapper_wheels
-      assert_equal other_wheels, @car.mongo_mapper_wheels(true)
-    end
-
-    should "be able to specify the class name of the associated class" do
-      vent_1 = MongoMapperVent.create
-      vent_2 = MongoMapperVent.create
-      vent_3 = MongoMapperVent.create
-      dashboard = MongoMapperDashboard.create
-      dashboard.vents = [vent_1, vent_2, vent_3]
-      dashboard.save
-      assert_set_equal [vent_1, vent_2, vent_3], MongoMapperDashboard.find(dashboard.id).vents
-    end
-
-    should "be able to specify the foreign key to use for the class" do
-      car = ActiveRecordCar.create
-      door_1 = CouchRestDoor.create({})
-      door_2 = CouchRestDoor.create({})
-      door_3 = CouchRestDoor.create({})
-      car.couch_rest_doors = [door_1, door_2, door_3]
-      car.save
-
-      assert_set_equal [door_1, door_2, door_3], ActiveRecordCar.find(car.id).couch_rest_doors
-      assert_set_equal [door_1.id.to_s, door_2.id.to_s, door_3.id.to_s], ActiveRecordCar.find(car.id).couch_rest_door_ids
-    end
-  end
-
-  context "An ActiveRecord class with a has_many association to a MongoMapper class" do
-    setup do
-      setup_fixtures
-      @car = ActiveRecordCar.create
-      @wheel_1 = MongoMapperWheel.create
-      @wheel_2 = MongoMapperWheel.create
-      @wheel_3 = MongoMapperWheel.create
-    end
-
-    should "be able to set the associated objects by their ids" do
-      @car.mongo_mapper_wheel_ids = [@wheel_1.id, @wheel_2.id, @wheel_3.id]
-      @car.save
-      assert_set_equal [@wheel_1, @wheel_2, @wheel_3], ActiveRecordCar.find(@car.id).mongo_mapper_wheels
-      assert_set_equal [@wheel_1.id.to_s, @wheel_2.id.to_s, @wheel_3.id.to_s], ActiveRecordCar.find(@car.id).mongo_mapper_wheel_ids
-    end
-
-    context "that works with associated objects" do
+  for_each_orm_extension_combination do |source, target|
+    context "A #{source} class with a has_many association to a #{target} class" do
       setup do
-        @car.mongo_mapper_wheels = [@wheel_1, @wheel_2, @wheel_3]
-        @car.save
-      end
-
-      should "be able to set the associated objects" do
-        assert_set_equal [@wheel_1, @wheel_2, @wheel_3], ActiveRecordCar.find(@car.id).mongo_mapper_wheels
-      end
-
-      should "be able to add an associated object using the << operator" do
-        wheel_4 = MongoMapperWheel.create
-        @car.mongo_mapper_wheels << wheel_4
-        @car.save
-        assert_set_equal [@wheel_1, @wheel_2, @wheel_3, wheel_4], ActiveRecordCar.find(@car.id).mongo_mapper_wheels
-      end
-
-      should "be able to remove an associated object using the delete method" do
-        @car.mongo_mapper_wheels.delete(@wheel_3)
-        @car.save
-        assert_set_equal [@wheel_1, @wheel_2], ActiveRecordCar.find(@car.id).mongo_mapper_wheels
-      end
-
-      should "be able to clear all associated objects using the clear method" do
-        @car.mongo_mapper_wheels.clear
-        @car.save
-        assert_equal [], ActiveRecordCar.find(@car.id).mongo_mapper_wheels
-      end
-
-      should "return an empty array if the association is not set" do
-        car = ActiveRecordCar.create
-        assert_set_equal [], ActiveRecordCar.find(car.id).mongo_mapper_wheels
-      end
-    end
-  end
-
-  context "A MongoMapper class with a has_many association to an ActiveRecord class" do
-    setup do
-      setup_fixtures
-      @wheel = MongoMapperWheel.create
-      @nut_1 = ActiveRecordNut.create
-      @nut_2 = ActiveRecordNut.create
-      @nut_3 = ActiveRecordNut.create
-    end
-
-    should "be able to set the associated objects by their ids" do
-      @wheel.active_record_nut_ids = [@nut_1.id, @nut_2.id, @nut_3.id]
-      @wheel.save
-      assert_set_equal [@nut_1, @nut_2, @nut_3], MongoMapperWheel.find(@wheel.id).active_record_nuts
-      assert_set_equal [@nut_1.id.to_s, @nut_2.id.to_s, @nut_3.id.to_s], MongoMapperWheel.find(@wheel.id).active_record_nut_ids
-    end
-
-    context "that works with associated objects" do
-      setup do
-        @wheel.active_record_nuts = [@nut_1, @nut_2, @nut_3]
-        @wheel.save
-      end
-
-      should "be able to set the associated objects" do
-        assert_set_equal [@nut_1, @nut_2, @nut_3], MongoMapperWheel.find(@wheel.id).active_record_nuts
-      end
-
-      should "be able to add an associated object using the << operator" do
-        nut_4 = ActiveRecordNut.create
-        @wheel.active_record_nuts << nut_4
-        @wheel.save
-        assert_set_equal [@nut_1, @nut_2, @nut_3, nut_4], MongoMapperWheel.find(@wheel.id).active_record_nuts
-      end
-
-      should "be able to remove an associated object using the delete method" do
-        @wheel.active_record_nuts.delete(@nut_3)
-        @wheel.save
-        assert_set_equal [@nut_1, @nut_2], MongoMapperWheel.find(@wheel.id).active_record_nuts
-      end
-
-      should "be able to clear all associated objects using the clear method" do
-        @wheel.active_record_nuts.clear
-        @wheel.save
-        assert_set_equal [], MongoMapperWheel.find(@wheel.id).active_record_nuts
-      end
-
-      should "return an empty array if the association is not set" do
-        wheel = MongoMapperWheel.create
-        assert_set_equal [], MongoMapperWheel.find(wheel.id).active_record_nuts
-      end
-    end
-  end
-
-  context "A CouchRest class with a has_many association to a MongoMapper class" do
-    setup do
-      setup_all_fixtures
-      @radio = CouchRestRadio.create({})
-      @button_1 = MongoMapperButton.create
-      @button_2 = MongoMapperButton.create
-      @button_3 = MongoMapperButton.create
-    end
-
-    should "be able to set the associated objects by their ids" do
-      @radio.mongo_mapper_button_ids = [@button_1.id, @button_2.id, @button_3.id]
-      @radio.save
-      assert_set_equal [@button_1, @button_2, @button_3], CouchRestRadio.get(@radio.id).mongo_mapper_buttons
-      assert_set_equal [@button_1.id.to_s, @button_2.id.to_s, @button_3.id.to_s], CouchRestRadio.get(@radio.id).mongo_mapper_button_ids
-    end
-
-    context "that works with associated objects" do
-      setup do
-        @radio.mongo_mapper_buttons = [@button_1, @button_2, @button_3]
-        @radio.save
-      end
-
-      should "be able to set the associated objects" do
-        assert_set_equal [@button_1, @button_2, @button_3], CouchRestRadio.get(@radio.id).mongo_mapper_buttons
-      end
-
-      should "be able to add an associated object using the << operator" do
-        button_4 = MongoMapperButton.create
-        @radio.mongo_mapper_buttons << button_4
-        @radio.save
-        assert_set_equal [@button_1, @button_2, @button_3, button_4], CouchRestRadio.get(@radio.id).mongo_mapper_buttons
-      end
-
-      should "be able to remove an associated object using the delete method" do
-        @radio.mongo_mapper_buttons.delete(@button_3)
-        @radio.save
-        assert_set_equal [@button_1, @button_2], CouchRestRadio.get(@radio.id).mongo_mapper_buttons
-      end
-
-      should "be able to clear all associated objects using the clear method" do
-        @radio.mongo_mapper_buttons.clear
-        @radio.save
-        assert_equal [], CouchRestRadio.get(@radio.id).mongo_mapper_buttons
-      end
-
-      should "return an empty array if the association is not set" do
-        radio = CouchRestRadio.create({})
-        assert_set_equal [], CouchRestRadio.get(radio.id).mongo_mapper_buttons
+        setup_fixtures_for(source, target)
+
+        @source_class = class_for_extension(source)
+        @source = @source_class.create({})
+        @target_class = class_for_extension(target, :has_many)
+        @target_1 = @target_class.create({})
+        @target_2 = @target_class.create({})
+        @target_3 = @target_class.create({})
+
+        @foreign_key = foreign_key_for(target, :has_many)
+        @foreign_key_id = foreign_key_id_for(target, :has_many)
+      end
+
+      should "be able to set the associated objects by their ids" do
+        @source.send("#{@foreign_key_id}=", [@target_1.id, @target_2.id, @target_3.id])
+        @source.save
+        [@target_1, @target_2, @target_3].each { |t| t._t_reload }
+        assert_set_equal [@target_1, @target_2, @target_3], @source_class._t_find(@source.id.to_s).send(@foreign_key)
+        assert_set_equal [@target_1.id.to_s, @target_2.id.to_s, @target_3.id.to_s], @source_class._t_find(@source.id.to_s).send(@foreign_key_id)
+      end
+
+      context "that works with associated objects" do
+        setup do
+          @source.send("#{@foreign_key}=", [@target_1, @target_2, @target_3])
+          @source.save
+        end
+
+        should "be able to set the associated objects" do
+          [@target_1, @target_2, @target_3].each { |t| t._t_reload }
+          assert_set_equal [@target_1, @target_2, @target_3], @source_class._t_find(@source.id.to_s).send(@foreign_key)
+        end
+
+        should "be able to add an associated object using the << operator" do
+          target_4 = @target_class.create({})
+          @source.send(@foreign_key) << target_4
+          @source.save
+          [@target_1, @target_2, @target_3, target_4].each { |t| t._t_reload }
+          assert_set_equal [@target_1, @target_2, @target_3, target_4], @source_class._t_find(@source.id.to_s).send(@foreign_key)
+        end
+
+        should "be able to add an associated object using the push method" do
+          target_4 = @target_class.create({})
+          target_5 = @target_class.create({})
+          @source.send(@foreign_key).push(target_4, target_5)
+          @source.save
+          [@target_1, @target_2, @target_3, target_4, target_5].each { |t| t._t_reload }
+          assert_set_equal [@target_1, @target_2, @target_3, target_4, target_5], @source_class._t_find(@source.id.to_s).send(@foreign_key)
+        end
+
+        should "be able to add an associated object using the concat method" do
+          target_4 = @target_class.create({})
+          target_5 = @target_class.create({})
+          @source.send(@foreign_key).concat([target_4, target_5])
+          @source.save
+          [@target_1, @target_2, @target_3, target_4, target_5].each { |t| t._t_reload }
+          assert_set_equal [@target_1, @target_2, @target_3, target_4, target_5], @source_class._t_find(@source.id.to_s).send(@foreign_key)
+        end
+
+        should "be able to remove an associated object using the delete method" do
+          @source.send(@foreign_key).delete(@target_3)
+          @source.save
+          [@target_1, @target_2].each { |t| t._t_reload }
+          assert_set_equal [@target_1, @target_2], @source_class._t_find(@source.id.to_s).send(@foreign_key)
+        end
+
+        should "be able to clear all associated objects using the clear method" do
+          @source.send(@foreign_key).clear
+          @source.save
+          assert_set_equal [], @source_class._t_find(@source.id.to_s).send(@foreign_key)
+        end
+
+        should "return an empty array if the association is not set" do
+          source = @source_class.create({})
+          assert_set_equal [], @source_class._t_find(source.id.to_s).send(@foreign_key)
+        end
       end
     end
   end

data/test/associations/has_one_test.rb

@@ -2,90 +2,28 @@ require 'test_helper'
 
 class HasOneTest < Test::Unit::TestCase
 
-  context "A class with a has_one association to another class" do
-    setup do
-      setup_fixtures
-      @climate_control_unit = ActiveRecordClimateControlUnit.create
-      @dashboard = MongoMapperDashboard.create(:active_record_climate_control_unit => @climate_control_unit)
-    end
-
-    should "memoize the association" do
-      assert_equal @climate_control_unit, @dashboard.active_record_climate_control_unit
-
-      other_climate_control_unit = ActiveRecordClimateControlUnit.create
-      assert_equal @climate_control_unit, MongoMapperDashboard.find(@dashboard.id).active_record_climate_control_unit
-      ActiveRecordClimateControlUnit.update(@climate_control_unit.id, :mongo_mapper_dashboard_id => nil)
-      ActiveRecordClimateControlUnit.update(other_climate_control_unit.id, :mongo_mapper_dashboard_id => @dashboard.id)
-      assert_equal other_climate_control_unit, MongoMapperDashboard.find(@dashboard.id).active_record_climate_control_unit
-
-      assert_equal @climate_control_unit, @dashboard.active_record_climate_control_unit
-      assert_equal other_climate_control_unit, @dashboard.active_record_climate_control_unit(true)
-    end
-
-    should "be able to specify the class name of the associated class" do
-      ash_tray = MongoMapperAshTray.create
-      dashboard = MongoMapperDashboard.create(:ash_tray => ash_tray)
-      assert_equal ash_tray, dashboard.ash_tray
-    end
-
-    should "be able to specify the foreign key to use for the class" do
-      car = ActiveRecordCar.create
-      windshield = CouchRestWindshield.create(:active_record_car => car)
-      assert_equal windshield, car.couch_rest_windshield
-
-      engine = ActiveRecordEngine.create(:active_record_car => car)
-      assert_equal engine, car.active_record_engine
-    end
-  end
-
-  context "A MongoMapper class with a has_one association to an ActiveRecord class" do
-    setup do
-      setup_fixtures
-      @climate_control_unit = ActiveRecordClimateControlUnit.create
-      @dashboard = MongoMapperDashboard.create
-    end
-
-    should "be able to set and get the associated object" do
-      @dashboard.active_record_climate_control_unit = @climate_control_unit
-      assert_equal @climate_control_unit, MongoMapperDashboard.find(@dashboard.id).active_record_climate_control_unit
-    end
-
-    should "return nil if no association is set" do
-      assert_nil MongoMapperDashboard.find(@dashboard.id).active_record_climate_control_unit
-    end
-  end
-
-  context "An ActiveRecord class with a has_one association to a MongoMapper class" do
-    setup do
-      setup_fixtures
-      @dashboard = MongoMapperDashboard.create
-      @car = ActiveRecordCar.create
-    end
-
-    should "be able to set and get the associated object" do
-      @car.mongo_mapper_dashboard = @dashboard
-      assert_equal @dashboard, ActiveRecordCar.find(@car.id).mongo_mapper_dashboard
-    end
-
-    should "return nil if no association is set" do
-      assert_nil ActiveRecordCar.find(@car.id).mongo_mapper_dashboard
-    end
-  end
-
-  context "A CouchRest class with a has_one association to a MongoMapper class" do
-    setup do
-      setup_all_fixtures
-      @dashboard = MongoMapperDashboard.create
-      @radio = CouchRestRadio.create({})
-    end
-
-    should "be able to set and get the associated object" do
-      @dashboard.couch_rest_radio = @radio
-      assert_equal @radio, MongoMapperDashboard.find(@dashboard.id).couch_rest_radio
-    end
-
-    should "return nil if no association is set" do
-      assert_nil MongoMapperDashboard.find(@dashboard.id).couch_rest_radio
+  for_each_orm_extension_combination do |source, target|
+    context "A #{source} class with a has_one association to a #{target} class" do
+      setup do
+        setup_fixtures_for(source, target)
+
+        @source_class = class_for_extension(source)
+        @source = @source_class.create({})
+        @target_class = class_for_extension(target, :has_one)
+        @target = @target_class.create({})
+
+        @foreign_key = foreign_key_for(target, :has_one)
+        @foreign_key_id = foreign_key_id_for(target, :has_one)
+      end
+
+      should "be able to set and get the associated object" do
+        @source.send("#{@foreign_key}=", @target)
+        assert_equal @target, @source_class._t_find(@source.id.to_s).send(@foreign_key)
+      end
+
+      should "return nil if no association is set" do
+        assert_nil @source_class._t_find(@source.id.to_s).send(@foreign_key)
+      end
     end
   end