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
|
import os
import yatest.common
import pytest
from test_sql2yql import get_sql2yql_cmd
from test_utils import pytest_generate_tests_by_template, DATA_PATH
def pytest_generate_tests(metafunc):
return pytest_generate_tests_by_template('.sql', metafunc)
def test(suite, case, tmpdir):
files = []
# case can contain slash because of nested suites
out_dir = tmpdir.mkdir(suite).mkdir(case.replace('/', '_')).dirname
case_file = os.path.join(DATA_PATH, suite, '%s.sql' % case)
ansi_lexer = False
with open(case_file, 'r') as f:
content = f.read()
ansi_lexer = 'ansi_lexer' in content
test_double_format = 'skip double format' not in content
if 'syntax_pg' in content:
pytest.skip('syntax_pg')
cmd = get_sql2yql_cmd(suite, case, case_file, out_dir=out_dir, ansi_lexer=ansi_lexer, test_format=True, test_double_format=test_double_format)
yatest.common.process.execute(cmd, env={'YQL_DETERMINISTIC_MODE': '1'})
files.append(os.path.join(out_dir, 'formatted.sql'))
return [yatest.common.canonical_file(file_name) for file_name in files]
|
