seed-fu-ndo 0.0.1

data/Rakefile

@@ -0,0 +1,33 @@
+#!/usr/bin/env rake
+begin
+  require 'bundler/setup'
+rescue LoadError
+  puts 'You must `gem install bundler` and `bundle install` to run rake tasks'
+end
+
+begin
+  require 'rdoc/task'
+rescue LoadError
+  require 'rdoc/rdoc'
+  require 'rake/rdoctask'
+  RDoc::Task = Rake::RDocTask
+end
+
+RDoc::Task.new(:rdoc) do |rdoc|
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title    = 'Seed-Fu-ndo'
+  rdoc.options << '--line-numbers'
+  rdoc.rdoc_files.include('README.rdoc')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end
+
+
+Bundler::GemHelper.install_tasks
+
+require 'rspec/core/rake_task'
+
+RSpec::Core::RakeTask.new do |t|
+end
+
+
+task :default => :spec

data/lib/seed-fu-ndo.rb

@@ -0,0 +1,21 @@
+require 'seed-fu'
+require 'seed-fu-ndo/seeder'
+require 'seed-fu-ndo/railtie' if defined?(Rails) && Rails.version >= "3"
+
+module SeedFuNdo
+  autoload :Recorder, 'seed-fu-ndo/recorder'
+  
+  mattr_reader :recorder
+  
+  # Unload seed data from files. Main entry point for third-party libraries.
+  # [fixture_paths] The paths to look for seed files in.
+  # [filter] A regexp. If given, only filenames matching this expression will be loaded.
+  def self.unseed(fixture_paths = SeedFu.fixture_paths, filter = nil)
+    @@recorder = Recorder.new
+    SeedFu::Runner.new(fixture_paths, filter).run
+    @@recorder.unseed
+  ensure
+    @@recorder = nil
+  end
+  
+end

data/lib/seed-fu-ndo/railtie.rb

@@ -0,0 +1,8 @@
+module SeedFuNdo
+  # Integrate unseed task into Rails.
+  class Railtie < Rails::Railtie
+    rake_tasks do
+      load "tasks/seed_fu_ndo.rake"
+    end
+  end
+end

data/lib/seed-fu-ndo/recorder.rb

@@ -0,0 +1,27 @@
+module SeedFuNdo
+
+  # Records seeder instances created by Seed Fu.
+  class Recorder
+    attr_reader :seeders
+    
+    def initialize
+      @seeders = []
+    end
+    
+    # Record a single seeder.
+    def record(seeder)
+      seeders << seeder
+    end
+    
+    # Unseed the data of all the recorded seeders in reverse order.
+    def unseed
+      ActiveRecord::Base.transaction do 
+        seeders.reverse.each do |seeder|
+          seeder.unseed
+        end
+      end
+    end
+    
+  end
+  
+end

data/lib/seed-fu-ndo/seeder.rb

@@ -0,0 +1,42 @@
+module SeedFu
+  
+  # Enhanced class from Seed Fu to allow for recording and destroying seed data.
+  class Seeder
+    
+    # Record instead of inserting the data if in recording mode. 
+    def seed_with_undo
+      if r = SeedFuNdo.recorder
+        r.record self
+        
+        # return existing records in case they are processed by the caller
+        @data.map { |record_data| find_record(record_data) }
+      else
+        seed_without_undo
+      end
+    end
+    alias_method_chain :seed, :undo
+    
+    # Destroy the seed data.
+    def unseed
+      @model_class.transaction do
+        @data.reverse.map { |record_data| unseed_record(record_data.symbolize_keys) }
+      end
+    end
+    
+    private
+    
+    def unseed_record(data)
+        record = find_record(data)
+        return if record.nil?
+
+        puts " - Remove #{@model_class} #{data.inspect}" unless @options[:quiet]
+
+        record.destroy || raise(ActiveRecord::ActiveRecordError, record.errors.full_messages.join(", ").presence)
+    end
+    
+    def find_record(data)
+      @model_class.where(constraint_conditions(data)).first
+    end
+  end
+  
+end

data/lib/seed-fu-ndo/version.rb

@@ -0,0 +1,4 @@
+module SeedFuNdo
+  # The current version of Seed Fu-ndo
+  VERSION = '0.0.1'
+end

data/lib/tasks/seed_fu_ndo.rake

@@ -0,0 +1,19 @@
+require 'seed-fu-ndo'
+
+namespace :db do
+  desc "Unloads seed data for the current environment."
+  task :unseed_fu => :environment do
+    if ENV["FILTER"]
+      filter = /#{ENV["FILTER"].gsub(/,/, "|")}/
+    end
+
+    if ENV["FIXTURE_PATH"]
+      fixture_paths = [ENV["FIXTURE_PATH"], ENV["FIXTURE_PATH"] + '/' + Rails.env]
+    end
+
+    SeedFuNdo.unseed(fixture_paths, filter)
+  end
+  
+  # alias
+  task :unseed => 'db:unseed_fu'
+end

data/spec/seeder_spec.rb

@@ -0,0 +1,80 @@
+require 'spec_helper'
+
+describe SeedFu::Seeder do
+  
+  context "in regular seed" do
+    it "should run single seeds" do
+      SeededModel.seed(:title) do |s|
+        s.title = 'Foo'
+        s.login = 'Foo'
+        s.first_name = 'Bar'
+      end
+      
+      SeededModel.count.should == 1
+    end
+    
+    it "should run multiple seeds" do
+      SeededModel.seed(:title, 
+        {:title => 'Foo', :login => 'Foo', :first_name => 'Foo'},
+        {:title => 'Bar', :login => 'Bar', :first_name => 'Bar'}
+      )
+      
+      SeededModel.count.should == 2
+    end
+  end
+  
+  context "in unseed" do
+  
+    before do
+      SeedFuNdo.send(:class_variable_set, :@@recorder, SeedFuNdo::Recorder.new)
+    end
+    
+    context "for single seeds" do
+      
+      before do
+        SeededModel.seed(:title) do |s|
+          s.title = 'Foo'
+          s.login = 'Foo'
+          s.first_name = 'Bar'
+        end
+      end
+          
+      it "should only record" do
+        SeededModel.count.should == 0
+      end
+    
+      it "should ignore non-existing records" do
+        SeedFuNdo.recorder.unseed
+        SeededModel.count.should == 0
+      end
+      
+      it "should unseed existing records" do
+        SeededModel.create!(:title => 'Foo')
+        SeedFuNdo.recorder.unseed
+        SeededModel.count.should == 0
+      end
+    end
+    
+    context "for multiple seeds" do
+      
+      before do
+        SeededModel.seed(:title, 
+          {:title => 'Foo', :login => 'Foo', :first_name => 'Foo'},
+          {:title => 'Bar', :login => 'Bar', :first_name => 'Bar'}
+        )
+      end
+          
+      it "should only record" do
+        SeededModel.count.should == 0
+      end
+      
+      it "should rollback if one record cannot be destroyed" do
+        SeededModel.create!(:title => 'Foo', :fail_to_destroy => true)
+        SeededModel.create!(:title => 'Bar')
+        expect { SeedFuNdo.recorder.unseed }.to raise_error(ActiveRecord::ActiveRecordError)
+        SeededModel.count.should == 2
+      end
+    end
+
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,36 @@
+require 'rubygems'
+require 'bundler/setup'
+require 'seed-fu-ndo'
+require 'logger'
+
+SeedFu.quiet = true
+
+ActiveRecord::Base.logger = Logger.new(File.dirname(__FILE__) + "/../debug.log")
+
+ActiveRecord::Base.establish_connection(
+  :adapter => "sqlite3",
+  :database => File.dirname(__FILE__) + "/test.sqlite3"
+)
+
+ActiveRecord::Schema.define :version => 0 do
+  create_table :seeded_models, :force => true do |t|
+    t.column :login, :string
+    t.column :first_name, :string
+    t.column :last_name, :string
+    t.column :title, :string
+    t.column :fail_to_destroy, :boolean, :default => false
+  end
+end
+
+class SeededModel < ActiveRecord::Base
+  validates_presence_of :title
+  attr_protected :first_name
+
+  before_destroy { false if fail_to_destroy }
+end
+
+RSpec.configure do |config|
+  config.before do
+    SeededModel.delete_all
+  end
+end

metadata

@@ -0,0 +1,120 @@
+--- !ruby/object:Gem::Specification 
+name: seed-fu-ndo
+version: !ruby/object:Gem::Version 
+  hash: 29
+  prerelease: 
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Pascal Zumkehr
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2012-06-06 00:00:00 Z
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  version_requirements: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 7
+        segments: 
+        - 2
+        - 2
+        - 0
+        version: 2.2.0
+  prerelease: false
+  requirement: *id001
+  name: seed-fu
+  type: :runtime
+- !ruby/object:Gem::Dependency 
+  version_requirements: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  prerelease: false
+  requirement: *id002
+  name: rspec
+  type: :development
+- !ruby/object:Gem::Dependency 
+  version_requirements: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  prerelease: false
+  requirement: *id003
+  name: sqlite3
+  type: :development
+description: While Seed Fu solves the problem for inserting and maintaining seed data in a database, Seed Fu-ndo adds the possiblity to remove this data again.
+email: 
+- spam@codez.ch
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/tasks/seed_fu_ndo.rake
+- lib/seed-fu-ndo.rb
+- lib/seed-fu-ndo/seeder.rb
+- lib/seed-fu-ndo/recorder.rb
+- lib/seed-fu-ndo/version.rb
+- lib/seed-fu-ndo/railtie.rb
+- Rakefile
+- spec/test.sqlite3
+- spec/seeder_spec.rb
+- spec/spec_helper.rb
+homepage: http://github.com/codez/seed-fu-ndo
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.8.17
+signing_key: 
+specification_version: 3
+summary: Seed Fu-ndo adds undo functionality to Seed Fu
+test_files: 
+- spec/test.sqlite3
+- spec/seeder_spec.rb
+- spec/spec_helper.rb