seedomatic 0.4.0 → 0.5.0

data/lib/seedomatic/seeder.rb

@@ -14,8 +14,8 @@ module SeedOMatic
       new_records = 0
       updated_records = 0
 
-      items.each do |i|
-        model = model_class.send(create_method, *create_args(i))
+      items.each do |attrs|
+        model = model_class.send(create_method, *create_args(attrs))
 
         if model.new_record?
           new_records += 1
@@ -24,7 +24,7 @@ module SeedOMatic
         end
 
         if model.new_record? || seed_mode == 'always'
-          model.attributes = i
+          model.attributes = process_lookups(attrs)
           model.save!
         end
       end
@@ -34,6 +34,16 @@ module SeedOMatic
 
   protected
 
+    def process_lookups(attrs)
+      attrs.select{|k| k.ends_with? "_lookup"}.each do |key, value|
+        association = key.gsub("_lookup", "")
+        lookup_class = model_class.reflect_on_association(association).klass
+
+        attrs[association] = lookup_class.where(value).first
+      end
+      attrs
+    end
+
     def create_method
       match_on.empty? ? 'new' : "find_or_initialize_by_#{match_on.join('_and_')}"
     end

data/lib/seedomatic/version.rb

@@ -1,3 +1,3 @@
 module Seedomatic
-  VERSION = "0.4.0"
+  VERSION = "0.5.0"
 end

data/readme.markdown

@@ -14,6 +14,8 @@ specify another directory when you run your seeds. Here's the structure of a typ
       items:
         - name: My Model 1
           code: my_model
+          other_model_lookup:
+            code: om_1
         - name: My Model 2
           code: my_model_2
     other_model:
@@ -63,6 +65,27 @@ fields). The following options are available:
 
 The default seed mode is `always`.
 
+### Lookups
+
+If your seeds depend on other seeded data, you can use *lookups* to locate other records. Just add _lookup to the name of any property, and then provide parameters to look up that data. Seed-O-Matic will search for a record that matches the criteria you specified. Seed-O-Matic will automatically determine the type of record to look up based on the association in your model.
+
+*Warning*: Currently, lookups will only work correctly if the record exists at the time that the seed was run. If the item you're looking up is located in another seed file, ensure that it is imported first (by putting it in a higher alphabetical order).
+
+Examples:
+
+#### With key / value lookup parameters
+
+    category_lookup:
+      code: lookup_code
+
+This will set the `category` of the lookup to the first category with a `code` of `lookup_code`
+
+#### With a string
+
+    category_lookup: name ilike '%code%'
+
+This will look for category records matching the SQL fragment you passed in.
+
 ### Importing specific directories, or specific files.
 
 To load seed files from a specific directory, call:

data/spec/lib/seeder_spec.rb

@@ -1,6 +1,9 @@
 require 'spec_helper'
 
 describe SeedOMatic::Seeder do
+
+  before { MyModel.reset }
+
   let!(:mock_model) { mock(MyModel).as_null_object }
   let(:existing_mock) { MyModel.create(:name => 'Existing') }
   let(:new_mock) { MyModel.new(:name => 'New') }
@@ -38,6 +41,17 @@ describe SeedOMatic::Seeder do
           MyModel[1].name.should == 'bar'
         end
       end
+
+      context "lookup fields" do
+        let(:items) { [{'name' => 'foo', 'category_lookup' => { 'code' => 'bar' }}]}
+        let(:category) { MyCategory.new }
+        specify {
+          MyModel.should_receive(:reflect_on_association).with('category').and_return(OpenStruct.new(:klass => MyCategory))
+          MyCategory.should_receive(:where).with(hash_including('code' => 'bar')).and_return(OpenStruct.new(:first => category))
+          subject
+          MyModel[0].category.should == category
+        }
+      end
     end
 
     describe "checking uniqueness on a specified field" do

data/spec/spec_helper.rb

@@ -1,3 +1,4 @@
 require 'seedomatic'
 
-require 'support/my_model.rb'
+require 'support/my_model.rb'
+require 'ostruct'

data/spec/support/my_model.rb

@@ -1,13 +1,22 @@
+class MyCategory
+
+end
+
 class MyModel
-  attr_accessor :name, :new_record
+  attr_accessor :name, :new_record, :category
 
   @@models = []
 
   def initialize(params = {})
     self.name = params['name']
+    self.category = params['category']
     self.new_record = true
   end
 
+  def self.reset
+    @@models = []
+  end
+
   def self.create(params = {})
     @@models << m = new(params)
     m.new_record = false
@@ -16,6 +25,7 @@ class MyModel
 
   def attributes=(attr)
     self.name = attr['name']
+    self.category = attr['category']
   end
 
   def save!

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: seedomatic
 version: !ruby/object:Gem::Version
-  version: 0.4.0
+  version: 0.5.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,11 +9,11 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-04-26 00:00:00.000000000 Z
+date: 2012-04-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec
-  requirement: &70296477233420 !ruby/object:Gem::Requirement
+  requirement: &70286621376940 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
@@ -21,10 +21,10 @@ dependencies:
         version: 2.9.0
   type: :development
   prerelease: false
-  version_requirements: *70296477233420
+  version_requirements: *70286621376940
 - !ruby/object:Gem::Dependency
   name: activesupport
-  requirement: &70296477231760 !ruby/object:Gem::Requirement
+  requirement: &70286621376520 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -32,7 +32,7 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70296477231760
+  version_requirements: *70286621376520
 description: Create repeatable seed fixtures that you can continually use and deploy
   to multiple environments and tenants.
 email: