require 'test_helper'
require 'active_record'

module Elasticsearch
  module Model
    class ActiveRecordImportIntegrationTest < Elasticsearch::Test::IntegrationTestCase
      context "ActiveRecord importing" do
        setup do
          ActiveRecord::Schema.define(:version => 1) do
            create_table :import_articles do |t|
              t.string   :title
              t.integer  :views
              t.string   :numeric # For the sake of invalid data sent to Elasticsearch
              t.datetime :created_at, :default => 'NOW()'
            end
          end

          class ::ImportArticle < ActiveRecord::Base
            include Elasticsearch::Model

            scope :popular, -> { where('views >= 50') }

            mapping do
              indexes :title,      type: 'string'
              indexes :views,      type: 'integer'
              indexes :numeric,    type: 'integer'
              indexes :created_at, type: 'date'
            end
          end

          ImportArticle.delete_all
          ImportArticle.__elasticsearch__.create_index! force: true
          ImportArticle.__elasticsearch__.client.cluster.health wait_for_status: 'yellow'

          100.times { |i| ImportArticle.create! title: "Test #{i}", views: i }
        end

        should "import all the documents" do
          assert_equal 100, ImportArticle.count

          ImportArticle.__elasticsearch__.refresh_index!
          assert_equal 0, ImportArticle.search('*').results.total

          batches = 0
          errors  = ImportArticle.import(batch_size: 10) do |response|
            batches += 1
          end

          assert_equal 0, errors
          assert_equal 10, batches

          ImportArticle.__elasticsearch__.refresh_index!
          assert_equal 100, ImportArticle.search('*').results.total
        end

        should "import only documents from a specific scope" do
          assert_equal 100, ImportArticle.count

          assert_equal 0, ImportArticle.import(scope: 'popular')

          ImportArticle.__elasticsearch__.refresh_index!
          assert_equal 50, ImportArticle.search('*').results.total
        end

        should "import only documents from a specific query" do
          assert_equal 100, ImportArticle.count

          assert_equal 0, ImportArticle.import(query: -> { where('views >= 30') })

          ImportArticle.__elasticsearch__.refresh_index!
          assert_equal 70, ImportArticle.search('*').results.total
        end

        should "report and not store/index invalid documents" do
          ImportArticle.create! title: "Test INVALID", numeric: "INVALID"

          assert_equal 101, ImportArticle.count

          ImportArticle.__elasticsearch__.refresh_index!
          assert_equal 0, ImportArticle.search('*').results.total

          batches = 0
          errors  = ImportArticle.__elasticsearch__.import(batch_size: 10) do |response|
            batches += 1
          end

          assert_equal 1, errors
          assert_equal 11, batches

          ImportArticle.__elasticsearch__.refresh_index!
          assert_equal 100, ImportArticle.search('*').results.total
        end

        should "transform documents with the option" do
          assert_equal 100, ImportArticle.count

          assert_equal 0, ImportArticle.import( transform: ->(a) {{ index: { data: { name: a.title, foo: 'BAR' } }}} )

          ImportArticle.__elasticsearch__.refresh_index!
          assert_contains ImportArticle.search('*').results.first._source.keys, 'name'
          assert_contains ImportArticle.search('*').results.first._source.keys, 'foo'
          assert_equal 100, ImportArticle.search('test').results.total
          assert_equal 100, ImportArticle.search('bar').results.total
        end
      end

    end
  end
end