def test_set_schema(completer):
text = ('SET SCHEMA ')
result = result_set(completer, text)
assert result == set([
schema(u"'blog'"),
schema(u"'Custom'"),
schema(u"'custom'"),
schema(u"'public'")
])
def test_cased_joins(completer, text):
result = result_set(completer, text)
assert result == set([schema('PUBLIC')] + cased_rels + [
join('"Users" ON "Users".UserID = Users.ID'),
join('Users Users2 ON Users2.ID = Users.PARENTID'),
join('Users Users2 ON Users2.PARENTID = Users.ID'),
])
def test_duplicate_aliases_with_casing(completer, text):
result = result_set(completer, text)
assert result == set([
schema('PUBLIC'),
table('Orders O2'),
table('Users U'),
table('"Users" U'),
table('"select" s'),
view('User_Emails UE'),
view('Functions F'),
function('_custom_fun() cf'),
function('Custom_Fun() CF'),
function('Custom_Func1() CF'),
function('custom_func2() cf'),
function('set_returning_func(x := , y := ) srf',
display='set_returning_func(x, y) srf'),
])
('blog', 'entries', 'entryid', 'blog', 'entacclog', 'entryid'),
('blog', 'entries', 'entryid', 'blog', 'entrytags', 'entryid'),
('blog', 'tags', 'tagid', 'blog', 'entrytags', 'tagid'),
],
},
'defaults': {
'public': {
('orders', 'id'): "nextval('orders_id_seq'::regclass)",
('orders', 'datestamp'): "now()",
('orders', 'status'): "'PENDING'::text",
}
},
}
testdata = MetaData(metadata)
cased_schemas = [schema(x) for x in ('public', 'blog', 'CUSTOM', '"Custom"')]
casing = ('SELECT', 'Orders', 'User_Emails', 'CUSTOM', 'Func1', 'Entries',
'Tags', 'EntryTags', 'EntAccLog', 'EntryID', 'EntryTitle',
'EntryText')
completers = testdata.get_completers(casing)
@parametrize('completer', completers(filtr=True, casing=False,
qualify=no_qual))
@parametrize('table', ['users', '"users"'])
def test_suggested_column_names_from_shadowed_visible_table(completer, table):
result = result_set(completer, 'SELECT FROM ' + table, len('SELECT '))
assert result == set(testdata.columns_functions_and_keywords('users'))
@parametrize('completer', completers(filtr=True, casing=False,
def test_set_schema(completer):
text = ('SET SCHEMA ')
result = result_set(completer, text)
expected = set([schema(u"'public'")])
assert result == expected
def test_table_casing(completer, text):
result = result_set(completer, text)
assert result == set([schema('PUBLIC')] + cased_rels)
def test_aliases_with_casing(completer, text):
result = result_set(completer, text)
assert result == set([schema('PUBLIC')] + cased_aliased_rels)
def test_list_functions_for_special(completer):
result = result_set(completer, r'\df ')
assert result == set([schema('PUBLIC')] +
[function(f) for f in cased_func_names])