sprig 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 243e9a9198023fecad0dbbbd51c7a7d6e3269572
-  data.tar.gz: 6f59f121d41be507cede08fde695874fa98416f2
+  metadata.gz: 45e7ad607a25ae6b5ddd621ba63d2dc51d97ecf9
+  data.tar.gz: b156960667081a75408bab2b2f5363a89b320d35
 SHA512:
-  metadata.gz: 7654b318cbabfa4b542849e3245a33205ecb6370eda6ac1dec799b39c0b12935c76328456616aeafe4e8bcbd940573a630cc150bf8b610bedd8d0dab2cdd1bbb
-  data.tar.gz: 6ec8691e22fa991d01ad88382bbd90620e4e114d215cff3b21f76e0f3d1bcdcf75b3c98122612094f3a7200c412d1f1dba4bd9b65bed6b29696a26da9d8ffa85
+  metadata.gz: 69ab1c4161a19017678b2e03bbd6af6edb62ef498a56cc0d7c8e16b6097e9b72e66890f17605786eaa78bbbc9eb264e9034075728de8275e12f02ff2a28a112c
+  data.tar.gz: 231648356a91e8f181161a6c9077161d4eb1d2e178ad1e04c96290029f11fdbb9b9a1f7459f774db6ce143cc2a23514d99faa3a8da65a419e347652adde26b1e

data/README.md

@@ -1,4 +1,4 @@
-#Sprig
+![Sprig](http://i.imgur.com/XCu3iVO.png)
 
 [![Code Climate](https://codeclimate.com/github/vigetlabs/sprig.png)](https://codeclimate.com/github/vigetlabs/sprig) [![Build Status](https://travis-ci.org/vigetlabs/sprig.png?branch=master)](https://travis-ci.org/vigetlabs/sprig) [![Gem Version](https://badge.fury.io/rb/sprig.png)](http://badge.fury.io/rb/sprig)
 
@@ -6,6 +6,11 @@ Seed Rails application by convention, not configuration.
 
 Provides support for common files types: *csv*, *yaml*, and *json*.  Extensible for the rest!
 
+Learn more about Sprig and view documentation at [http://vigetlabs.github.io/sprig/](http://vigetlabs.github.io/sprig/).
+
+## Installation
+
+Use `rails generate sprig:install` to create environment-specific seed directories.
 
 ##The Sprig Directive
 
@@ -21,7 +26,6 @@ sprig [User, Post, Comment]
 
 This directive tells Sprig to go find your datafiles for the `User`, `Post`, and `Comment` seed resources, build records from the data entries, and insert them into the database. Sprig will automatically detect known datafile types like `.yml`, `.json`, or `.csv` within your environment-specific seed directory.
 
-
 ##Environment
 
 Seed files are unique to the environment in which your Rails application is running. Within `db/seeds` create an environment-specific directory (i.e. `/development` for your 'development' environment).
@@ -81,6 +85,8 @@ records:
     body: "Yaml Comment body"
 ```
 
+**Note: For namespaced or STI classes, you'll need to include the namespace with the class name in the seed file name. For example `Users::HeadManager` would need to be `users_head_managers.yml`**
+
 ### Special Options
 
 These are provided in a `options:` key for *yaml* and *json* files.
@@ -100,6 +106,19 @@ options:
   find_existing_by: ['title', 'user_id']
 ```
 
+### Computed Values
+
+It's common to want seed values that are dynamic.  Sprig supports an ERB style syntax for computing seed attributes.
+
+```yaml
+# posts.yml
+
+records:
+  - sprig_id: 1
+    body: "Yaml Post body"
+    published_at: "<%= 1.week.ago %>"
+```
+
 ##Custom Sources and Parsers
 
 If all your data is in `.wat` files, fear not. You can tell Sprig where to look for your data, and point it toward a custom parser class for turning your data into records. The example below tells Sprig to read `User` seed data from a Google Spreadsheet, and parse it accordingly.

data/lib/generators/sprig/install_generator.rb

@@ -0,0 +1,31 @@
+require 'rails/generators/base'
+
+module Sprig
+  module Generators
+    class InstallGenerator < Rails::Generators::Base
+      argument :arg_envs, :type => :array, :optional => true
+
+      desc "Install Sprig seed directories"
+
+      def create_enviroment_directories
+        empty_directory 'db/seeds'
+
+        envs.each { |env| empty_directory "db/seeds/#{env}" }
+      end
+
+      private
+
+      def envs
+        arg_envs ? arg_envs : default_envs
+      end
+
+      def default_envs
+        env_configs.map { |p| File.basename(p, File.extname(p)) }
+      end
+
+      def env_configs
+        Dir[Rails.root.join('config/environments', '*.rb')]
+      end
+    end
+  end
+end

data/lib/sprig.rb

@@ -10,20 +10,28 @@ module Sprig
   autoload :Parser,               'sprig/parser'
   autoload :Helpers,              'sprig/helpers'
   autoload :Planter,              'sprig/planter'
-  autoload :SprigLogger,            'sprig/sprig_logger'
-  autoload :SprigRecordStore,      'sprig/sprig_record_store'
+  autoload :ProcessNotifier,      'sprig/process_notifier'
+  autoload :Logging,              'sprig/logging'
+  autoload :NullRecord,           'sprig/null_record'
+  autoload :SprigRecordStore,     'sprig/sprig_record_store'
   autoload :Data,                 'sprig/data'
   autoload :Seed,                 'sprig/seed'
 
-  def self.configuration
-    @@configuration ||= Sprig::Configuration.new
-  end
+  class << self
+    def configuration
+      @@configuration ||= Sprig::Configuration.new
+    end
 
-  def self.configure
-    yield configuration
-  end
+    def configure
+      yield configuration
+    end
+
+    def reset_configuration
+      @@configuration = nil
+    end
 
-  def self.reset_configuration
-    @@configuration = nil
+    def logger
+      configuration.logger
+    end
   end
 end

data/lib/sprig/configuration.rb

@@ -1,12 +1,16 @@
 module Sprig
   class Configuration
 
-    attr_writer :directory
+    attr_writer :directory, :logger
 
     def directory
       Rails.root.join(@directory || default_directory, Rails.env)
     end
 
+    def logger
+      @logger ||= Logger.new($stdout)
+    end
+
     private
 
     def default_directory

data/lib/sprig/directive.rb

@@ -24,7 +24,7 @@ module Sprig
     end
 
     def datasource
-      @datasource ||= Source.new(klass.to_s.tableize, options)
+      @datasource ||= Source.new(klass.to_s.tableize.gsub("/", "_"), options)
     end
 
     private

data/lib/sprig/helpers.rb

@@ -16,6 +16,8 @@ module Sprig
 
     def sprig_record(klass, seed_id)
       SprigRecordStore.instance.get(klass, seed_id)
+    rescue SprigRecordStore::RecordNotFoundError => error
+      NullRecord.new(error)
     end
 
     def sprig_file(relative_path)

data/lib/sprig/logging.rb

@@ -0,0 +1,35 @@
+module Sprig
+  module Logging
+    LOG_LEVELS = {
+      debug: :blue,
+      info:  :green,
+      warn:  :orange,
+      error: :red
+    }
+
+    LOG_COLORS = {
+      blue:   34,
+      green:  32,
+      orange: 33,
+      red:    31
+    }
+
+    LOG_LEVELS.each do |level, color|
+      define_method("log_#{level}") do |message|
+        Sprig.logger.send(level, send(color, message.to_s))
+      end
+    end
+
+    private
+
+    def colorize(message, color_code)
+      "\e[#{color_code}m#{message}\e[0m"
+    end
+
+    LOG_COLORS.each do |name, color_code|
+      define_method(name) do |message|
+        colorize(message, color_code)
+      end
+    end
+  end
+end

data/lib/sprig/null_record.rb

@@ -0,0 +1,13 @@
+module Sprig
+  class NullRecord
+    include Logging
+
+    def initialize(error)
+      log_error "#{error} (Substituted with NullRecord)"
+    end
+
+    def method_missing(*)
+      nil
+    end
+  end
+end

data/lib/sprig/planter.rb

@@ -9,7 +9,7 @@ module Sprig
         plant(seed)
       end
 
-      logger.log_summary
+      notifier.finished
     end
 
     private
@@ -20,19 +20,19 @@ module Sprig
       DependencySorter.new(seeds).sorted_items
     end
 
-    def logger
-      @logger ||= SprigLogger.new
+    def notifier
+      @notifier ||= ProcessNotifier.new
     end
 
     def plant(seed)
-      logger.processing
+      notifier.in_progress
       seed.before_save
 
       if seed.save_record
         seed.save_to_store
-        logger.log_success(seed)
+        notifier.success(seed)
       else
-        logger.log_error(seed)
+        notifier.error(seed)
       end
     end
   end

data/lib/sprig/process_notifier.rb

@@ -0,0 +1,59 @@
+require 'active_support/inflector'
+
+module Sprig
+  class ProcessNotifier
+    include Logging
+
+    def initialize
+      @success_count = 0
+      @error_count = 0
+      @errors = []
+    end
+
+    def success(seed)
+      log_info seed.success_log_text
+      @success_count += 1
+    end
+
+    def error(seed)
+      @errors << seed.record
+      log_error seed.error_log_text
+      log_error seed.record
+      log_error seed.record.errors.messages
+      @error_count += 1
+    end
+
+    def finished
+      log_debug 'Seeding complete.'
+
+      if @success_count > 0
+        log_info success_summary
+      else
+        log_error success_summary
+      end
+
+      if @error_count > 0
+        log_error error_summary
+
+        @errors.each do |error|
+          log_error error
+          log_error "#{error.errors.messages}\n"
+        end
+      end
+    end
+
+    def in_progress
+      log_debug "Planting those seeds...\r"
+    end
+
+    private
+
+    def success_summary
+      "#{@success_count} #{'seed'.pluralize(@success_count)} successfully planted."
+    end
+
+    def error_summary
+      "#{@error_count} #{'seed'.pluralize(@error_count)} couldn't be planted:"
+    end
+  end
+end

data/lib/sprig/seed/factory.rb

@@ -4,17 +4,17 @@ module Sprig
       def self.new_from_directive(directive)
         raise ArgumentError, 'Must provide a Directive' unless directive.is_a? Directive
 
-        klass      = directive.klass
-        datasource = directive.datasource
-        options    = directive.options
-
-        new(klass, datasource, options)
+        new(
+          directive.klass,
+          directive.datasource,
+          directive.options
+        )
       end
 
       def initialize(klass, datasource, options)
-        self.klass             = klass
-        self.datasource        = datasource
-        self.initial_options   = options
+        self.klass           = klass
+        self.datasource      = datasource
+        self.initial_options = options
       end
 
       def add_seeds_to_hopper(hopper)
@@ -30,6 +30,7 @@ module Sprig
       def klass=(klass)
         raise ArgumentError, 'Must provide a Class as first argument' unless klass.is_a? Class
 
+        klass.reset_column_information
         @klass = klass
       end
 

data/lib/sprig/sprig_record_store.rb

@@ -11,7 +11,7 @@ module Sprig
     end
 
     def get(klass, sprig_id)
-      records_of_klass(klass)[sprig_id.to_s] || record_not_found
+      records_of_klass(klass)[sprig_id.to_s] || record_not_found(klass, sprig_id)
     end
 
     private
@@ -24,7 +24,7 @@ module Sprig
       @records ||= {}
     end
 
-    def record_not_found
+    def record_not_found(klass, sprig_id)
       raise(RecordNotFoundError, "Record for class #{klass} and sprig_id #{sprig_id} could not be found.")
     end
   end

data/lib/sprig/version.rb

@@ -1,3 +1,3 @@
 module Sprig
-  VERSION = "0.1.1"
+  VERSION = "0.1.2"
 end

data/spec/fixtures/config/environments/development.rb

@@ -0,0 +1 @@
+# Rails environment config

data/spec/fixtures/config/environments/production.rb

@@ -0,0 +1 @@
+# Rails environment config

data/spec/fixtures/config/environments/test.rb

@@ -0,0 +1 @@
+# Rails environment config

data/spec/fixtures/models/user.rb

@@ -0,0 +1,3 @@
+class User < ActiveRecord::Base
+  validates :first_name, :last_name, :presence => true
+end

data/spec/fixtures/seeds/test/invalid_users.yml

@@ -0,0 +1,6 @@
+# users.yml
+
+records:
+  - sprig_id: 1
+    first_name: "Clark"
+    # intentionally leaving out last_name so record fails to create

data/spec/fixtures/seeds/test/posts_missing_record.yml

@@ -0,0 +1,7 @@
+# posts_missing_dependency.yml
+
+records:
+  - sprig_id: 1
+    title: 'Yaml title'
+    content: 'Yaml content'
+    user_id: "<%= sprig_record(User, 1).id %>" # This sprig record did not save.

data/spec/lib/generators/sprig/install_generator_spec.rb

@@ -0,0 +1,59 @@
+require 'spec_helper'
+require 'generators/sprig/install_generator'
+
+describe Sprig::Generators::InstallGenerator do
+  destination File.expand_path("../../tmp", __FILE__)
+
+  before do
+    stub_rails_root
+    prepare_destination
+    run_generator
+  end
+
+  it "creates empty db/seeds directory" do
+    assert_directory "db/seeds"
+  end
+
+  it "creates empty seed environment directories" do
+    [
+      :development,
+      :test,
+      :production
+    ].each do |env|
+      assert_directory "db/seeds/#{env}"
+    end
+  end
+end
+
+
+# Generator arguments are set on a class basis. We need to open
+# a new describe block to make these examples work.
+
+describe Sprig::Generators::InstallGenerator do
+  context "with arguments" do
+    destination File.expand_path("../../tmp", __FILE__)
+    arguments %w(development test integration)
+
+    before do
+      stub_rails_root
+      prepare_destination
+      run_generator
+    end
+
+    it "creates empty seed directories from arguments" do
+      [
+        :development,
+        :test,
+        :integration
+      ].each do |env|
+        assert_directory "db/seeds/#{env}"
+      end
+
+      [
+        :production
+      ].each do |env|
+        assert_no_directory "db/seeds/#{env}"
+      end
+    end
+  end
+end

data/spec/lib/sprig/configuration_spec.rb

@@ -7,7 +7,6 @@ describe Sprig::Configuration do
   end
 
   describe "#directory" do
-
     it "returns db/seeds/:env by default" do
       subject.directory.to_path.should == '~/db/seeds/development'
     end
@@ -18,4 +17,20 @@ describe Sprig::Configuration do
       subject.directory.to_path.should == '~/seed_files/development'
     end
   end
+
+  describe "#logger" do
+    it "returns an stdout logger by default" do
+      logger = double('Logger')
+      Logger.stub(:new).with($stdout).and_return(logger)
+
+      subject.logger.should == logger
+    end
+
+    it "returns a custom logger" do
+      logger = double('Logger')
+      subject.logger = logger
+
+      subject.logger.should == logger
+    end
+  end
 end

data/spec/lib/sprig/directive_spec.rb

@@ -2,6 +2,11 @@ require 'spec_helper'
 
 describe Sprig::Directive do
 
+  module Users
+    class Admin < User
+    end
+  end
+
   describe "#klass" do
     context "given a class" do
       subject { described_class.new(Post) }
@@ -9,6 +14,12 @@ describe Sprig::Directive do
       its(:klass) { should == Post }
     end
 
+    context "given a class within a module" do
+      subject { described_class.new(Users::Admin) }
+
+      its(:klass) { should == Users::Admin }
+    end
+
     context "given options with a class" do
       subject { described_class.new(:class => Post) }
 
@@ -47,14 +58,26 @@ describe Sprig::Directive do
   describe "#datasource" do
     let(:datasource) { double('datasource') }
 
-    subject { described_class.new(:class => Post, :source => 'source') }
+    context "with a class" do
+      subject { described_class.new(:class => Post, :source => 'source') }
 
-    before do
-      Sprig::Source.stub(:new).with('posts', { :source => 'source' }).and_return(datasource)
+      before do
+        Sprig::Source.stub(:new).with('posts', { :source => 'source' }).and_return(datasource)
+      end
+
+      it "returns a sprig data source" do
+        subject.datasource.should == datasource
+      end
     end
 
-    it "returns a sprig data source" do
-      subject.datasource.should == datasource
+    context "with a class within a module" do
+      subject { described_class.new(Users::Admin) }
+
+      it "passes the correct path to Source" do
+        Sprig::Source.should_receive(:new).with("users_admins", {})
+
+        subject.datasource
+      end
     end
   end
 end

data/spec/lib/sprig/null_record_spec.rb

@@ -0,0 +1,24 @@
+require 'spec_helper'
+
+describe Sprig::NullRecord do
+  let(:error_msg) { 'Something bad happened.' }
+  subject { described_class.new(error_msg) }
+
+  describe "#new" do
+    it "logs an error upon initialization" do
+      log_should_receive(:error, with: "#{error_msg} (Substituted with NullRecord)")
+
+      subject
+    end
+  end
+
+  describe "#method_missing" do
+    it "returns nil for undefined method calls" do
+      subject.enhance_your_calm.should == nil
+    end
+  end
+
+  it_behaves_like "a logging entity" do
+    subject! { described_class.new(error_msg) }
+  end
+end

data/spec/lib/sprig/process_notifier_spec.rb

@@ -0,0 +1,88 @@
+require 'spec_helper'
+
+describe Sprig::ProcessNotifier do
+  it_behaves_like "a logging entity" do
+    subject { described_class.new }
+  end
+
+  describe "#success" do
+    let(:seed) { double('Seed', success_log_text: 'I am a teapot.') }
+
+    it "logs the seed's success message" do
+      log_should_receive(:info, with: 'I am a teapot.')
+
+      subject.success(seed)
+    end
+  end
+
+  describe "#error" do
+    let(:errors) { double('Errors', messages: 'error messages') }
+    let(:seed_record) { double('Record', to_s: 'Seed Record', errors: errors) }
+    let(:seed) { double('Seed', error_log_text: 'I am a teapot.', record: seed_record) }
+
+    it "logs the seed's error message and error details" do
+      log_should_receive(:error, with: 'I am a teapot.').ordered
+      log_should_receive(:error, with: 'Seed Record').ordered
+      log_should_receive(:error, with: 'error messages').ordered
+
+      subject.error(seed)
+    end
+  end
+
+  describe "#finished" do
+    it "logs a complete message" do
+      log_should_receive(:debug, with: 'Seeding complete.')
+
+      subject.finished
+    end
+
+    context "when records are saved successfully" do
+      let(:seed) { double('Seed', success_log_text: 'I am a teapot.') }
+
+      before do
+        subject.success(seed)
+      end
+
+      it "logs a summery of successful saves" do
+        log_should_receive(:info, with: '1 seed successfully planted.')
+
+        subject.finished
+      end
+    end
+
+    context "when no records are saved successfully" do
+      it "logs a summery of successful saves" do
+        log_should_receive(:error, with: '0 seeds successfully planted.')
+
+        subject.finished
+      end
+    end
+
+    context "when there is an error saving a record" do
+      let(:errors) { double('Errors', messages: 'error messages') }
+      let(:seed_record) { double('Record', to_s: 'Seed Record', errors: errors) }
+      let(:seed) { double('Seed', error_log_text: 'I am a teapot.', record: seed_record) }
+
+      before do
+        subject.error(seed)
+      end
+
+      it "logs a summary of errors" do
+        log_should_receive(:error, with: '0 seeds successfully planted.').ordered
+        log_should_receive(:error, with: "1 seed couldn't be planted:").ordered
+        log_should_receive(:error, with: 'Seed Record').ordered
+        log_should_receive(:error, with: "error messages\n").ordered
+
+        subject.finished
+      end
+    end
+  end
+
+  describe "#in_progress" do
+    it "logs an in-progress message" do
+      log_should_receive(:debug, with: "Planting those seeds...\r")
+
+      subject.in_progress
+    end
+  end
+end

data/spec/lib/sprig_spec.rb

@@ -0,0 +1,52 @@
+require 'spec_helper'
+
+describe Sprig do
+  let(:configuration) { double('Configuration') }
+
+  before do
+    Sprig::Configuration.stub(:new).and_return(configuration)
+  end
+
+  describe ".configuration" do
+    context "when there is not yet a Configuration instance" do
+      it "returns a new Configuration instance" do
+        described_class.configuration.should == configuration
+      end
+    end
+
+    context "when there is an existing Configuration instance" do
+      before do
+        described_class.configuration
+      end
+
+      it "returns the existing Configuration instance" do
+        new_configuration = double('Configuration')
+        Sprig::Configuration.stub(:new).and_return(new_configuration)
+
+        described_class.configuration.should == configuration
+      end
+    end
+  end
+
+  describe ".reset_configuration" do
+    before do
+      described_class.configuration
+    end
+
+    it "clears the existing configuration" do
+      described_class.reset_configuration
+      new_configuration = double('Configuration')
+      Sprig::Configuration.stub(:new).and_return(new_configuration)
+
+      described_class.configuration.should == new_configuration
+    end
+  end
+
+  describe ".configure" do
+    it "yields the configuration" do
+      described_class.configure do |config|
+        config.should == configuration
+      end
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -6,13 +6,18 @@ require "database_cleaner"
 require "webmock"
 require "vcr"
 require "pry"
+require "generator_spec"
 
 require "sprig"
 include Sprig::Helpers
 
 Dir[File.dirname(__FILE__) + '/fixtures/models/*.rb'].each {|file| require file }
+Dir[File.dirname(__FILE__) + '/support/**/*.rb'].each {|file| require file}
 
 RSpec.configure do |c|
+  c.include ColoredText
+  c.include LoggerMock
+
   c.before(:suite) do
     DatabaseCleaner.strategy = :transaction
     DatabaseCleaner.clean_with(:truncation)
@@ -38,11 +43,11 @@ end
 # Database
 ActiveRecord::Base.establish_connection(:adapter => "sqlite3", :database => "spec/db/activerecord.db")
 
-#User.connection.execute "DROP TABLE IF EXISTS users;"
-#User.connection.execute "CREATE TABLE users (id INTEGER PRIMARY KEY , first_name VARCHAR(255), last_name VARCHAR(255));"
+User.connection.execute "DROP TABLE IF EXISTS users;"
+User.connection.execute "CREATE TABLE users (id INTEGER PRIMARY KEY , first_name VARCHAR(255), last_name VARCHAR(255), type VARCHAR(255));"
 
 Post.connection.execute "DROP TABLE IF EXISTS posts;"
-Post.connection.execute "CREATE TABLE posts (id INTEGER PRIMARY KEY , title VARCHAR(255), content VARCHAR(255), published BOOLEAN);"
+Post.connection.execute "CREATE TABLE posts (id INTEGER PRIMARY KEY , title VARCHAR(255), content VARCHAR(255), published BOOLEAN , user_id INTEGER);"
 
 Comment.connection.execute "DROP TABLE IF EXISTS comments;"
 Comment.connection.execute "CREATE TABLE comments (id INTEGER PRIMARY KEY , post_id INTEGER, body VARCHAR(255));"

data/spec/sprig_spec.rb

@@ -112,6 +112,27 @@ describe "Seeding an application" do
     end
   end
 
+  context "with a relationship to a record that didn't save" do
+    around do |example|
+      load_seeds('invalid_users.yml', 'posts_missing_record.yml', &example)
+    end
+
+    it "does not error, but carries on with the seeding" do
+      expect {
+        sprig [
+          {
+            :class  => Post,
+            :source => open('spec/fixtures/seeds/test/posts_missing_record.yml')
+          },
+          {
+            :class  => User,
+            :source => open('spec/fixtures/seeds/test/invalid_users.yml')
+          }
+        ]
+      }.to_not raise_error
+    end
+  end
+
   context "with multiple files for a class" do
     around do |example|
       load_seeds('posts.yml', 'legacy_posts.yml', &example)

data/spec/support/helpers/colored_text.rb

@@ -0,0 +1,17 @@
+module ColoredText
+  def blue_text(text)
+    "\e[34m#{text}\e[0m"
+  end
+
+  def green_text(text)
+    "\e[32m#{text}\e[0m"
+  end
+
+  def orange_text(text)
+    "\e[33m#{text}\e[0m"
+  end
+
+  def red_text(text)
+    "\e[31m#{text}\e[0m"
+  end
+end

data/spec/support/helpers/logger_mock.rb

@@ -0,0 +1,21 @@
+module LoggerMock
+  def log_should_receive(level, options)
+    Sprig.logger.should_receive(level).with(send("log_#{level}_text", options.fetch(:with)))
+  end
+
+  def log_debug_text(text)
+    blue_text(text)
+  end
+
+  def log_info_text(text)
+    green_text(text)
+  end
+
+  def log_warn_text(text)
+    orange_text(text)
+  end
+
+  def log_error_text(text)
+    red_text(text)
+  end
+end

data/spec/support/shared_examples/a_logging_entity.rb

@@ -0,0 +1,16 @@
+shared_examples_for "a logging entity" do
+  let(:message) { 'A log message.' }
+
+  [
+    :debug,
+    :info,
+    :warn,
+    :error
+  ].each do |level|
+    it "can log a #{level}" do
+      log_should_receive(level, with: message)
+
+      subject.send("log_#{level}", message)
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: sprig
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - Lawson Kurtz
@@ -9,104 +9,118 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-01-28 00:00:00.000000000 Z
+date: 2014-05-14 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ~>
+    - - "~>"
       - !ruby/object:Gem::Version
         version: '3.1'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ~>
+    - - "~>"
       - !ruby/object:Gem::Version
         version: '3.1'
 - !ruby/object:Gem::Dependency
   name: sqlite3
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ~>
+    - - "~>"
       - !ruby/object:Gem::Version
         version: 1.3.8
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ~>
+    - - "~>"
       - !ruby/object:Gem::Version
         version: 1.3.8
 - !ruby/object:Gem::Dependency
   name: rspec
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ~>
+    - - "~>"
       - !ruby/object:Gem::Version
         version: 2.14.0
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ~>
+    - - "~>"
       - !ruby/object:Gem::Version
         version: 2.14.0
 - !ruby/object:Gem::Dependency
   name: database_cleaner
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ~>
+    - - "~>"
       - !ruby/object:Gem::Version
         version: 1.2.0
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ~>
+    - - "~>"
       - !ruby/object:Gem::Version
         version: 1.2.0
 - !ruby/object:Gem::Dependency
   name: webmock
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ~>
+    - - "~>"
       - !ruby/object:Gem::Version
         version: 1.15.0
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ~>
+    - - "~>"
       - !ruby/object:Gem::Version
         version: 1.15.0
 - !ruby/object:Gem::Dependency
   name: vcr
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ~>
+    - - "~>"
       - !ruby/object:Gem::Version
         version: 2.8.0
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ~>
+    - - "~>"
       - !ruby/object:Gem::Version
         version: 2.8.0
 - !ruby/object:Gem::Dependency
   name: pry
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: generator_spec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
 description: Sprig is a library for managing interconnected, environment-specific
@@ -121,6 +135,7 @@ files:
 - MIT-LICENSE
 - README.md
 - Rakefile
+- lib/generators/sprig/install_generator.rb
 - lib/sprig.rb
 - lib/sprig/configuration.rb
 - lib/sprig/data.rb
@@ -130,6 +145,8 @@ files:
 - lib/sprig/directive.rb
 - lib/sprig/directive_list.rb
 - lib/sprig/helpers.rb
+- lib/sprig/logging.rb
+- lib/sprig/null_record.rb
 - lib/sprig/parser.rb
 - lib/sprig/parser/base.rb
 - lib/sprig/parser/csv.rb
@@ -137,6 +154,7 @@ files:
 - lib/sprig/parser/json.rb
 - lib/sprig/parser/yml.rb
 - lib/sprig/planter.rb
+- lib/sprig/process_notifier.rb
 - lib/sprig/seed.rb
 - lib/sprig/seed/attribute.rb
 - lib/sprig/seed/attribute_collection.rb
@@ -144,16 +162,20 @@ files:
 - lib/sprig/seed/factory.rb
 - lib/sprig/seed/record.rb
 - lib/sprig/source.rb
-- lib/sprig/sprig_logger.rb
 - lib/sprig/sprig_record_store.rb
 - lib/sprig/version.rb
 - spec/db/activerecord.db
 - spec/feature/configurations_spec.rb
 - spec/fixtures/cassettes/google_spreadsheet_json_posts.yml
+- spec/fixtures/config/environments/development.rb
+- spec/fixtures/config/environments/production.rb
+- spec/fixtures/config/environments/test.rb
 - spec/fixtures/models/comment.rb
 - spec/fixtures/models/post.rb
+- spec/fixtures/models/user.rb
 - spec/fixtures/seeds/staging/posts.yml
 - spec/fixtures/seeds/test/comments.yml
+- spec/fixtures/seeds/test/invalid_users.yml
 - spec/fixtures/seeds/test/legacy_posts.yml
 - spec/fixtures/seeds/test/posts.csv
 - spec/fixtures/seeds/test/posts.json
@@ -161,11 +183,19 @@ files:
 - spec/fixtures/seeds/test/posts_find_existing_by_multiple.yml
 - spec/fixtures/seeds/test/posts_find_existing_by_single.yml
 - spec/fixtures/seeds/test/posts_missing_dependency.yml
+- spec/fixtures/seeds/test/posts_missing_record.yml
+- spec/lib/generators/sprig/install_generator_spec.rb
 - spec/lib/sprig/configuration_spec.rb
 - spec/lib/sprig/directive_list_spec.rb
 - spec/lib/sprig/directive_spec.rb
+- spec/lib/sprig/null_record_spec.rb
+- spec/lib/sprig/process_notifier_spec.rb
+- spec/lib/sprig_spec.rb
 - spec/spec_helper.rb
 - spec/sprig_spec.rb
+- spec/support/helpers/colored_text.rb
+- spec/support/helpers/logger_mock.rb
+- spec/support/shared_examples/a_logging_entity.rb
 homepage: http://www.github.com/vigetlabs/sprig
 licenses:
 - MIT
@@ -176,17 +206,17 @@ require_paths:
 - lib
 required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
-  - - '>='
+  - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
-  - - '>='
+  - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.2.1
+rubygems_version: 2.2.2
 signing_key: 
 specification_version: 4
 summary: Relational, environment-specific seeding for Rails apps.
@@ -194,10 +224,15 @@ test_files:
 - spec/db/activerecord.db
 - spec/feature/configurations_spec.rb
 - spec/fixtures/cassettes/google_spreadsheet_json_posts.yml
+- spec/fixtures/config/environments/development.rb
+- spec/fixtures/config/environments/production.rb
+- spec/fixtures/config/environments/test.rb
 - spec/fixtures/models/comment.rb
 - spec/fixtures/models/post.rb
+- spec/fixtures/models/user.rb
 - spec/fixtures/seeds/staging/posts.yml
 - spec/fixtures/seeds/test/comments.yml
+- spec/fixtures/seeds/test/invalid_users.yml
 - spec/fixtures/seeds/test/legacy_posts.yml
 - spec/fixtures/seeds/test/posts.csv
 - spec/fixtures/seeds/test/posts.json
@@ -205,9 +240,16 @@ test_files:
 - spec/fixtures/seeds/test/posts_find_existing_by_multiple.yml
 - spec/fixtures/seeds/test/posts_find_existing_by_single.yml
 - spec/fixtures/seeds/test/posts_missing_dependency.yml
+- spec/fixtures/seeds/test/posts_missing_record.yml
+- spec/lib/generators/sprig/install_generator_spec.rb
 - spec/lib/sprig/configuration_spec.rb
 - spec/lib/sprig/directive_list_spec.rb
 - spec/lib/sprig/directive_spec.rb
+- spec/lib/sprig/null_record_spec.rb
+- spec/lib/sprig/process_notifier_spec.rb
+- spec/lib/sprig_spec.rb
 - spec/spec_helper.rb
 - spec/sprig_spec.rb
-has_rdoc: 
+- spec/support/helpers/colored_text.rb
+- spec/support/helpers/logger_mock.rb
+- spec/support/shared_examples/a_logging_entity.rb

data/lib/sprig/sprig_logger.rb

@@ -1,68 +0,0 @@
-require 'active_support/inflector'
-
-module Sprig
-  class SprigLogger
-    def initialize
-      @success_count = 0
-      @error_count = 0
-      @errors = []
-    end
-
-    def log_success(seed)
-      message = seed.success_log_text
-      puts green(message)
-      @success_count += 1
-    end
-
-    def log_error(seed)
-      message = seed.error_log_text
-      @errors << seed.record
-      puts red(message)
-      @error_count += 1
-    end
-
-    def log_summary
-      puts 'Seeding complete.'
-
-      if @success_count > 0
-        puts green(success_summary)
-      else
-        puts red(success_summary)
-      end
-
-      if @error_count > 0
-        puts red(error_summary)
-
-        @errors.each do |error|
-          puts red("#{error}\n#{error.errors.messages}\n\n")
-        end
-      end
-    end
-
-    def processing
-      print "Planting those seeds...\r"
-    end
-
-    private
-
-    def colorize(message, color_code)
-      "\e[#{color_code}m#{message}\e[0m"
-    end
-
-    def red(message)
-      colorize(message, 31)
-    end
-
-    def green(message)
-      colorize(message, 32)
-    end
-
-    def success_summary
-      "#{@success_count} #{'seed'.pluralize(@success_count)} successfully planted."
-    end
-
-    def error_summary
-      "#{@error_count} #{'seed'.pluralize(@error_count)} couldn't be planted:"
-    end
-  end
-end