def test_wildcard_column_expansion_with_insert(completer, complete_event, text): pos = text.index('*') + 1 completions = completer.get_completions( Document(text=text, cursor_position=pos), complete_event) expected = [wildcard_expansion('id, ordered_date, status')] assert expected == completions
def test_wildcard_column_expansion_with_table_qualifier(completer, text, expected): position = len("SELECT users.*") completions = get_result(completer, text, position) expected = [wildcard_expansion(expected)] assert expected == completions
def test_wildcard_column_expansion_with_function(completer, text): position = len('SELECT *') completions = get_result(completer, text, position) col_list = 'x' expected = [wildcard_expansion(col_list)] assert expected == completions
def test_wildcard_column_expansion(completer, text): position = text.find("*") + 1 completions = get_result(completer, text, position) col_list = "id, parentid, email, first_name, last_name" expected = [wildcard_expansion(col_list)] assert expected == completions
def test_wildcard_column_expansion_with_table_qualifier(completer, complete_event, text, expected): pos = len('SELECT users.*') completions = completer.get_completions( Document(text=text, cursor_position=pos), complete_event) expected = [wildcard_expansion(expected)] assert expected == completions
def test_wildcard_column_expansion_with_alias(completer, text): position = text.find('*') + 1 completions = get_result(completer, text, position) col_list = 'id, u.parentid, u.email, u.first_name, u.last_name' expected = [wildcard_expansion(col_list)] assert expected == completions
def test_wildcard_column_expansion_with_alias(completer, complete_event, sql): pos = sql.find('*') + 1 completions = completer.get_completions( Document(text=sql, cursor_position=pos), complete_event) col_list = 'id, u.parentid, u.email, u.first_name, u.last_name' expected = [wildcard_expansion(col_list)] assert expected == completions
def test_wildcard_column_expansion_with_two_tables(completer): text = 'SELECT * FROM public."select" JOIN custom.users ON true' position = len('SELECT *') completions = get_result(completer, text, position) cols = ('"select".id, "select"."localtime", "select"."ABC", ' 'users.id, users.phone_number') expected = [wildcard_expansion(cols)] assert completions == expected
def test_wildcard_column_expansion_with_table_qualifier(completer): text = 'SELECT "select".* FROM public."select"' position = len('SELECT "select".*') completions = get_result(completer, text, position) col_list = 'id, "select"."localtime", "select"."ABC"' expected = [wildcard_expansion(col_list)] assert expected == completions
def test_wildcard_column_expansion_with_alias_qualifier(completer): text = 'SELECT p.* FROM custom.products p' position = len('SELECT p.*') completions = get_result(completer, text, position) col_list = 'id, p.product_name, p.price' expected = [wildcard_expansion(col_list)] assert expected == completions
def test_wildcard_column_expansion_with_function(completer, complete_event, text): pos = len('SELECT *') completions = completer.get_completions( Document(text=text, cursor_position=pos), complete_event) col_list = 'x' expected = [wildcard_expansion(col_list)] assert expected == completions
def test_wildcard_column_expansion_with_table_qualifier( completer, text, expected ): position = len('SELECT users.*') completions = get_result(completer, text, position) expected = [wildcard_expansion(expected)] assert expected == completions
def test_wildcard_column_expansion_with_two_tables(completer): text = 'SELECT * FROM "select" JOIN users u ON true' position = len('SELECT *') completions = get_result(completer, text, position) cols = ('"select".id, "select".insert, "select"."ABC", ' 'u.id, u.parentid, u.email, u.first_name, u.last_name') expected = [wildcard_expansion(cols)] assert completions == expected
def test_wildcard_column_expansion_with_two_tables_and_parent(completer): text = 'SELECT "select".* FROM "select" JOIN users u ON true' position = len('SELECT "select".*') completions = get_result(completer, text, position) col_list = 'id, "select".insert, "select"."ABC"' expected = [wildcard_expansion(col_list)] assert expected == completions
def test_wildcard_column_expansion_with_two_tables_and_parent(completer, complete_event): sql = 'SELECT "select".* FROM "select" JOIN users u ON true' pos = len('SELECT "select".*') completions = completer.get_completions( Document(text=sql, cursor_position=pos), complete_event) col_list = 'id, "select"."insert", "select"."ABC"' expected = [wildcard_expansion(col_list)] assert expected == completions
def test_wildcard_column_expansion_with_two_tables(completer, complete_event): sql = 'SELECT * FROM "select" JOIN users u ON true' pos = len('SELECT *') completions = completer.get_completions( Document(text=sql, cursor_position=pos), complete_event) cols = ('"select".id, "select"."insert", "select"."ABC", ' 'u.id, u.parentid, u.email, u.first_name, u.last_name') expected = [wildcard_expansion(cols)] assert completions == expected
def test_wildcard_column_expansion_with_alias_qualifier(completer, complete_event): sql = 'SELECT u.* FROM users u' pos = len('SELECT u.*') completions = completer.get_completions( Document(text=sql, cursor_position=pos), complete_event) col_list = 'id, u.parentid, u.email, u.first_name, u.last_name' expected = [wildcard_expansion(col_list)] assert expected == completions
def test_wildcard_column_expansion(completer, complete_event): sql = 'SELECT * FROM users' pos = len('SELECT *') completions = completer.get_completions( Document(text=sql, cursor_position=pos), complete_event) col_list = 'id, parentid, email, first_name, last_name' expected = [wildcard_expansion(col_list)] assert expected == completions
def test_wildcard_column_expansion_with_alias_qualifier(completer, complete_event): sql = 'SELECT p.* FROM custom.products p' pos = len('SELECT p.*') completions = completer.get_completions( Document(text=sql, cursor_position=pos), complete_event) col_list = 'id, p.product_name, p.price' expected = [wildcard_expansion(col_list)] assert expected == completions
def test_wildcard_column_expansion_with_two_tables(completer, complete_event): sql = 'SELECT * FROM public."select" JOIN custom.users ON true' pos = len('SELECT *') completions = completer.get_completions( Document(text=sql, cursor_position=pos), complete_event) cols = ('"select".id, "select"."insert", "select"."ABC", ' 'users.id, users.phone_number') expected = [wildcard_expansion(cols)] assert completions == expected
def test_wildcard_column_expansion_with_two_tables_and_parent(completer, complete_event): sql = 'SELECT "select".* FROM public."select" JOIN custom.users u ON true' pos = len('SELECT "select".*') completions = completer.get_completions( Document(text=sql, cursor_position=pos), complete_event) col_list = 'id, "select"."insert", "select"."ABC"' expected = [wildcard_expansion(col_list)] assert expected == completions
def test_wildcard_column_expansion_with_table_qualifier(completer, complete_event): sql = 'SELECT "select".* FROM public."select"' pos = len('SELECT "select".*') completions = completer.get_completions( Document(text=sql, cursor_position=pos), complete_event) col_list = 'id, "select"."insert", "select"."ABC"' expected = [wildcard_expansion(col_list)] assert expected == completions
def test_wildcard_column_expansion_with_insert(completer, text): position = text.index('*') + 1 completions = get_result(completer, text, position) expected = [wildcard_expansion('ordered_date, status')] assert expected == completions