Function TestCollectionSearchRequest

pindex_bleve_test.go:713–774

(t *testing.T)

}

func TestCollectionSearchRequest(t *testing.T) {
	testCache := getTestCache()
	tests := []struct {
		indexName     string
		collections   []string
		qNumDisjuncts int
		qField        string
		qTerm         []string
	}{
		{
			indexName:     "ftsIndexA",
			collections:   []string{"colA"},
			qField:        "_$scope_$collection",
			qTerm:         []string{"_$suid_$cuidA"},
			qNumDisjuncts: 1,
		},
		{
			indexName:     "ftsIndexB",
			collections:   []string{"colA", "colB", "colC"},
			qField:        "_$scope_$collection",
			qTerm:         []string{"_$suid_$cuidA", "_$suid_$cuidB", "_$suid_$cuidC"},
			qNumDisjuncts: 3,
		},
	}

	var sr *SearchRequest
	err := json.Unmarshal([]byte(`{"query": {"query": "california"}, "size": 4, "from": 5}`), &sr)
	if err != nil {
		t.Fatal(err)
	}

	for _, test := range tests {
		sr.Collections = test.collections
		bsr, err := sr.ConvertToBleveSearchRequest()
		_, bsr.Query = sr.decorateQuery(test.indexName, bsr.Query, testCache)
		if err != nil {
			t.Fatal(err)
		}
		switch qq := bsr.Query.(type) {
		case *query.ConjunctionQuery:
			if len(qq.Conjuncts) != 2 {
				t.Fatalf("Exception in conjunction query, number of conjunct clauses: %v",
					len(qq.Conjuncts))
			}
			dqs := qq.Conjuncts[1].(*query.DisjunctionQuery)
			if len(dqs.Disjuncts) != test.qNumDisjuncts {
				t.Fatalf("Exception in disjunction query, number of disjunct clauses: %v",
					len(dqs.Disjuncts))
			}

			for i, dq := range dqs.Disjuncts {
				mq := dq.(*query.MatchQuery)
				if mq.Match != test.qTerm[i] || mq.FieldVal != test.qField {
					t.Fatalf("Exception in disjunction should query: %v, %v",
						mq.Match, mq.FieldVal)
				}
			}
		default: