Merge pull request #170 from darikg/better_dots

Better handle schema qualifications in current word

2 files changed, 7 insertions, 3 deletions

diff --git a/tests/test_smart_completion_multiple_schemata.py b/tests/test_smart_completion_multiple_schemata.py
index 66653f62..6190d774 100644
--- a/tests/test_smart_completion_multiple_schemata.py
+++ b/tests/test_smart_completion_multiple_schemata.py
@@ -145,9 +145,7 @@ def test_suggested_table_names_with_schema_dot(completer, complete_event):
     assert set(result) == set([
         Completion(text='users', start_position=0),
         Completion(text='products', start_position=0),
-        Completion(text='shipments', start_position=0),
-        Completion(text='func3', start_position=0),
-        Completion(text='func4', start_position=0)])
+        Completion(text='shipments', start_position=0)])
 
 def test_suggested_column_names_with_qualified_alias(completer, complete_event):
     """
diff --git a/tests/test_sqlcompletion.py b/tests/test_sqlcompletion.py
index 1e3ae66b..063bf28c 100644
--- a/tests/test_sqlcompletion.py
+++ b/tests/test_sqlcompletion.py
@@ -253,3 +253,9 @@ def test_specials_not_included_after_initial_token():
                                'create table foo (dt d')
 
     assert sorted_dicts(suggestions) == sorted_dicts([{'type': 'keyword'}])
+
+
+def test_drop_schema_qualified_table_suggests_only_tables():
+    text = 'DROP TABLE schema_name.table_name'
+    suggestions = suggest_type(text, text)
+    assert suggestions == [{'type': 'table', 'schema': 'schema_name'}]
\ No newline at end of file