superseeder 0.9.2 → 0.9.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 48b9be1f436d7ee497deb42090c6f44a37dbec8b
-  data.tar.gz: fffa66f6b4ff0183a9d0b41aaf3063796d182841
+  metadata.gz: 5ac1535106365fed57579024bed2fffd848749ce
+  data.tar.gz: ce9deb018e56f473cc7ddfbb1c84194064cb05c6
 SHA512:
-  metadata.gz: da5f5ac3f1e05524e921259f063c3d5c74619679263947403da8f4c66affd82f2c34e50ca33cd0450b743b2f23839f299d06750570d69c631f645883638b8a3e
-  data.tar.gz: 6827ee70a77780a5e9e9364affc9a1753ad7ebea13a5fc3a5a50584c08f850dd38c68c4132de82df2fe7fae2b966baf7a0c345facc6cb0c5d3ce369275636f4d
+  metadata.gz: 2474335519980b1a92432b879876e62b34f29b58670758943e12535bbf2e0863c2fbdbc485170fff26e6e5989f11471d07ffbcc2fc2ef627eaee529027de6101
+  data.tar.gz: 941980a27de9ee42dd0d6e0c4a36f9a3a6eaab69b447e25334c66314180d6e1fa18e1867593f403b5e61612ba391554714fcfef8104b8397baf34ef2bd7fb396

data/Rakefile

@@ -14,19 +14,24 @@ RDoc::Task.new(:rdoc) do |rdoc|
   rdoc.rdoc_files.include('lib/**/*.rb')
 end
 
-
-
-
 Bundler::GemHelper.install_tasks
 
 require 'rake/testtask'
 
 Rake::TestTask.new(:test) do |t|
-  t.libs << 'lib'
-  t.libs << 'test'
-  t.pattern = 'test/**/*_test.rb'
-  t.verbose = false
+  if ENV['ADAPTER'].nil?
+    require 'superseeder/adapter'
+    ::Superseeder::Adapter.adapters.each do |adapter|
+      ENV['ADAPTER'] = adapter
+      Rake::Task['test'].execute
+    end
+  else
+    puts "Testing with #{ENV['ADAPTER']} adapter..."
+    t.libs << 'lib'
+    t.libs << 'test'
+    t.pattern = 'test/**/*_test.rb'
+    t.verbose = false
+  end
 end
 
-
 task default: :test

data/lib/superseeder/adapter.rb

@@ -0,0 +1,30 @@
+module Superseeder
+  class Adapter
+
+
+    def self.adapters
+      %w(active_record mongoid)
+    end
+
+    def initialize(instance)
+      @instance = instance
+      if defined? ActiveRecord::Base
+        if instance.kind_of? ActiveRecord::Base
+          require 'superseeder/adapters/active_record'
+          self.singleton_class.include ::Superseeder::Adapters::ActiveRecord
+        end
+      end
+      if defined? Mongoid::Document
+        if instance.kind_of? Mongoid::Document
+          require 'superseeder/adapters/mongoid'
+          self.singleton_class.include ::Superseeder::Adapters::Mongoid
+        end
+      end
+    end
+
+    protected
+
+    attr_reader :instance
+
+  end
+end

data/lib/superseeder/adapters/active_record.rb

@@ -0,0 +1,31 @@
+module Superseeder
+  module Adapters
+    module ActiveRecord
+
+      def each_relation
+        self.instance.singleton_class.reflections.each do |key, val|
+          yield key, self.is_array_relation?(val), val.class_name.constantize
+        end
+      end
+
+      def each_field(row)
+        self.instance.attributes.each do |field, val|
+          next unless row.key? field
+          yield field, false #self.is_array_field?(options) TODO inquiry
+        end
+      end
+
+      protected
+
+      def is_array_field?(options)
+        options.type == Array
+      end
+
+      def is_array_relation?(reflection)
+        [::ActiveRecord::Reflection::HasManyReflection,
+         ::ActiveRecord::Reflection::HasAndBelongsToManyReflection].any?{ |c| reflection.kind_of? c }
+      end
+
+    end
+  end
+end

data/lib/superseeder/adapters/mongoid.rb

@@ -0,0 +1,31 @@
+module Superseeder
+  module Adapters
+    module Mongoid
+
+      def each_relation
+        self.instance.relations.each do |key, val|
+          yield key, self.is_array_relation?(val), val.class_name.constantize
+        end
+      end
+
+      def each_field(row)
+        self.instance.fields.select{ |f, o| row.key? o.options[:as] || f }.each do |field, options|
+          yield options.options[:as] || field, self.is_array_field?(options)
+        end
+      end
+
+      protected
+
+      def is_array_field?(options)
+        options.type == Array
+      end
+
+      def is_array_relation?(options)
+        [::Mongoid::Relations::Referenced::Many,
+         ::Mongoid::Relations::Embedded::Many,
+         ::Mongoid::Relations::Referenced::ManyToMany].include? options[:relation]
+      end
+
+    end
+  end
+end

data/lib/superseeder/formats/roo.rb

@@ -33,29 +33,29 @@ module Superseeder
                          row['_type'].constantize
                        end.new
 
+            require 'superseeder/adapter'
+            adapter = ::Superseeder::Adapter.new instance
+
             # Set relations
-            instance.class.reflections.each do |key, val|
-              attrs = row.select{ |k, _| k =~ /\A#{key}_/ }
+            adapter.each_relation do |name, is_array, class_name|
+              attrs = row.select{ |k, _| k =~ /\A#{name}_/ }
               next if attrs.empty?
-              attrs = attrs.inject({}){ |h, (k, v)| h[k.sub /\A#{key}_/, ''] = v; h }
-              if [Mongoid::Relations::Referenced::Many,
-                  Mongoid::Relations::Embedded::Many,
-                  Mongoid::Relations::Referenced::ManyToMany].include? val[:relation]
+              attrs = attrs.inject({}){ |h, (k, v)| h[k.sub /\A#{name}_/, ''] = v; h }
+              if is_array
                 vals = attrs.map do |k, v|
-                  v.split(many_sep).map{ val.class_name.constantize.find_by k => v } unless v.nil?
+                  v.split(many_sep).map{ |u| class_name.find_by k => u } unless v.nil?
                 end
                 vals.flatten!
                 vals.compact!
-                instance.send "#{key}=", vals
+                instance.send "#{name}=", vals
               else
-                instance.send "#{key}=", val.class_name.constantize.find_by(attrs)
+                instance.send "#{name}=", class_name.find_by(attrs)
               end
             end
 
-            # Set attributes
-            instance.fields.select{ |f, o| row.key? o.options[:as] || f }.each do |f, o|
-              name = o.options[:as] || f
-              val = if o.type == Array
+            # Set fields
+            adapter.each_field(row) do |name, is_array|
+              val = if is_array
                       row[name].try(:split, many_sep)
                     else
                       row[name]
@@ -66,7 +66,7 @@ module Superseeder
             if instance.valid?
               instance.save
             else
-              logger.debug "Skipped #{row} : #{instance.errors.full_messages}"
+              Rails.logger.debug "Skipped #{row} : #{instance.errors.full_messages}"
             end
           end
         end

data/lib/superseeder/seedable.rb

@@ -2,8 +2,12 @@ module Superseeder
   module Seedable
 
     def seed(*args, &block)
-      opts  = args.extract_options!
-      path = Rails.root.join('db', 'seeds', 'data')
+      opts = args.extract_options!
+      path = if opts[:path].blank?
+               Rails.root.join('db', 'seeds', 'data')
+             else
+               Pathname(opts.delete :path)
+             end
       filename = opts.delete :filename
       path = if filename.blank?
                tmp = path.join "#{self.name.underscore.pluralize}*"
@@ -12,18 +16,20 @@ module Superseeder
                raise ArgumentError.new "No candidate files for #{self.name}. Either create a #{tmp} file or specify the filename you want to use with the :filename option." if candidates.length == 0
                candidates.first
              else
-               path.join filename
+               tmp = path.join filename
+               raise ArgumentError.new "#{filename} does not exist." unless File.exist? tmp
+               tmp
              end
       ext = File.extname path
       Dir[Pathname(Gem::Specification.find_by_name('superseeder').gem_dir).join('lib','superseeder','formats','*.rb')].each{ |f| require(f) }
       modules = Superseeder::Formats.constants.select{ |c| Module === Superseeder::Formats.const_get(c) }.map{ |c| Superseeder::Formats.const_get(c) }
       mod = modules.find{ |m| m.send(:extensions).include? ext }
       raise ArgumentError.new "No registered module to support #{ext} extension." if mod.nil?
-      logger.debug "Seeding #{self.name.downcase.pluralize.humanize} from #{path}..."
+      Rails.logger.debug "Seeding #{self.name.downcase.pluralize.humanize} from #{path}..."
       count = self.count
       self.extend mod
       self.__process path, opts, &block
-      logger.debug "Done. (#{self.count - count} entries added)"
+      Rails.logger.debug "Done. (#{self.count - count} entries added)"
     end
   end
 

data/lib/superseeder/version.rb

@@ -1,3 +1,3 @@
 module Superseeder
-  VERSION = '0.9.2'
+  VERSION = '0.9.3'
 end

data/test/adapters/active_record.rb

@@ -0,0 +1,26 @@
+require 'active_record'
+require 'rake'
+
+ActiveRecord::Base.establish_connection :adapter => 'sqlite3', :database => ':memory:'
+
+class Car < ActiveRecord::Base
+
+  belongs_to :parking
+
+  validates :name, :presence => true
+end
+
+class Parking < ActiveRecord::Base
+
+  has_many :cars
+
+  validates :name, :presence => true, :uniqueness => true
+end
+
+load File.dirname(__FILE__) + '/schema.rb'
+#ActiveRecord::Tasks::DatabaseTasks.migrate
+#Rake.load_rakefile 'active_record/railties/databases.rake'
+#require 'active_record/railties/databases.rake'
+#Rake::Task['db:migrate'].invoke
+
+require_relative 'all'

data/test/adapters/all.rb

@@ -0,0 +1,31 @@
+class Whatever
+
+  def self.count
+    0
+  end
+
+end
+
+class Multiple
+
+  def self.count
+    0
+  end
+
+end
+
+class Nofile
+
+  def self.count
+    0
+  end
+
+end
+
+class Noformat
+
+  def self.count
+    0
+  end
+
+end

data/test/adapters/mongoid.rb

@@ -0,0 +1,25 @@
+require 'mongoid'
+
+Mongoid.load! 'test/adapters/mongoid.yml', :test
+Mongoid.raise_not_found_error = false
+
+class Car
+  include Mongoid::Document
+
+  field :name, :type => String
+
+  belongs_to :parking
+
+  validates :name, :presence => true
+end
+
+class Parking
+  include Mongoid::Document
+
+  field :name, :type => String
+  has_many :cars
+
+  validates :name, :presence => true, :uniqueness => true
+end
+
+require_relative 'all'

data/test/adapters/mongoid.yml

@@ -0,0 +1,6 @@
+test:
+  sessions:
+    default:
+      database: superseeder_test
+      hosts:
+        - localhost:27017

data/test/adapters/schema.rb

@@ -0,0 +1,12 @@
+ActiveRecord::Schema.define do
+
+  create_table :parkings do |t|
+    t.string :name
+  end
+
+  create_table :cars do |t|
+    t.string :name
+    t.references :parking
+  end
+
+end

data/test/dummy/config/application.rb

@@ -1,9 +1,23 @@
 require File.expand_path('../boot', __FILE__)
 
-require 'rails/all'
+require 'rails'
+
+%w(
+  action_controller
+  action_view
+  action_mailer
+  active_job
+  rails/test_unit
+  sprockets
+).each do |framework|
+  begin
+    require "#{framework}/railtie"
+  rescue LoadError
+  end
+end
 
 Bundler.require(*Rails.groups)
-require "superseeder"
+require 'superseeder'
 
 module Dummy
   class Application < Rails::Application

data/test/dummy/config/environments/test.rb

@@ -13,7 +13,7 @@ Rails.application.configure do
   config.eager_load = false
 
   # Configure static asset server for tests with Cache-Control for performance.
-  config.serve_static_assets  = true
+  config.serve_static_files  = true
   config.static_cache_control = 'public, max-age=3600'
 
   # Show full error reports and disable caching.
@@ -34,6 +34,8 @@ Rails.application.configure do
   # Print deprecation notices to the stderr.
   config.active_support.deprecation = :stderr
 
+  config.active_support.test_order = :random
+
   # Raises error for missing translations
   # config.action_view.raise_on_missing_translations = true
 end

data/test/dummy/db/seeds/data/cars_with_parkings.csv

@@ -0,0 +1,4 @@
+name,parking_name
+Mustang,North
+Corvette,South
+BMW,South

data/test/dummy/db/seeds/data/cars_without_parkings.csv

@@ -0,0 +1,4 @@
+name
+Mustang
+Corvette
+BMW

data/test/dummy/db/seeds/data/parkings_with_cars.csv

@@ -0,0 +1,5 @@
+name,cars_name
+North,Mustang
+East,
+South,"Corvette,BMW"
+West,

data/test/dummy/db/seeds/data/parkings_without_cars.csv

@@ -0,0 +1,5 @@
+name
+North
+East
+South
+West