dqlx/query_operation_test.go at main · fenos/dqlx · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
package dqlx_test

import (
	"testing"

	dql "github.com/fenos/dqlx"
	"github.com/stretchr/testify/require"
)

func Test_Multiple_Blocks(t *testing.T) {
	query1 := dql.
		QueryEdge("bladerunner", dql.EqFn("item", "value")).
		Fields(`
			uid
			super_alias:name
			initial_release_date
			d as netflix_id
		`).
		Filter(dql.Eq{"field1": "value1"})

	query2 := dql.
		QueryEdge("bladerunner2", dql.EqFn("item", "value")).
		Fields(`
			uid
			name
		`).
		Filter(dql.Eq{"field1": "value1"})

	query, variables, err := dql.QueriesToDQL(query1, query2)

	require.NoError(t, err)
	require.Equal(t, map[string]string{
		"$0": "value",
		"$1": "value1",
		"$2": "value",
		"$3": "value1",
	}, variables)

	expected := dql.Minify(`
		query Bladerunner_Bladerunner2($0:string, $1:string, $2:string, $3:string) {
			<bladerunner>(func: eq(<item>,$0)) @filter(eq(<field1>,$1)) {
				<uid>
				<super_alias>:<name>
				<initial_release_date>
				d as <netflix_id>
			}

			<bladerunner2>(func: eq(<item>,$2)) @filter(eq(<field1>,$3)) {
				<uid>
				<name>
			}
		}
	`)

	require.Equal(t, expected, query)
}

func Test_Multiple_Blocks_With_Select(t *testing.T) {
	q1 := dql.Query(dql.EqFn("id", "id_a")).
		Select(dql.As("id_a", "id"))

	q2 := dql.Query(dql.EqFn("id", "id_b")).
		Select(dql.As("id_b", "id"))

	query, variables, err := dql.QueriesToDQL(q1, q2)

	require.NoError(t, err)

	require.Equal(t, map[string]string{
		"$0": "id_a",
		"$1": "id_b",
	}, variables)

	expected := dql.Minify(`
		query Rootquery_Rootquery_1($0:string, $1:string) {
          <rootQuery>(func: eq(<id>,$0)) {
            id_a as <id>
          }
          <rootQuery_1>(func: eq(<id>,$1)) {
            id_b as <id>
          }
        }
	`)

	require.Equal(t, expected, query)
}