require 'spec_helper'

describe ActiveRecord::Turntable::ConnectionProxy do
  before(:all) do
    reload_turntable!(File.join(File.dirname(__FILE__), "../../config/turntable.yml"))
  end

  context "When initialized" do
    before do
      establish_connection_to("test")
      truncate_shard
    end
    let(:cluster) { ActiveRecord::Turntable::Cluster.new(User, ActiveRecord::Base.turntable_config[:clusters][:user_cluster]) }
    subject { ActiveRecord::Turntable::ConnectionProxy.new(cluster) }
    its(:master_connection) { should == ActiveRecord::Base.connection }
  end

  context "AR3.1" do
    it "should proxies columns" do
      pending "spec not implemented yet"
    end

    it "should proxies columns_hash" do
      pending "spec not implemented yet"
    end
  end

  context "User insert with id" do
    before do
      establish_connection_to("test")
      truncate_shard
      ActiveRecord::Base.logger = Logger.new(STDOUT)
    end

    it "should be saved to user_shard_1 with id = 1" do
      user = User.new
      user.id = 1
      # mock(User.turntable_cluster).select_shard(1) { User.turntable_cluster.shards[:user_shard_1] }
      lambda {
        user.save!
      }.should_not raise_error
    end

    it "should be saved to user_shard_2 with id = 30000" do
      user = User.new
      user.id = 30000
      # mock(User.turntable_cluster).select_shard(30000) { User.turntable_cluster.shards[:user_shard_2] }
      lambda {
        user.save!
      }.should_not raise_error
    end

    it "should be saved to user_shard_2 with id = 30000 with SQL injection attack" do
      user = User.new
      user.id = 30000
      user.nickname = "hogehgoge'00"
      # mock(User.turntable_cluster).select_shard(30000) { User.turntable_cluster.shards[:user_shard_2] }
      lambda {
        user.save!
      }.should_not raise_error
      user.reload

    end

    it "should should be saved the same string when includes escaped string" do
      user = User.new
      user.id = 30000
      user.nickname = "hoge@\n@\\@@\\nhoge\\\nhoge\\n"
      user.save!
      user.reload
      user.nickname.should == "hoge@\n@\\@@\\nhoge\\\nhoge\\n"
    end
  end

  context "When have 2 Users in different shards" do
    before do
      establish_connection_to("test")
      truncate_shard
      @user1 = User.new
      @user1.id = 1
      @user1.save!
      @user2 = User.new
      @user2.id = 30000
      @user2.save!
    end

    it "should be saved to user_shard_1 with id = 1" do
      @user1.nickname = "foobar"
      lambda {
        @user1.save!
      }.should_not raise_error

    end

    it "should be saved to user_shard_2 with id = 30000" do
      @user2.nickname = "hogehoge"
      lambda {
        @user2.save!
      }.should_not raise_error
    end

    it "User.where('id IN (1, 30000)') returns 2 record" do
      User.where(:id => [1, 30000]).all.size.should == 2
    end

    it "count should be 2" do
      User.count.should == 2
    end

    it "User.all returns 2 User object" do
      User.all.size.should == 2
    end
  end

  context "When calling with_all" do
    before do
      establish_connection_to("test")
      truncate_shard
      @user1 = User.new
      @user1.id = 1
      @user1.nickname = 'user1'
      @user1.save!
      @user2 = User.new
      @user2.id = 30000
      @user2.nickname = 'user2'
      @user2.save!
    end

    context "do; User.count; end" do
      subject {
        User.connection.with_all do
          User.count
        end
      }
      it { should have(3).items }

      it "returns User.count of each shards" do
        subject[0].should == 1
        subject[1].should == 1
        subject[2].should == 0
      end
    end

    context "call with true" do
      context "block raises error" do
        subject {
          User.connection.with_all(true) do
            raise "Unko Error"
          end
        }
        it { lambda { subject }.should_not raise_error }
        it { should have(3).items }
        it "collection " do
          subject.each do |s|
            s.should be_instance_of(RuntimeError)
          end
        end
      end
    end
  end

  context "When calling exists? with shard_key" do
    subject { User.exists?(id: 1) }
    it { should be_true }
  end

  context "When calling exists? with non-existed shard_key" do
    subject { User.exists?(id: 3) }
    it { should be_false }
  end

  context "When calling exists? with non shard_key" do
    subject { User.exists?(nickname: 'user2') }
    it { should be_true }
  end

  context "When calling exists? with non-existed non shard_key" do
    subject { User.exists?(nickname: 'user999') }
    it { should be_false }
  end

  context "#table_exists?" do
    subject { User.connection.table_exists?(:users) }
    it { should be_true }
  end
end