sprig 0.1.6 → 0.1.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: a11f717b26ca547b63e3a10d77e90e5fb0b34f99
-  data.tar.gz: 875a60cf93f12dda5e3cf04f17dca18c1201f53c
+  metadata.gz: e9deb6a8c1f64f83ed2fbeb9c4c7a7de28a90be5
+  data.tar.gz: 38ab2e9a1eee0cd9a43f5578cd5a049859ae6a21
 SHA512:
-  metadata.gz: 3c5b80e9fbc98aeca33b2f535336a4ac577f4837ecc88b5315351f5cef5582bf67b6fad52a4af4b706c045bdb3120461c1c74b5e118eb177406f66a824e5cf99
-  data.tar.gz: 7660d42b5f0ca71849c63de2abaef6cc4ad324ee08598b00676b4196c80050eeb49264589258e601504ad5bac97514edfe42d5ec0624dbdb87193a903d0db875
+  metadata.gz: b0dcb9273bb4bffb8cef23386f31e23bdfbbb5484a194a1809565514137b95ca6076927526d99d6f83779c280197c1c2d228dd9f94660869cbcaed8996870535
+  data.tar.gz: afbc6277ada442ce7bd424b33abd4b7d38f8e02c7112c0e1281f5fdeac8c3ed0a45533746e04168664c5110875c87eece488d43a7b61835c3732d21d57ccc0be

data/README.md

@@ -32,7 +32,6 @@ Seed files are unique to the environment in which your Rails application is runn
 
 Todo: [Support for shared seed files]
 
-
 ##Seed files
 
 Hang your seed definitions on a `records` key for *yaml* and *json* files.
@@ -74,6 +73,8 @@ records:
 
 Each seed record needs a `sprig_id` defined that must be *unique across all seed files per class*.  It can be an integer, string, whatever you prefer; as long as it is unique, Sprig can sort your seeds for insertion and detect any cyclic relationships.
 
+### Relationships
+
 Create relationships between seed records with the `sprig_record` helper:
 
 ```yaml
@@ -85,6 +86,28 @@ records:
     body: "Yaml Comment body"
 ```
 
+#### Has and Belongs to Many
+For `has_and_belongs_to_many` (HABTM) relationships, you may define relation ids in array format. So if `Post` `has_and_belongs_to_many :tags`, you could write:
+```yaml
+#posts.yml
+
+records:
+  - sprig_id: 1
+    title: 'All About Brains'
+    content: 'Lorem ipsum...'
+    tag_ids:
+      - '<%= sprig_record(Tag, 1).id %>'
+      - '<%= sprig_record(Tag, 2).id %>'
+```
+```yaml
+#tags.yml
+
+records:
+  - sprig_id: 1
+    name: 'Biology'
+  - sprig_id: 2
+    name: 'Neuroscience'
+```
 **Note: For namespaced or STI classes, you'll need to include the namespace with the class name in the seed file name. For example `Users::HeadManager` would need to be `users_head_managers.yml`**
 
 ### Special Options
@@ -124,6 +147,8 @@ records:
 If all your data is in `.wat` files, fear not. You can tell Sprig where to look for your data, and point it toward a custom parser class for turning your data into records. The example below tells Sprig to read `User` seed data from a Google Spreadsheet, and parse it accordingly.
 
 ```ruby
+require 'open-uri'
+
 fanciness = {
   :class  => User,
   :source => open('https://spreadsheets.google.com/feeds/list/somerandomtoken/1/public/values?alt=json'),
@@ -150,7 +175,7 @@ end
 ## Populate Seed Files from Database
 
 Want to create Sprig seed files from the records in your database? Well,
-[Sprig::Reap](http://www.rubygems.org/sprig-reap) can create them for you! Check out the gem's
+[Sprig::Reap](https://rubygems.org/gems/sprig-reap) can create them for you! Check out the gem's
 [README](https://github.com/vigetlabs/sprig-reap#sprigreap) for installation instructions and
 details on usage.
 

data/lib/sprig.rb

@@ -19,6 +19,23 @@ module Sprig
   autoload :Seed,                 'sprig/seed'
 
   class << self
+    attr_writer :adapter
+
+    def adapter
+      @adapter ||= :active_record
+    end
+
+    def adapter_model_class
+      @adapter_model_class ||= case adapter
+      when :active_record
+        ActiveRecord::Base
+      when :mongoid
+        Mongoid::Document
+      else
+        raise "Unknown model class for adapter #{adapter}"
+      end
+    end
+
     def configuration
       @@configuration ||= Sprig::Configuration.new
     end

data/lib/sprig/directive.rb

@@ -7,7 +7,7 @@ module Sprig
         case
         when args.is_a?(Hash)
           args
-        when args.is_a?(Class) && args < ActiveRecord::Base
+        when args.is_a?(Class) && args < Sprig.adapter_model_class
           { :class => args }
         else
           raise ArgumentError, argument_error_message
@@ -30,8 +30,8 @@ module Sprig
     private
 
     def argument_error_message
-      'Sprig::Directive must be instantiated with an '\
-      'ActiveRecord subclass or a Hash with :class defined'
+      'Sprig::Directive must be instantiated with a(n) '\
+      "#{Sprig.adapter_model_class} class or a Hash with :class defined"
     end
   end
 end

data/lib/sprig/seed/factory.rb

@@ -30,7 +30,7 @@ module Sprig
       def klass=(klass)
         raise ArgumentError, 'Must provide a Class as first argument' unless klass.is_a? Class
 
-        klass.reset_column_information
+        klass.reset_column_information if defined?(ActiveRecord) && klass < ActiveRecord::Base
         @klass = klass
       end
 

data/lib/sprig/version.rb

@@ -1,3 +1,3 @@
 module Sprig
-  VERSION = "0.1.6"
+  VERSION = "0.1.7"
 end

data/spec/adapters/active_record.rb

@@ -0,0 +1,34 @@
+require "database_cleaner"
+
+RSpec.configure do |c|
+  c.before(:suite) do
+    DatabaseCleaner[:active_record].strategy = :transaction
+    DatabaseCleaner[:active_record].clean_with(:truncation)
+  end
+
+  c.before(:each) do
+    DatabaseCleaner[:active_record].start
+  end
+
+  c.after(:each) do
+    DatabaseCleaner[:active_record].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 AUTOINCREMENT, title VARCHAR(255), content VARCHAR(255), photo 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));"
+
+Tag.connection.execute "DROP TABLE IF EXISTS tags;"
+Tag.connection.execute "CREATE TABLE tags (id INTEGER PRIMARY KEY , name VARCHAR(255));"
+
+Tag.connection.execute "DROP TABLE IF EXISTS posts_tags;"
+Tag.connection.execute "CREATE TABLE posts_tags (id INTEGER PRIMARY KEY , post_id INTEGER, tag_id INTEGER);"

data/spec/adapters/mongoid.rb

@@ -0,0 +1,19 @@
+require "database_cleaner"
+
+RSpec.configure do |c|
+  c.before(:suite) do
+    DatabaseCleaner[:mongoid].strategy = :truncation
+    DatabaseCleaner[:mongoid].clean_with(:truncation)
+  end
+
+  c.before(:each) do
+    DatabaseCleaner[:mongoid].start
+  end
+
+  c.after(:each) do
+    DatabaseCleaner[:mongoid].clean
+  end
+end
+
+# Datastore
+Mongoid.load!(File.join File.dirname(__FILE__), 'mongoid.yml')

data/spec/adapters/mongoid.yml

@@ -0,0 +1,12 @@
+test:
+  sessions:
+    default:
+      database: sprig_test
+      hosts:
+        - localhost:27017
+      options:
+        read: primary
+        # In the test environment we lower the retries and retry interval to
+        # low amounts for fast failures.
+        max_retries: 1
+        retry_interval: 0

data/spec/fixtures/models/mongoid/comment.rb

@@ -0,0 +1,9 @@
+class Comment
+  include Mongoid::Document
+
+  field :body, :type => String
+
+  belongs_to :post
+
+  validates :post, :presence => true
+end

data/spec/fixtures/models/mongoid/post.rb

@@ -0,0 +1,18 @@
+class Post
+  include Mongoid::Document
+
+  if Mongoid::VERSION.split('.').first == "3"
+    field :user_id,   :type => Moped::BSON::ObjectId
+  else
+    field :user_id,   :type => BSON::ObjectId
+  end
+  field :title,     :type => String
+  field :content,   :type => String
+  field :published, :type => Boolean
+
+  has_and_belongs_to_many :tags
+
+  def photo=(file)
+    write_attribute(:photo, File.basename(file.path))
+  end
+end

data/spec/fixtures/models/mongoid/tag.rb

@@ -0,0 +1,7 @@
+class Tag
+  include Mongoid::Document
+
+  field :name, :type => String
+
+  validates :name, :presence => true
+end

data/spec/fixtures/models/mongoid/user.rb

@@ -0,0 +1,9 @@
+class User
+  include Mongoid::Document
+
+  field :first_name, :type => String
+  field :last_name,  :type => String
+  field :type,       :type => String
+
+  validates :first_name, :last_name, :presence => true
+end

data/spec/lib/generators/sprig/install_generator_spec.rb

@@ -1,7 +1,7 @@
 require 'spec_helper'
 require 'generators/sprig/install_generator'
 
-describe Sprig::Generators::InstallGenerator do
+describe Sprig::Generators::InstallGenerator, type: :generator do
   destination File.expand_path("../../tmp", __FILE__)
 
   before do
@@ -29,7 +29,7 @@ end
 # Generator arguments are set on a class basis. We need to open
 # a new describe block to make these examples work.
 
-describe Sprig::Generators::InstallGenerator do
+describe Sprig::Generators::InstallGenerator, type: :generator do
   context "with arguments" do
     destination File.expand_path("../../tmp", __FILE__)
     arguments %w(development test integration)

data/spec/lib/sprig/directive_spec.rb

@@ -34,8 +34,8 @@ describe Sprig::Directive do
           subject.klass
         }.to raise_error(
           ArgumentError,
-          'Sprig::Directive must be instantiated with an '\
-          'ActiveRecord subclass or a Hash with :class defined'
+          'Sprig::Directive must be instantiated with a(n) '\
+          "#{Sprig.adapter_model_class} class or a Hash with :class defined"
         )
       end
     end

data/spec/lib/sprig_spec.rb

@@ -7,6 +7,57 @@ describe Sprig do
     Sprig::Configuration.stub(:new).and_return(configuration)
   end
 
+  describe '.adapter' do
+    it { expect(described_class).to respond_to(:adapter).with(0).arguments }
+
+    it { expect(described_class.adapter).not_to be nil }
+  end
+
+  describe '.adapter=' do
+    let(:value) { :mongo_mapper }
+
+    around(:each) do |example|
+      previous_value = described_class.adapter
+
+      example.run
+
+      described_class.adapter = previous_value
+    end
+
+    it { expect(described_class).to respond_to(:adapter=).with(1).argument }
+
+    it 'changes the value' do
+      expect { described_class.adapter = value }.to change(described_class, :adapter).to(value)
+    end
+  end
+
+  describe '.adapter_model_class' do
+    let(:expected_class) do
+      case Sprig.adapter
+      when :active_record
+        ActiveRecord::Base
+      when :mongoid
+        Mongoid::Document
+      end
+    end
+
+    before(:each) { described_class.instance_variable_set(:@adapter_model_class, nil) }
+
+    it { expect(described_class).to respond_to(:adapter_model_class).with(0).arguments }
+
+    it { expect(described_class.adapter_model_class).to be expected_class }
+
+    describe 'with an unrecognized adapter' do
+      let(:value) { :mongo_mapper }
+
+      it 'raises an error' do
+        allow(described_class).to receive(:adapter).and_return(value)
+
+        expect { described_class.adapter_model_class }.to raise_error(/unknown model class/i)
+      end
+    end
+  end
+
   describe ".configuration" do
     context "when there is not yet a Configuration instance" do
       it "returns a new Configuration instance" do

data/spec/spec_helper.rb

@@ -7,8 +7,6 @@ SimpleCov.formatter = Coveralls::SimpleCov::Formatter
 SimpleCov.start "rails"
 
 require "rails"
-require "active_record"
-require "database_cleaner"
 require "webmock"
 require "vcr"
 require "pry"
@@ -17,25 +15,13 @@ require "generator_spec"
 require "sprig"
 include Sprig::Helpers
 
-Dir[File.dirname(__FILE__) + '/fixtures/models/*.rb'].each {|file| require file }
 Dir[File.dirname(__FILE__) + '/support/**/*.rb'].each {|file| require file}
 
 RSpec.configure do |c|
   c.include ColoredText
   c.include LoggerMock
 
-  c.before(:suite) do
-    DatabaseCleaner.strategy = :transaction
-    DatabaseCleaner.clean_with(:truncation)
-  end
-
-  c.before(:each) do
-    DatabaseCleaner.start
-  end
-
   c.after(:each) do
-    DatabaseCleaner.clean
-
     Sprig.reset_configuration
   end
 end
@@ -46,23 +32,25 @@ VCR.configure do |c|
   c.hook_into :webmock
 end
 
-# Database
-ActiveRecord::Base.establish_connection(:adapter => "sqlite3", :database => "spec/db/activerecord.db")
+# ActiveRecord (via SQlite3)
+begin
+  require 'active_record'
+  require 'sqlite3'
 
-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));"
+  Sprig.adapter = :active_record
+rescue LoadError; end
 
-Post.connection.execute "DROP TABLE IF EXISTS posts;"
-Post.connection.execute "CREATE TABLE posts (id INTEGER PRIMARY KEY AUTOINCREMENT, title VARCHAR(255), content VARCHAR(255), photo VARCHAR(255), published BOOLEAN , user_id INTEGER);"
+# Mongoid
+begin
+  require 'mongoid'
 
-Comment.connection.execute "DROP TABLE IF EXISTS comments;"
-Comment.connection.execute "CREATE TABLE comments (id INTEGER PRIMARY KEY , post_id INTEGER, body VARCHAR(255));"
+  Sprig.adapter = :mongoid
+rescue LoadError; end
 
-Tag.connection.execute "DROP TABLE IF EXISTS tags;"
-Tag.connection.execute "CREATE TABLE tags (id INTEGER PRIMARY KEY , name VARCHAR(255));"
+# Require model files.
+Dir[File.dirname(__FILE__) + "/fixtures/models/#{Sprig.adapter}/*.rb"].each {|file| require file}
 
-Tag.connection.execute "DROP TABLE IF EXISTS posts_tags;"
-Tag.connection.execute "CREATE TABLE posts_tags (id INTEGER PRIMARY KEY , post_id INTEGER, tag_id INTEGER);"
+require "adapters/#{Sprig.adapter}.rb"
 
 # Helpers
 #

data/spec/sprig_spec.rb

@@ -2,6 +2,14 @@ require 'spec_helper'
 require 'open-uri'
 
 describe "Seeding an application" do
+  let(:missing_record_error) do
+    if defined?(ActiveRecord) && Post < ActiveRecord::Base
+      ActiveRecord::RecordNotFound
+    elsif defined?(Mongoid) && Post < Mongoid::Document
+      Mongoid::Errors::DocumentNotFound
+    end
+  end
+
   before do
     stub_rails_root
   end
@@ -272,7 +280,7 @@ describe "Seeding an application" do
   end
 
   context "with a malformed directive" do
-    let(:expected_error_message) { 'Sprig::Directive must be instantiated with an ActiveRecord subclass or a Hash with :class defined' }
+    let(:expected_error_message) { "Sprig::Directive must be instantiated with a(n) #{Sprig.adapter_model_class} class or a Hash with :class defined" }
 
     context "including a class that is not a subclass of AR" do
       it "raises an argument error" do
@@ -327,7 +335,7 @@ describe "Seeding an application" do
 
           expect {
             existing_match.reload
-          }.to raise_error(ActiveRecord::RecordNotFound)
+          }.to raise_error(missing_record_error)
 
           expect {
             existing_nonmatch.reload
@@ -408,5 +416,33 @@ describe "Seeding an application" do
         end
       end
     end
+
+    context "defined within the directive" do
+      let!(:existing) do
+        Post.create(
+          :title    => "Yaml title",
+          :content  => "Existing content")
+      end
+
+      around do |example|
+        load_seeds('posts.yml', &example)
+      end
+
+      it "respects the directive option" do
+        sprig [
+          {
+            :class   => Post,
+            :source  => open("spec/fixtures/seeds/test/posts.yml"),
+            :delete_existing_by => :title
+          }
+        ]
+
+        Post.count.should == 1
+
+        expect {
+          existing.reload
+        }.to raise_error(missing_record_error)
+      end
+    end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: sprig
 version: !ruby/object:Gem::Version
-  version: 0.1.6
+  version: 0.1.7
 platform: ruby
 authors:
 - Lawson Kurtz
@@ -9,36 +9,42 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-06-27 00:00:00.000000000 Z
+date: 2015-03-20 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
-  name: rails
+  name: appraisal
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ~>
       - !ruby/object:Gem::Version
-        version: '3.1'
+        version: '1.0'
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 1.0.3
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ~>
       - !ruby/object:Gem::Version
-        version: '3.1'
+        version: '1.0'
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 1.0.3
 - !ruby/object:Gem::Dependency
-  name: sqlite3
+  name: rails
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ~>
       - !ruby/object:Gem::Version
-        version: 1.3.8
+        version: '3.1'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ~>
       - !ruby/object:Gem::Version
-        version: 1.3.8
+        version: '3.1'
 - !ruby/object:Gem::Dependency
   name: rspec
   requirement: !ruby/object:Gem::Requirement
@@ -193,16 +199,23 @@ files:
 - lib/sprig/sprig_record_store.rb
 - lib/sprig/tsortable_hash.rb
 - lib/sprig/version.rb
+- spec/adapters/active_record.rb
+- spec/adapters/mongoid.rb
+- spec/adapters/mongoid.yml
 - spec/db/activerecord.db
 - spec/feature/configurations_spec.rb
 - spec/fixtures/cassettes/google_spreadsheet_json_posts.yml
 - spec/fixtures/config/environments/development.rb
 - spec/fixtures/config/environments/production.rb
 - spec/fixtures/config/environments/test.rb
-- spec/fixtures/models/comment.rb
-- spec/fixtures/models/post.rb
-- spec/fixtures/models/tag.rb
-- spec/fixtures/models/user.rb
+- spec/fixtures/models/active_record/comment.rb
+- spec/fixtures/models/active_record/post.rb
+- spec/fixtures/models/active_record/tag.rb
+- spec/fixtures/models/active_record/user.rb
+- spec/fixtures/models/mongoid/comment.rb
+- spec/fixtures/models/mongoid/post.rb
+- spec/fixtures/models/mongoid/tag.rb
+- spec/fixtures/models/mongoid/user.rb
 - spec/fixtures/seeds/staging/posts.yml
 - spec/fixtures/seeds/test/comments.yml
 - spec/fixtures/seeds/test/files/cat.png
@@ -263,16 +276,23 @@ signing_key:
 specification_version: 4
 summary: Relational, environment-specific seeding for Rails apps.
 test_files:
+- spec/adapters/active_record.rb
+- spec/adapters/mongoid.rb
+- spec/adapters/mongoid.yml
 - spec/db/activerecord.db
 - spec/feature/configurations_spec.rb
 - spec/fixtures/cassettes/google_spreadsheet_json_posts.yml
 - spec/fixtures/config/environments/development.rb
 - spec/fixtures/config/environments/production.rb
 - spec/fixtures/config/environments/test.rb
-- spec/fixtures/models/comment.rb
-- spec/fixtures/models/post.rb
-- spec/fixtures/models/tag.rb
-- spec/fixtures/models/user.rb
+- spec/fixtures/models/active_record/comment.rb
+- spec/fixtures/models/active_record/post.rb
+- spec/fixtures/models/active_record/tag.rb
+- spec/fixtures/models/active_record/user.rb
+- spec/fixtures/models/mongoid/comment.rb
+- spec/fixtures/models/mongoid/post.rb
+- spec/fixtures/models/mongoid/tag.rb
+- spec/fixtures/models/mongoid/user.rb
 - spec/fixtures/seeds/staging/posts.yml
 - spec/fixtures/seeds/test/comments.yml
 - spec/fixtures/seeds/test/files/cat.png