rdf_result_test.go

//go:build integration || cloud || upgrade

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

package query

import (
	"context"
	"testing"

	"github.com/stretchr/testify/require"
)

func TestRDFResult(t *testing.T) {
	query := `{
		friends_15_and_19(func: uid(1)) {
		  name
		  friend @filter(ge(age, 15) AND lt(age, 19)) {
			  name
			  age
		  }
		}
	  }`

	rdf, err := processQueryRDF(context.Background(), query)
	require.NoError(t, err)
	require.Equal(t, rdf, `<0x1> <name> "Michonne" .
<0x1> <friend> <0x17> .
<0x1> <friend> <0x18> .
<0x1> <friend> <0x19> .
<0x17> <name> "Rick Grimes" .
<0x18> <name> "Glenn Rhee" .
<0x19> <name> "Daryl Dixon" .
<0x17> <age> "15" .
<0x18> <age> "15" .
<0x19> <age> "17" .
`)
}

func TestRDFNormalize(t *testing.T) {
	query := `
	{
		me(func: uid(0x01)) @normalize {
			mn: name
			gender
			friend {
				n: name
				d: dob
				friend {
					fn : name
				}
			}
			son {
				sn: name
			}
		}
	}`
	_, err := processQueryRDF(context.Background(), query)
	require.Error(t, err, "normalize directive is not supported in the rdf output format")
}

func TestRDFGroupBy(t *testing.T) {
	query := `
	{
		me(func: uid(1, 23, 24, 25, 31)) @groupby(age) {
				count(uid)
		}
	}`
	_, err := processQueryRDF(context.Background(), query)
	require.Contains(t, err.Error(), "groupby is not supported in rdf output format")
}

func TestRDFUidCount(t *testing.T) {
	query := `
	{
		me(func: gt(count(friend), 0)) {
			count(uid)
		}
	}`
	_, err := processQueryRDF(context.Background(), query)
	require.Contains(t, err.Error(), "uid count is not supported in the rdf output format")
}

func TestRDFIngoreReflex(t *testing.T) {
	query := `
	{
		me(func:anyofterms(name, "Michonne Rick Daryl")) @ignoreReflex {
			name
			friend {
				name
				friend {
					name
				}
			}
		}
	}`
	_, err := processQueryRDF(context.Background(), query)
	require.Contains(t, err.Error(),
		"ignorereflex directive is not supported in the rdf output format")
}

func TestRDFRecurse(t *testing.T) {
	query := `
	{
		me(func: anyofterms(name, "Michonne Rick Daryl")) @recurse(depth: 1, loop: true) {
			name
			friend
		}
	}`
	rdf, err := processQueryRDF(context.Background(), query)
	require.NoError(t, err)
	require.Equal(t, rdf, `<0x1> <name> "Michonne" .
<0x17> <name> "Rick Grimes" .
<0x19> <name> "Daryl Dixon" .
`)
}

func TestRDFIgnoreUid(t *testing.T) {
	query := `
	{
		me(func: anyofterms(name, "Michonne Rick Daryl")) {
			uid
			name
		}
	}`
	rdf, err := processQueryRDF(context.Background(), query)
	require.NoError(t, err)
	require.Equal(t, rdf, `<0x1> <name> "Michonne" .
<0x17> <name> "Rick Grimes" .
<0x19> <name> "Daryl Dixon" .
`)
}

func TestRDFCheckPwd(t *testing.T) {
	query := `
    {
        me(func: uid(0x01)) {
			expand(_all_)
			checkpwd(password, "12345")
		}
    }
	`
	_, err := processQueryRDF(context.Background(), query)
	require.Contains(t, err.Error(),
		"chkpwd function is not supported in the rdf output format")
}

func TestRDFPredicateCount(t *testing.T) {
	query := `
    {
		me(func:anyofterms(name, "Michonne Rick Daryl")) {
			name
			count(friend)
			friend {
				name
			}
		}
    }
	`

	rdf, err := processQueryRDF(context.Background(), query)
	require.NoError(t, err)
	require.Equal(t, `<0x1> <name> "Michonne" .
<0x17> <name> "Rick Grimes" .
<0x19> <name> "Daryl Dixon" .
<0x1> <count(friend)> "5" .
<0x17> <count(friend)> "1" .
<0x19> <count(friend)> "0" .
<0x1> <friend> <0x17> .
<0x1> <friend> <0x18> .
<0x1> <friend> <0x19> .
<0x1> <friend> <0x1f> .
<0x1> <friend> <0x65> .
<0x17> <friend> <0x1> .
<0x1> <name> "Michonne" .
<0x17> <name> "Rick Grimes" .
<0x18> <name> "Glenn Rhee" .
<0x19> <name> "Daryl Dixon" .
<0x1f> <name> "Andrea" .
`, rdf)
}

func TestRDFFacets(t *testing.T) {
	query := `
		{
			shortest(from: 1, to:1001, numpaths: 4) {
				path @facets(weight)
			}
		}`
	_, err := processQueryRDF(context.Background(), query)
	require.Contains(t, err.Error(),
		"facets are not supported in the rdf output format")
}

func TestDateRDF(t *testing.T) {
	query := `
		{
			me(func: uid(0x01)) {
				name
				gender
				friend(orderdesc: film.film.initial_release_date) {
					name
					film.film.initial_release_date
				}
			}
		}
	`
	rdf, err := processQueryRDF(context.Background(), query)
	require.NoError(t, err)
	expected := `<0x1> <name> "Michonne" .
<0x1> <gender> "female" .
<0x1> <friend> <0x19> .
<0x1> <friend> <0x18> .
<0x1> <friend> <0x17> .
<0x1> <friend> <0x1f> .
<0x1> <friend> <0x65> .
<0x17> <name> "Rick Grimes" .
<0x18> <name> "Glenn Rhee" .
<0x19> <name> "Daryl Dixon" .
<0x1f> <name> "Andrea" .
<0x17> <film.film.initial_release_date> "1900-01-02T00:00:00Z" .
<0x18> <film.film.initial_release_date> "1909-05-05T00:00:00Z" .
<0x19> <film.film.initial_release_date> "1929-01-10T00:00:00Z" .
<0x1f> <film.film.initial_release_date> "1801-01-15T00:00:00Z" .
`
	require.Equal(t, expected, rdf)
}