def test_keyword_upper_casing(self):
new_completer = PGCompleter(smart_completion=True,
settings={'keyword_casing': 'upper'})
text = 'sel'
position = len(text)
result = new_completer.get_completions(
Document(text=text, cursor_position=position), self.complete_event)
# then completions should now be lower case
self.assertEqual(result, [
Completion(
text='SELECT', start_position=-3, display_meta="keyword")
])
def test_keyword_auto_casing(self):
new_completer = PGCompleter(smart_completion=True,
settings={'keyword_casing': 'auto'})
# if text is lower case
text = 'sel'
position = len(text)
result = new_completer.get_completions(
Document(text=text, cursor_position=position), self.complete_event)
# then completions should be lower case as well
self.assertEqual(result, [
Completion(
text='select', start_position=-3, display_meta="keyword")
])
class TestNaiveCompletion(unittest.TestCase):
"""Methods for testing non-smart completion"""
def setUp(self):
self.completer = PGCompleter(smart_completion=False)
self.complete_event = Mock()
def test_empty_string_completion(self):
text = ''
position = 0
result = set(
self.completer.get_completions(
Document(text=text, cursor_position=position),
self.complete_event))
self.assertSetEqual(
result, set(map(Completion, self.completer.all_completions)))
def test_select_keyword_completion(self):
text = 'SEL'
position = len('SEL')
result = set(
self.completer.get_completions(
Document(text=text, cursor_position=position),
self.complete_event))
self.assertSetEqual(
result, set([Completion(text='SELECT', start_position=-3)]))
def test_function_name_completion(self):
text = 'SELECT MA'
position = len('SELECT MA')
result = set(
self.completer.get_completions(
Document(text=text, cursor_position=position),
self.complete_event))
self.assertSetEqual(
result,
set([
Completion(text='MATERIALIZED VIEW', start_position=-2),
Completion(text='MAX', start_position=-2),
Completion(text='MAXEXTENTS', start_position=-2)
]))
def test_column_name_completion(self):
text = 'SELECT FROM users'
position = len('SELECT ')
result = set(
self.completer.get_completions(
Document(text=text, cursor_position=position),
self.complete_event))
self.assertSetEqual(
result, set(map(Completion, self.completer.all_completions)))
# {{ PGToolsService EDIT }}
# Disabling as we will not support cli-only features
# def test_paths_completion(self):
# text = '\i '
# position = len(text)
# result = set(self.completer.get_completions(
# Document(text=text, cursor_position=position),
# self.complete_event,
# smart_completion=True))
# # Set comparison: > means "is superset"
# self.assertTrue(result > set([Completion(text="setup.py", start_position=0)]))
def test_alter_well_known_keywords_completion(self):
text = 'ALTER '
position = len(text)
result = set(
self.completer.get_completions(Document(text=text,
cursor_position=position),
self.complete_event,
smart_completion=True))
# Set comparison: > means "is superset"
self.assertTrue(result > set([
Completion(text="DATABASE", display_meta='keyword'),
Completion(text="TABLE", display_meta='keyword'),
Completion(text="SYSTEM", display_meta='keyword'),
]))
self.assertTrue(
Completion(text="CREATE", display_meta="keyword") not in result)
def test_keyword_lower_casing(self):
new_completer = PGCompleter(smart_completion=True,
settings={'keyword_casing': 'lower'})
text = 'SEL'
position = len(text)
result = set(
new_completer.get_completions(
Document(text=text, cursor_position=position),
self.complete_event))
# then completions should now be lower case
self.assertSetEqual(
result,
set([
Completion(text='select',
start_position=-3,
display_meta="keyword")
]))
def test_keyword_upper_casing(self):
new_completer = PGCompleter(smart_completion=True,
settings={'keyword_casing': 'upper'})
text = 'sel'
position = len(text)
result = set(
new_completer.get_completions(
Document(text=text, cursor_position=position),
self.complete_event))
# then completions should now be lower case
self.assertSetEqual(
result,
set([
Completion(text='SELECT',
start_position=-3,
display_meta="keyword")
]))
def test_keyword_auto_casing(self):
new_completer = PGCompleter(smart_completion=True,
settings={'keyword_casing': 'auto'})
# if text is lower case
text = 'sel'
position = len(text)
result = set(
new_completer.get_completions(
Document(text=text, cursor_position=position),
self.complete_event))
# then completions should be lower case as well
self.assertSetEqual(
result,
set([
Completion(text='select',
start_position=-3,
display_meta="keyword")
]))