inner_hits_test.go

package es_test

import (
	"testing"

	ScriptLanguage "github.com/Trendyol/es-query-builder/es/enums/script-language"
	Order "github.com/Trendyol/es-query-builder/es/enums/sort/order"

	"github.com/Trendyol/es-query-builder/es"
	"github.com/Trendyol/es-query-builder/test/assert"
)

////   Inner Hits   ////

func Test_InnerHits_should_exist_on_es_package(t *testing.T) {
	t.Parallel()
	// Given When Then
	assert.NotNil(t, es.InnerHits)
}

func Test_InnerHits_method_should_create_innerHitsType(t *testing.T) {
	t.Parallel()
	// Given
	ih := es.InnerHits()

	// Then
	assert.NotNil(t, ih)
	assert.IsTypeString(t, "es.innerHitsType", ih)
}

func Test_InnerHits_should_have_Explain_method(t *testing.T) {
	t.Parallel()
	// Given
	ih := es.InnerHits()

	// When Then
	assert.NotNil(t, ih.Explain)
}

func Test_InnerHits_Explain_should_create_json_with_explain_field_inside_inner_hits(t *testing.T) {
	t.Parallel()
	// Given
	ih := es.InnerHits().Explain(true)
	// When Then
	assert.NotNil(t, ih)
	bodyJSON := assert.MarshalWithoutError(t, ih)
	assert.Equal(t, "{\"explain\":true}", bodyJSON)
}

func Test_InnerHits_should_have_Fields_method(t *testing.T) {
	t.Parallel()
	// Given
	ih := es.InnerHits()

	// When Then
	assert.NotNil(t, ih.Fields)
}

func Test_InnerHits_Fields_should_create_json_with_fields_field_inside_inner_hits(t *testing.T) {
	t.Parallel()
	// Given
	ih := es.InnerHits().Fields("f1", "g2", "h3")
	// When Then
	assert.NotNil(t, ih)
	bodyJSON := assert.MarshalWithoutError(t, ih)
	assert.Equal(t, "{\"fields\":[\"f1\",\"g2\",\"h3\"]}", bodyJSON)
}

func Test_InnerHits_should_have_Size_method(t *testing.T) {
	t.Parallel()
	// Given
	ih := es.InnerHits()

	// When Then
	assert.NotNil(t, ih.Size)
}

func Test_InnerHits_Size_should_create_json_with_size_field_inside_inner_hits(t *testing.T) {
	t.Parallel()
	// Given
	ih := es.InnerHits().Size(100)
	// When Then
	assert.NotNil(t, ih)
	bodyJSON := assert.MarshalWithoutError(t, ih)
	assert.Equal(t, "{\"size\":100}", bodyJSON)
}

func Test_InnerHits_should_have_From_method(t *testing.T) {
	t.Parallel()
	// Given
	ih := es.InnerHits()

	// When Then
	assert.NotNil(t, ih.From)
}

func Test_InnerHits_From_should_create_json_with_from_field_inside_inner_hits(t *testing.T) {
	t.Parallel()
	// Given
	ih := es.InnerHits().From(5_000)
	// When Then
	assert.NotNil(t, ih)
	bodyJSON := assert.MarshalWithoutError(t, ih)
	assert.Equal(t, "{\"from\":5000}", bodyJSON)
}

func Test_InnerHits_should_have_Name_method(t *testing.T) {
	t.Parallel()
	// Given
	ih := es.InnerHits()

	// When Then
	assert.NotNil(t, ih.Name)
}

func Test_InnerHits_Name_should_create_json_with_name_field_inside_inner_hits(t *testing.T) {
	t.Parallel()
	// Given
	ih := es.InnerHits().Name("Göksel")
	// When Then
	assert.NotNil(t, ih)
	bodyJSON := assert.MarshalWithoutError(t, ih)
	assert.Equal(t, "{\"name\":\"Göksel\"}", bodyJSON)
}

func Test_InnerHits_should_have_SourceFalse_method(t *testing.T) {
	t.Parallel()
	// Given
	ih := es.InnerHits()

	// When Then
	assert.NotNil(t, ih.SourceFalse)
}

func Test_InnerHits_SourceFalse_should_create_json_with_source_field_inside_inner_hits_with_false(t *testing.T) {
	t.Parallel()
	// Given
	ih := es.InnerHits().SourceFalse()
	// When Then
	assert.NotNil(t, ih)
	bodyJSON := assert.MarshalWithoutError(t, ih)
	assert.Equal(t, "{\"_source\":false}", bodyJSON)
}

func Test_InnerHits_should_have_SourceIncludes_method(t *testing.T) {
	t.Parallel()
	// Given
	ih := es.InnerHits()

	// When Then
	assert.NotNil(t, ih.SourceIncludes)
}

func Test_InnerHits_SourceIncludes_should_create_json_with_source_field_inside_inner_hits(t *testing.T) {
	t.Parallel()
	// Given
	ih := es.InnerHits().SourceIncludes("id", "name", "description")
	// When Then
	assert.NotNil(t, ih)
	bodyJSON := assert.MarshalWithoutError(t, ih)
	assert.Equal(t, "{\"_source\":{\"includes\":[\"id\",\"name\",\"description\"]}}", bodyJSON)
}

func Test_InnerHits_SourceIncludes_should_not_create_json_with_source_field_inside_inner_hits_when_empty(t *testing.T) {
	t.Parallel()
	// Given
	ih := es.InnerHits().SourceIncludes()
	// When Then
	assert.NotNil(t, ih)
	bodyJSON := assert.MarshalWithoutError(t, ih)
	assert.Equal(t, "{}", bodyJSON)
}

func Test_InnerHits_should_have_SourceExcludes_method(t *testing.T) {
	t.Parallel()
	// Given
	ih := es.InnerHits()

	// When Then
	assert.NotNil(t, ih.SourceExcludes)
}

func Test_InnerHits_SourceExcludes_should_create_json_with_source_field_inside_inner_hits(t *testing.T) {
	t.Parallel()
	// Given
	ih := es.InnerHits().SourceExcludes("secret", "key", "partition")
	// When Then
	assert.NotNil(t, ih)
	bodyJSON := assert.MarshalWithoutError(t, ih)
	assert.Equal(t, "{\"_source\":{\"excludes\":[\"secret\",\"key\",\"partition\"]}}", bodyJSON)
}

func Test_InnerHits_SourceExcludes_should_not_create_json_with_source_field_inside_inner_hits_when_empty(t *testing.T) {
	t.Parallel()
	// Given
	ih := es.InnerHits().SourceExcludes()
	// When Then
	assert.NotNil(t, ih)
	bodyJSON := assert.MarshalWithoutError(t, ih)
	assert.Equal(t, "{}", bodyJSON)
}

func Test_InnerHits_should_have_StoredFields_method(t *testing.T) {
	t.Parallel()
	// Given
	ih := es.InnerHits()

	// When Then
	assert.NotNil(t, ih.StoredFields)
}

func Test_InnerHits_StoredFields_should_create_json_with_stored_fields_field_inside_inner_hits(t *testing.T) {
	t.Parallel()
	// Given
	ih := es.InnerHits().StoredFields("x1", "y2", "z3")
	// When Then
	assert.NotNil(t, ih)
	bodyJSON := assert.MarshalWithoutError(t, ih)
	assert.Equal(t, "{\"stored_fields\":[\"x1\",\"y2\",\"z3\"]}", bodyJSON)
}

func Test_InnerHits_should_have_TrackScores_method(t *testing.T) {
	t.Parallel()
	// Given
	ih := es.InnerHits()

	// When Then
	assert.NotNil(t, ih.TrackScores)
}

func Test_InnerHits_TrackScores_should_create_json_with_track_scores_field_inside_inner_hits(t *testing.T) {
	t.Parallel()
	// Given
	ih := es.InnerHits().TrackScores(false)
	// When Then
	assert.NotNil(t, ih)
	bodyJSON := assert.MarshalWithoutError(t, ih)
	assert.Equal(t, "{\"track_scores\":false}", bodyJSON)
}

func Test_InnerHits_should_have_Version_method(t *testing.T) {
	t.Parallel()
	// Given
	ih := es.InnerHits()

	// When Then
	assert.NotNil(t, ih.Version)
}

func Test_InnerHits_Version_should_create_json_with_version_field_inside_inner_hits(t *testing.T) {
	t.Parallel()
	// Given
	ih := es.InnerHits().Version(true)
	// When Then
	assert.NotNil(t, ih)
	bodyJSON := assert.MarshalWithoutError(t, ih)
	assert.Equal(t, "{\"version\":true}", bodyJSON)
}

func Test_InnerHits_should_have_IgnoreUnmapped_method(t *testing.T) {
	t.Parallel()
	// Given
	ih := es.InnerHits()

	// When Then
	assert.NotNil(t, ih.IgnoreUnmapped)
}

func Test_InnerHits_IgnoreUnmapped_should_create_json_with_ignore_unmapped_field_inside_inner_hits(t *testing.T) {
	t.Parallel()
	// Given
	ih := es.InnerHits().IgnoreUnmapped(true)
	// When Then
	assert.NotNil(t, ih)
	bodyJSON := assert.MarshalWithoutError(t, ih)
	assert.Equal(t, "{\"ignore_unmapped\":true}", bodyJSON)
}

func Test_InnerHits_should_have_SeqNoPrimaryTerm_method(t *testing.T) {
	t.Parallel()
	// Given
	ih := es.InnerHits()

	// When Then
	assert.NotNil(t, ih.SeqNoPrimaryTerm)
}

func Test_InnerHits_SeqNoPrimaryTerm_should_create_json_with_seq_no_primary_term_field_inside_inner_hits(t *testing.T) {
	t.Parallel()
	// Given
	ih := es.InnerHits().SeqNoPrimaryTerm(false)
	// When Then
	assert.NotNil(t, ih)
	bodyJSON := assert.MarshalWithoutError(t, ih)
	assert.Equal(t, "{\"seq_no_primary_term\":false}", bodyJSON)
}

func Test_InnerHits_should_have_Sort_method(t *testing.T) {
	t.Parallel()
	// Given
	ih := es.InnerHits()

	// When Then
	assert.NotNil(t, ih.Sort)
}

func Test_InnerHits_Sort_should_create_json_with_sort_field_inside_inner_hits(t *testing.T) {
	t.Parallel()
	// Given
	ih := es.InnerHits().Sort(es.Sort("indexedAt").Order(Order.Desc))
	// When Then
	assert.NotNil(t, ih)
	bodyJSON := assert.MarshalWithoutError(t, ih)
	assert.Equal(t, "{\"sort\":[{\"indexedAt\":{\"order\":\"desc\"}}]}", bodyJSON)
}

func Test_InnerHits_should_have_Collapse_method(t *testing.T) {
	t.Parallel()
	// Given
	ih := es.InnerHits()

	// When Then
	assert.NotNil(t, ih.Collapse)
}

func Test_InnerHits_Collapse_should_create_json_with_collapse_field_inside_inner_hits(t *testing.T) {
	t.Parallel()
	// Given
	ih := es.InnerHits().Collapse(es.FieldCollapse("yelle"))
	// When Then
	assert.NotNil(t, ih)
	bodyJSON := assert.MarshalWithoutError(t, ih)
	assert.Equal(t, "{\"collapse\":{\"field\":\"yelle\"}}", bodyJSON)
}

func Test_InnerHits_should_have_DocvalueFields_method(t *testing.T) {
	t.Parallel()
	// Given
	ih := es.InnerHits()

	// When Then
	assert.NotNil(t, ih.DocvalueFields)
}

func Test_InnerHits_DocvalueFields_should_create_json_with_docvalue_fields_field_inside_inner_hits(t *testing.T) {
	t.Parallel()
	// Given
	ih := es.InnerHits().DocvalueFields(
		es.FieldAndFormat("id"),
		es.FieldAndFormat("name"),
		es.FieldAndFormat("partition"),
	)
	// When Then
	assert.NotNil(t, ih)
	bodyJSON := assert.MarshalWithoutError(t, ih)
	assert.Equal(t, "{\"docvalue_fields\":[{\"field\":\"id\"},{\"field\":\"name\"},{\"field\":\"partition\"}]}", bodyJSON)
}

func Test_InnerHits_should_have_ScriptField_method(t *testing.T) {
	t.Parallel()
	// Given
	ih := es.InnerHits()

	// When Then
	assert.NotNil(t, ih.ScriptField)
}

func Test_InnerHits_ScriptField_should_create_json_with_script_fields_field_inside_inner_hits(t *testing.T) {
	t.Parallel()
	// Given
	ih := es.InnerHits().
		ScriptField("first_script",
			es.ScriptField(es.ScriptID("key", ScriptLanguage.Painless).
				Option("retry", "3").
				Option("timeout", "300s").
				Parameter("p1", 100).
				Parameter("p2", "hello"),
			).IgnoreFailure(false),
		).
		ScriptField("second_script",
			es.ScriptField(es.ScriptID("value", ScriptLanguage.Mustache).
				Option("retry", "5").
				Parameter("p1", "world"),
			).IgnoreFailure(true),
		)
	// When Then
	assert.NotNil(t, ih)
	bodyJSON := assert.MarshalWithoutError(t, ih)
	// nolint:golint,lll
	assert.Equal(t, "{\"script_fields\":{\"first_script\":{\"ignore_failure\":false,\"script\":{\"id\":\"key\",\"lang\":\"painless\",\"options\":{\"retry\":\"3\",\"timeout\":\"300s\"},\"params\":{\"p1\":100,\"p2\":\"hello\"}}},\"second_script\":{\"ignore_failure\":true,\"script\":{\"id\":\"value\",\"lang\":\"mustache\",\"options\":{\"retry\":\"5\"},\"params\":{\"p1\":\"world\"}}}}}", bodyJSON)
}

////   Script Field   ////

func Test_ScriptField_should_exist_on_es_package(t *testing.T) {
	t.Parallel()
	// Given When Then
	assert.NotNil(t, es.ScriptField)
}

func Test_ScriptField_method_should_create_scriptFieldType(t *testing.T) {
	t.Parallel()
	// Given
	sf := es.ScriptField(es.ScriptID("key", ScriptLanguage.Painless))

	// Then
	assert.NotNil(t, sf)
	assert.IsTypeString(t, "es.scriptFieldType", sf)
}

func Test_ScriptField_should_have_IgnoreFailure_method(t *testing.T) {
	t.Parallel()
	// Given
	sf := es.ScriptField(es.ScriptID("key", ScriptLanguage.Painless))

	// When Then
	assert.NotNil(t, sf.IgnoreFailure)
}

func Test_ScriptField_IgnoreFailure_should_create_json_with_collapse_field_inside_field_collapse(t *testing.T) {
	t.Parallel()
	// Given
	sf := es.ScriptField(es.ScriptID("key", ScriptLanguage.Painless)).IgnoreFailure(true)
	// When Then
	assert.NotNil(t, sf)
	bodyJSON := assert.MarshalWithoutError(t, sf)
	assert.Equal(t, "{\"ignore_failure\":true,\"script\":{\"id\":\"key\",\"lang\":\"painless\"}}", bodyJSON)
}

////   Field Collapse   ////

func Test_FieldCollapse_should_exist_on_es_package(t *testing.T) {
	t.Parallel()
	// Given When Then
	assert.NotNil(t, es.FieldCollapse)
}

func Test_FieldCollapse_method_should_create_fieldCollapseType(t *testing.T) {
	t.Parallel()
	// Given
	fc := es.FieldCollapse("name")

	// Then
	assert.NotNil(t, fc)
	assert.IsTypeString(t, "es.fieldCollapseType", fc)
}

func Test_FieldCollapse_should_have_Collapse_method(t *testing.T) {
	t.Parallel()
	// Given
	fc := es.FieldCollapse("id")

	// When Then
	assert.NotNil(t, fc.Collapse)
}

func Test_FieldCollapse_Collapse_should_create_json_with_collapse_field_inside_field_collapse(t *testing.T) {
	t.Parallel()
	// Given
	fc := es.FieldCollapse("id").Collapse(es.FieldCollapse("index"))
	// When Then
	assert.NotNil(t, fc)
	bodyJSON := assert.MarshalWithoutError(t, fc)
	assert.Equal(t, "{\"collapse\":{\"field\":\"index\"},\"field\":\"id\"}", bodyJSON)
}

func Test_FieldCollapse_should_have_InnerHits_method(t *testing.T) {
	t.Parallel()
	// Given
	fc := es.FieldCollapse("id")

	// When Then
	assert.NotNil(t, fc.InnerHits)
}

func Test_FieldCollapse_InnerHits_should_create_json_with_inner_hits_field_inside_field_collapse(t *testing.T) {
	t.Parallel()
	// Given
	fc := es.FieldCollapse("display").InnerHits(es.InnerHits().Fields("name", "surname"))
	// When Then
	assert.NotNil(t, fc)
	bodyJSON := assert.MarshalWithoutError(t, fc)
	assert.Equal(t, "{\"field\":\"display\",\"inner_hits\":[{\"fields\":[\"name\",\"surname\"]}]}", bodyJSON)
}

func Test_FieldCollapse_should_have_MaxConcurrentGroupSearches_method(t *testing.T) {
	t.Parallel()
	// Given
	fc := es.FieldCollapse("id")

	// When Then
	assert.NotNil(t, fc.MaxConcurrentGroupSearches)
}

// nolint:golint,lll
func Test_FieldCollapse_MaxConcurrentGroupSearches_should_create_json_with_max_concurrent_group_searches_field_inside_field_collapse(t *testing.T) {
	t.Parallel()
	// Given
	fc := es.FieldCollapse("display").MaxConcurrentGroupSearches(67)
	// When Then
	assert.NotNil(t, fc)
	bodyJSON := assert.MarshalWithoutError(t, fc)
	assert.Equal(t, "{\"field\":\"display\",\"max_concurrent_group_searches\":67}", bodyJSON)
}

////   Field And Format   ////

func Test_FieldAndFormat_should_exist_on_es_package(t *testing.T) {
	t.Parallel()
	// Given When Then
	assert.NotNil(t, es.FieldAndFormat)
}

func Test_FieldAndFormat_method_should_create_fieldAndFormatType(t *testing.T) {
	t.Parallel()
	// Given
	fnf := es.FieldAndFormat("name")

	// Then
	assert.NotNil(t, fnf)
	assert.IsTypeString(t, "es.fieldAndFormatType", fnf)
}

func Test_FieldAndFormat_should_have_Format_method(t *testing.T) {
	t.Parallel()
	// Given
	fnf := es.FieldAndFormat("name")

	// When Then
	assert.NotNil(t, fnf.Format)
}

func Test_FieldAndFormat_Format_should_create_json_with_format_field_inside_field_and_format(t *testing.T) {
	t.Parallel()
	// Given
	fnf := es.FieldAndFormat("xyz").Format("yyyy-MM-dd")
	// When Then
	assert.NotNil(t, fnf)
	bodyJSON := assert.MarshalWithoutError(t, fnf)
	assert.Equal(t, "{\"field\":\"xyz\",\"format\":\"yyyy-MM-dd\"}", bodyJSON)
}

func Test_FieldAndFormat_should_have_IncludeUnmapped_method(t *testing.T) {
	t.Parallel()
	// Given
	fnf := es.FieldAndFormat("name")

	// When Then
	assert.NotNil(t, fnf.IncludeUnmapped)
}

func Test_FieldAndFormat_IncludeUnmapped_should_create_json_with_include_unmapped_field_inside_field_and_format(t *testing.T) {
	t.Parallel()
	// Given
	fnf := es.FieldAndFormat("hello").IncludeUnmapped(false)
	// When Then
	assert.NotNil(t, fnf)
	bodyJSON := assert.MarshalWithoutError(t, fnf)
	assert.Equal(t, "{\"field\":\"hello\",\"include_unmapped\":false}", bodyJSON)
}