Database rewinder with sharding using ar-octopus

database_rewinder.rb

# spec/support/database_rewinder.rb

RSpec.configure do |config|
  config.before(:suite) do
    DatabaseRewinder['test']

    Octopus.config[Rails.env].each do |shard, shard_config|
      DatabaseRewinder.cleaners << DatabaseRewinder::Cleaner.new(
        config:          shard_config,
        connection_name: shard,
      )
    end

    DatabaseRewinder.clean_all
  end

  config.after(:each) do
    DatabaseRewinder.clean
  end
end