database_rewinder 0.9.1 → 0.9.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b65d98970a4cf3742c0ea0254aedb60cc3d64260f9b3bbca199bad7880a178aa
-  data.tar.gz: 131690c4430bc96b9fb06d00eaf6474bbe23f9bb038e44f18c00240f5ff9372d
+  metadata.gz: cdfa4a22f839cc429f7334f0c175a8f51a1c398e67cb10ebe831f9ba18761fed
+  data.tar.gz: f7bc66b8f216587fdc6af220a5033d973c6da09c2035a0c63c8a5741b31c5fd4
 SHA512:
-  metadata.gz: 02cdab1001bce6f92944c765d696ee94e6ea2cdddb4ffe9b438a184d62fd0adb8ba9cb2521d1c3a655cfbe52a653ab62f56f3259244cc55ea5f420240aa814be
-  data.tar.gz: 11a8155b683374275680fbb16c03b2d6f5f8af4fa51310e6e05e49b1330f6dbf5ba8793643617085cb6d2765372c61d5ef7a0887639e351554b52492c68e0417
+  metadata.gz: 83b2df863231cef40e644e8a6e567862842bb630cbebd965769a19e365c23dd1877675c548e81150e797b95e8f069bc5b45119c3f7186944d44cd9f889c42ed8
+  data.tar.gz: 34aa414d520248f0989087acb94d1e2c01f861c84baceb6cfa89c8bbb424018374fc75e261d52e18d651e994a25ae62e2ed447215f8cbd81b19ae729e7de301d

data/.travis.yml

@@ -1,47 +1,90 @@
 language: ruby
 
+addons:
+  postgresql: '9.4'
+  mysql: '5.7'
+
 before_install:
-  - gem update --system
-  - gem install bundler
+  # install older versions of rubygems and bundler only on Ruby < 2.7
+  - if [ "${TRAVIS_RUBY_VERSION}" != "ruby-head" ] && [ `echo "${TRAVIS_RUBY_VERSION:0:3} < 2.7" | bc` == 1 ]; then gem i rubygems-update -v '<3' && update_rubygems; fi;
+  - if [ "${TRAVIS_RUBY_VERSION}" != "ruby-head" ] && [ `echo "${TRAVIS_RUBY_VERSION:0:3} < 2.7" | bc` == 1 ]; then gem i bundler -v '<2'; fi;
+  - sudo service mysql restart
+
 
 cache: bundler
 
 env:
-  - DB=sqlite3
-  - DB=mysql
-  - DB=postgresql
+  matrix:
+    - DB=sqlite3
+    - DB=mysql
+    - DB=postgresql
 
 rvm:
-  - 2.5.1
-  - 2.4.4
-  - 2.3.7
-  - 2.2.10
+  - 2.7.3
 
 gemfile:
   - gemfiles/rails_52.gemfile
-  - gemfiles/rails_51.gemfile
-  - gemfiles/rails_50.gemfile
-  - gemfiles/rails_42.gemfile
+  - gemfiles/rails_60.gemfile
+  - gemfiles/rails_61.gemfile
 
 sudo: false
 
+dist: xenial
+
 matrix:
   include:
-    - rvm: 2.5.1
+    - rvm: ruby-head
+      gemfile: gemfiles/rails_edge.gemfile
+      env: DB=postgresql
+    - rvm: ruby-head
+      gemfile: gemfiles/rails_61.gemfile
+      env: DB=postgresql
+    - rvm: 3.0.1
+      gemfile: gemfiles/rails_edge.gemfile
+      env: DB=postgresql
+    - rvm: 3.0.1
+      gemfile: gemfiles/rails_edge.gemfile
+      env: DB=mysql
+    - rvm: 3.0.1
       gemfile: gemfiles/rails_edge.gemfile
       env: DB=sqlite3
-    - rvm: 2.3.7
-      gemfile: gemfiles/rails_41.gemfile
+    - rvm: 3.0.1
+      gemfile: gemfiles/rails_61.gemfile
+      env: DB=postgresql
+    - rvm: 3.0.1
+      gemfile: gemfiles/rails_61.gemfile
+      env: DB=mysql
+    - rvm: 3.0.1
+      gemfile: gemfiles/rails_61.gemfile
       env: DB=sqlite3
-    - rvm: 2.3.7
+    - rvm: 2.6.7
+      gemfile: gemfiles/rails_60.gemfile
+      env: DB=postgresql
+    - rvm: 2.6.7
+      gemfile: gemfiles/rails_52.gemfile
+      env: DB=postgresql
+    - rvm: 2.5.9
+      gemfile: gemfiles/rails_51.gemfile
+      env: DB=postgresql
+    - rvm: 2.4.10
+      gemfile: gemfiles/rails_50.gemfile
+      env: DB=postgresql
+    - rvm: 2.3.8
+      gemfile: gemfiles/rails_41.gemfile
+      env: DB=postgresql
+    - rvm: 2.3.8
       gemfile: gemfiles/rails_40.gemfile
-      env: DB=sqlite3
+      env: DB=postgresql
+    - rvm: 2.2.10
+      gemfile: gemfiles/rails_52.gemfile
+      env: DB=postgresql
     - rvm: 2.1.10
       gemfile: gemfiles/rails_42.gemfile
-      env: DB=sqlite3
+      env: DB=postgresql
     - rvm: 2.0.0
       gemfile: gemfiles/rails_42.gemfile
-      env: DB=sqlite3
+      env: DB=postgresql
 
   allow_failures:
+    - rvm: ruby-head
     - gemfile: gemfiles/rails_edge.gemfile

data/README.md

@@ -23,9 +23,9 @@ This strategy was originally devised and implemented by Shingo Morita (@eudoxa)
 
 ## Supported versions
 
-* ActiveRecord 4.0, 4.1, 4.2, 5.0, 5.1, 5.2, 6.0 (edge)
+* ActiveRecord 4.0, 4.1, 4.2, 5.0, 5.1, 5.2, 6.0, 6.1, 7.0 (edge)
 
-* Ruby 2.0, 2.1, 2.2, 2.3, 2.4, 2.5, 2.6 (trunk)
+* Ruby 2.0, 2.1, 2.2, 2.3, 2.4, 2.5, 2.6, 2.7, 3.0, 3.1 (trunk)
 
 ## Installation
 
@@ -98,7 +98,7 @@ In cases of this situation, bundle database\_rewinder and add the following conf
 
 ```ruby
 RSpec.configure do |config|
-  config.use_transactional_examples = false
+  config.use_transactional_tests = false
 
   ...
 end

data/database_rewinder.gemspec

@@ -6,10 +6,10 @@ $LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
 
 Gem::Specification.new do |spec|
   spec.name          = "database_rewinder"
-  spec.version       = '0.9.1'
+  spec.version       = '0.9.6'
   spec.authors       = ["Akira Matsuda"]
   spec.email         = ["ronnie@dio.jp"]
-  spec.description   = "A minimalist's tiny and ultra-fast database cleaner"
+  spec.description   = "A minimalist's tiny and ultra-fast database cleaner for Active Record"
   spec.summary       = "A minimalist's tiny and ultra-fast database cleaner"
   spec.homepage      = 'https://github.com/amatsuda/database_rewinder'
   spec.license       = "MIT"
@@ -19,7 +19,7 @@ Gem::Specification.new do |spec|
   spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
   spec.require_paths = ["lib"]
 
-  spec.add_development_dependency "bundler", "~> 1.3"
+  spec.add_development_dependency 'bundler'
   spec.add_development_dependency "rake"
   spec.add_development_dependency 'test-unit-rails'
   spec.add_development_dependency 'rails'

data/gemfiles/rails_40.gemfile

@@ -3,6 +3,7 @@ source 'https://rubygems.org'
 gem 'rails', '~> 4.0.0'
 gem 'pg', '~> 0.21'
 gem 'mysql2', '~> 0.3.10'
+gem 'sqlite3', '< 1.4'
 gem 'nokogiri', RUBY_VERSION < '2.1' ? '~> 1.6.0' : '>= 1.7'
 
 gemspec path: '../'

data/gemfiles/rails_41.gemfile

@@ -3,6 +3,7 @@ source 'https://rubygems.org'
 gem 'rails', '~> 4.1.0'
 gem 'pg', '~> 0.21'
 gem 'mysql2', '~> 0.3.13'
+gem 'sqlite3', '< 1.4'
 gem 'nokogiri', RUBY_VERSION < '2.1' ? '~> 1.6.0' : '>= 1.7'
 
 gemspec path: '../'

data/gemfiles/rails_42.gemfile

@@ -3,6 +3,7 @@ source 'https://rubygems.org'
 gem 'rails', '~> 4.2.0'
 gem 'pg', '~> 0.21'
 gem 'mysql2', '~> 0.4.0'
+gem 'sqlite3', '< 1.4'
 gem 'nokogiri', RUBY_VERSION < '2.1' ? '~> 1.6.0' : '>= 1.7'
 
 gemspec path: '../'

data/gemfiles/rails_50.gemfile

@@ -2,5 +2,6 @@ source 'https://rubygems.org'
 
 gem 'rails', '~> 5.0.0'
 gem 'pg', '~> 0.21'
+gem 'sqlite3', '< 1.4'
 
 gemspec path: '../'

data/gemfiles/rails_52.gemfile

@@ -1,6 +1,6 @@
 source 'https://rubygems.org'
 
-gem 'rails', '~> 5.2.0'
+gem 'rails', '~> 5.2.0', git: 'https://github.com/rails/rails', branch: '5-2-stable'
 gem 'pg', '~> 0.21'
 
 gemspec path: '../'

data/gemfiles/rails_60.gemfile

@@ -0,0 +1,6 @@
+source 'https://rubygems.org'
+
+gem 'rails', '~> 6.0.0'
+gem 'selenium-webdriver'
+
+gemspec path: '../'

data/gemfiles/rails_61.gemfile

@@ -0,0 +1,6 @@
+source 'https://rubygems.org'
+
+gem 'rails', '~> 6.1.0'
+gem 'selenium-webdriver'
+
+gemspec path: '../'

data/gemfiles/rails_edge.gemfile

@@ -4,6 +4,7 @@ git_source(:github) do |repo_name|
   "https://github.com/#{repo_name}.git"
 end
 
-gem 'rails', github: 'rails/rails'
+gem 'rails', github: 'rails/rails', branch: 'main'
+gem 'selenium-webdriver'
 
 gemspec path: '../'

data/lib/database_rewinder.rb

@@ -16,7 +16,7 @@ module DatabaseRewinder
     end
 
     def create_cleaner(connection_name)
-      config = database_configuration[connection_name] or raise %Q[Database configuration named "#{connection_name}" is not configured.]
+      config = configuration_hash_for(connection_name) or raise %Q[Database configuration named "#{connection_name}" is not configured.]
 
       Cleaner.new(config: config, connection_name: connection_name, only: @only, except: @except).tap {|c| @cleaners << c}
     end
@@ -48,7 +48,7 @@ module DatabaseRewinder
       end or return
 
       sql.split(';').each do |statement|
-        match = statement.match(/\A\s*INSERT(?:\s+IGNORE)?(?:\s+INTO)?\s+(?:\.*[`"]?([^.\s`"]+)[`"]?)*/i)
+        match = statement.match(/\A\s*INSERT(?:\s+IGNORE)?(?:\s+INTO)?\s+(?:\.*[`"]?([^.\s`"(]+)[`"]?)*/i)
         next unless match
 
         table = match[1]
@@ -73,7 +73,7 @@ module DatabaseRewinder
 
     # cache AR connection.tables
     def all_table_names(connection)
-      cache_key = connection.pool.spec.config
+      cache_key = get_cache_key(connection.pool)
       #NOTE connection.tables warns on AR 5 with some adapters
       tables = ActiveSupport::Deprecation.silence { connection.tables }
       @table_names_cache[cache_key] ||= tables.reject do |t|
@@ -81,7 +81,48 @@ module DatabaseRewinder
         (ActiveRecord::Base.respond_to?(:internal_metadata_table_name) && (t == ActiveRecord::Base.internal_metadata_table_name))
       end
     end
+
+    def get_cache_key(connection_pool)
+      if connection_pool.respond_to?(:db_config) # ActiveRecord >= 6.1
+        connection_pool.db_config.configuration_hash
+      else
+        connection_pool.spec.config
+      end
+    end
+
+    def configuration_hash_for(connection_name)
+      if database_configuration.respond_to?(:configs_for)
+        hash_config = database_configuration_for(connection_name)
+        if hash_config
+          if hash_config.respond_to?(:configuration_hash)
+            hash_config.configuration_hash.stringify_keys
+          else
+            hash_config.config
+          end
+        end
+      else
+        database_configuration[connection_name]
+      end
+    end
+
+    def database_configuration_for(connection_name)
+      traditional_configuration_for(connection_name) || multiple_database_configuration_for(connection_name)
+    end
+
+    def traditional_configuration_for(connection_name)
+      database_configuration.configs_for(env_name: connection_name).first
+    end
+
+    def multiple_database_configuration_for(connection_name)
+      if (ActiveRecord::VERSION::MAJOR >= 6) && (ActiveRecord::VERSION::MINOR >= 1)
+        database_configuration.configs_for(name: connection_name)
+      else
+        database_configuration.configs_for(spec_name: connection_name)
+      end
+    end
   end
+
+  private_class_method :configuration_hash_for, :get_cache_key
 end
 
 begin

data/lib/database_rewinder/active_record_monkey.rb

@@ -2,22 +2,66 @@
 
 module DatabaseRewinder
   module InsertRecorder
-    def execute(sql, *)
-      DatabaseRewinder.record_inserted_table self, sql
-      super
+    module Execute
+      module NoKwargs
+        def execute(sql, *)
+          DatabaseRewinder.record_inserted_table self, sql
+          super
+        end
+      end
+
+      module WithKwargs
+        def execute(sql, *, **)
+          DatabaseRewinder.record_inserted_table self, sql
+          super
+        end
+      end
+    end
+
+    module ExecQuery
+      module NoKwargs
+        def exec_query(sql, *)
+          DatabaseRewinder.record_inserted_table self, sql
+          super
+        end
+      end
+
+      module WithKwargs
+        def exec_query(sql, *, **)
+          DatabaseRewinder.record_inserted_table self, sql
+          super
+        end
+      end
     end
 
-    def exec_query(sql, *)
-      DatabaseRewinder.record_inserted_table self, sql
-      super
+    # This method actually no longer has to be a `prepended` hook because InsertRecorder is a module without a direct method now, but still doing this just for compatibility
+    def self.prepended(mod)
+      if meth = mod.instance_method(:execute)
+        if meth.parameters.any? {|type, _name| [:key, :keyreq, :keyrest].include? type }
+          mod.send :prepend, Execute::WithKwargs
+        else
+          mod.send :prepend, Execute::NoKwargs
+        end
+      end
+      if meth = mod.instance_method(:exec_query)
+        if meth.parameters.any? {|type, _name| [:key, :keyreq, :keyrest].include? type }
+          mod.send :prepend, ExecQuery::WithKwargs
+        else
+          mod.send :prepend, ExecQuery::NoKwargs
+        end
+      end
     end
   end
 end
 
-::ActiveRecord::ConnectionAdapters::SQLite3Adapter.send :prepend, DatabaseRewinder::InsertRecorder if defined? ::ActiveRecord::ConnectionAdapters::SQLite3Adapter
-::ActiveRecord::ConnectionAdapters::PostgreSQLAdapter.send :prepend, DatabaseRewinder::InsertRecorder if defined? ::ActiveRecord::ConnectionAdapters::PostgreSQLAdapter
-::ActiveRecord::ConnectionAdapters::AbstractMysqlAdapter.send :prepend, DatabaseRewinder::InsertRecorder if defined? ::ActiveRecord::ConnectionAdapters::AbstractMysqlAdapter
 
+# Already loaded adapters (SQLite3Adapter, PostgreSQLAdapter, AbstractMysqlAdapter, and possibly another third party adapter)
+::ActiveRecord::ConnectionAdapters::AbstractAdapter.descendants.each do |adapter|
+  # Note: this would only prepend on AbstractMysqlAdapter and not on Mysql2Adapter because ```Mysql2Adapter < InsertRecorder``` becomes true immediately after AbstractMysqlAdapter prepends InsertRecorder
+  adapter.send :prepend, DatabaseRewinder::InsertRecorder unless adapter < DatabaseRewinder::InsertRecorder
+end
+
+# Third party adapters that might be loaded in the future
 def (::ActiveRecord::ConnectionAdapters::AbstractAdapter).inherited(adapter)
-  adapter.prepend DatabaseRewinder::InsertRecorder
+  adapter.send :prepend, DatabaseRewinder::InsertRecorder
 end

data/lib/database_rewinder/compatibility.rb

@@ -2,8 +2,8 @@
 
 module DatabaseRewinder
   module Compatibility
-    def clean_with(*args)
-      cleaners.each {|c| c.clean_with(*args)}
+    def clean_with(*args, **opts)
+      cleaners.each {|c| c.clean_with(*args, **opts)}
     end
 
     def cleaning
@@ -36,7 +36,7 @@ module DatabaseRewinder
     #    DatabaseRewinder[:active_record, connection: 'the_db_name']
     #
     # You can cleanup multiple databases for each test using this configuration.
-    def [](orm, connection: nil, **)
+    def [](orm = nil, connection: nil, **)
       if connection.nil?
         if orm.is_a? String
           connection = orm

data/test/config/database.yml

@@ -1,45 +1,44 @@
-sqlite3: &sqlite3
+<% case ENV['DB']
+   when 'sqlite3' %>
+test:
   adapter: sqlite3
   database: db/database_rewinder_test.sqlite3
   pool: 5
   timeout: 5000
 
-sqlite3_2: &sqlite3_2
+test2:
   adapter: sqlite3
   database: db/database_rewinder_test2.sqlite3
   pool: 5
   timeout: 5000
 
-mysql: &mysql
+<% when 'mysql' %>
+test:
   adapter: mysql2
   host: localhost
   username: root
   password:
   database: database_rewinder_test
 
-mysql_2: &mysql_2
+test2:
   adapter: mysql2
   host: localhost
   username: root
   password:
   database: database_rewinder_test2
 
-postgresql: &postgresql
+<% when 'postgresql' %>
+test:
   adapter: postgresql
   host: localhost
   username: postgres
   password:
   database: database_rewinder_test
 
-postgresql_2: &postgresql_2
+test2:
   adapter: postgresql
   host: localhost
   username: postgres
   password:
   database: database_rewinder_test2
-
-test:
-  <<: *<%= ENV['DB'] %>
-
-test2:
-  <<: *<%= "#{ENV['DB']}_2" %>
+<% end %>

data/test/database_rewinder_test.rb

@@ -3,7 +3,7 @@
 require 'test_helper'
 
 class DatabaseRewinder::DatabaseRewinderTest < ActiveSupport::TestCase
-  if ActiveRecord::VERSION::STRING >= '5'
+  if ActiveRecord::VERSION::MAJOR >= 5
     self.use_transactional_tests = false
   else
     self.use_transactional_fixtures = false
@@ -44,6 +44,54 @@ class DatabaseRewinder::DatabaseRewinderTest < ActiveSupport::TestCase
           DatabaseRewinder[:aho, connection: 'ccc']
         end
       end
+
+      if ActiveRecord::VERSION::MAJOR >= 6
+        sub_test_case 'with traditional configurations' do
+          test 'simply giving a connection name only' do
+            assert_cleaners_added ['aaa'] do
+              DatabaseRewinder.database_configuration = ActiveRecord::DatabaseConfigurations.new({'aaa' => {'adapter' => 'sqlite3', 'database' => ':memory:'}})
+              DatabaseRewinder['aaa']
+            end
+          end
+
+          test 'giving a connection name via Hash with :connection key' do
+            assert_cleaners_added ['bbb'] do
+              DatabaseRewinder.database_configuration = ActiveRecord::DatabaseConfigurations.new({'bbb' => {'adapter' => 'sqlite3', 'database' => ':memory:'}})
+              DatabaseRewinder[connection: 'bbb']
+            end
+          end
+
+          test 'the Cleaner compatible syntax' do
+            assert_cleaners_added ['ccc'] do
+              DatabaseRewinder.database_configuration = ActiveRecord::DatabaseConfigurations.new({'ccc' => {'adapter' => 'sqlite3', 'database' => ':memory:'}})
+              DatabaseRewinder[:aho, connection: 'ccc']
+            end
+          end
+        end
+
+        sub_test_case 'with multiple database configurations' do
+          test 'simply giving a connection name only' do
+            assert_cleaners_added ['aaa'] do
+              DatabaseRewinder.database_configuration = ActiveRecord::DatabaseConfigurations.new({'test' => {'aaa' => {'adapter' => 'sqlite3', 'database' => ':memory:'}}})
+              DatabaseRewinder['aaa']
+            end
+          end
+
+          test 'giving a connection name via Hash with :connection key' do
+            assert_cleaners_added ['bbb'] do
+              DatabaseRewinder.database_configuration = ActiveRecord::DatabaseConfigurations.new({'test' => {'bbb' => {'adapter' => 'sqlite3', 'database' => ':memory:'}}})
+              DatabaseRewinder[connection: 'bbb']
+            end
+          end
+
+          test 'the Cleaner compatible syntax' do
+            assert_cleaners_added ['ccc'] do
+              DatabaseRewinder.database_configuration = ActiveRecord::DatabaseConfigurations.new({'test' => {'ccc' => {'adapter' => 'sqlite3', 'database' => ':memory:'}}})
+              DatabaseRewinder[:aho, connection: 'ccc']
+            end
+          end
+        end
+      end
     end
 
     test 'for connecting to multiple databases' do
@@ -76,6 +124,25 @@ class DatabaseRewinder::DatabaseRewinderTest < ActiveSupport::TestCase
       DatabaseRewinder.database_configuration = nil
     end
 
+    sub_test_case 'via General Active Record insertions' do
+      setup do
+        DatabaseRewinder.cleaners
+        @cleaner = DatabaseRewinder.instance_variable_get(:'@cleaners').detect {|c| c.db == (ENV['DB'] == 'sqlite3' ? 'db/database_rewinder_test.sqlite3' : 'database_rewinder_test')}
+      end
+
+      test 'create' do
+        Bar.create name: 'bar1'
+        assert_equal ['bars'], @cleaner.inserted_tables
+      end
+
+      if ActiveRecord::VERSION::MAJOR >= 6
+        test 'insert_all' do
+          Bar.insert_all! [{name: 'bar1'}]
+          assert_equal ['bars'], @cleaner.inserted_tables
+        end
+      end
+    end
+
     sub_test_case 'common database' do
       test 'include database name' do
         perform_insert 'INSERT INTO "database"."foos" ("name") VALUES (?)'
@@ -95,6 +162,10 @@ class DatabaseRewinder::DatabaseRewinderTest < ActiveSupport::TestCase
         SQL
         assert_equal ['foos'], @cleaner.inserted_tables
       end
+      test 'without spaces between table name and columns list' do
+        perform_insert 'INSERT INTO foos(name) VALUES (?)'
+        assert_equal ['foos'], @cleaner.inserted_tables
+      end
 
       test 'with multi statement query' do
         perform_insert <<-SQL
@@ -136,7 +207,7 @@ class DatabaseRewinder::DatabaseRewinderTest < ActiveSupport::TestCase
     assert_equal 0, Bar.count
   end
 
-  if ActiveRecord::VERSION::STRING >= '4'
+  if ActiveRecord::VERSION::MAJOR >= 4
     sub_test_case 'migrations' do
       test '.clean_all should not touch AR::SchemaMigration' do
         begin
@@ -160,7 +231,7 @@ class DatabaseRewinder::DatabaseRewinderTest < ActiveSupport::TestCase
       @except = @cleaner.instance_variable_get(:@except)
       Foo.create! name: 'foo1'
       Bar.create! name: 'bar1'
-      DatabaseRewinder.clean_with :truncation, options
+      DatabaseRewinder.clean_with :truncation, **options
     end
 
     test 'with only option' do

data/test/fake_app.rb

@@ -17,30 +17,30 @@ load 'active_record/railties/databases.rake'
 
 require 'active_record/base'
 ActiveRecord::Tasks::DatabaseTasks.root ||= Rails.root
-ActiveRecord::Tasks::DatabaseTasks.drop_current ENV['DB']
-ActiveRecord::Tasks::DatabaseTasks.drop_current "#{ENV['DB']}_2"
-ActiveRecord::Tasks::DatabaseTasks.create_current ENV['DB']
-ActiveRecord::Tasks::DatabaseTasks.create_current "#{ENV['DB']}_2"
+ActiveRecord::Tasks::DatabaseTasks.drop_current 'test'
+ActiveRecord::Tasks::DatabaseTasks.drop_current 'test2'
+ActiveRecord::Tasks::DatabaseTasks.create_current 'test'
+ActiveRecord::Tasks::DatabaseTasks.create_current 'test2'
 
 # models
 class Foo < ActiveRecord::Base; end
 class Bar < ActiveRecord::Base; end
 class Baz < ActiveRecord::Base; end
 class Quu < ActiveRecord::Base
-  establish_connection "#{ENV['DB']}_2".to_sym
+  establish_connection :test2
 end
 
 # migrations
 class CreateAllTables < ActiveRecord::VERSION::MAJOR >= 5 ? ActiveRecord::Migration[5.0] : ActiveRecord::Migration
   def self.up
-    ActiveRecord::Base.establish_connection ENV['DB'].to_sym
-    create_table(:bars) {|t| t.string :name }
+    ActiveRecord::Base.establish_connection :test
+    create_table(:bars) {|t| t.string :name; t.index :name, unique: true }
     create_table(:foos) {|t| t.string :name; t.references :bar, foreign_key: true }
     create_table(:bazs) {|t| t.string :name }
 
-    test2_connection = ActiveRecord::Base.establish_connection("#{ENV['DB']}_2".to_sym).connection
+    test2_connection = ActiveRecord::Base.establish_connection(:test2).connection
     test2_connection.create_table(:quus) {|t| t.string :name }
-    ActiveRecord::Base.establish_connection ENV['DB'].to_sym
+    ActiveRecord::Base.establish_connection :test
   end
 
   def self.down
@@ -48,8 +48,8 @@ class CreateAllTables < ActiveRecord::VERSION::MAJOR >= 5 ? ActiveRecord::Migrat
     drop_table(:bars) {|t| t.string :name }
     drop_table(:bazs) {|t| t.string :name }
 
-    test2_connection = ActiveRecord::Base.establish_connection("#{ENV['DB']}_2".to_sym).connection
+    test2_connection = ActiveRecord::Base.establish_connection(:test2).connection
     test2_connection.drop_table :quus
-    ActiveRecord::Base.establish_connection ENV['DB'].to_sym
+    ActiveRecord::Base.establish_connection :test
   end
 end

data/test/test_helper.rb

@@ -10,6 +10,10 @@ require 'active_record'
 require 'database_rewinder'
 require 'fake_app'
 require 'test/unit/rails/test_help'
+begin
+  require 'selenium/webdriver'  # rails 6
+rescue LoadError
+end
 
 migrated = ActiveRecord::Base.connection.respond_to?(:data_source_exists?) ? ActiveRecord::Base.connection.data_source_exists?('foos') : ActiveRecord::Base.connection.table_exists?('foos')
 CreateAllTables.up unless migrated

metadata

@@ -1,29 +1,29 @@
 --- !ruby/object:Gem::Specification
 name: database_rewinder
 version: !ruby/object:Gem::Version
-  version: 0.9.1
+  version: 0.9.6
 platform: ruby
 authors:
 - Akira Matsuda
-autorequire: 
+autorequire:
 bindir: bin
 cert_chain: []
-date: 2018-12-18 00:00:00.000000000 Z
+date: 2021-05-31 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '1.3'
+        version: '0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '1.3'
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: rake
   requirement: !ruby/object:Gem::Requirement
@@ -108,7 +108,7 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
-description: A minimalist's tiny and ultra-fast database cleaner
+description: A minimalist's tiny and ultra-fast database cleaner for Active Record
 email:
 - ronnie@dio.jp
 executables: []
@@ -128,6 +128,8 @@ files:
 - gemfiles/rails_50.gemfile
 - gemfiles/rails_51.gemfile
 - gemfiles/rails_52.gemfile
+- gemfiles/rails_60.gemfile
+- gemfiles/rails_61.gemfile
 - gemfiles/rails_edge.gemfile
 - lib/database_rewinder.rb
 - lib/database_rewinder/active_record_monkey.rb
@@ -147,7 +149,7 @@ homepage: https://github.com/amatsuda/database_rewinder
 licenses:
 - MIT
 metadata: {}
-post_install_message: 
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -162,8 +164,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.0.0.beta3
-signing_key: 
+rubygems_version: 3.2.15
+signing_key:
 specification_version: 4
 summary: A minimalist's tiny and ultra-fast database cleaner
 test_files: