normalize_feature_flag_test.go

//go:build integration2

/*
 * SPDX-FileCopyrightText: © Hypermode Inc. <hello@hypermode.com>
 * SPDX-License-Identifier: Apache-2.0
 */

package query

import (
	"testing"

	"github.com/stretchr/testify/require"

	"github.com/dgraph-io/dgo/v250/protos/api"
	"github.com/hypermodeinc/dgraph/v25/dgraphtest"
)

func TestNormalizeDirectiveWithNoListResponse(t *testing.T) {
	conf := dgraphtest.NewClusterConfig().WithNumAlphas(1).WithNumZeros(1).
		WithReplicas(1).WithNormalizeCompatibilityMode("v20")
	c, err := dgraphtest.NewLocalCluster(conf)
	require.NoError(t, err)
	defer func() { c.Cleanup(t.Failed()) }()
	require.NoError(t, c.Start())

	gc, cleanup, err := c.Client()
	require.NoError(t, err)
	defer cleanup()
	require.NoError(t, c.AssignUids(gc.Dgraph, 100))

	const dataSchema = `
        friend : [uid] @reverse @count .
        name   : string @index(term, exact, trigram) @count @lang .
        dob    : dateTime @index(year) .`
	require.NoError(t, gc.SetupSchema(dataSchema))

	triples := []byte(`
        <1> <friend> <23> .
        <1> <friend> <24> .
        <1> <friend> <25> .
        <1> <friend> <31> .
        <1> <friend> <101>.
        <23> <friend> <1> .
        <31> <friend> <1> .
        <31> <friend> <25> .
        <1> <dob> "1910-01-01" .
		<23> <dob> "1910-01-02" .
		<24> <dob> "1909-05-05" .
		<25> <dob> "1909-01-10" .
		<31> <dob> "1901-01-15" .
        <1> <name> "Michonne" .
        <23> <name> "Rick Grimes" .
        <24> <name> "Glenn Rhee" .`)
	_, err = gc.Mutate(&api.Mutation{SetNquads: triples, CommitNow: true})
	require.NoError(t, err)

	query := `
		{
			me(func: uid(0x01)) @recurse @normalize {
				n: name
				d: dob
				friend
			}
		}`
	js, err := gc.Query(query)
	require.NoError(t, err)
	require.JSONEq(t, `
        {
            "me": [
                {
                    "n": "Michonne",
                    "d": "1910-01-01T00:00:00Z",
                    "n": "Rick Grimes",
                    "d": "1910-01-02T00:00:00Z",
                    "n": "Michonne",
                    "d": "1910-01-01T00:00:00Z"
                },
                {
                    "n": "Michonne",
                    "d": "1910-01-01T00:00:00Z",
                    "n": "Glenn Rhee",
                    "d": "1909-05-05T00:00:00Z"
                },
                {
                    "n": "Michonne",
                    "d": [
                        "1910-01-01T00:00:00Z",
                        "1909-01-10T00:00:00Z"
                    ]
                },
                {
                    "n": "Michonne",
                    "d": [
                        "1910-01-01T00:00:00Z",
                        "1901-01-15T00:00:00Z"
                    ],
                    "n": "Michonne",
                    "d": "1910-01-01T00:00:00Z"
                },
                {
                    "n": "Michonne",
                    "d": [
                        "1910-01-01T00:00:00Z",
                        "1901-01-15T00:00:00Z",
                        "1909-01-10T00:00:00Z"
                    ]
                }
            ]
        }`, string(js.Json))
}