test/acceptance/graphql_resolvers/local_aggregate_hybrid_search_test.go

//                           _       _
// __      _____  __ ___   ___  __ _| |_ ___
// \ \ /\ / / _ \/ _` \ \ / / |/ _` | __/ _ \
//  \ V  V /  __/ (_| |\ V /| | (_| | ||  __/
//   \_/\_/ \___|\__,_| \_/ |_|\__,_|\__\___|
//
//  Copyright © 2016 - 2024 Weaviate B.V. All rights reserved.
//
//  CONTACT: [email protected]
//

package test

import (
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
	"github.com/weaviate/weaviate/test/helper"
	graphqlhelper "github.com/weaviate/weaviate/test/helper/graphql"
)

func aggregationWithHybridSearch(t *testing.T) {
	t.Run("without search vector", func(t *testing.T) {
		query := `
		{
			Aggregate {
				Company
				(
					objectLimit: 3
      				hybrid: {
        				alpha: 0.5
        				query: "Apple"
      				}
				)
				{
					name {
						topOccurrences {
							value
						}
					}
				}
			}
		}`
		result := graphqlhelper.AssertGraphQL(t, helper.RootAuth, query).Get("Aggregate", "Company").AsSlice()
		require.Len(t, result, 1)
		topOccur := result[0].(map[string]interface{})["name"].(map[string]interface{})["topOccurrences"].([]interface{})
		require.Len(t, topOccur, 3)
		assert.Contains(t, topOccur, map[string]interface{}{"value": "Apple"})
		assert.Contains(t, topOccur, map[string]interface{}{"value": "Apple Inc."})
		assert.Contains(t, topOccur, map[string]interface{}{"value": "Apple Incorporated"})
	})

	t.Run("with grouping, sparse search only", func(t *testing.T) {
		query := `
		{
			Aggregate {
				Company
    			(
				  	groupBy: "name"
				  	hybrid: {
						alpha: 0
        				query: "Google"
					}
    			)
				{
					name {
        				topOccurrences {
          					value
        				}
      				}
				}
			}
		}`

		type object = map[string]interface{}

		result := graphqlhelper.AssertGraphQL(t, helper.RootAuth, query).Get("Aggregate", "Company").AsSlice()
		require.Len(t, result, 3)
		assert.Contains(t, result, object{
			"name": object{
				"topOccurrences": []interface{}{
					object{"value": "Google"},
				},
			},
		})
		assert.Contains(t, result, object{
			"name": object{
				"topOccurrences": []interface{}{
					object{"value": "Google Inc."},
				},
			},
		})
		assert.Contains(t, result, object{
			"name": object{
				"topOccurrences": []interface{}{
					object{"value": "Google Incorporated"},
				},
			},
		})
	})
}