cosmium/parsers/nosql/subquery_test.go

package nosql_test

import (
	"testing"

	"github.com/pikami/cosmium/parsers"
)

func Test_Parse_SubQuery(t *testing.T) {

	t.Run("Should parse FROM subquery", func(t *testing.T) {
		testQueryParse(
			t,
			`SELECT c.id FROM (SELECT VALUE cc["info"] FROM cc) AS c`,
			parsers.SelectStmt{
				SelectItems: []parsers.SelectItem{
					{Path: []string{"c", "id"}},
				},
				Table: parsers.Table{
					Value: "c",
					SelectItem: parsers.SelectItem{
						Alias: "c",
						Type:  parsers.SelectItemTypeSubQuery,
						Value: parsers.SelectStmt{
							Table: parsers.Table{Value: "cc"},
							SelectItems: []parsers.SelectItem{
								{Path: []string{"cc", "info"}, IsTopLevel: true},
							},
						},
					},
				},
			},
		)
	})

	t.Run("Should parse JOIN subquery", func(t *testing.T) {
		testQueryParse(
			t,
			`SELECT c.id, cc.name FROM c JOIN (SELECT tag.name FROM tag IN c.tags) AS cc`,
			parsers.SelectStmt{
				SelectItems: []parsers.SelectItem{
					{Path: []string{"c", "id"}},
					{Path: []string{"cc", "name"}},
				},
				Table: parsers.Table{
					Value: "c",
				},
				JoinItems: []parsers.JoinItem{
					{
						Table: parsers.Table{
							Value: "cc",
						},
						SelectItem: parsers.SelectItem{
							Alias: "cc",
							Type:  parsers.SelectItemTypeSubQuery,
							Value: parsers.SelectStmt{
								SelectItems: []parsers.SelectItem{
									{Path: []string{"tag", "name"}},
								},
								Table: parsers.Table{
									Value: "tag",
									SelectItem: parsers.SelectItem{
										Path: []string{"c", "tags"},
									},
								},
							},
						},
					},
				},
			},
		)
	})

	t.Run("Should parse JOIN EXISTS subquery", func(t *testing.T) {
		testQueryParse(
			t,
			`SELECT c.id
			FROM c
			JOIN (
				SELECT VALUE EXISTS(SELECT tag.name FROM tag IN c.tags)
			) AS hasTags
			WHERE hasTags`,
			parsers.SelectStmt{
				SelectItems: []parsers.SelectItem{
					{Path: []string{"c", "id"}},
				},
				Table: parsers.Table{
					Value: "c",
				},
				JoinItems: []parsers.JoinItem{
					{
						Table: parsers.Table{Value: "hasTags"},
						SelectItem: parsers.SelectItem{
							Alias: "hasTags",
							Type:  parsers.SelectItemTypeSubQuery,
							Value: parsers.SelectStmt{
								SelectItems: []parsers.SelectItem{
									{
										IsTopLevel: true,
										Type:       parsers.SelectItemTypeSubQuery,
										Value: parsers.SelectStmt{
											SelectItems: []parsers.SelectItem{
												{Path: []string{"tag", "name"}},
											},
											Table: parsers.Table{
												Value: "tag",
												SelectItem: parsers.SelectItem{
													Path: []string{"c", "tags"},
												},
											},
											Exists: true,
										},
									},
								},
							},
						},
					},
				},
				Filters: parsers.SelectItem{
					Path: []string{"hasTags"},
				},
			},
		)
	})
}