Function TestAST_RewriteWithFunctionRef

runtime/pkg/duckdbsql/ast_test.go:477–615

(t *testing.T)

}

func TestAST_RewriteWithFunctionRef(t *testing.T) {
	sqlVariations := []struct {
		title       string
		sql         string
		replace     []*TableRef
		expectedSql string
	}{
		{
			"with single path and literal prop",
			`select * from AdBids`,
			[]*TableRef{
				{
					Function: "read_csv",
					Paths:    []string{"/path/to/AdBids.csv"},
					Properties: map[string]any{
						"delim": "|",
					},
				},
			},
			`SELECT * FROM read_csv(main.list_value('/path/to/AdBids.csv'), (delim = '|'))`,
		},
		{
			"with single path and float prop",
			`select * from AdBids`,
			[]*TableRef{
				{
					Function: "read_csv",
					Paths:    []string{"/path/to/AdBids.csv"},
					Properties: map[string]any{
						"sample_size": float64(-1),
					},
				},
			},
			`SELECT * FROM read_csv(main.list_value('/path/to/AdBids.csv'), (sample_size = -1))`,
		},
		{
			"with multiple paths with map prop",
			`select * from AdBids`,
			[]*TableRef{
				{
					Function: "read_csv",
					Paths:    []string{"/path/to/AdBids1.csv", "/path/to/AdBids2.csv"},
					Properties: map[string]any{
						"columns": map[string]any{
							"A": "Date",
						},
					},
				},
			},
			`SELECT * FROM read_csv(main.list_value('/path/to/AdBids1.csv', '/path/to/AdBids2.csv'), ("columns" = main.struct_pack(A := 'Date')))`,
		},
		{
			"with deep map prop",
			`select * from AdBids`,
			[]*TableRef{
				{
					Function: "read_csv",
					Paths:    []string{"/path/to/AdBids.csv"},