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
|
#include "yql.h"
#include <yql/essentials/ast/yql_ast.h>
#include <yql/essentials/providers/common/provider/yql_provider_names.h>
#include <yql/essentials/sql/settings/translation_settings.h>
#include <yql/essentials/sql/v1/sql.h>
#include <yql/essentials/sql/v1/lexer/lexer.h>
#include <yql/essentials/sql/v1/lexer/antlr4/lexer.h>
#include <yql/essentials/sql/v1/proto_parser/proto_parser.h>
#include <yql/essentials/sql/v1/proto_parser/antlr4/proto_parser.h>
#include <library/cpp/testing/unittest/registar.h>
using namespace NSQLComplete;
class TSQLv1Parser {
public:
TSQLv1Parser() {
Settings_.Arena = &Arena_;
Settings_.ClusterMapping = {
{"socrates", TString(NYql::YtProviderName)},
{"plato", TString(NYql::YtProviderName)},
};
Settings_.SyntaxVersion = 1;
Lexers_.Antlr4 = NSQLTranslationV1::MakeAntlr4LexerFactory();
Parsers_.Antlr4 = NSQLTranslationV1::MakeAntlr4ParserFactory(/*isAmbiguityError=*/ true);
}
NYql::TAstParseResult Parse(const TString& query) {
Arena_.Reset();
auto result = NSQLTranslationV1::SqlToYql(Lexers_, Parsers_, query, Settings_);
Y_ENSURE(result.IsOk());
return result;
}
private:
google::protobuf::Arena Arena_;
NSQLTranslation::TTranslationSettings Settings_;
NSQLTranslationV1::TLexers Lexers_;
NSQLTranslationV1::TParsers Parsers_;
};
TYqlContext Analyze(const TString& query) {
auto ast = TSQLv1Parser().Parse(query);
NYql::TIssues issues;
return *MakeYqlAnalysis()->Analyze(*ast.Root, issues);
}
Y_UNIT_TEST_SUITE(YqlAnalysisTests) {
Y_UNIT_TEST(NamesAreCollected) {
TString input = R"(
USE yt:socrates;
SELECT * FROM Input;
CREATE TABLE Newbie (x Unit);
INSERT INTO plato.Input (id) VALUES (1);
)";
THashMap<TString, THashSet<TString>> expected = {
{"socrates", {"Input", "Newbie"}},
{"plato", {"Input"}},
};
UNIT_ASSERT_VALUES_EQUAL(Analyze(input).TablesByCluster, expected);
}
} // Y_UNIT_TEST_SUITE(YqlAnalysisTests)
|
