summaryrefslogtreecommitdiffstats
path: root/tests/test_smart_completion_public_schema_only.py
diff options
context:
space:
mode:
authorkoljonen <koljonen@outlook.com>2016-06-18 21:21:17 +0200
committerkoljonen <koljonen@outlook.com>2016-06-18 21:21:17 +0200
commit42cf384ede0d6114311c3145082a8aa529e26c03 (patch)
treebd7a9576f86b1172a39ac7744a5e41bb14442a62 /tests/test_smart_completion_public_schema_only.py
parentc3c18bcebe125dffd6c4773f2ce0f910362b4fac (diff)
In completion tests, get datatypes from metadata instead of manually listing them
Diffstat (limited to 'tests/test_smart_completion_public_schema_only.py')
-rw-r--r--tests/test_smart_completion_public_schema_only.py5
1 files changed, 1 insertions, 4 deletions
diff --git a/tests/test_smart_completion_public_schema_only.py b/tests/test_smart_completion_public_schema_only.py
index e1729240..57f138dd 100644
--- a/tests/test_smart_completion_public_schema_only.py
+++ b/tests/test_smart_completion_public_schema_only.py
@@ -5,7 +5,6 @@ from metadata import (MetaData, alias, name_join, fk_join, join, keyword,
view,
function,
column,
- datatype,
wildcard_expansion)
from prompt_toolkit.document import Document
from pgcli.packages.function_metadata import FunctionMetadata, ForeignKey
@@ -582,9 +581,7 @@ def test_suggest_datatype(text, completer, complete_event):
pos = len(text)
result = completer.get_completions(
Document(text=text, cursor_position=pos), complete_event)
- assert set(result) == set(testdata.schemas() + [
- datatype('custom_type1'),
- datatype('custom_type2'),
+ assert set(result) == set(testdata.schemas() + testdata.datatypes() + [
table('users'),
table('"Users"'),
table('orders'),