summaryrefslogtreecommitdiffstats
path: root/test/test_naive_completion.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/test_naive_completion.py')
-rw-r--r--test/test_naive_completion.py42
1 files changed, 17 insertions, 25 deletions
diff --git a/test/test_naive_completion.py b/test/test_naive_completion.py
index 0bc3bf8..31ac165 100644
--- a/test/test_naive_completion.py
+++ b/test/test_naive_completion.py
@@ -6,56 +6,48 @@ from prompt_toolkit.document import Document
@pytest.fixture
def completer():
import mycli.sqlcompleter as sqlcompleter
+
return sqlcompleter.SQLCompleter(smart_completion=False)
@pytest.fixture
def complete_event():
from unittest.mock import Mock
+
return Mock()
def test_empty_string_completion(completer, complete_event):
- text = ''
+ text = ""
position = 0
- result = list(completer.get_completions(
- Document(text=text, cursor_position=position),
- complete_event))
+ result = list(completer.get_completions(Document(text=text, cursor_position=position), complete_event))
assert result == list(map(Completion, completer.all_completions))
def test_select_keyword_completion(completer, complete_event):
- text = 'SEL'
- position = len('SEL')
- result = list(completer.get_completions(
- Document(text=text, cursor_position=position),
- complete_event))
- assert result == list([Completion(text='SELECT', start_position=-3)])
+ text = "SEL"
+ position = len("SEL")
+ result = list(completer.get_completions(Document(text=text, cursor_position=position), complete_event))
+ assert result == list([Completion(text="SELECT", start_position=-3)])
def test_function_name_completion(completer, complete_event):
- text = 'SELECT MA'
- position = len('SELECT MA')
- result = list(completer.get_completions(
- Document(text=text, cursor_position=position),
- complete_event))
+ text = "SELECT MA"
+ position = len("SELECT MA")
+ result = list(completer.get_completions(Document(text=text, cursor_position=position), complete_event))
assert sorted(x.text for x in result) == ["MASTER", "MAX"]
def test_column_name_completion(completer, complete_event):
- text = 'SELECT FROM users'
- position = len('SELECT ')
- result = list(completer.get_completions(
- Document(text=text, cursor_position=position),
- complete_event))
+ text = "SELECT FROM users"
+ position = len("SELECT ")
+ result = list(completer.get_completions(Document(text=text, cursor_position=position), complete_event))
assert result == list(map(Completion, completer.all_completions))
def test_special_name_completion(completer, complete_event):
- text = '\\'
- position = len('\\')
- result = set(completer.get_completions(
- Document(text=text, cursor_position=position),
- complete_event))
+ text = "\\"
+ position = len("\\")
+ result = set(completer.get_completions(Document(text=text, cursor_position=position), complete_event))
# Special commands will NOT be suggested during naive completion mode.
assert result == set()