query_parser_test.go

package solr_test

import (
	"testing"

	"github.com/stretchr/testify/assert"

	"github.com/stevenferrer/solr-go"
)

func TestQueryParsers(t *testing.T) {
	t.Run("standard query parser", func(t *testing.T) {
		a := assert.New(t)

		got := solr.NewStandardQueryParser().
			Tag("certain").BuildParser()
		a.Equal("{!lucene tag=certain}", got)

		got = solr.NewStandardQueryParser().Query("'solr rocks'").
			Df("text").Op("AND").Sow().BuildParser()
		expect := "{!lucene df=text q.op=AND sow=true v='solr rocks'}"
		a.Equal(expect, got)

		got = solr.NewStandardQueryParser().
			Query("'solr rocks'").BuildParser()
		expect = "{!lucene v='solr rocks'}"
		a.Equal(expect, got)
	})

	t.Run("dismax query parser", func(t *testing.T) {
		a := assert.New(t)

		got := solr.NewDisMaxQueryParser().BuildParser()
		a.Equal("{!dismax}", got)

		got = solr.NewDisMaxQueryParser().
			Query("'solr rocks'").
			Alt("*:*").
			Qf("'one^2.3 two three^0.4'").
			Mm("75%").
			Pf("'one^2.3 two three^0.4'").
			Ps("1").
			Qs("1").
			Tie("0.1").
			Bq("category:food^10").
			Bf("div(1,sum(1,price))^1.5").
			BuildParser()
		expect := `{!dismax q.alt=*:* qf='one^2.3 two three^0.4' mm=75% pf='one^2.3 two three^0.4' ps=1 qs=1 tie=0.1 bq=category:food^10 bf=div(1,sum(1,price))^1.5 v='solr rocks'}`
		a.Equal(expect, got)

		got = solr.NewDisMaxQueryParser().
			Query("'solr rocks'").BuildParser()
		expect = "{!dismax v='solr rocks'}"
		a.Equal(expect, got)
	})

	t.Run("parent query parser", func(t *testing.T) {
		a := assert.New(t)
		got := solr.NewParentQueryParser().
			Query("comment:SolrCloud").
			Which("content_type:parent").
			Filters("$childfq").
			ExcludeTags("certain").
			Score("total").
			Tag("top").
			BuildParser()
		expect := `{!parent which=content_type:parent tag=top filters=$childfq excludeTags=certain score=total v=comment:SolrCloud}`
		a.Equal(expect, got)
	})

	t.Run("children query parser", func(t *testing.T) {
		a := assert.New(t)
		got := solr.NewChildrenQueryParser().
			Query("$parent").
			Of("$parent").
			Filters("$someFilters").
			ExcludeTags("certain").
			BuildParser()
		expect := `{!child of=$parent filters=$someFilters excludeTags=certain v=$parent}`
		a.Equal(expect, got)
	})

	t.Run("filters query parser", func(t *testing.T) {
		a := assert.New(t)
		got := solr.NewFiltersQueryParser().
			Query("field:text").
			Param("$fqs").
			ExcludeTags("sample").
			BuildParser()
		expect := `{!filters param=$fqs excludeTags=sample v=field:text}`
		a.Equal(expect, got)
	})

}