activerecord-turntable 2.5.0 → 3.0.0.alpha1

data/spec/active_record/turntable/active_record_ext/association_spec.rb

@@ -1,4 +1,4 @@
-require 'spec_helper'
+require "spec_helper"
 
 describe ActiveRecord::Turntable::ActiveRecordExt::Association do
   before(:all) do
@@ -11,7 +11,7 @@ describe ActiveRecord::Turntable::ActiveRecordExt::Association do
   end
 
   let!(:user) do
-    user = User.new({:nickname => 'user1'})
+    user = User.new({ nickname: "user1" })
     user.id = 1
     user.save
     user
@@ -47,20 +47,11 @@ describe ActiveRecord::Turntable::ActiveRecordExt::Association do
       ActiveRecord::Base.turntable_config.instance_variable_get(:@config)[:raise_on_not_specified_shard_query] = true
     end
     let(:cards_user) { CardsUser.where(user: user).first }
-    let(:cards_users_history) { cards_users_histories.find { |history| history.user_id == user.id } }
 
     context "associated objects has same turntable_key" do
-      context "AssociationRelation#to_a" do
-        subject { cards_user.cards_users_histories.to_a }
-        it { expect { subject }.to_not raise_error }
-        it { is_expected.to include(*cards_users_histories.select { |history| history.cards_user_id == cards_user.id }) }
-      end
-
-      context "AssociationRelation#where" do
-        subject { cards_user.cards_users_histories.where(id: cards_users_history.id).to_a }
-        it { expect { subject }.to_not raise_error }
-        it { is_expected.to include(cards_users_history) }
-      end
+      subject { cards_user.cards_users_histories.to_a }
+      it { expect { subject }.to_not raise_error }
+      it { is_expected.to include(*cards_users_histories.select { |history| history.cards_user_id == cards_user.id }) }
     end
 
     context "associated objects has different turntable_key" do

data/spec/active_record/turntable/active_record_ext/clever_load_spec.rb

@@ -1,5 +1,5 @@
-require 'spec_helper'
-require 'logger'
+require "spec_helper"
+require "logger"
 
 describe ActiveRecord::Turntable::ActiveRecordExt::CleverLoad do
   before(:all) do
@@ -10,14 +10,14 @@ describe ActiveRecord::Turntable::ActiveRecordExt::CleverLoad do
     establish_connection_to(:test)
     truncate_shard
 
-    @user1 = User.new({:nickname => 'user1'})
+    @user1 = User.new({ nickname: "user1" })
     @user1.id = 1
     @user1.save
-    @user1_status = @user1.create_user_status(:hp => 10, :mp => 10)
-    @user2 = User.new({:nickname => 'user2'})
+    @user1_status = @user1.create_user_status(hp: 10, mp: 10)
+    @user2 = User.new({ nickname: "user2" })
     @user2.id = 2
     @user2.save
-    @user2_status = @user2.create_user_status(:hp => 20, :mp => 10)
+    @user2_status = @user2.create_user_status(hp: 20, mp: 10)
   end
 
   context "When a model has has_one relation" do

data/spec/active_record/turntable/active_record_ext/fixture_set_spec.rb

@@ -1,7 +1,7 @@
-require 'spec_helper'
+require "spec_helper"
 
-require 'active_record'
-require 'active_record/turntable/active_record_ext/fixtures'
+require "active_record"
+require "active_record/turntable/active_record_ext/fixtures"
 
 describe ActiveRecord::FixtureSet do
   before(:all) do
@@ -21,7 +21,7 @@ describe ActiveRecord::FixtureSet do
     subject { ActiveRecord::FixtureSet.create_fixtures(fixtures_root, "cards") }
     it { is_expected.to be_instance_of(Array) }
     it "creates card records" do
-      expect {subject}.to change {Card.count}.from(0).to(cards.size)
+      expect { subject }.to change { Card.count }.from(0).to(cards.size)
     end
   end
 end

data/spec/active_record/turntable/active_record_ext/locking_optimistic_spec.rb

@@ -1,4 +1,4 @@
-require 'spec_helper'
+require "spec_helper"
 
 describe ActiveRecord::Turntable::ActiveRecordExt::LockingOptimistic do
   before(:all) do
@@ -10,14 +10,10 @@ describe ActiveRecord::Turntable::ActiveRecordExt::LockingOptimistic do
     truncate_shard
   end
 
-  before(:each) do
+  before do
     ActiveRecord::Base.turntable_config.instance_variable_get(:@config)[:raise_on_not_specified_shard_update] = true
   end
 
-  after(:each) do
-    ActiveRecord::Base.turntable_config.instance_variable_get(:@config)[:raise_on_not_specified_shard_update] = false
-  end
-
   let!(:user_status) do
     user_status = UserStatus.new(user_id: 1)
     user_status.id = 10
@@ -29,13 +25,4 @@ describe ActiveRecord::Turntable::ActiveRecordExt::LockingOptimistic do
     subject { user_status.update_attributes(hp: 20) }
     it { expect { subject }.to change(user_status, :lock_version).by(1) }
   end
-
-  describe "Json serialized column is saved" do
-    before do
-      user_status.update_attributes(data: {foo: 'bar'})
-      user_status.reload
-    end
-    subject { user_status.data }
-    it { expect { subject }.to_not raise_error }
-  end
 end

data/spec/active_record/turntable/active_record_ext/migration_spec.rb

@@ -1,4 +1,4 @@
-require 'spec_helper'
+require "spec_helper"
 
 describe ActiveRecord::Turntable::Migration do
   before(:all) do
@@ -16,11 +16,11 @@ describe ActiveRecord::Turntable::Migration do
     context "With clusters definitions" do
       let(:migration_class) {
         klass = Class.new(ActiveRecord::Migration) {
-        clusters :user_cluster
+          clusters :user_cluster
         }
       }
-      let(:cluster_config) { ActiveRecord::Base.turntable_config["clusters"]["user_cluster"] }
-      let(:user_cluster_shards) { cluster_config["shards"].map { |s| s["connection"] } }
+      let(:cluster_config) { ActiveRecord::Base.turntable_config[:clusters][:user_cluster] }
+      let(:user_cluster_shards) { cluster_config[:shards].map { |s| s[:connection] } }
 
       it { is_expected.to eq(user_cluster_shards) }
     end

data/spec/active_record/turntable/active_record_ext/persistence_spec.rb

@@ -1,5 +1,5 @@
-require 'spec_helper'
-require 'logger'
+require "spec_helper"
+require "logger"
 
 describe ActiveRecord::Turntable::ActiveRecordExt::Persistence do
   before(:all) do
@@ -19,20 +19,20 @@ describe ActiveRecord::Turntable::ActiveRecordExt::Persistence do
   end
 
   let(:user) {
-    u = User.new({:nickname => 'foobar'})
+    u = User.new({ nickname: "foobar" })
     u.id = 1
     u.updated_at = Time.current - 1.day
     u.save
     u
   }
   let(:user_status){
-    stat = user.create_user_status(:hp => 10, :mp => 10)
+    stat = user.create_user_status(hp: 10, mp: 10)
     stat.updated_at = Time.current - 1.day
     stat.save
     stat
   }
   let(:card){
-    Card.create!(:name => 'foobar')
+    Card.create!(name: "foobar")
   }
   let(:cards_user){
     user.cards_users.create(card: card)
@@ -41,7 +41,7 @@ describe ActiveRecord::Turntable::ActiveRecordExt::Persistence do
     context "with blob column" do
       let(:blob_value) { "\123\123\123" }
       let(:user) {
-        u = User.new(:nickname => 'x', :blob => blob_value )
+        u = User.new(nickname: "x", blob: blob_value)
         u.id = 1
         u.save
         u
@@ -121,11 +121,14 @@ describe ActiveRecord::Turntable::ActiveRecordExt::Persistence do
     end
 
     it "should change updated_at when updating" do
-      cards_user.num = 2
+      cards_user
 
-      expect {
-        cards_user.save!
-      }.to change(cards_user, :updated_at)
+      Timecop.travel(1.day.from_now) do
+        expect {
+          cards_user.num = 2
+          cards_user.save!
+        }.to change(cards_user, :updated_at)
+      end
     end
 
     it "should send shard_key condition when destroying" do
@@ -148,7 +151,7 @@ describe ActiveRecord::Turntable::ActiveRecordExt::Persistence do
 
       expect { cards_user.reload }.to_not raise_error
 
-      expect(strio.string.split("\n").select {|stmt| stmt =~ /SELECT/ and stmt !~ /Turntable/ }).to have(1).items
+      expect(strio.string.split("\n").select { |stmt| stmt =~ /SELECT/ and stmt !~ /Turntable/ }).to have(1).items
     end
 
     it "should execute one query when touching" do
@@ -157,7 +160,7 @@ describe ActiveRecord::Turntable::ActiveRecordExt::Persistence do
       ActiveRecord::Base.logger = Logger.new(strio)
 
       expect { cards_user.touch }.to_not raise_error
-      expect(strio.string.split("\n").select {|stmt| stmt =~ /UPDATE/ and stmt !~ /Turntable/ }).to have(1).items
+      expect(strio.string.split("\n").select { |stmt| stmt =~ /UPDATE/ and stmt !~ /Turntable/ }).to have(1).items
     end
 
     it "should execute one query when locking" do
@@ -166,7 +169,7 @@ describe ActiveRecord::Turntable::ActiveRecordExt::Persistence do
       ActiveRecord::Base.logger = Logger.new(strio)
 
       expect { cards_user.lock! }.to_not raise_error
-      expect(strio.string.split("\n").select {|stmt| stmt =~ /SELECT/ and stmt !~ /Turntable/ }).to have(1).items
+      expect(strio.string.split("\n").select { |stmt| stmt =~ /SELECT/ and stmt !~ /Turntable/ }).to have(1).items
     end
 
     it "should execute one query when update_columns" do
@@ -175,7 +178,7 @@ describe ActiveRecord::Turntable::ActiveRecordExt::Persistence do
       ActiveRecord::Base.logger = Logger.new(strio)
 
       expect { cards_user.update_columns(num: 10) }.to_not raise_error
-      expect(strio.string.split("\n").select {|stmt| stmt =~ /UPDATE/ and stmt !~ /Turntable/ }).to have(1).items
+      expect(strio.string.split("\n").select { |stmt| stmt =~ /UPDATE/ and stmt !~ /Turntable/ }).to have(1).items
     end
   end
 
@@ -205,5 +208,4 @@ describe ActiveRecord::Turntable::ActiveRecordExt::Persistence do
     it { is_expected.to be_instance_of(CardsUser) }
     it { is_expected.to eq(cards_user) }
   end
-
 end

data/spec/active_record/turntable/active_record_ext/sequencer_spec.rb

@@ -1,4 +1,4 @@
-require 'spec_helper'
+require "spec_helper"
 
 describe ActiveRecord::Turntable::ActiveRecordExt::Sequencer do
   before(:all) do

data/spec/active_record/turntable/active_record_ext/test_fixtures_spec.rb

@@ -1,7 +1,7 @@
-require 'spec_helper'
+require "spec_helper"
 
-require 'active_record'
-require 'active_record/turntable/active_record_ext/fixtures'
+require "active_record"
+require "active_record/turntable/active_record_ext/fixtures"
 
 describe ActiveRecord::TestFixtures do
   before(:all) do

data/spec/active_record/turntable/algorithm/modulo_algorithm_spec.rb

@@ -1,4 +1,4 @@
-require 'spec_helper'
+require "spec_helper"
 
 describe ActiveRecord::Turntable::Algorithm::ModuloAlgorithm do
   before(:all) do
@@ -27,9 +27,8 @@ describe ActiveRecord::Turntable::Algorithm::ModuloAlgorithm do
 
     context "#calculate with a value that is not a number" do
       it "raises ActiveRecord::Turntable::CannotSpecifyShardError" do
-        expect { @alg.calculate('a') }.to raise_error(ActiveRecord::Turntable::CannotSpecifyShardError)
+        expect { @alg.calculate("a") }.to raise_error(ActiveRecord::Turntable::CannotSpecifyShardError)
       end
     end
   end
-
 end

data/spec/active_record/turntable/algorithm/range_algorithm_spec.rb

@@ -1,4 +1,4 @@
-require 'spec_helper'
+require "spec_helper"
 
 describe ActiveRecord::Turntable::Algorithm::RangeAlgorithm do
   before(:all) do
@@ -27,9 +27,8 @@ describe ActiveRecord::Turntable::Algorithm::RangeAlgorithm do
 
     context "#calculate with 10000000" do
       it "raises ActiveRecord::Turntable::CannotSpecifyShardError" do
-        expect { @alg.calculate(10000000) }.to raise_error(ActiveRecord::Turntable::CannotSpecifyShardError)
+        expect { @alg.calculate(10_000_000) }.to raise_error(ActiveRecord::Turntable::CannotSpecifyShardError)
       end
     end
   end
-
 end

data/spec/active_record/turntable/algorithm/range_bsearch_algorithm_spec.rb

@@ -1,4 +1,4 @@
-require 'spec_helper'
+require "spec_helper"
 
 describe ActiveRecord::Turntable::Algorithm::RangeBsearchAlgorithm do
   before(:all) do
@@ -27,9 +27,8 @@ describe ActiveRecord::Turntable::Algorithm::RangeBsearchAlgorithm do
 
     context "#calculate with 10000000" do
       it "raises ActiveRecord::Turntable::CannotSpecifyShardError" do
-        expect { @alg.calculate(10000000) }.to raise_error(ActiveRecord::Turntable::CannotSpecifyShardError)
+        expect { @alg.calculate(10_000_000) }.to raise_error(ActiveRecord::Turntable::CannotSpecifyShardError)
       end
     end
   end
-
 end

data/spec/active_record/turntable/algorithm_spec.rb

@@ -1,4 +1,4 @@
-require 'spec_helper'
+require "spec_helper"
 
 describe ActiveRecord::Turntable::Algorithm do
   before(:all) do
@@ -31,7 +31,7 @@ describe ActiveRecord::Turntable::Algorithm do
       end
 
       it "called with 10 returns {\"user_shard_1\" => 10}" do
-        expect(algorithm.calculate_used_shards_with_weight(10)).to eq({"user_shard_1" => 10})
+        expect(algorithm.calculate_used_shards_with_weight(10)).to eq({ "user_shard_1" => 10 })
       end
 
       it "called with 65000 returns 2 items" do
@@ -39,7 +39,7 @@ describe ActiveRecord::Turntable::Algorithm do
       end
 
       it "called with 65000 returns {\"user_shard_1\" => 39999, \"user_shard_2\" => 25001}" do
-        expect(algorithm.calculate_used_shards_with_weight(65000)).to eq({"user_shard_1" => 39999, "user_shard_2" => 25001})
+        expect(algorithm.calculate_used_shards_with_weight(65000)).to eq({ "user_shard_1" => 39999, "user_shard_2" => 25001 })
       end
     end
   end
@@ -70,7 +70,7 @@ describe ActiveRecord::Turntable::Algorithm do
       end
 
       it "called with 10 returns {\"user_shard_1\" => 10}" do
-        expect(algorithm.calculate_used_shards_with_weight(10)).to eq({"user_shard_1" => 10})
+        expect(algorithm.calculate_used_shards_with_weight(10)).to eq({ "user_shard_1" => 10 })
       end
 
       it "called with 65000 returns 2 items" do
@@ -78,7 +78,7 @@ describe ActiveRecord::Turntable::Algorithm do
       end
 
       it "called with 65000 returns {\"user_shard_1\" => 39999, \"user_shard_2\" => 25001}" do
-        expect(algorithm.calculate_used_shards_with_weight(65000)).to eq({"user_shard_1" => 39999, "user_shard_2" => 25001})
+        expect(algorithm.calculate_used_shards_with_weight(65000)).to eq({ "user_shard_1" => 39999, "user_shard_2" => 25001 })
       end
     end
   end

data/spec/active_record/turntable/base_spec.rb

@@ -1,4 +1,4 @@
-require 'spec_helper'
+require "spec_helper"
 
 describe ActiveRecord::Turntable::Base do
   before(:all) do

data/spec/active_record/turntable/cluster_spec.rb

@@ -1,4 +1,4 @@
-require 'spec_helper'
+require "spec_helper"
 
 describe ActiveRecord::Turntable::Cluster do
   before(:all) do

data/spec/active_record/turntable/config_spec.rb

@@ -1,4 +1,4 @@
-require 'spec_helper'
+require "spec_helper"
 
 describe ActiveRecord::Turntable::Config do
   before(:all) do

data/spec/active_record/turntable/connection_proxy_spec.rb

@@ -1,4 +1,4 @@
-require 'spec_helper'
+require "spec_helper"
 
 describe ActiveRecord::Turntable::ConnectionProxy do
   before(:all) do
@@ -91,7 +91,6 @@ describe ActiveRecord::Turntable::ConnectionProxy do
       expect {
         @user1.save!
       }.not_to raise_error
-
     end
 
     it "should be saved to user_shard_2 with id = 30000" do
@@ -102,7 +101,7 @@ describe ActiveRecord::Turntable::ConnectionProxy do
     end
 
     it "User.where('id IN (1, 30000)') returns 2 record" do
-      expect(User.where(:id => [1, 30000]).all.size).to eq(2)
+      expect(User.where(id: [1, 30000]).all.size).to eq(2)
     end
 
     it "count should be 2" do
@@ -120,11 +119,11 @@ describe ActiveRecord::Turntable::ConnectionProxy do
       truncate_shard
       @user1 = User.new
       @user1.id = 1
-      @user1.nickname = 'user1'
+      @user1.nickname = "user1"
       @user1.save!
       @user2 = User.new
       @user2.id = 30000
-      @user2.nickname = 'user2'
+      @user2.nickname = "user2"
       @user2.save!
     end
 
@@ -167,11 +166,11 @@ describe ActiveRecord::Turntable::ConnectionProxy do
       truncate_shard
       @user1 = User.new
       @user1.id = 1
-      @user1.nickname = 'user1'
+      @user1.nickname = "user1"
       @user1.save!
       @user2 = User.new
       @user2.id = 30000
-      @user2.nickname = 'user2'
+      @user2.nickname = "user2"
       @user2.save!
     end
 
@@ -185,11 +184,11 @@ describe ActiveRecord::Turntable::ConnectionProxy do
       truncate_shard
       @user1 = User.new
       @user1.id = 1
-      @user1.nickname = 'user1'
+      @user1.nickname = "user1"
       @user1.save!
       @user2 = User.new
       @user2.id = 30000
-      @user2.nickname = 'user2'
+      @user2.nickname = "user2"
       @user2.save!
     end
 
@@ -203,15 +202,15 @@ describe ActiveRecord::Turntable::ConnectionProxy do
       truncate_shard
       @user1 = User.new
       @user1.id = 1
-      @user1.nickname = 'user1'
+      @user1.nickname = "user1"
       @user1.save!
       @user2 = User.new
       @user2.id = 30000
-      @user2.nickname = 'user2'
+      @user2.nickname = "user2"
       @user2.save!
     end
 
-    subject { User.exists?(nickname: 'user2') }
+    subject { User.exists?(nickname: "user2") }
     it { is_expected.to be_truthy }
   end
 
@@ -221,15 +220,15 @@ describe ActiveRecord::Turntable::ConnectionProxy do
       truncate_shard
       @user1 = User.new
       @user1.id = 1
-      @user1.nickname = 'user1'
+      @user1.nickname = "user1"
       @user1.save!
       @user2 = User.new
       @user2.id = 30000
-      @user2.nickname = 'user2'
+      @user2.nickname = "user2"
       @user2.save!
     end
 
-    subject { User.exists?(nickname: 'user999') }
+    subject { User.exists?(nickname: "user999") }
     it { is_expected.to be_falsey }
   end
 
@@ -239,11 +238,11 @@ describe ActiveRecord::Turntable::ConnectionProxy do
       truncate_shard
       @user1 = User.new
       @user1.id = 1
-      @user1.nickname = 'user1'
+      @user1.nickname = "user1"
       @user1.save!
       @user2 = User.new
       @user2.id = 30000
-      @user2.nickname = 'user2'
+      @user2.nickname = "user2"
       @user2.save!
     end
 

data/spec/active_record/turntable/finder_spec.rb

@@ -1,4 +1,4 @@
-require 'spec_helper'
+require "spec_helper"
 
 describe "ActiveRecord::FinderMethods" do
   before(:all) do

data/spec/active_record/turntable/mixer/fader_spec.rb

@@ -1,4 +1,4 @@
-require 'spec_helper'
+require "spec_helper"
 
 describe ActiveRecord::Turntable::Mixer::Fader do
 end

data/spec/active_record/turntable/mixer_spec.rb

@@ -1,4 +1,4 @@
-require 'spec_helper'
+require "spec_helper"
 
 describe ActiveRecord::Turntable::Mixer do
   before(:all) do
@@ -85,7 +85,7 @@ describe ActiveRecord::Turntable::Mixer do
         }
 
         it { is_expected.to be_instance_of Array }
-        it { is_expected.to eq([1,2,3,4,5]) }
+        it { is_expected.to eq([1, 2, 3, 4, 5]) }
       end
 
       context "When call find_shard_keys with not determine shardkey condition" do
@@ -108,7 +108,5 @@ describe ActiveRecord::Turntable::Mixer do
         it { is_expected.to eq([]) }
       end
     end
-
   end
-
 end

data/spec/active_record/turntable/query_cache_spec.rb

@@ -0,0 +1,28 @@
+require "spec_helper"
+require "active_support/executor"
+
+describe ActiveRecord::Turntable::QueryCache do
+  before(:all) do
+    reload_turntable!(File.join(File.dirname(__FILE__), "../../config/turntable.yml"))
+  end
+
+  before do
+    establish_connection_to(:test)
+    truncate_shard
+  end
+
+  let(:mw) {
+    executor = Class.new(ActiveSupport::Executor)
+    ActiveRecord::Turntable::QueryCache.install_executor_hooks executor
+    lambda { |env|
+      executor.wrap {
+        [200, {}, nil]
+      }
+    }
+  }
+  subject { mw.call({}) }
+
+  it "should returns 200 response" do
+    expect(subject.first).to eq(200)
+  end
+end

data/spec/active_record/turntable/sequencer/api_spec.rb

@@ -1,4 +1,4 @@
-require 'spec_helper'
+require "spec_helper"
 
 describe ActiveRecord::Turntable::Sequencer::Api do
   before(:all) do
@@ -10,7 +10,7 @@ describe ActiveRecord::Turntable::Sequencer::Api do
   let(:klass) { Class.new }
   let(:api_host) { "example.example" }
   let(:api_port) { 80 }
-  let(:options) { { "api_host" => api_host, "api_port" => api_port } }
+  let(:options) { { api_host: api_host, api_port: api_port }.with_indifferent_access }
   let(:api_response) { 1024 }
 
   let(:next_sequence_uri) { "http://#{api_host}/sequences/#{sequence_name}/new" }
@@ -18,7 +18,7 @@ describe ActiveRecord::Turntable::Sequencer::Api do
 
   describe "#next_sequence_value" do
     before do
-      stub_request(:get, next_sequence_uri).to_return(:body => api_response.to_s)
+      stub_request(:get, next_sequence_uri).to_return(body: api_response.to_s)
     end
 
     subject { sequencer.next_sequence_value(sequence_name) }
@@ -28,7 +28,7 @@ describe ActiveRecord::Turntable::Sequencer::Api do
 
   describe "#current_sequence_value" do
     before do
-      stub_request(:get, current_sequence_uri).to_return(:body => api_response.to_s)
+      stub_request(:get, current_sequence_uri).to_return(body: api_response.to_s)
     end
 
     subject { sequencer.current_sequence_value(sequence_name) }

data/spec/active_record/turntable/sequencer/barrage_spec.rb

@@ -1,4 +1,4 @@
-require 'spec_helper'
+require "spec_helper"
 
 describe ActiveRecord::Turntable::Sequencer::Barrage do
   before(:all) do
@@ -7,7 +7,7 @@ describe ActiveRecord::Turntable::Sequencer::Barrage do
 
   let(:sequencer) { ActiveRecord::Turntable::Sequencer::Barrage.new(klass, options) }
   let(:sequence_name) { "hogefuga" }
-  let(:options) { { "options" => { "generators" => [ {"name" => "sequence", "length" => 16} ] } } }
+  let(:options) { { options: { generators: [{ name: "sequence", length: 16 }] } }.with_indifferent_access }
   let(:klass) { Class.new }
 
   describe "#next_sequence_value" do

data/spec/active_record/turntable/sequencer/mysql_spec.rb

@@ -1,4 +1,4 @@
-require 'spec_helper'
+require "spec_helper"
 
 describe ActiveRecord::Turntable::Sequencer::Mysql do
   before(:all) do

data/spec/active_record/turntable/shard_spec.rb

@@ -1,4 +1,4 @@
-require 'spec_helper'
+require "spec_helper"
 
 describe ActiveRecord::Turntable::Shard do
   before(:all) do

data/spec/active_record/turntable/sql_tree_patch_spec.rb

@@ -1,5 +1,5 @@
-require 'spec_helper'
-require 'active_record/turntable/sql_tree_patch'
+require "spec_helper"
+require "active_record/turntable/sql_tree_patch"
 
 describe SQLTree do
   before(:all) do

data/spec/active_record/turntable/transaction_spec.rb

@@ -1,4 +1,4 @@
-require 'spec_helper'
+require "spec_helper"
 
 describe "transaction" do
   before(:all) do
@@ -13,7 +13,7 @@ describe "transaction" do
 
   describe "all_cluster_transaction" do
     let(:all_clusters) { clusters.values }
-    let(:shards) { all_clusters.map { |c| c.shards.values }.flatten(1) }
+    let(:shards) { all_clusters.flat_map { |c| c.shards.values } }
 
     it "all shards should begin transaction" do
       User.all_cluster_transaction {