has_alter_ego 0.0.4 → 0.0.5

data/Changelog.md

@@ -1,3 +1,10 @@
+# 0.0.5 (2010-06-13)
+* Implemented smart associations, e.g.
+    category_by_name: Sport                          # => @car.category = Category.find_by_name("Sport")
+    sellers_by_name_and_active: [Hugo, true]         # @car.sellers = Seller.find_all_by_name_and_active("Hugo", true)
+    sellers_by_name_and_active: [[Hugo, Egon], true] # @car.sellers = Seller.find_all_by_name_and_active(["Hugo", "Egon"], true)
+* mark objects as modified when saved with save!
+
 # 0.0.4 (2010-06-11)
 * fix some bugs esp. wrt associations
 * added a on_seed hook which allows for custom login on seed

data/README.md

@@ -124,6 +124,24 @@ If you don't want to inherit changes for an object without actually modifying it
     @car.alter_ego_state
     => "default"
 
+# Smart associations
+
+It's possible to define dynamic associations in the seed data which is helpful if the IDs of the associated objects are
+not known or the associations depends on the state of the objects. This is done by appending *_by* clauses to the
+association name, similar to the dynamic finders in ActiveRecord::Base.
+
+**Example:**
+db/fixtures/car.yml:
+    1:
+      brand: Lotus
+      model: Elise
+      category_id: 3                                   # Static way of specifying associations
+      category_by_name: Sport                          # => @car.category = Category.find_by_name("Sport")
+
+      sellers_by_name_and_active: [Hugo, true]         # @car.sellers = Seller.find_all_by_name_and_active("Hugo", true)
+      sellers_by_name_and_active: [[Hugo, Egon], true] # @car.sellers = Seller.find_all_by_name_and_active(["Hugo", "Egon"], true)
+
+
 # Custom logic on seed
 
 has_alter_ego provides a hook for adding custom logic when an object is created or updated from the seed definitions.

data/lib/has_alter_ego.rb

@@ -46,10 +46,7 @@ module HasAlterEgo
         if db_object
           raise "There is already a #{db_object.class} with id #{db_object.id} in the database." unless db_object.has_alter_ego?
           if db_object.alter_ego.state == 'default'
-            yml[primary_key].keys.each do |attr|
-              next unless db_object.respond_to?(attr)
-              db_object.send(attr+"=", yml[primary_key][attr])
-            end
+            assign_attributes db_object, yml[primary_key]
             db_object.on_seed(yml[primary_key])
             db_object.save_without_alter_ego
           end
@@ -60,10 +57,7 @@ module HasAlterEgo
 
           db_object = self.new
           db_object[self.primary_key] = primary_key
-          yml[primary_key].keys.each do |attr|
-            next unless db_object.respond_to?(attr)
-            db_object.send(attr+"=" , yml[primary_key][attr])
-          end
+          assign_attributes db_object, yml[primary_key]
           db_object.build_alter_ego
           db_object.alter_ego.state = 'default'
           db_object.on_seed(yml[primary_key])
@@ -71,7 +65,25 @@ module HasAlterEgo
         end
       end
 
-      private
+      def assign_attributes obj, yml
+        reflections = self.reflect_on_all_associations
+        yml.keys.each do |attr|
+          if obj.respond_to?(attr)
+            obj.send(attr+"=" , yml[attr])
+          else
+            # Handle smart associations
+            reflections.select{|r| attr.start_with?(r.name.to_s+"_by_")}.each do |r|
+              finder = [:has_one, :belongs_to].include?(r.macro) ? "find" : "find_all"
+              if attr.index("_and_")
+                objects = r.klass.send(attr.gsub(r.name.to_s, finder), *(yml[attr]))
+              else
+                objects = r.klass.send(attr.gsub(r.name.to_s, finder), yml[attr])
+              end
+              obj.send(r.name.to_s+"=", objects) if objects.present?
+            end
+          end
+        end
+      end
 
       def get_yml
         filename = File.join(RAILS_ROOT, "db", "fixtures", "alter_egos", self.table_name + ".yml")
@@ -103,6 +115,10 @@ module HasAlterEgo
         save_without_alter_ego perform_validation
       end
 
+      def save!
+        save || raise(RecordNotSaved)
+      end
+
       def destroy
         if self.alter_ego
           self.alter_ego.state = 'destroyed'

data/test/db/fixtures/alter_egos/sellers.yml

@@ -2,4 +2,7 @@
   name: Bruno
 
 2:
-  name: Harald
+  name: Harald
+
+3:
+  name: Hugo

data/test/has_alter_ego_test.rb

@@ -1,36 +1,5 @@
 require 'test_helper'
 
-
-class Seller < ActiveRecord::Base
-  has_and_belongs_to_many :cars
-  has_alter_ego
-end
-
-class Car < ActiveRecord::Base
-  has_many :tires
-  has_and_belongs_to_many :sellers
-  has_alter_ego
-  attr_accessor :tires_count
-
-  def on_seed attributes
-    self[:tires_count] = attributes["custom_data"]["tires"] if attributes["custom_data"]
-  end
-end
-
-class Bike < ActiveRecord::Base
-end
-
-class Scooter < ActiveRecord::Base
-end
-
-class Tire < ActiveRecord::Base
-end
-
-class Drink < ActiveRecord::Base
-  set_primary_key :name
-  has_alter_ego
-end
-
 class HasAlterEgoTest < Test::Unit::TestCase
   def test_space_gets_reserved
     c = Car.create
@@ -206,4 +175,68 @@ class HasAlterEgoTest < Test::Unit::TestCase
 
     assert_equal [1,2], c.seller_ids
   end
+
+  def test_smart_habtm
+    Car.destroy_all
+    AlterEgo.find_all_by_alter_ego_object_type("Car").map(&:destroy)
+    Car.class_eval do
+      def self.get_yml
+        return {1 => {
+                  "brand" => "Lotus",
+                  "model" => "Elise",
+                  "sellers_by_name" => ["Harald", "Hugo"]}}
+      end
+    end
+    Car.parse_yml
+    assert_equal 1, Car.count
+    assert_equal 2, Car.first.sellers.size
+    assert_equal [2,3].sort, Car.first.seller_ids.sort
+  end
+
+  def test_smart_has_many
+    Light.create(:color => "blue", :active => true)
+    Light.create(:color => "yellow", :active => true)
+    Light.create(:color => "red", :active => true)
+    Light.create(:color => "green", :active => false)
+    Bike.destroy_all
+    Bike.class_eval do
+      def self.get_yml
+        return {1 => {
+                  "lights_by_color_and_active" => [["blue", "yellow", "green"], true]}}
+      end
+    end
+    Bike.parse_yml
+    assert_equal 1, Bike.count
+    assert_equal 2, Bike.first.lights.size
+    assert_equal Light.find_all_by_color(["blue", "yellow"]), Bike.first.lights
+  end
+
+  def test_smart_has_one
+    Creator.create(:name => "Martin")
+    Creator.create(:name => "Mike")
+    Drink.destroy_all
+    Drink.class_eval do
+      def self.get_yml
+        return {1 => {
+                  "creator_by_name" => ["Mike"]}}
+      end
+    end
+    Drink.parse_yml
+    assert_equal 1, Drink.count
+    assert_equal Creator.find_by_name("Mike"), Drink.first.creator
+  end
+
+  def test_smart_belongs_to
+
+    Creator.class_eval do
+      def self.get_yml
+        return {1 => {
+                  "name" => "Bernd",
+                  "drink_by_name" => "water"}}
+      end
+    end
+    Creator.parse_yml
+    c = Creator.find_by_name "Bernd"
+    assert_equal Drink.find_by_name("water"), c.drink
+  end
 end

data/test/test_helper.rb

@@ -25,6 +25,12 @@ silence_stream(STDOUT) do
     create_table :bikes
     create_table :scooters
 
+    create_table :lights do |t|
+      t.string :color
+      t.boolean :active
+      t.integer :bike_id
+    end
+
     create_table :tires do |t|
       t.integer :car_id
     end
@@ -34,6 +40,11 @@ silence_stream(STDOUT) do
       t.string :color
     end
 
+    create_table :creators do |t|
+      t.string :name
+      t.string :drink_id
+    end
+
     create_table :sellers do |t|
       t.string :name
     end
@@ -43,4 +54,45 @@ silence_stream(STDOUT) do
     end
 
   end
+end
+
+
+class Seller < ActiveRecord::Base
+  has_and_belongs_to_many :cars
+  has_alter_ego
+end
+
+class Car < ActiveRecord::Base
+  has_many :tires
+  has_and_belongs_to_many :sellers
+  has_alter_ego
+  attr_accessor :tires_count
+
+  def on_seed attributes
+    self[:tires_count] = attributes["custom_data"]["tires"] if attributes["custom_data"]
+  end
+end
+
+class Bike < ActiveRecord::Base
+  has_many :lights
+end
+
+class Scooter < ActiveRecord::Base
+end
+
+class Light < ActiveRecord::Base
+end
+
+class Tire < ActiveRecord::Base
+end
+
+class Drink < ActiveRecord::Base
+  set_primary_key :name
+  has_one :creator
+  has_alter_ego
+end
+
+class Creator < ActiveRecord::Base
+  belongs_to :drink
+  has_alter_ego
 end

metadata

@@ -1,13 +1,12 @@
 --- !ruby/object:Gem::Specification 
 name: has_alter_ego
 version: !ruby/object:Gem::Version 
-  hash: 23
   prerelease: false
   segments: 
   - 0
   - 0
-  - 4
-  version: 0.0.4
+  - 5
+  version: 0.0.5
 platform: ruby
 authors: 
 - "Andr\xC3\xA9 Duffeck"
@@ -15,7 +14,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-06-11 00:00:00 +02:00
+date: 2010-06-13 00:00:00 +02:00
 default_executable: 
 dependencies: []
 
@@ -29,24 +28,24 @@ extensions: []
 extra_rdoc_files: []
 
 files: 
-- lib/has_alter_ego/alter_ego.rb
-- lib/has_alter_ego/dumper.rb
 - lib/generators/has_alter_ego_generator.rb
 - lib/has_alter_ego.rb
+- lib/has_alter_ego/dumper.rb
+- lib/has_alter_ego/alter_ego.rb
 - lib/tasks/has_alter_ego.rake
-- Changelog.md
 - README.md
+- Changelog.md
 - Rakefile
 - rails/init.rb
 - generators/has_alter_ego/templates/create_alter_egos.rb
 - generators/has_alter_ego/has_alter_ego_generator.rb
+- test/db/fixtures/alter_egos/drinks.yml
+- test/db/fixtures/alter_egos/cars.yml
 - test/db/fixtures/alter_egos/sellers.yml
 - test/db/fixtures/alter_egos/bikes.yml
-- test/db/fixtures/alter_egos/cars.yml
-- test/db/fixtures/alter_egos/drinks.yml
 - test/test_helper.rb
 - test/has_alter_ego_test.rb
-has_rdoc: false
+has_rdoc: true
 homepage: http://github.com/aduffeck/has_alter_ego
 licenses: []
 
@@ -56,27 +55,23 @@ rdoc_options: []
 require_paths: 
 - lib
 required_ruby_version: !ruby/object:Gem::Requirement 
-  none: false
   requirements: 
   - - ">="
     - !ruby/object:Gem::Version 
-      hash: 3
       segments: 
       - 0
       version: "0"
 required_rubygems_version: !ruby/object:Gem::Requirement 
-  none: false
   requirements: 
   - - ">="
     - !ruby/object:Gem::Version 
-      hash: 3
       segments: 
       - 0
       version: "0"
 requirements: []
 
 rubyforge_project: 
-rubygems_version: 1.3.7
+rubygems_version: 1.3.6
 signing_key: 
 specification_version: 3
 summary: A Rails plugin which makes it possible to keep seed and live data in parallel