sprig 0.1.2 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 45e7ad607a25ae6b5ddd621ba63d2dc51d97ecf9
-  data.tar.gz: b156960667081a75408bab2b2f5363a89b320d35
+  metadata.gz: d2748e6392c09851ed00b725700d826a95fd3568
+  data.tar.gz: b826aa49873ab82d3a738d03328d444d19e67ff1
 SHA512:
-  metadata.gz: 69ab1c4161a19017678b2e03bbd6af6edb62ef498a56cc0d7c8e16b6097e9b72e66890f17605786eaa78bbbc9eb264e9034075728de8275e12f02ff2a28a112c
-  data.tar.gz: 231648356a91e8f181161a6c9077161d4eb1d2e178ad1e04c96290029f11fdbb9b9a1f7459f774db6ce143cc2a23514d99faa3a8da65a419e347652adde26b1e
+  metadata.gz: 007edb3280e9bd30a4ddebe436a3d07e9a7c0d72525dab0e0ea3b22754027f7551fb857838b076411f682a27e3bc3b1a0a322642d1f00f50f5f981cff66c563b
+  data.tar.gz: fd2563c0b6bfb9086a396d1c6b914b38c9263084a9056dc0551686393165f69a56d046e5a228de92d54bf0066e18c276a782bf1d57832d7c1ba611b61ba23bb8

data/README.md

@@ -147,4 +147,52 @@ Sprig.configure do |c|
 end
 ```
 
+## Populate Seed Files from Database
+
+Don't want to write Sprig seed files from scratch?  Well, Sprig can create them for you!
+
+Via a rake task:
+```
+rake db:seed:reap
+```
+Or from the Rails console:
+```
+Sprig::Harvest.reap
+```
+
+By default, Sprig will create seed files (currently in `.yaml` only) for every model in your Rails
+application.  The seed files will be placed in a folder in `db/seeds` named after the current
+`Rails.env`.
+
+If any of the models in your application are using STI, Sprig will create a single seed file named
+after the STI base model.  STI sub-type records will all be written to that file.
+
+### Additional Configuration
+
+Don't like the defaults when reaping Sprig records? You may specify the environment (`db/seeds`
+target folder) or models (`ActiveRecord::Base.subclasses`-only) you want seed files for.
+
+Example (rake task):
+```
+rake db:seed:reap ENV=integration MODELS=User, Post
+```
+
+Example (Rails console):
+```
+Sprig::Harvest.reap(env: 'integration', models: [User, Post])
+```
+
+### Adding to Existing Seed Files (`.yaml` only)
+
+Already have some seed files set up?  No worries!  Sprig will detect existing seed files and append
+to them with the records from your database with no extra work needed.  Sprig will automatically
+assign unique `sprig_ids` so you won't have to deal with pesky duplicates.
+
+NOTE: Sprig does not account for your application or database validations.  If you reap seed files
+from your database multiple times in a row without deleting the previous seed files or sprig
+records, you'll end up with duplicate sprig records (but they'll all have unique `sprig_ids`).  This
+may cause validation issues when you seed your database.
+
+## License
+
 This project rocks and uses MIT-LICENSE.

data/lib/sprig.rb

@@ -1,6 +1,7 @@
 module Sprig
   autoload :Configuration,        'sprig/configuration'
   autoload :Planter,              'sprig/planter'
+  autoload :TsortableHash,        'sprig/tsortable_hash'
   autoload :Dependency,           'sprig/dependency'
   autoload :DependencyCollection, 'sprig/dependency_collection'
   autoload :DependencySorter,     'sprig/dependency_sorter'
@@ -16,6 +17,8 @@ module Sprig
   autoload :SprigRecordStore,     'sprig/sprig_record_store'
   autoload :Data,                 'sprig/data'
   autoload :Seed,                 'sprig/seed'
+  autoload :Harvest,              'sprig/harvest'
+  autoload :Railtie,              'sprig/railtie'
 
   class << self
     def configuration

data/lib/sprig/dependency_sorter.rb

@@ -54,16 +54,6 @@ module Sprig
       MissingDependencyError.new(missing_dependency)
     end
 
-    class TsortableHash < Hash
-      include TSort
-
-      alias tsort_each_node each_key
-
-      def tsort_each_child(node, &block)
-        fetch(node).each(&block)
-      end
-    end
-
     class MissingDependencyError < StandardError
       def initialize(missing_dependency = nil)
         super message_for(missing_dependency)

data/lib/sprig/harvest.rb

@@ -0,0 +1,33 @@
+module Sprig
+  module Harvest
+    autoload :Configuration, 'sprig/harvest/configuration'
+    autoload :Model,         'sprig/harvest/model'
+    autoload :Record,        'sprig/harvest/record'
+    autoload :SeedFile,      'sprig/harvest/seed_file'
+
+    class << self
+      def reap(options = {})
+        configure do |config|
+          config.env     = options[:env]    || options['ENV']
+          config.classes = options[:models] || options['MODELS']
+        end
+
+        Model.all.each { |model| SeedFile.new(model).write }
+      end
+
+      private
+
+      cattr_reader :configuration
+
+      delegate :env, :classes, to: :configuration
+
+      def configuration
+        @@configuration ||= Configuration.new
+      end
+
+      def configure
+        yield configuration
+      end
+    end
+  end
+end

data/lib/sprig/harvest/configuration.rb

@@ -0,0 +1,63 @@
+module Sprig
+  module Harvest
+    class Configuration
+      VALID_CLASSES = ActiveRecord::Base.subclasses
+
+      def env
+        @env ||= Rails.env
+      end
+
+      def env=(given_env)
+        parse_valid_env_from given_env do |environment|
+          @env = environment
+        end
+      end
+
+      def classes
+        @classes ||= VALID_CLASSES
+      end
+
+      def classes=(given_classes)
+        parse_valid_classes_from given_classes do |classes|
+          @classes = classes
+        end
+      end
+
+      private
+
+      def parse_valid_env_from(input)
+        return if input.nil?
+        environment = input.strip.downcase
+        create_seeds_folder(environment)
+        yield environment
+      end
+
+      def create_seeds_folder(env)
+        folder = Rails.root.join('db', 'seeds', env)
+        FileUtils.mkdir_p(folder) unless File.directory? folder
+      end
+
+      def parse_valid_classes_from(input)
+        return if input.nil?
+
+        classes = if input.is_a? String
+          input.split(',').map { |klass_string| klass_string.strip.classify.constantize }
+        else
+          input
+        end
+
+        validate_classes(classes)
+
+        yield classes
+      end
+
+      def validate_classes(classes)
+        classes.each do |klass|
+          unless VALID_CLASSES.include? klass
+            raise ArgumentError, "Cannot create a seed file for #{klass} because it is not a subclass of ActiveRecord::Base."
+          end
+        end
+      end
+    end
+  end
+end

data/lib/sprig/harvest/model.rb

@@ -0,0 +1,77 @@
+module Sprig
+  module Harvest
+    class Model
+      def self.all
+        @@all ||= begin
+          models = Sprig::Harvest.classes.map { |klass| new(klass) }
+          
+          tsorted_classes(models).map do |klass|
+            models.find { |model| model.klass == klass }
+          end
+        end
+      end
+
+      def self.find(klass, id)
+        all.find { |model| model.klass == klass }.find(id)
+      end
+
+      attr_reader :klass
+      attr_writer :existing_sprig_ids
+
+      def initialize(klass)
+        @klass = klass
+      end
+
+      def attributes
+        klass.column_names
+      end
+
+      def dependencies
+        @dependencies ||= klass.reflect_on_all_associations(:belongs_to).map do |association|
+          association.name.to_s.classify.constantize
+        end
+      end
+
+      def existing_sprig_ids
+        @existing_sprig_ids ||= []
+      end
+
+      def generate_sprig_id
+        existing_sprig_ids.select { |i| i.is_a? Integer }.sort.last + 1
+      end
+
+      def find(id)
+        records.find { |record| record.id == id }
+      end
+
+      def to_s
+        klass.to_s
+      end
+
+      def to_yaml(options = {})
+        namespace         = options[:namespace]
+        formatted_records = records.map(&:to_hash)
+
+        yaml = if namespace
+          { namespace => formatted_records }.to_yaml
+        else
+          formatted_records.to_yaml
+        end
+
+        yaml.gsub("---\n", '') # Remove annoying YAML separator
+      end
+
+      def records
+        @records ||= klass.all.map { |record| Record.new(record, self) }
+      end
+
+      private
+
+      def self.tsorted_classes(models)
+        models.reduce(TsortableHash.new) do |hash, model|
+          hash.merge(model.klass => model.dependencies)
+        end.tsort
+      end
+    end
+  end
+end

data/lib/sprig/harvest/record.rb

@@ -0,0 +1,61 @@
+module Sprig
+  module Harvest
+    class Record
+      attr_reader :record, :model
+      attr_writer :sprig_id
+
+      def initialize(record, model)
+        @record = record
+        @model  = model
+      end
+
+      def attributes
+        @attributes ||= Array.new.replace(model.attributes).tap do |attrs|
+          attrs[0] = 'sprig_id'
+        end
+      end
+
+      def to_hash
+        attributes.reduce(Hash.new) { |hash, attr| hash.merge(attr => send(attr)) }
+      end
+
+      def sprig_id
+        @sprig_id ||= model.existing_sprig_ids.include?(record.id) ? model.generate_sprig_id : record.id
+      end
+
+      private
+
+      def method_missing(method, *args, &block)
+        attr = model.attributes.find { |attr| attr == method.to_s }
+
+        if attr.nil?
+          super
+        elsif dependency_finder.match(attr)
+          klass    = klass_for(attr)
+          id       = record.send(attr)
+          sprig_id = Model.find(klass, id).sprig_id
+
+          sprig_record(klass, sprig_id)
+        else
+          record.send(attr)
+        end
+      end
+
+      def respond_to_missing?(method, include_private = false)
+        model.attributes.include?(method.to_s) || super
+      end
+
+      def dependency_finder
+        /_id/
+      end
+
+      def klass_for(attr)
+        attr.gsub(dependency_finder, '').classify.constantize
+      end
+
+      def sprig_record(klass, sprig_id)
+        "<%= sprig_record(#{klass}, #{sprig_id}).id %>"
+      end
+    end
+  end
+end

data/lib/sprig/harvest/seed_file.rb

@@ -0,0 +1,54 @@
+module Sprig
+  module Harvest
+    class SeedFile
+      DEFAULT_NAMESPACE = 'records'
+
+      attr_reader :model
+
+      def initialize(model)
+        raise ArgumentError, 'Must initialize with a Sprig::Harvest::Model' unless model.is_a? Model
+
+        @model = model
+      end
+
+      def path
+        Rails.root.join('db', 'seeds', Sprig::Harvest.env, "#{model.to_s.tableize.gsub('/', '_')}.yml")
+      end
+
+      def exists?
+        File.exists?(path)
+      end
+
+      def write
+        initialize_file do |file, namespace|
+          file.write model.to_yaml(:namespace => namespace)
+        end
+      end
+
+      private
+
+      def initialize_file
+        existing_file = exists?
+        access_type = existing_file ? 'a+' : 'w'
+
+        File.open(path, access_type) do |file|
+          namespace = DEFAULT_NAMESPACE
+
+          if existing_file
+            model.existing_sprig_ids = existing_sprig_ids(file.read)
+            namespace = nil
+            file.write("\n")
+          end
+
+          yield file, namespace
+        end
+      end
+
+      def existing_sprig_ids(yaml)
+        YAML.load(yaml).fetch(DEFAULT_NAMESPACE).to_a.map do |record|
+          record.fetch('sprig_id')
+        end
+      end
+    end
+  end
+end

data/lib/sprig/railtie.rb

@@ -0,0 +1,7 @@
+module Sprig
+  class Railtie < Rails::Railtie
+    rake_tasks do
+      load 'tasks/reap.rake'
+    end
+  end
+end

data/lib/sprig/seed/entry.rb

@@ -35,7 +35,11 @@ module Sprig
       end
 
       def success_log_text
-        "#{klass.name} with sprig_id #{sprig_id} successfully saved."
+        "#{klass.name} with sprig_id #{sprig_id} successfully #{success_log_status_text}."
+      end
+
+      def success_log_status_text
+        record.existing? ? "updated" : "saved"
       end
 
       def error_log_text

data/lib/sprig/seed/record.rb

@@ -11,9 +11,10 @@ module Sprig
       end
 
       def initialize(klass, attributes, orm_record = nil)
-        @klass = klass
+        @klass      = klass
         @attributes = attributes
         @orm_record = orm_record || klass.new
+        @existing   = @orm_record.persisted?
       end
 
       def save
@@ -21,6 +22,10 @@ module Sprig
         orm_record.save
       end
 
+      def existing?
+        @existing
+      end
+
       private
 
       attr_reader :attributes, :klass

data/lib/sprig/tsortable_hash.rb

@@ -0,0 +1,11 @@
+module Sprig
+  class TsortableHash < Hash
+    include TSort
+
+    alias tsort_each_node each_key
+
+    def tsort_each_child(node, &block)
+      fetch(node).each(&block)
+    end
+  end
+end

data/lib/sprig/version.rb

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

data/lib/tasks/reap.rake

@@ -0,0 +1,9 @@
+namespace :db do
+  namespace :seed do
+    desc 'Create Sprig seed files from database records'
+    task :reap => :environment do
+      Rails.application.eager_load!
+      Sprig::Harvest.reap(ENV)
+    end
+  end
+end

data/spec/fixtures/yaml/comment_seeds.yml

@@ -0,0 +1,7 @@
+records:
+  - sprig_id: 10
+    post_id: <%= sprig_record(Post, 2).id %>
+    body: pork chop sandwiches
+  - sprig_id: 20
+    post_id: <%= sprig_record(Post, 1).id %>
+    body: do you play baseball?

data/spec/fixtures/yaml/post_seeds.yml

@@ -0,0 +1,9 @@
+records:
+  - sprig_id: 1
+    title: such post
+    content: Once upon a time BOOM POSTED
+    published: true
+  - sprig_id: 2
+    title: very words
+    content:  Word cannon has been loaded up real good.
+    published: true

data/spec/fixtures/yaml/records_with_namespace.yml

@@ -0,0 +1,7 @@
+records:
+- sprig_id: 1
+  post_id: <%= sprig_record(Post, 1).id %>
+  body: 
+- sprig_id: 2
+  post_id: <%= sprig_record(Post, 2).id %>
+  body: 

data/spec/fixtures/yaml/records_without_namespace.yml

@@ -0,0 +1,6 @@
+- sprig_id: 1
+  post_id: <%= sprig_record(Post, 1).id %>
+  body: 
+- sprig_id: 2
+  post_id: <%= sprig_record(Post, 2).id %>
+  body: 

data/spec/fixtures/yaml/user_seeds.yml

@@ -0,0 +1,7 @@
+records:
+  - sprig_id: 5
+    first_name: Bo
+    last_name: Janglez
+  - sprig_id: 9
+    first_name: Jimmy
+    last_name: Russel

data/spec/lib/sprig/harvest/configuration_spec.rb

@@ -0,0 +1,99 @@
+require 'spec_helper'
+
+describe Sprig::Harvest::Configuration do
+  subject { described_class.new }
+
+  before do
+    stub_rails_root
+  end
+
+  describe "#env" do
+    context "from a fresh configuration" do
+      its(:env) { should == Rails.env }
+    end
+  end
+
+  describe "#env=" do
+    context "when given nil" do
+      it "does not change the env" do
+        subject.env = nil
+
+        subject.env.should_not == nil
+      end
+    end
+
+    context "given a non-nil value" do
+      let(:input) { ' ShaBOOSH' }
+
+      it "formats the given value and then sets the environment" do
+        subject.env = input
+
+        subject.env.should == 'shaboosh'
+      end
+
+      context "and the corresponding seeds folder does not yet exist" do
+        after do
+          FileUtils.remove_dir('./spec/fixtures/db/seeds/shaboosh')
+        end
+
+        it "creates the seeds folder" do
+          subject.env = input
+
+          File.directory?('./spec/fixtures/db/seeds/shaboosh').should == true
+        end
+      end
+    end
+  end
+
+  describe "#classes" do
+    context "from a fresh configuration" do
+      its(:classes) { should == ActiveRecord::Base.subclasses }
+    end
+  end
+
+  describe "#classes=" do
+    context "when given nil" do
+      it "does not set classes to nil" do
+        subject.classes = nil
+
+        subject.classes.should_not == nil
+      end
+    end
+
+    context "when given an array of classes" do
+      context "where one or more classes are not subclasses of ActiveRecord::Base" do
+        it "raises an error" do
+          expect {
+            subject.classes = [Comment, Sprig::Harvest::Model]
+          }.to raise_error ArgumentError, 'Cannot create a seed file for Sprig::Harvest::Model because it is not a subclass of ActiveRecord::Base.'
+        end
+      end
+
+      context "where all classes are subclasses of ActiveRecord::Base" do
+        it "sets classes to the given input" do
+          subject.classes = [Comment, Post]
+
+          subject.classes.should == [Comment, Post]
+        end
+      end
+    end
+
+    context "when given a string" do
+      context "where one or more classes are not subclasses of ActiveRecord::Base" do
+        it "raises an error" do
+          expect {
+            subject.classes = 'Sprig::Harvest::Model'
+          }.to raise_error ArgumentError, 'Cannot create a seed file for Sprig::Harvest::Model because it is not a subclass of ActiveRecord::Base.'
+        end
+      end
+
+      context "where all classes are subclasses of ActiveRecord::Base" do
+        it "sets classes to the parsed input" do
+          subject.classes = ' comment, post'
+
+          subject.classes.should == [Comment, Post]
+        end
+      end
+    end
+  end
+end

data/spec/lib/sprig/harvest/model_spec.rb

@@ -0,0 +1,120 @@
+require 'spec_helper'
+
+describe Sprig::Harvest::Model do
+  describe ".all" do
+    let(:all_models) do
+      [
+        described_class.new(Post),
+        described_class.new(Comment),
+        described_class.new(User)
+      ]
+    end
+
+    before do
+      Sprig::Harvest.stub(:classes).and_return([Comment, Post, User])
+    end
+
+    it "returns an dependency-sorted array of Sprig::Harvest::Models" do
+      described_class.all.all? { |model| model.is_a? Sprig::Harvest::Model }.should == true
+      described_class.all.map(&:klass).should == all_models.map(&:klass)
+    end
+  end
+
+  describe ".find" do
+    let!(:user)      { User.create(:first_name => 'Bo', :last_name => 'Janglez') }
+    let!(:post1)     { Post.create }
+    let!(:post2)     { Post.create }
+    let!(:comment1)  { Comment.create(:post => post1) }
+    let!(:comment2)  { Comment.create(:post => post2) }
+
+    subject { described_class }
+
+    it "returns the Sprig::Harvest::Record for the given class and id" do
+      subject.find(User, 1).record.should == user
+      subject.find(Post, 1).record.should == post1
+      subject.find(Post, 2).record.should == post2
+      subject.find(Comment, 1).record.should == comment1
+      subject.find(Comment, 2).record.should == comment2
+    end
+  end
+
+  describe "#attributes" do
+    subject { described_class.new(User) }
+
+    its(:attributes) { should == User.column_names }
+  end
+
+  describe "#dependencies" do
+    subject { described_class.new(Comment) }
+
+    its(:dependencies) { should == [Post] }
+  end
+
+  describe "#find" do
+    let!(:post1) { Post.create }
+    let!(:post2) { Post.create }
+
+    subject { described_class.new(Post) }
+
+    it "returns the Sprig::Harvest::Record with the given id" do
+      harvest_record = subject.find(2)
+      harvest_record.should be_an_instance_of Sprig::Harvest::Record
+      harvest_record.record.should == post2
+    end
+  end
+
+  describe "#generate_sprig_id" do
+    subject { described_class.new(Comment) }
+
+    context "when the existing sprig_ids are all integers" do
+      before do
+        subject.existing_sprig_ids = [5, 20, 8]
+      end
+
+      it "returns an integer-type sprig_id that is not taken" do
+        subject.generate_sprig_id.should == 21
+      end
+    end
+
+    context "when the existing sprig ids contain non-integer values" do
+      before do
+        subject.existing_sprig_ids = [1, 5, 'l_2', 'l_10', 'such_sprigs', 10.9]
+      end
+      it "returns an integer-type sprig_id that is not taken" do
+        subject.generate_sprig_id.should == 6
+      end
+    end
+  end
+
+  describe "#to_s" do
+    subject { described_class.new(Comment) }
+
+    its(:to_s) { should == "Comment" }
+  end
+
+  describe "#to_yaml" do
+    let!(:user)      { User.create(:first_name => 'Bo', :last_name => 'Janglez') }
+    let!(:post1)     { Post.create }
+    let!(:post2)     { Post.create }
+    let!(:comment1)  { Comment.create(:post => post1) }
+    let!(:comment2)  { Comment.create(:post => post2) }
+
+    subject { described_class.new(Comment) }
+
+    context "when passed a value for the namespace" do
+      it "returns the correct yaml" do
+        subject.to_yaml(:namespace => 'records').should == yaml_from_file('records_with_namespace.yml')
+      end
+    end
+
+    context "when no namespace is given" do
+      it "returns the correct yaml" do
+        subject.to_yaml.should == yaml_from_file('records_without_namespace.yml')
+      end
+    end
+  end
+
+  def yaml_from_file(basename)
+    File.read('spec/fixtures/yaml/' + basename)
+  end
+end

data/spec/lib/sprig/harvest/record_spec.rb

@@ -0,0 +1,70 @@
+require 'spec_helper'
+
+class Very; end
+
+describe Sprig::Harvest::Record do
+  let(:record)         { double('ActiveRecord::Base Instance') }
+  let(:model)          { double('Sprig::Harvest::Model') }
+  let(:harvest_record) { double('Sprig::Harvest::Record for class Very', :sprig_id => 5) }
+  let(:sprig_record)   { "<%= sprig_record(Very, 5).id %>" }
+
+  subject { described_class.new(record, model) }
+
+  before do
+    attrs = { 
+      'id'      => 0,
+      'such'    => 1,
+      'wow'     => 2,
+      'very_id' => 3
+    }
+
+    attrs.each_pair do |attr, val|
+      record.stub(attr).and_return(val)
+    end
+
+    model.stub(:attributes).and_return(attrs.keys)
+    model.stub(:existing_sprig_ids).and_return([])
+
+    Sprig::Harvest::Model.stub(:find).and_return(harvest_record)
+  end
+
+  its(:id)      { should == 0 }
+  its(:such)    { should == 1 }
+  its(:wow)     { should == 2 }
+  its(:very_id) { should == sprig_record }
+
+  describe "#attributes" do
+    it "returns an array of attributes from the given model with sprig_id swapped out for id" do
+      subject.attributes.should == %w(
+        sprig_id
+        such
+        wow 
+        very_id
+      )
+    end
+  end
+
+  describe "#to_hash" do
+    it "returns its attributes and their values in a hash" do
+      subject.to_hash.should == {
+        'sprig_id' => 0,
+        'such'     => 1,
+        'wow'      => 2,
+        'very_id'  => sprig_record
+      }
+    end
+  end
+
+  describe "#sprig_id" do
+    its(:sprig_id) { should == record.id }
+
+    context "when an existing seed record has a sprig_id equal to the record's id" do
+      before do
+        model.stub(:existing_sprig_ids).and_return([record.id])
+        model.stub(:generate_sprig_id).and_return(25)
+      end
+
+      its(:sprig_id) { should == 25 }
+    end
+  end
+end

data/spec/lib/sprig/harvest/seed_file_spec.rb

@@ -0,0 +1,109 @@
+require 'spec_helper'
+
+describe Sprig::Harvest::SeedFile do
+  let(:model) { Sprig::Harvest::Model.new(Comment) }
+
+  subject { described_class.new(model) }
+
+  before do
+    stub_rails_root
+    Sprig::Harvest.stub(:env).and_return('dreamland')
+  end
+
+  describe "#initialize" do
+    context "given a non-Sprig::Harvest::Model" do
+      it "raises an error" do
+        expect {
+          described_class.new(User)
+        }.to raise_error ArgumentError, 'Must initialize with a Sprig::Harvest::Model'
+      end
+    end
+  end
+
+  describe "#path" do
+    around do |example|
+      setup_seed_folder('./spec/fixtures/db/seeds/dreamland', &example)
+    end
+
+    its(:path) { should == Rails.root.join('db', 'seeds', 'dreamland', 'comments.yml') }
+  end
+
+  describe "#exists?" do
+    subject { described_class.new(model) }
+
+    around do |example|
+      setup_seed_folder('./spec/fixtures/db/seeds/dreamland', &example)
+    end
+
+    context "when the seed file already exists" do
+      before do
+        File.stub(:exists?).with(subject.path).and_return(true)
+      end
+
+      its(:exists?) { should == true }
+    end
+
+    context "when the seed file does not exist" do
+      its(:exists?) { should == false }
+    end
+  end
+
+  describe "#write" do
+    let!(:user)      { User.create(:first_name => 'Bo', :last_name => 'Janglez') }
+    let!(:post1)     { Post.create }
+    let!(:post2)     { Post.create }
+    let!(:comment1)  { Comment.create(:post => post1) }
+    let!(:comment2)  { Comment.create(:post => post2) }
+
+    around do |example|
+      setup_seed_folder('./spec/fixtures/db/seeds/dreamland', &example)
+    end
+
+    context "when the seed file already exists" do
+      before do
+        yaml = File.read('./spec/fixtures/yaml/comment_seeds.yml')
+        File.open(subject.path, 'w') { |file| file.write(yaml) }
+      end
+
+      it "pulls out the existing sprig ids and stores them on the given model" do
+        model.should_receive(:existing_sprig_ids=).with([10, 20])
+
+        subject.write
+      end
+
+      it "grabs the yaml for the given model without a namespace" do
+        model.should_receive(:to_yaml).with(:namespace => nil)
+
+        subject.write
+      end
+
+      it "populates the file" do
+        starting_size = File.size(subject.path)
+
+        subject.write
+
+        File.size?(subject.path).should > starting_size
+      end
+    end
+
+    context "when the seed file does not yet exist" do
+      it "does not pass any existing sprig ids to the given model" do
+        model.should_not_receive(:existing_sprig_ids=)
+
+        subject.write
+      end
+
+      it "grabs the yaml for the given model with the 'records' namespace" do
+        model.should_receive(:to_yaml).with(:namespace => 'records')
+
+        subject.write
+      end
+
+      it "populates the file" do
+        subject.write
+
+        File.size?(subject.path).should > 0
+      end
+    end
+  end
+end

data/spec/lib/sprig/harvest_spec.rb

@@ -0,0 +1,55 @@
+require 'spec_helper'
+
+describe Sprig::Harvest do
+  describe ".reap" do
+    let(:seed_file) { double('Sprig::Harvest::SeedFile', :write => 'such seeds') }
+
+    before do
+      stub_rails_root
+      stub_rails_env('dreamland')
+      Sprig::Harvest::SeedFile.stub(:new).and_return(seed_file)
+    end
+
+    around do |example|
+      setup_seed_folder('./spec/fixtures/db/seeds/dreamland', &example)
+    end
+
+    it "generates a seed file for each class" do
+      seed_file.should_receive(:write).exactly(3).times
+
+      subject.reap
+    end
+
+    context "when passed an environment in the options hash" do
+      context "in :env" do
+        it "sets the environment" do
+          subject.reap(:env => 'integration')
+          subject.env.should == 'integration'
+        end
+      end
+
+      context "in 'ENV'" do
+        it "sets the environment" do
+          subject.reap('ENV' => ' Integration')
+          subject.env.should == 'integration'
+        end
+      end
+    end
+
+    context "when passed a set of classes in the options hash" do
+      context "in :classes" do
+        it "sets the classes" do
+          subject.reap(:models => [User, Post])
+          subject.classes.should == [User, Post]
+        end
+      end
+
+      context "sets the classes" do
+        it "passes the value to its configuration" do
+          subject.reap('MODELS' => 'User, Post')
+          subject.classes.should == [User, Post]
+        end
+      end
+    end
+  end
+end

data/spec/lib/sprig/seed/entry_spec.rb

@@ -0,0 +1,31 @@
+require 'spec_helper'
+
+describe Sprig::Seed::Entry do
+  describe ".success_log_text" do
+    context "on a new record" do
+      it "indicates the record was 'saved'" do
+        subject = described_class.new(Post, { title: "Hello World!", content: "Stuff", sprig_id: 1 }, {})
+        subject.save_record
+
+        subject.success_log_text.should == "Post with sprig_id 1 successfully saved."
+      end
+    end
+
+    context "on an existing record" do
+      let!(:existing) do
+        Post.create(
+          :title      => "Existing title",
+          :content    => "Existing content",
+          :published  => false
+        )
+      end
+
+      it "indicates the record was 'updated'" do
+        subject = described_class.new(Post, { title: "Existing title", content: "Existing content", sprig_id: 1 }, { find_existing_by: [:title] })
+        subject.save_record
+
+        subject.success_log_text.should == "Post with sprig_id 1 successfully updated."
+      end
+    end
+  end
+end

data/spec/lib/sprig/seed/record_spec.rb

@@ -0,0 +1,25 @@
+require 'spec_helper'
+
+describe Sprig::Seed::Record do
+  describe ".existing?" do
+    let!(:existing) do
+      Post.create(
+        :title      => "Existing title",
+        :content    => "Existing content",
+        :published  => false
+      )
+    end
+
+    it "returns true if the record has already been saved to the database" do
+      subject = described_class.new_or_existing(Post, { title: "Existing title" }, { title: "Existing title" })
+
+      subject.existing?.should == true
+    end
+
+    it "returns false if the record is new" do
+      subject = described_class.new_or_existing(Post, { title: "New title" }, { title: "New title" })
+
+      subject.existing?.should == false
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -78,3 +78,12 @@ def load_seeds(*files)
     `rm ./spec/fixtures/db/seeds/#{env}/#{file}`
   end
 end
+
+# Create and remove seed folder around a spec
+def setup_seed_folder(path)
+  FileUtils.mkdir_p(path)
+
+  yield
+
+  FileUtils.remove_dir(path)
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: sprig
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.3
 platform: ruby
 authors:
 - Lawson Kurtz
@@ -9,118 +9,118 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-05-14 00:00:00.000000000 Z
+date: 2014-06-04 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
@@ -144,6 +144,11 @@ files:
 - lib/sprig/dependency_sorter.rb
 - lib/sprig/directive.rb
 - lib/sprig/directive_list.rb
+- lib/sprig/harvest.rb
+- lib/sprig/harvest/configuration.rb
+- lib/sprig/harvest/model.rb
+- lib/sprig/harvest/record.rb
+- lib/sprig/harvest/seed_file.rb
 - lib/sprig/helpers.rb
 - lib/sprig/logging.rb
 - lib/sprig/null_record.rb
@@ -155,6 +160,7 @@ files:
 - lib/sprig/parser/yml.rb
 - lib/sprig/planter.rb
 - lib/sprig/process_notifier.rb
+- lib/sprig/railtie.rb
 - lib/sprig/seed.rb
 - lib/sprig/seed/attribute.rb
 - lib/sprig/seed/attribute_collection.rb
@@ -163,7 +169,9 @@ files:
 - lib/sprig/seed/record.rb
 - lib/sprig/source.rb
 - lib/sprig/sprig_record_store.rb
+- lib/sprig/tsortable_hash.rb
 - lib/sprig/version.rb
+- lib/tasks/reap.rake
 - spec/db/activerecord.db
 - spec/feature/configurations_spec.rb
 - spec/fixtures/cassettes/google_spreadsheet_json_posts.yml
@@ -184,12 +192,24 @@ files:
 - 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/fixtures/yaml/comment_seeds.yml
+- spec/fixtures/yaml/post_seeds.yml
+- spec/fixtures/yaml/records_with_namespace.yml
+- spec/fixtures/yaml/records_without_namespace.yml
+- spec/fixtures/yaml/user_seeds.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/harvest/configuration_spec.rb
+- spec/lib/sprig/harvest/model_spec.rb
+- spec/lib/sprig/harvest/record_spec.rb
+- spec/lib/sprig/harvest/seed_file_spec.rb
+- spec/lib/sprig/harvest_spec.rb
 - spec/lib/sprig/null_record_spec.rb
 - spec/lib/sprig/process_notifier_spec.rb
+- spec/lib/sprig/seed/entry_spec.rb
+- spec/lib/sprig/seed/record_spec.rb
 - spec/lib/sprig_spec.rb
 - spec/spec_helper.rb
 - spec/sprig_spec.rb
@@ -206,17 +226,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.2
+rubygems_version: 2.2.1
 signing_key: 
 specification_version: 4
 summary: Relational, environment-specific seeding for Rails apps.
@@ -241,15 +261,28 @@ test_files:
 - 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/fixtures/yaml/comment_seeds.yml
+- spec/fixtures/yaml/post_seeds.yml
+- spec/fixtures/yaml/records_with_namespace.yml
+- spec/fixtures/yaml/records_without_namespace.yml
+- spec/fixtures/yaml/user_seeds.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/harvest/configuration_spec.rb
+- spec/lib/sprig/harvest/model_spec.rb
+- spec/lib/sprig/harvest/record_spec.rb
+- spec/lib/sprig/harvest/seed_file_spec.rb
+- spec/lib/sprig/harvest_spec.rb
 - spec/lib/sprig/null_record_spec.rb
 - spec/lib/sprig/process_notifier_spec.rb
+- spec/lib/sprig/seed/entry_spec.rb
+- spec/lib/sprig/seed/record_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
+has_rdoc: