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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
#include "global.h"
#include <library/cpp/testing/unittest/registar.h>
using namespace NSQLComplete;
Y_UNIT_TEST_SUITE(GlobalAnalysisTests) {
Y_UNIT_TEST(TopLevelNamesCollected) {
IGlobalAnalysis::TPtr global = MakeGlobalAnalysis();
TString query = R"(
DECLARE $cluster_name AS String;
IMPORT math SYMBOLS $sqrt, $pow;
$sqrt = 0;
DEFINE ACTION $hello_world($name, $suffix?) AS
$name = $name ?? ($suffix ?? "world");
SELECT "Hello, " || $name || "!";
END DEFINE;
$first, $second, $_ = AsTuple(1, 2, 3);
)";
TGlobalContext ctx = global->Analyze({query}, {});
Sort(ctx.Names);
TVector<TString> expected = {
"cluster_name",
"first",
"hello_world",
"pow",
"second",
"sqrt",
};
UNIT_ASSERT_VALUES_EQUAL(ctx.Names, expected);
}
Y_UNIT_TEST(LocalNamesCollected) {
IGlobalAnalysis::TPtr global = MakeGlobalAnalysis();
TString query = R"(
DEFINE ACTION $sum($x, $y) AS
$acc = 0;
EVALUATE FOR $i IN AsList($x, $y) DO BEGIN
$plus = ($a, $b) -> (#);
$acc = $plus($acc, $i);
END DO;
END DEFINE;
)";
TCompletionInput input = SharpedInput(query);
TGlobalContext ctx = global->Analyze(input, {});
Sort(ctx.Names);
TVector<TString> expected = {
"a",
"acc",
"b",
"i",
"plus",
"sum",
"x",
"y",
};
UNIT_ASSERT_VALUES_EQUAL(ctx.Names, expected);
}
Y_UNIT_TEST(EnclosingFunctionName) {
IGlobalAnalysis::TPtr global = MakeGlobalAnalysis();
{
TString query = "SELECT * FROM Concat(#)";
TGlobalContext ctx = global->Analyze(SharpedInput(query), {});
UNIT_ASSERT_VALUES_EQUAL(ctx.EnclosingFunction, "Concat");
}
{
TString query = "SELECT * FROM Concat(a, #)";
TGlobalContext ctx = global->Analyze(SharpedInput(query), {});
UNIT_ASSERT_VALUES_EQUAL(ctx.EnclosingFunction, "Concat");
}
{
TString query = "SELECT * FROM Concat(a#)";
TGlobalContext ctx = global->Analyze(SharpedInput(query), {});
UNIT_ASSERT_VALUES_EQUAL(ctx.EnclosingFunction, "Concat");
}
{
TString query = "SELECT * FROM Concat(#";
TGlobalContext ctx = global->Analyze(SharpedInput(query), {});
UNIT_ASSERT_VALUES_EQUAL(ctx.EnclosingFunction, Nothing());
}
{
TString query = "SELECT * FROM (#)";
TGlobalContext ctx = global->Analyze(SharpedInput(query), {});
UNIT_ASSERT_VALUES_EQUAL(ctx.EnclosingFunction, Nothing());
}
}
Y_UNIT_TEST(SimpleSelectFrom) {
IGlobalAnalysis::TPtr global = MakeGlobalAnalysis();
{
TString query = "SELECT # FROM plato.Input";
TGlobalContext ctx = global->Analyze(SharpedInput(query), {});
TColumnContext expected = {.Tables = {TTableId{"plato", "Input"}}};
UNIT_ASSERT_VALUES_EQUAL(ctx.Column, expected);
}
{
TString query = "SELECT # FROM plato.`//home/input`";
TGlobalContext ctx = global->Analyze(SharpedInput(query), {});
TColumnContext expected = {.Tables = {TTableId{"plato", "//home/input"}}};
UNIT_ASSERT_VALUES_EQUAL(ctx.Column, expected);
}
{
TString query = "SELECT # FROM plato.Input AS x";
TGlobalContext ctx = global->Analyze(SharpedInput(query), {});
TColumnContext expected = {.Tables = {TAliased<TTableId>("x", TTableId{"plato", "Input"})}};
UNIT_ASSERT_VALUES_EQUAL(ctx.Column, expected);
}
}
} // Y_UNIT_TEST_SUITE(GlobalAnalysisTests)
|
