seedomatic 0.1.0

data/.gitignore

@@ -0,0 +1,4 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*

data/.rspec

@@ -0,0 +1 @@
+--color

data/Gemfile

@@ -0,0 +1,4 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in seedomatic.gemspec
+gemspec

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/lib/seedomatic/runner.rb

@@ -0,0 +1,47 @@
+
+
+module SeedOMatic
+  module Runner
+    extend self
+
+    def run (opts = {})
+      results = {}
+
+      files_to_import(opts).each do |file|
+        file_info = load_file(file)
+
+        if should_import?(opts, file_info)
+          results[file_info[:model_name]] = Seeder.new(file_info).import
+        end
+      end
+
+      results
+    end
+
+  protected
+
+    def should_import?(import_options, file_info)
+      (!import_options[:tagged_with]  || ([*import_options[:tagged_with]] - [*file_info[:tags]]).empty?) &&
+      (!import_options[:not_tagged_with] || ([*import_options[:not_tagged_with]] & [*file_info[:tags]]).empty?)
+    end
+
+    def files_to_import(options)
+      if options[:file]
+        [options[:file]]
+      else
+        dir = options[:dir] || "config/seeds"
+        Dir.open(dir).map{|f| "#{dir}/#{f}"}.select{|file| File.file? file}
+      end
+    end
+
+    def load_file(file)
+      data = YAML.load_file(file)
+      {
+        :model_name => data.keys.first,
+        :match_on => data.first[1]['match_on'],
+        :items => data.first[1]['items']
+      }
+    end
+
+  end
+end

data/lib/seedomatic/seeder.rb

@@ -0,0 +1,48 @@
+module SeedOMatic
+  class Seeder
+
+    attr_accessor :model_name, :items, :match_on
+
+    def initialize(data)
+      @model_name = data[:model_name]
+      @items = data[:items]
+      @match_on = [*data[:match_on]]
+    end
+
+    def import
+      new_records = 0
+      updated_records = 0
+
+      items.each do |i|
+        model = model_class.send(create_method, *create_args(i))
+
+        if model.new_record?
+          new_records += 1
+        else
+          updated_records += 1
+        end
+
+        model.attributes = i
+        model.save!
+      end
+
+      { :count => items.length, :new => new_records, :updated => updated_records}
+    end
+
+  protected
+
+    def create_method
+      match_on.empty? ? 'new' : "find_or_initialize_by_#{match_on.join('_and_')}"
+    end
+
+    def create_args(item)
+      match_on.map{|m| item[m] }
+    end
+
+    def model_class
+      return model_name if model_name.is_a? Class
+      model_name.to_s.classify.constantize
+    end
+
+  end
+end

data/lib/seedomatic/version.rb

@@ -0,0 +1,3 @@
+module Seedomatic
+  VERSION = "0.1.0"
+end

data/lib/seedomatic.rb

@@ -0,0 +1,11 @@
+require "seedomatic/version"
+
+require 'yaml'
+require 'active_support/core_ext'
+
+module SeedOMatic
+  autoload :Seeder, 'seedomatic/seeder'
+  autoload :Runner, 'seedomatic/runner'
+
+  extend Runner
+end

data/readme.markdown

@@ -0,0 +1,2 @@
+# Seed-O-Matic
+*Run repeatable seeds across a variety of environments*

data/seedomatic.gemspec

@@ -0,0 +1,24 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "seedomatic/version"
+
+Gem::Specification.new do |s|
+  s.name        = "seedomatic"
+  s.version     = Seedomatic::VERSION
+  s.authors     = ["Ryan Brunner"]
+  s.email       = ["ryan@ryanbrunner.com"]
+  s.homepage    = "http://github.com/ryanbrunner/seedomatic"
+  s.summary     = %q{Seed-O-Matic makes seeding easier.}
+  s.description = %q{Create repeatable seed fixtures that you can continually use and deploy to multiple environments and tenants.}
+
+  s.rubyforge_project = "seedomatic"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  # specify any dependencies here; for example:
+  s.add_development_dependency "rspec", '~> 2.9.0'
+  s.add_runtime_dependency "active_support"
+end

data/spec/lib/seed_o_matic_spec.rb

@@ -0,0 +1,77 @@
+require 'spec_helper'
+
+describe SeedOMatic do
+  let!(:mock_seeder) { mock(SeedOMatic::Seeder).as_null_object }
+
+  describe "run" do
+    subject { described_class.run opts }
+
+    context "specified file" do
+      let(:opts) { { :file => 'spec/support/single_model.yml' } }
+
+      specify {
+        SeedOMatic::Seeder.should_receive(:new).with(hash_including(:model_name => 'model_name',
+                                                                    :items => ['name' => 'Name 1', 'code' => 'code_1'])).and_return(mock_seeder)
+        subject
+      }
+    end
+
+    context "specified directory" do
+      let(:opts) { { :dir => 'spec/support/seed_directory' } }
+
+      specify {
+        SeedOMatic::Seeder.should_receive(:new).with(hash_including(:model_name => 'dir1')).and_return(mock_seeder)
+        SeedOMatic::Seeder.should_receive(:new).with(hash_including(:model_name => 'dir2')).and_return(mock_seeder)
+
+        subject
+      }
+    end
+
+    context "no options (assumed config / seeds directory)" do
+
+    end
+  end
+
+  describe "should_import?" do
+    subject { SeedOMatic.send(:should_import?, run_options, file_info)}
+
+    describe "tagged_with" do
+      let(:run_options) { { :tagged_with => 'tag1'}}
+
+      context "tag exists" do
+        let(:file_info) { {:tags => ['tag1', 'tag2']}}
+        it { should be_true }
+      end
+      context "tag does not exist" do
+        let(:file_info) { {:tags => ['tag3', 'tag2']}}
+        it { should be_false }
+      end
+      context "multiple tags" do
+        let(:run_options) { { :tagged_with => ['tag1', 'tag2'] }}
+        context "all match" do
+          let(:file_info) { {:tags => ['tag1', 'tag2', 'tag3']}}
+          it { should be_true }
+        end
+        context "some match" do
+          let(:file_info) { {:tags => ['tag4', 'tag2', 'tag3']}}
+          it { should be_false}
+        end
+      end
+    end
+    describe "not_tagged_with" do
+      let(:run_options) { { :not_tagged_with => 'tag1'}}
+      context "tag exists" do
+        let(:file_info) { {:tags => ['tag1', 'tag2']}}
+        it { should be_false }
+      end
+      context "tag does not exist" do
+        let(:file_info) { {:tags => ['tag3']}}
+        it { should be_true }
+      end
+      context "no tags" do
+        let(:file_info) { {:tags => []}}
+        it { should be_true }
+      end
+    end
+  end
+end

data/spec/lib/seeder_spec.rb

@@ -0,0 +1,120 @@
+require 'spec_helper'
+
+describe SeedOMatic::Seeder do
+  let!(:mock_model) { mock(MyModel).as_null_object }
+
+  let(:seeder) { described_class.new(data) }
+  let(:data) { { :model_name => model_name, :items => items, :match_on => match_on } }
+  let(:model_name) { "MyModel" }
+  let(:items) {
+    [{'name' => 'foo', 'code' => 'uniquecode', 'code_category' => 'more_unique'} ]
+  }
+  let(:match_on) { nil }
+
+  describe "import" do
+    subject { seeder.import }
+
+    describe "creating records" do
+      context "single item" do
+        let(:items) { [{'name' => 'foo'}] }
+        specify {
+          MyModel.should_receive(:new).and_return(mock_model)
+          subject
+        }
+      end
+
+      context "multiple items" do
+        let(:items) { [{'name' => 'foo'}, {'name' => 'bar'} ] }
+        specify {
+          MyModel.should_receive(:new).twice.and_return(mock_model)
+          subject
+        }
+        it "should set fields properly" do
+          subject
+          MyModel[0].name.should == 'foo'
+          MyModel[1].name.should == 'bar'
+        end
+      end
+    end
+
+    describe "checking uniqueness on a specified field" do
+      context "single field" do
+        let(:match_on) { 'code' }
+
+        specify {
+          MyModel.should_receive(:find_or_initialize_by_code).with('uniquecode').and_return(MyModel.new)
+          subject
+        }
+      end
+      context "multiple fields" do
+        let(:match_on) { ['code', 'code_category'] }
+
+        specify {
+          MyModel.should_receive(:find_or_initialize_by_code_and_code_category).with('uniquecode', 'more_unique').and_return(MyModel.new)
+          subject
+        }
+      end
+    end
+  end
+
+  describe "create_method" do
+    subject { seeder.send(:create_method) }
+
+    context "no matching fields" do
+      let(:match_on) { nil }
+      it { should == 'new' }
+    end
+
+    context "one matching field" do
+      let(:match_on) { 'code' }
+      it { should == 'find_or_initialize_by_code' }
+    end
+
+    context "multiple matching fields" do
+      let(:match_on) { ['code', 'code_category'] }
+      it { should == 'find_or_initialize_by_code_and_code_category' }
+    end
+  end
+
+  describe "create_args" do
+    subject { seeder.send(:create_args, {'code' => 1, 'code_category' => 2}) }
+
+    context "no matching fields" do
+      let(:match_on) { nil }
+      it { should == [] }
+    end
+
+    context "one matching field" do
+      let(:match_on) { 'code' }
+      it { should == [1] }
+    end
+
+    context "multiple matching fields" do
+      let(:match_on) { ['code', 'code_category'] }
+      it { should == [1,2] }
+    end
+
+    context "order sensitivity" do
+      let(:match_on) { ['code_category', 'code'] }
+      it { should == [2,1] }
+    end
+  end
+
+  describe "model class" do
+    subject { seeder.send(:model_class) }
+
+    ["MyModel", MyModel, :my_model, 'my_model'].each do |name|
+      describe name do
+        let(:model_name) { name }
+        it { should == MyModel }
+      end
+    end
+
+    ["Blah", "blah", :not_a_model].each do |name|
+      describe name do
+        let(:model_name) { name }
+        specify { expect { subject }.to raise_error }
+      end
+    end
+  end
+end

data/spec/spec_helper.rb

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

data/spec/support/my_model.rb

@@ -0,0 +1,39 @@
+class MyModel
+  attr_accessor :name, :new_record
+
+  @@models = []
+
+  def initialize(params = {})
+    self.name = params['name']
+    self.new_record = true
+  end
+
+  def self.create(params = {})
+    @@models << m = new(params)
+    m.new_record = false
+    m
+  end
+
+  def attributes=(attr)
+    self.name = attr['name']
+  end
+
+  def save!
+    @@models << self
+    new_record = false
+    self
+  end
+
+  def new_record?
+    @new_record
+  end
+
+  def self.[] (index)
+    @@models[index]
+  end
+
+  def self.models
+    @@models
+  end
+
+end

data/spec/support/seed_directory/dir1.yml

@@ -0,0 +1,4 @@
+dir1:
+  items:
+    - name: Name 1
+      code: code_1

data/spec/support/seed_directory/dir2.yml

@@ -0,0 +1,4 @@
+dir2:
+  items:
+    - name: Name 1
+      code: code_1

data/spec/support/single_model.yml

@@ -0,0 +1,4 @@
+model_name:
+  items:
+    - name: Name 1
+      code: code_1

metadata

@@ -0,0 +1,92 @@
+--- !ruby/object:Gem::Specification
+name: seedomatic
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+  prerelease: 
+platform: ruby
+authors:
+- Ryan Brunner
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-04-23 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: &70246141530860 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 2.9.0
+  type: :development
+  prerelease: false
+  version_requirements: *70246141530860
+- !ruby/object:Gem::Dependency
+  name: active_support
+  requirement: &70246141530440 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70246141530440
+description: Create repeatable seed fixtures that you can continually use and deploy
+  to multiple environments and tenants.
+email:
+- ryan@ryanbrunner.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- .rspec
+- Gemfile
+- Rakefile
+- lib/seedomatic.rb
+- lib/seedomatic/runner.rb
+- lib/seedomatic/seeder.rb
+- lib/seedomatic/version.rb
+- readme.markdown
+- seedomatic.gemspec
+- spec/lib/seed_o_matic_spec.rb
+- spec/lib/seeder_spec.rb
+- spec/spec_helper.rb
+- spec/support/my_model.rb
+- spec/support/seed_directory/dir1.yml
+- spec/support/seed_directory/dir2.yml
+- spec/support/single_model.yml
+homepage: http://github.com/ryanbrunner/seedomatic
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: seedomatic
+rubygems_version: 1.8.15
+signing_key: 
+specification_version: 3
+summary: Seed-O-Matic makes seeding easier.
+test_files:
+- spec/lib/seed_o_matic_spec.rb
+- spec/lib/seeder_spec.rb
+- spec/spec_helper.rb
+- spec/support/my_model.rb
+- spec/support/seed_directory/dir1.yml
+- spec/support/seed_directory/dir2.yml
+- spec/support/single_model.yml