sprig-reap 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: d9274dc514b03b7804de43aba385e8cc68d954af
+  data.tar.gz: f9c7b62a38c72b9745ef5a982b7ae26fc43ced61
+SHA512:
+  metadata.gz: d0f68d527c8ae07b07a1fb7b6e3f2874343f3b62f4f340383a2dd90c354914d13a821ce2898c6cc364cf8b3b7b4e805b208104465198498cb5e3ee97b750ef6c
+  data.tar.gz: 50f0b95641ecf1de92a336f302b6bd33657e7f7f15db17527ef888aaf7fd28f031a8d63ca60bbea01ae37fbf8eb7a3f33c2e1eb319bc454f1ba589fdfbe8a6a1

data/MIT-LICENSE

@@ -0,0 +1,20 @@
+Copyright 2014 Viget
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,51 @@
+# Sprig::Reap
+
+Automatic capture and output of your application's data state to Sprig-formatted seed files.
+
+## Populate Seed Files from Database
+
+Don't want to write Sprig seed files from scratch?  Well, Sprig::Reap can create them for you!
+
+Via a rake task:
+```
+rake db:seed:reap
+```
+Or from the Rails console:
+```
+Sprig::Reap.reap
+```
+
+By default, Sprig::Reap 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::Reap 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::Reap records? You may specify the environment (`db/seeds`
+target folder) or models (`ActiveRecord::Base.subclasses`-only) you want seed files for.
+
+```
+# Rake Task
+rake db:seed:reap ENV=integration MODELS=User,Post
+
+# Rails Console
+Sprig::Reap.reap(env: 'integration', models: [User, Post])
+```
+
+### Adding to Existing Seed Files (`.yaml` only)
+
+Already have some seed files set up?  No worries!  Sprig::Reap will detect existing seed files and append
+to them with the records from your database with no extra work needed.  Sprig::Reap will automatically
+assign unique `sprig_ids` so you won't have to deal with pesky duplicates.
+
+NOTE: Sprig::Reap 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/Rakefile

@@ -0,0 +1,5 @@
+require 'rspec/core/rake_task'
+
+RSpec::Core::RakeTask.new(:spec)
+
+task :default => :spec

data/lib/sprig/reap/configuration.rb

@@ -0,0 +1,61 @@
+module Sprig::Reap
+  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

data/lib/sprig/reap/model.rb

@@ -0,0 +1,75 @@
+module Sprig::Reap
+  class Model
+    def self.all
+      @@all ||= begin
+        models = Sprig::Reap.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

data/lib/sprig/reap/record.rb

@@ -0,0 +1,59 @@
+module Sprig::Reap
+  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

data/lib/sprig/reap/seed_file.rb

@@ -0,0 +1,52 @@
+module Sprig::Reap
+  class SeedFile
+    DEFAULT_NAMESPACE = 'records'
+
+    attr_reader :model
+
+    def initialize(model)
+      raise ArgumentError, 'Must initialize with a Sprig::Reap::Model' unless model.is_a? Model
+
+      @model = model
+    end
+
+    def path
+      Rails.root.join('db', 'seeds', Sprig::Reap.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

data/lib/sprig/reap/tsortable_hash.rb

@@ -0,0 +1,11 @@
+module Sprig::Reap
+  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/reap/version.rb

@@ -0,0 +1,5 @@
+module Sprig
+  module Reap
+    VERSION = "0.0.1"
+  end
+end

data/lib/sprig/reap.rb

@@ -0,0 +1,35 @@
+require "sprig/reap/version"
+
+module Sprig::Reap
+  autoload :TsortableHash, 'sprig/reap/tsortable_hash'
+  autoload :Railtie,       'sprig/reap/railtie'
+  autoload :Configuration, 'sprig/reap/configuration'
+  autoload :Model,         'sprig/reap/model'
+  autoload :Record,        'sprig/reap/record'
+  autoload :SeedFile,      'sprig/reap/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

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::Reap.reap(ENV)
+    end
+  end
+end

data/spec/fixtures/models/comment.rb

@@ -0,0 +1,5 @@
+class Comment < ActiveRecord::Base
+  belongs_to :post
+
+  validates :post, :presence => true
+end

data/spec/fixtures/models/post.rb

@@ -0,0 +1,2 @@
+class Post < ActiveRecord::Base
+end

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/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/reap/configuration_spec.rb

@@ -0,0 +1,99 @@
+require 'spec_helper'
+
+describe Sprig::Reap::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::Reap::Model]
+          }.to raise_error ArgumentError, 'Cannot create a seed file for Sprig::Reap::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::Reap::Model'
+          }.to raise_error ArgumentError, 'Cannot create a seed file for Sprig::Reap::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/reap/model_spec.rb

@@ -0,0 +1,120 @@
+require 'spec_helper'
+
+describe Sprig::Reap::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::Reap.stub(:classes).and_return([Comment, Post, User])
+    end
+
+    it "returns an dependency-sorted array of Sprig::Reap::Models" do
+      described_class.all.all? { |model| model.is_a? Sprig::Reap::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::Reap::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::Reap::Record with the given id" do
+      reap_record = subject.find(2)
+      reap_record.should be_an_instance_of Sprig::Reap::Record
+      reap_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/reap/record_spec.rb

@@ -0,0 +1,70 @@
+require 'spec_helper'
+
+class Very; end
+
+describe Sprig::Reap::Record do
+  let(:record)       { double('ActiveRecord::Base Instance') }
+  let(:model)        { double('Sprig::Reap::Model') }
+  let(:reap_record)  { double('Sprig::Reap::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::Reap::Model.stub(:find).and_return(reap_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/reap/seed_file_spec.rb

@@ -0,0 +1,109 @@
+require 'spec_helper'
+
+describe Sprig::Reap::SeedFile do
+  let(:model) { Sprig::Reap::Model.new(Comment) }
+
+  subject { described_class.new(model) }
+
+  before do
+    stub_rails_root
+    Sprig::Reap.stub(:env).and_return('dreamland')
+  end
+
+  describe "#initialize" do
+    context "given a non-Sprig::Reap::Model" do
+      it "raises an error" do
+        expect {
+          described_class.new(User)
+        }.to raise_error ArgumentError, 'Must initialize with a Sprig::Reap::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/reap_spec.rb

@@ -0,0 +1,55 @@
+require 'spec_helper'
+
+describe Sprig::Reap do
+  describe ".reap" do
+    let(:seed_file) { double('Sprig::Reap::SeedFile', :write => 'such seeds') }
+
+    before do
+      stub_rails_root
+      stub_rails_env
+      Sprig::Reap::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 => 'dreamland')
+          subject.env.should == 'dreamland'
+        end
+      end
+
+      context "in 'ENV'" do
+        it "sets the environment" do
+          subject.reap('ENV' => ' Dreamland')
+          subject.env.should == 'dreamland'
+        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/spec_helper.rb

@@ -0,0 +1,46 @@
+ENV["RAILS_ENV"] ||= 'test'
+
+require "rails"
+require "active_record"
+require "database_cleaner"
+require "pry"
+require "generator_spec"
+
+require "sprig/reap"
+
+%w(
+  /fixtures/models/*rb
+  /support/**/*.rb
+).each do |file_set|
+  Dir[File.dirname(__FILE__) + file_set].each { |file| require file }
+end
+
+RSpec.configure do |config|
+  config.include RailsStubs
+  config.include FileSetup
+
+  config.before(:suite) do
+    DatabaseCleaner.strategy = :transaction
+    DatabaseCleaner.clean_with(:truncation)
+  end
+
+  config.before(:each) do
+    DatabaseCleaner.start
+  end
+
+  config.after(:each) do
+    DatabaseCleaner.clean
+  end
+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), 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 , 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/support/file_setup.rb

@@ -0,0 +1,10 @@
+module FileSetup
+  # Create and remove seed folder around a spec
+  def setup_seed_folder(path)
+    FileUtils.mkdir_p(path)
+
+    yield
+
+    FileUtils.remove_dir(path)
+  end
+end

data/spec/support/rails_stubs.rb

@@ -0,0 +1,11 @@
+module RailsStubs
+  # Setup fake `Rails.root`
+  def stub_rails_root(path='./spec/fixtures')
+    Rails.stub(:root).and_return(Pathname.new(path))
+  end
+
+  # Setup fake `Rails.env`
+  def stub_rails_env(env='development')
+    Rails.stub(:env).and_return(env)
+  end
+end

metadata

@@ -0,0 +1,174 @@
+--- !ruby/object:Gem::Specification
+name: sprig-reap
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Ryan Stenberg
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-06-10 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: 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-Reap is a gem that allows you to output your application's data
+  state to seed files.
+email:
+- ryan.stenberg@viget.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/sprig/reap/configuration.rb
+- lib/sprig/reap/model.rb
+- lib/sprig/reap/record.rb
+- lib/sprig/reap/seed_file.rb
+- lib/sprig/reap/tsortable_hash.rb
+- lib/sprig/reap/version.rb
+- lib/sprig/reap.rb
+- lib/tasks/reap.rake
+- MIT-LICENSE
+- Rakefile
+- README.md
+- spec/db/activerecord.db
+- spec/fixtures/models/comment.rb
+- spec/fixtures/models/post.rb
+- spec/fixtures/models/user.rb
+- 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/sprig/reap/configuration_spec.rb
+- spec/lib/sprig/reap/model_spec.rb
+- spec/lib/sprig/reap/record_spec.rb
+- spec/lib/sprig/reap/seed_file_spec.rb
+- spec/lib/sprig/reap_spec.rb
+- spec/spec_helper.rb
+- spec/support/file_setup.rb
+- spec/support/rails_stubs.rb
+homepage: http://www.github.com/vigetlabs/sprig-reap
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+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.0.3
+signing_key: 
+specification_version: 4
+summary: Automatic seed file generation for Rails apps using Sprig.
+test_files:
+- spec/db/activerecord.db
+- spec/fixtures/models/comment.rb
+- spec/fixtures/models/post.rb
+- spec/fixtures/models/user.rb
+- 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/sprig/reap/configuration_spec.rb
+- spec/lib/sprig/reap/model_spec.rb
+- spec/lib/sprig/reap/record_spec.rb
+- spec/lib/sprig/reap/seed_file_spec.rb
+- spec/lib/sprig/reap_spec.rb
+- spec/spec_helper.rb
+- spec/support/file_setup.rb
+- spec/support/rails_stubs.rb