def test_wildcard_column_expansion_with_alias(self, 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)]
self.assertListEqual(expected, completions)
def test_wildcard_column_expansion_with_function(self, completer, text):
position = len('SELECT *')
completions = get_result(completer, text, position)
col_list = 'x'
expected = [wildcard_expansion(col_list)]
self.assertListEqual(expected, completions)
def test_wildcard_column_expansion_with_two_tables_and_parent(self, 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)]
self.assertListEqual(expected, completions)
def test_wildcard_column_expansion_with_two_tables(self, 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)]
self.assertListEqual(expected, completions)
def test_wildcard_column_expansion_with_table_qualifier(self, completer, text_and_expected):
text = text_and_expected[0]
expected = text_and_expected[1]
position = len('SELECT users.*')
completions = get_result(completer, text, position)
expected = [wildcard_expansion(expected)]
self.assertListEqual(expected, completions)
def test_wildcard_column_expansion_with_two_tables(self, 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)]
self.assertListEqual(expected, completions)
def test_wildcard_column_expansion_with_table_qualifier(self, 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)]
self.assertListEqual(expected, completions)
def test_wildcard_column_expansion_with_alias_qualifier(self, 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)]
self.assertListEqual(expected, completions)
def test_wildcard_column_expansion_with_insert(self, completer, text):
position = text.index('*') + 1
completions = get_result(completer, text, position)
expected = [wildcard_expansion('ordered_date, status')]
self.assertListEqual(expected, completions)