require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')

describe Octopus::Proxy do
  before(:each) do
    @proxy = Octopus::Proxy.new(Octopus.config())
  end

  describe "creating a new instance" do    
    it "should initialize all shards and groups" do
      @proxy.instance_variable_get(:@shards).keys.to_set.should == [:postgresql_shard, :alone_shard, :aug2011, :canada, :brazil, :aug2009, :russia, :aug2010, :master].to_set
      @proxy.instance_variable_get(:@groups).should == {:country_shards=>[:canada, :brazil, :russia], :history_shards=>[:aug2009, :aug2010, :aug2011]}
    end

    it "should initialize the block attribute as false" do
      @proxy.block.should be_false
    end    
    it "should initialize replicated attribute as false" do
      @proxy.instance_variable_get(:@replicated).should be_false      
    end

    describe "should raise error if you have duplicated shard names" do
      before(:each) do
        Octopus.stub!(:env).and_return("production_raise_error")        
      end

      it "should raise the error" do
        lambda { Octopus::Proxy.new(Octopus.config()) }.should raise_error("You have duplicated shard names!")        
      end
    end
  end

  describe "returning the correct connection" do
    describe "should return the shard name" do
      it "when current_shard is empty" do
        @proxy.shard_name.should == :master        
      end

      it "when current_shard is a single shard" do
        @proxy.current_shard = :canada
        @proxy.shard_name.should == :canada        
      end

      it "when current_shard is more than one shard" do
        @proxy.current_shard = [:russia, :brazil]
        @proxy.shard_name.should == :russia              
      end
    end

    describe "should return the connection based on shard_name" do
      it "when current_shard is empty" do
        @proxy.select_connection().should == @proxy.instance_variable_get(:@shards)[:master].connection()        
      end

      it "when current_shard is a single shard" do
        @proxy.current_shard = :canada
        @proxy.select_connection().should == @proxy.instance_variable_get(:@shards)[:canada].connection()       
      end
    end
  end
end