def test_whole_line_comments(
query_whole_line_comments, query_without_whole_line_comments
):
table_finder = TableFinder()
cleaned_query = table_finder._remove_inline_comments(query_whole_line_comments)
cleaned_query_str = normalize_query(" ".join(cleaned_query))
assert cleaned_query_str == query_without_whole_line_comments
def query_without_inline_comments():
query = """
SELECT a.val1, a.val2, b.val, c.val
FROM a JOIN b ON (a.key = b.key)
LEFT OUTER JOIN c ON (a.key = c.key)
"""
return normalize_query(query)
def query_without_whole_line_comments():
query = """
SELECT col1
FROM (
SELECT col1, SUM(col2) AS col2sum
FROM t1
GROUP BY col1
) t2
WHERE t2.col2sum > 10
"""
return normalize_query(query)
def test_inline_comments(query_inline_comments, query_without_inline_comments):
table_finder = TableFinder()
cleaned_query = table_finder._remove_trailing_characters(query_inline_comments)
cleaned_query_str = normalize_query(cleaned_query)
assert cleaned_query_str == query_without_inline_comments
def test_block_comments(query_block_comments, query_without_block_comments):
table_finder = TableFinder()
cleaned_query = table_finder._remove_block_comments(query_block_comments)
assert normalize_query(cleaned_query) == query_without_block_comments
def test_read_query(simple_query):
assert normalize_query(simple_query) == "SELECT * FROM mytable"
def query_without_block_comments():
query = """
SELECT *
FROM Customers;
"""
return normalize_query(query)