data_seeder 0.0.1

data/test/models/data_seeder_test.rb

@@ -0,0 +1,147 @@
+require 'test_helper'
+
+describe DataSeeder, :model do
+  describe 'when run with txt files' do
+    before do
+      @name = 'test.txt'
+      @seed_dir = setup_seed_dir(@name, 'countries.txt', 'states.txt')
+    end
+
+    after do
+      cleanup_seed_dir(@name)
+    end
+
+    it 'should load seed files' do
+      modify_seed_file(@name, 'states.txt') do |body|
+        body.sub('KY Kentucky', 'KV Kentucky').sub('VT Vermont', 'VT Vermount')
+      end
+      DataSeeder.run(seed_dir: @seed_dir)
+      assert_equal 50, State.count
+      assert_equal 'United States', Country.find_by(code: 'US').try(:name)
+      assert_equal 'Kentucky', State.find_by(code: 'KV').try(:name)
+      assert_equal 'Vermount', State.find_by(code: 'VT').try(:name)
+      modify_seed_file(@name, 'states.txt') do |body|
+        body.sub('KV Kentucky', 'KY Kentucky').sub('VT Vermount', 'VT Vermont')
+      end
+      DataSeeder.run(seed_dir: @seed_dir)
+      assert_equal 50, State.count
+      assert_equal 'Kentucky', State.find_by(code: 'KY').try(:name)
+      assert_equal 'Vermont', State.find_by(code: 'VT').try(:name)
+      assert_nil   State.find_by(code: 'KV')
+    end
+  end
+
+  %w(csv json yml).each do |loader_type|
+    describe "when run with #{loader_type} files" do
+      before do
+        @name     = "test.#{loader_type}"
+        @file     = "states.#{loader_type}"
+        @seed_dir = setup_seed_dir(@name, @file)
+      end
+
+      after do
+        cleanup_seed_dir(@name)
+      end
+
+      it 'should load seed files' do
+        modify_seed_file(@name, @file) do |body|
+          body.sub('Alaska', 'Alaskaska')
+        end
+        DataSeeder.run(seed_dir: @seed_dir)
+        assert_equal 50, State.count
+        assert_equal 'Alaskaska', State.find_by(code: 'AK').try(:name)
+        modify_seed_file(@name, @file) do |body|
+          body.sub('Alaskaska', 'Alaska')
+        end
+        DataSeeder.run(seed_dir: @seed_dir)
+        assert_equal 50, State.count
+        assert_equal 'Alaska', State.find_by(code: 'AK').try(:name)
+      end
+    end
+  end
+
+  describe 'when run with a custom loader' do
+    before do
+      @name = 'test.custom'
+      @seed_dir = setup_seed_dir(@name, 'states.txt', 'foo.err', 'bar.err')
+    end
+
+    after do
+      cleanup_seed_dir(@name)
+    end
+
+    it 'should load seed files' do
+      modify_seed_file(@name, 'states.txt') do |body|
+        body.sub('KY Kentucky', 'KV Kentucky').sub('VT Vermont', 'VT Vermount')
+      end
+      DataSeeder.run(seed_dir: @seed_dir, loaders: {'err' => AppErrorDataSeeder.new})
+      assert_equal 50, State.count
+      assert_equal 'Kentucky', State.find_by(code: 'KV').try(:name)
+      assert_equal 'Vermount', State.find_by(code: 'VT').try(:name)
+      assert_equal 2, App.count
+      assert App.find_by(name: 'foo')
+      bar = App.find_by(name: 'bar')
+      assert bar
+      assert 3, bar.app_errors.count
+      assert_equal 'Error message for B1', bar.app_errors.find_by(code: 'B1').try(:message)
+
+      modify_seed_file(@name, 'states.txt') do |body|
+        body.sub('KV Kentucky', 'KY Kentucky').sub('VT Vermount', 'VT Vermont')
+      end
+      modify_seed_file(@name, 'bar.err') do |body|
+        body.sub('B1 Error message for B1', 'C1 Error message for C1')
+      end
+      DataSeeder.run(seed_dir: @seed_dir, loaders: {'err' => AppErrorDataSeeder.new})
+      assert_equal 50, State.count
+      assert_equal 'Kentucky', State.find_by(code: 'KY').try(:name)
+      assert_equal 'Vermont', State.find_by(code: 'VT').try(:name)
+      assert_nil   State.find_by(code: 'KV')
+      assert_equal 2, App.count
+      assert App.find_by(name: 'foo')
+      bar = App.find_by(name: 'bar')
+      assert bar
+      assert 3, bar.app_errors.count
+      assert_nil bar.app_errors.find_by(code: 'B1')
+      assert_equal 'Error message for C1', bar.app_errors.find_by(code: 'C1').try(:message)
+
+      FileUtils.cp(Rails.root.join('db', 'seed.test', 'zulu.err'), @seed_dir)
+      DataSeeder.run(seed_dir: @seed_dir, loaders: {'err' => AppErrorDataSeeder.new})
+      assert_equal 50, State.count
+      assert_equal 3, App.count
+      zulu = App.find_by(name: 'zulu')
+      assert zulu
+      assert_equal 2, zulu.app_errors.count
+      assert zulu.app_errors.find_by(code: 'Z1')
+    end
+  end
+
+
+  def setup_seed_dir(name, *files)
+    dir_name = seed_dir_name(name)
+    FileUtils.mkdir_p(dir_name)
+    files.each do |file|
+      FileUtils.cp(Rails.root.join('db', 'seed.test', file), dir_name)
+    end
+    return dir_name
+  end
+
+  def cleanup_seed_dir(name)
+    FileUtils.rm_rf(seed_dir_name(name))
+  end
+
+  def modify_seed_file(name, file, &block)
+    file_name = seed_file_name(name, file)
+    body = File.read(file_name)
+    File.open(file_name, 'w') do |f|
+      f.write yield(body)
+    end
+  end
+
+  def seed_dir_name(name)
+    Rails.root.join('tmp', "db.seed.#{name}.#{$$}")
+  end
+
+  def seed_file_name(name, file)
+    File.join(seed_dir_name(name), file)
+  end
+end

data/test/test_helper.rb

@@ -0,0 +1,12 @@
+ENV["RAILS_ENV"] = "test"
+
+require File.expand_path("../../test/dummy/config/environment.rb",  __FILE__)
+#ActiveRecord::Migrator.migrations_paths = [File.expand_path("../../test/dummy/db/migrate", __FILE__)]
+#ActiveRecord::Migrator.migrations_paths << File.expand_path('../../db/migrate', __FILE__)
+require "rails/test_help"
+require "minitest/rails"
+
+# Filter out Minitest backtrace while allowing backtrace from other libraries
+# to be shown.
+Minitest.backtrace_filter = Minitest::BacktraceFilter.new
+ActiveRecord::Migration.maintain_test_schema!

metadata

@@ -0,0 +1,159 @@
+--- !ruby/object:Gem::Specification
+name: data_seeder
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Brad Pardee
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-03-15 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rails
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: sqlite3
+  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: Provides a simple methodology for seeding your Rails database
+email:
+- bradpardee@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- README.md
+- Rakefile
+- app/models/data_seeder/seed_file.rb
+- db/migrate/20150306195118_create_data_seeder_seed_files.rb
+- lib/data_seeder.rb
+- lib/data_seeder/config.rb
+- lib/data_seeder/engine.rb
+- lib/data_seeder/loader.rb
+- lib/data_seeder/loader/csv.rb
+- lib/data_seeder/loader/json.rb
+- lib/data_seeder/loader/txt.rb
+- lib/data_seeder/loader/yaml.rb
+- lib/data_seeder/logger.rb
+- lib/data_seeder/version.rb
+- test/dummy/Rakefile
+- test/dummy/app/models/app.rb
+- test/dummy/app/models/app_error.rb
+- test/dummy/app/models/app_error_data_seeder.rb
+- test/dummy/app/models/country.rb
+- test/dummy/app/models/state.rb
+- test/dummy/bin/bundle
+- test/dummy/bin/rails
+- test/dummy/bin/rake
+- test/dummy/config.ru
+- test/dummy/config/application.rb
+- test/dummy/config/boot.rb
+- test/dummy/config/database.yml
+- test/dummy/config/environment.rb
+- test/dummy/config/environments/development.rb
+- test/dummy/config/environments/test.rb
+- test/dummy/config/initializers/backtrace_silencers.rb
+- test/dummy/config/initializers/inflections.rb
+- test/dummy/db/migrate/20150313022149_create_countries.rb
+- test/dummy/db/migrate/20150313022228_create_states.rb
+- test/dummy/db/migrate/20150313172634_create_apps.rb
+- test/dummy/db/migrate/20150313172719_create_app_errors.rb
+- test/dummy/db/schema.rb
+- test/dummy/db/seed.test/bar.err
+- test/dummy/db/seed.test/countries.txt
+- test/dummy/db/seed.test/foo.err
+- test/dummy/db/seed.test/states.csv
+- test/dummy/db/seed.test/states.json
+- test/dummy/db/seed.test/states.txt
+- test/dummy/db/seed.test/states.yml
+- test/dummy/db/seed.test/zulu.err
+- test/dummy/db/test.sqlite3
+- test/dummy/log/development.log
+- test/dummy/log/test.log
+- test/models/data_seeder_test.rb
+- test/test_helper.rb
+homepage: http://github.com/bpardee/data_seeder
+licenses:
+- Apache License V2.0
+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.4.5
+signing_key: 
+specification_version: 4
+summary: Seed your Rails database
+test_files:
+- test/dummy/app/models/app.rb
+- test/dummy/app/models/app_error.rb
+- test/dummy/app/models/app_error_data_seeder.rb
+- test/dummy/app/models/country.rb
+- test/dummy/app/models/state.rb
+- test/dummy/bin/bundle
+- test/dummy/bin/rails
+- test/dummy/bin/rake
+- test/dummy/config/application.rb
+- test/dummy/config/boot.rb
+- test/dummy/config/database.yml
+- test/dummy/config/environment.rb
+- test/dummy/config/environments/development.rb
+- test/dummy/config/environments/test.rb
+- test/dummy/config/initializers/backtrace_silencers.rb
+- test/dummy/config/initializers/inflections.rb
+- test/dummy/config.ru
+- test/dummy/db/migrate/20150313022149_create_countries.rb
+- test/dummy/db/migrate/20150313022228_create_states.rb
+- test/dummy/db/migrate/20150313172634_create_apps.rb
+- test/dummy/db/migrate/20150313172719_create_app_errors.rb
+- test/dummy/db/schema.rb
+- test/dummy/db/seed.test/bar.err
+- test/dummy/db/seed.test/countries.txt
+- test/dummy/db/seed.test/foo.err
+- test/dummy/db/seed.test/states.csv
+- test/dummy/db/seed.test/states.json
+- test/dummy/db/seed.test/states.txt
+- test/dummy/db/seed.test/states.yml
+- test/dummy/db/seed.test/zulu.err
+- test/dummy/db/test.sqlite3
+- test/dummy/log/development.log
+- test/dummy/log/test.log
+- test/dummy/Rakefile
+- test/models/data_seeder_test.rb
+- test/test_helper.rb
+has_rdoc: