require_relative "test_helper"

class QueryTest < Minitest::Test
  def test_basic
    store_names ["Milk", "Apple"]
    query = Product.search("milk", execute: false)
    query.body[:query] = {match_all: {}}
    assert_equal ["Apple", "Milk"], query.map(&:name).sort
    assert_equal ["Apple", "Milk"], query.execute.map(&:name).sort
  end

  def test_body_options
    store_names ["Milk", "Milk2"]
    assert_search "milk", ["Milk", "Milk2"], body_options: {min_score: 0.0001}
    assert_search_relation ["Milk", "Milk2"], Product.search("milk").body_options(min_score: 0.0001)
  end

  def test_default_timeout
    assert_equal "6s", Product.search("*", execute: false).body[:timeout]
    assert_equal "6s", Product.search("*").body[:timeout]
  end

  def test_timeout_override
    assert_equal "1s", Product.search("*", body_options: {timeout: "1s"}, execute: false).body[:timeout]
    assert_equal "1s", Product.search("*").body_options(timeout: "1s").body[:timeout]
  end

  def test_request_params
    assert_equal "dfs_query_then_fetch", Product.search("*", request_params: {search_type: "dfs_query_then_fetch"}, execute: false).params[:search_type]
    assert_equal "dfs_query_then_fetch", Product.search("*").request_params(search_type: "dfs_query_then_fetch").params[:search_type]
  end

  def test_debug
    store_names ["Milk"]
    out, _ = capture_io do
      assert_search "milk", ["Milk"], debug: true
    end
    refute_includes out, "Error"
  end

  def test_debug_relation
    store_names ["Milk"]
    out, _ = capture_io do
      assert_search_relation ["Milk"], Product.search("milk").debug(true)
    end
    refute_includes out, "Error"
  end
end