YQ Connector: move tests from yql to ydb (OSS)

Перенос папки с тестами на Коннектор из папки yql в папку ydb (синхронизируется с github).

1 files changed, 29 insertions, 0 deletions

diff --git a/contrib/clickhouse/src/Storages/extractTableFunctionArgumentsFromSelectQuery.cpp b/contrib/clickhouse/src/Storages/extractTableFunctionArgumentsFromSelectQuery.cpp
new file mode 100644
index 0000000000..382964d9fe
--- /dev/null
+++ b/contrib/clickhouse/src/Storages/extractTableFunctionArgumentsFromSelectQuery.cpp
@@ -0,0 +1,29 @@
+#include <Storages/extractTableFunctionArgumentsFromSelectQuery.h>
+
+#include <Parsers/ASTExpressionList.h>
+#include <Parsers/ASTFunction.h>
+#include <Parsers/ASTLiteral.h>
+#include <Parsers/ASTSelectQuery.h>
+#include <Parsers/ASTTablesInSelectQuery.h>
+#include <Parsers/queryToString.h>
+
+
+namespace DB
+{
+
+ASTExpressionList * extractTableFunctionArgumentsFromSelectQuery(ASTPtr & query)
+{
+    auto * select_query = query->as<ASTSelectQuery>();
+    if (!select_query || !select_query->tables())
+        return nullptr;
+
+    auto * tables = select_query->tables()->as<ASTTablesInSelectQuery>();
+    auto * table_expression = tables->children[0]->as<ASTTablesInSelectQueryElement>()->table_expression->as<ASTTableExpression>();
+    if (!table_expression->table_function)
+        return nullptr;
+
+    auto * table_function = table_expression->table_function->as<ASTFunction>();
+    return table_function->arguments->as<ASTExpressionList>();
+}
+
+}