database_rewinder 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: db7ed1dc6ce9a93cb75fcbb82baf3e02654f9242
-  data.tar.gz: 5be39f4a597442871bc8e70d1d9fd2a5a4dceaf3
+  metadata.gz: b36e1dc99cdb3e75821ed24db79120cac63d3268
+  data.tar.gz: 8d125b3188fe1613c09b889a3d8570c757ab227b
 SHA512:
-  metadata.gz: 0f8b1ffbc5cc58969aaf7f028e70e1d87f67a6bf48597d2b97b9e80f2ee06fd0d1e1f2f490a7b12860bf7d4c04f58323606f579c5e019c515ae655785829cccd
-  data.tar.gz: ecf4ec4b7450800d551965fe1bf213023b11ed3bbf41ae7be3d9780fc088d5ba9c5b03db7bc235f9a8d6d451e18c376f6e1fdc7b44ba5059fd8ae3e0631178b7
+  metadata.gz: 182d904250271fe18edeec5f9484ea1a8412e99bf558b55e8a5f492afc8735ea74a1e07b335713d88b2f2bacd0b391eeebd6f276da32ae8bc30bcfca1262cc10
+  data.tar.gz: 99cfc24b2918823a43ec6678f95ac59096bb15fd9e5cc3ff385778af24d6a239ad74017cd8309dd9ecd143c21089a2a8a7ddc19a205c7e8da928e37a9dc633c1

data/Rakefile

@@ -1,3 +1,5 @@
+require 'bundler'
+require 'bundler/setup'
 require "bundler/gem_tasks"
 require 'rspec/core'
 require 'rspec/core/rake_task'

data/database_rewinder.gemspec

@@ -4,7 +4,7 @@ $LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
 
 Gem::Specification.new do |spec|
   spec.name          = "database_rewinder"
-  spec.version       = '0.0.2'
+  spec.version       = '0.0.3'
   spec.authors       = ["Akira Matsuda"]
   spec.email         = ["ronnie@dio.jp"]
   spec.description   = "A minimalist's tiny and ultra-fast database cleaner"

data/lib/database_rewinder.rb

@@ -7,7 +7,7 @@ module DatabaseRewinder
   class << self
     def init
       @cleaners, @table_names_cache, @clean_all, @only, @except = [], {}, false
-      @db_config = YAML::load(ERB.new(IO.read(Rails.root.join 'config/database.yml')).result)
+      @db_config = YAML::load(ERB.new(Rails.root.join('config/database.yml').read).result)
     end
 
     def create_cleaner(connection_name)
@@ -71,14 +71,15 @@ module DatabaseRewinder
 
     # for database_cleaner compat
     def start; end
-    def strategy=(_strategy, only: nil, except: nil, **)
-      @only, @except = only, except
+    def strategy=(args)
+      options = args.is_a?(Array) ? args.extract_options! : {}
+      @only, @except = options[:only], options[:except]
     end
 
     # cache AR connection.tables
     def all_table_names(connection)
       db = connection.instance_variable_get(:'@config')[:database]
-      @table_names_cache[db] ||= connection.tables
+      @table_names_cache[db] ||= connection.tables.reject{|t| t == ActiveRecord::Migrator.schema_migrations_table_name }
     end
   end
 end

data/lib/database_rewinder/cleaner.rb

@@ -29,9 +29,10 @@ module DatabaseRewinder
     end
 
     # for database_cleaner compat
-    def strategy=(_strategy, only: nil, except: nil, **)
-      @only += Array(only) unless only.blank?
-      @except += Array(except) unless except.blank?
+    def strategy=(args)
+      options = args.is_a?(Array) ? args.extract_options! : {}
+      @only += Array(options[:only]) unless options[:only].blank?
+      @except += Array(options[:except]) unless options[:except].blank?
     end
 
     private

data/spec/cleaner_spec.rb

@@ -0,0 +1,45 @@
+require 'spec_helper'
+
+module DatabaseRewinder
+  describe Cleaner do
+    describe '#strategy=' do
+      before { @cleaner = described_class.new }
+
+      context 'when only strategy is given' do
+        before { @cleaner.strategy = :truncation }
+
+        it 'should ignore strategy' do
+          expect(@cleaner.instance_variable_get(:@only)).to eq([])
+          expect(@cleaner.instance_variable_get(:@except)).to eq([])
+        end
+      end
+
+      context 'when only option is given' do
+        before { @cleaner.strategy = :truncation, { only: ['foos'] } }
+
+        it 'should set only option' do
+          expect(@cleaner.instance_variable_get(:@only)).to eq(['foos'])
+          expect(@cleaner.instance_variable_get(:@except)).to eq([])
+        end
+      end
+
+      context 'when except option is given' do
+        before { @cleaner.strategy = :truncation, { except: ['foos'] } }
+
+        it 'should set only option' do
+          expect(@cleaner.instance_variable_get(:@only)).to eq([])
+          expect(@cleaner.instance_variable_get(:@except)).to eq(['foos'])
+        end
+      end
+
+      context 'when only and except option are given' do
+        before { @cleaner.strategy = :truncation, { only: ['foos'], except: ['bars'] } }
+
+        it 'should set only option' do
+          expect(@cleaner.instance_variable_get(:@only)).to eq(['foos'])
+          expect(@cleaner.instance_variable_get(:@except)).to eq(['bars'])
+        end
+      end
+    end
+  end
+end

data/spec/database_rewinder_spec.rb

@@ -36,4 +36,52 @@ describe DatabaseRewinder do
       Bar.count.should == 0
     end
   end
+
+  describe '.clean_all' do
+    before do
+      ActiveRecord::SchemaMigration.create_table
+      ActiveRecord::SchemaMigration.create! version: '001'
+      Foo.create! name: 'foo1'
+      DatabaseRewinder.clean_all
+    end
+    after { ActiveRecord::SchemaMigration.drop_table }
+    it 'should clean except schema_migrations' do
+      Foo.count.should == 0
+      ActiveRecord::SchemaMigration.count.should == 1
+    end
+  end
+
+  describe '.strategy=' do
+    context 'when no option is specified' do
+      before { described_class.strategy = :truncate }
+      it 'should set strategy' do
+        expect(described_class.instance_variable_get(:@only)).to be_nil
+        expect(described_class.instance_variable_get(:@except)).to be_nil
+      end
+    end
+
+    context 'when only option is specified' do
+      before { described_class.strategy = :truncate, { only: %w{foos} } }
+      it 'should set only option' do
+        expect(described_class.instance_variable_get(:@only)).to eq(['foos'])
+        expect(described_class.instance_variable_get(:@except)).to be_nil
+      end
+    end
+
+    context 'when except option is specified' do
+      before { described_class.strategy = :truncate, { except: %w{foos} } }
+      it 'should set except option' do
+        expect(described_class.instance_variable_get(:@only)).to be_nil
+        expect(described_class.instance_variable_get(:@except)).to eq(['foos'])
+      end
+    end
+
+    context 'when only and except option are specified' do
+      before { described_class.strategy = :truncate, { only: %w{foos}, except: %w{bars} } }
+      it 'should set only and except options' do
+        expect(described_class.instance_variable_get(:@only)).to eq(['foos'])
+        expect(described_class.instance_variable_get(:@except)).to eq(['bars'])
+      end
+    end
+  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: database_rewinder
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Akira Matsuda
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-09-27 00:00:00.000000000 Z
+date: 2014-03-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -98,6 +98,7 @@ files:
 - lib/database_rewinder/cleaner.rb
 - lib/database_rewinder/railtie.rb
 - spec/active_record_monkey_spec.rb
+- spec/cleaner_spec.rb
 - spec/config/database.yml
 - spec/database_rewinder_spec.rb
 - spec/fake_app.rb
@@ -122,14 +123,14 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.0.3
+rubygems_version: 2.2.1
 signing_key: 
 specification_version: 4
 summary: A minimalist's tiny and ultra-fast database cleaner
 test_files:
 - spec/active_record_monkey_spec.rb
+- spec/cleaner_spec.rb
 - spec/config/database.yml
 - spec/database_rewinder_spec.rb
 - spec/fake_app.rb
 - spec/spec_helper.rb
-has_rdoc: