require 'spec_helper'

describe Waistband::Index do

  let(:index)   { Waistband::Index.new('events') }
  let(:index2)  { Waistband::Index.new('search') }
  let(:attrs)   { {'ok' => {'yeah' => true}} }

  it "initializes values" do
    expect(index.instance_variable_get('@stringify')).to eql true
  end

  it "creates the index" do
    index.delete!
    expect{ index.refresh }.to raise_error(Elasticsearch::Transport::Transport::Errors::NotFound)

    index.create!
    expect{ index.refresh }.to_not raise_error
  end

  it "deletes the index" do
    index.delete!
    expect{ index.refresh }.to raise_error(Elasticsearch::Transport::Transport::Errors::NotFound)
    index.create!
  end

  it "blows up when trying to delete an index that does not exist" do
    index.delete!
    expect { index.delete! }.to raise_error(::Waistband::Errors::IndexNotFound)
  end

  it "updates the index's mappings" do
    index.refresh
    response = index.update_mapping('event')
    expect(response['acknowledged']).to be_true
  end

  it "updates the index's settings" do
    index.refresh
    response = index.update_settings
    expect(response['acknowledged']).to be_true
  end

  it "proxies to the client's search" do
    result = index.search({})
    expect(result).to be_a Waistband::SearchResults
    expect(result.took).to be_present
    expect(result.hits).to be_an Array
  end

  describe "storing" do

    it "stores data" do
      expect(index.save('__test_write', {'ok' => 'yeah'})).to be_true
      expect(index.read('__test_write')).to eql({
        '_id' => '__test_write',
        '_index' => 'events_test',
        '_source' => {'ok' => 'yeah'},
        '_type' => 'event',
        '_version' => 1,
        'found' => true
      })
    end

    it "data is stringified" do
      index.save('__test_write', attrs)
      expect(index.read('__test_write')[:_source]).to eql({"ok"=>"{\"yeah\"=>true}"})
    end

    it "data is indirectly accessible when not stringified" do
      index2.save('__test_not_string', attrs)
      expect(index2.read('__test_not_string')[:_source][:ok][:yeah]).to eql true
    end

    it "deletes data" do
      index.save('__test_write', attrs)
      index.destroy('__test_write')
      expect(index.read('__test_write')).to be_nil
    end

    it "returns nil on 404" do
      expect(index.read('__not_here')).to be_nil
    end

    it "blows up on 404 when using the bang method" do
      expect {
        index.read!('__not_here')
      }.to raise_error(Elasticsearch::Transport::Transport::Errors::NotFound)
    end

    it "doesn't mix data between two indexes" do
      index.save('__test_write',  {'data' => 'index_1'})
      index2.save('__test_write', {'data' => 'index_2'})

      expect(index.read('__test_write')[:_source]).to   eql({'data' => 'index_1'})
      expect(index2.read('__test_write')[:_source]).to  eql({'data' => 'index_2'})
    end

  end

  describe 'paginating search results' do

    it "permits paginating" do
      results = index.search(page: 5, page_size: 15)
      expect(results.instance_variable_get('@page')).to eql 5
      expect(results.instance_variable_get('@page_size')).to eql 15
    end

    it "permits passing in page_size without page" do
      results = index.search(page_size: 999)
      expect(results.instance_variable_get('@page')).to eql 1
      expect(results.instance_variable_get('@page_size')).to eql 999
    end

  end

  describe 'subindexes' do

    let(:sharded_index) { Waistband::Index.new('events', subs: %w(2013 01)) }

    it "permits subbing the index" do
      expect(sharded_index.send(:config_name)).to eql 'events_test__2013_01'
    end

    it "permits sharding into singles" do
      index = Waistband::Index.new 'events', subs: '2013'
      expect(index.send(:config_name)).to eql 'events_test__2013'
    end

    it "retains the same options from the parent index config" do
      config = sharded_index.send(:config)

      expect(sharded_index.send(:base_config_name)).to eql 'events_test'
      expect(config['stringify']).to be_true
      expect(config['settings']).to be_present
    end

    it "creates the sharded index with the same mappings as the parent" do
      sharded_index.delete!

      expect {
        sharded_index.create!
      }.to_not raise_error
    end

    describe 'no configged index name' do

      it "gets a default name that makes sense for the index when not defined" do
        index = Waistband::Index.new 'events_no_name', subs: %w(2013 01)
        expect(index.send(:config_name)).to eql 'events_no_name_test__2013_01'
      end

    end

  end

  describe '#base_config_name' do

    it "gets a default name that makes sense for the index when not defined" do
      index = Waistband::Index.new 'events_no_name'
      expect(index.send(:base_config_name)).to eql 'events_no_name_test'
    end

  end

  describe 'aliases' do

    it "returns the alias name with the env" do
      expect(index.send(:full_alias_name, 'all_events')).to eql 'all_events_test'
    end

    it "if the index has a custom name, the alias name doesn't automatically append the env" do
      index.stub(:config).and_return({
        'name' => 'super_custom'
      })
      expect(index.send(:full_alias_name, 'all_events')).to eql 'all_events'
    end

    it "creates aliases" do
      expect(index.alias_exists?('events_alias_yo')).to be_false
      index.alias 'events_alias_yo'
      expect(index.alias_exists?('events_alias_yo')).to be_true
    end

  end

end