require 'spec_helper'

describe Stackster do
  before do
    @config_stub = stub 'Config', :logger => @logger_stub, :access_key => 'key', :secret_key => 'XXX', :region => 'us-west1'
    @response_stub = stub 'Excon::Response', :body => { 
      'RequestId' => 'rid',
      'Domains' => ['domain1', 'domain2'],
      'Items' => { 'item1' => { 'key' => ['value'] } },
      'NextToken' => nil
    }
    @multi_response_stub = stub 'Excon::Response', :body => { 
      'RequestId' => 'rid',
      'Domains' => ['domain1', 'domain2'],
      'Items' => { 'item1-2' => { 'key' => ['value'] } },
      'NextToken' => 'Chunk2'
    }

    @db_mock = mock 'SimpleDB'
    Fog::AWS::SimpleDB.stub(:new).and_return(@db_mock)
    @db_mock.stub(:list_domains).and_return(@response_stub)

    @db = Stackster::AWS::SimpleDB.new(:config => @config_stub)
  end

  describe 'domains' do
    it 'should return a list of domains' do
      @db.domains.should == ['domain1', 'domain2']
    end
  end

  describe 'domain_exists?' do
    it 'should return true for existing domains' do
      @db.domain_exists?('domain1').should be_true
    end

    it 'should return false for non-existent domains' do
      @db.domain_exists?('baddomain1').should_not be_true
    end
  end

  describe 'create_domain' do
    it 'should create a new domain' do
      @db_mock.should_receive(:create_domain).with('newdomain').and_return(@response_stub)

      @db.create_domain('newdomain').body['RequestId'].should == 'rid'
    end

    it 'should not create a duplicate domain' do
      @db_mock.should_not_receive(:create_domain)

      @db.create_domain('domain1').should be_nil
    end
  end

  describe 'put_attributes' do
    it 'should update the specified domain' do
      @db_mock.should_receive(:put_attributes).with('domain1', 'item1', { 'key' => 'value' }, {}).and_return(@response_stub)

      @db.put_attributes('domain1', 'item1', { 'key' => 'value' }, {}).body['RequestId'].should == 'rid'
    end
  end

  describe 'select' do
    it 'should return query items' do
      @db_mock.should_receive(:select).with('item1', { "ConsistentRead" => true, "NextToken" => nil } ).and_return(@response_stub)

      @db.select('item1').should == { 'item1' => { 'key' => ['value'] } }
    end

    it 'should return multiple chunks of query items' do
      @db_mock.should_receive(:select).with('item1', { "ConsistentRead" => true, "NextToken" => nil } ).and_return(@multi_response_stub)
      @db_mock.should_receive(:select).with('item1', { "ConsistentRead" => true, "NextToken" => 'Chunk2' } ).and_return(@response_stub)

      @db.select('item1').should == { 'item1' => { 'key' => ['value'] }, 'item1-2' => { 'key' => ['value'] } }
    end
  end

  describe 'delete' do
    it 'should delete the attributes identified by domain and key' do
      @db_mock.should_receive(:delete_attributes).with('domain1', 'item1').and_return(@response_stub)

      @db.delete('domain1', 'item1').body['RequestId'].should == 'rid'
    end
  end
end