diff options
| author | Darik Gamble <darik.gamble@gmail.com> | 2015-05-19 19:58:38 -0400 |
|---|---|---|
| committer | Darik Gamble <darik.gamble@gmail.com> | 2015-05-20 17:59:32 -0400 |
| commit | 70732f0025dffc0ae156cd9a7b56b872c3534b6d (patch) | |
| tree | 5909258c24f0631f6f26bd559bf485f49cae6557 | |
| parent | d0eef1bb1447737b7341da268e3dc6361d17a9c3 (diff) | |
`drop schema` and `create schema` should suggest schemas
| -rw-r--r-- | pgcli/packages/sqlcompletion.py | 3 | ||||
| -rw-r--r-- | tests/test_sqlcompletion.py | 5 |
2 files changed, 8 insertions, 0 deletions
diff --git a/pgcli/packages/sqlcompletion.py b/pgcli/packages/sqlcompletion.py index 1b2d1164..709540cd 100644 --- a/pgcli/packages/sqlcompletion.py +++ b/pgcli/packages/sqlcompletion.py @@ -277,6 +277,9 @@ def suggest_based_on_last_token(token, text_before_cursor, full_text, identifier # "\c <db", "use <db>", "DROP DATABASE <db>", # "CREATE DATABASE <newdb> WITH TEMPLATE <db>" return [{'type': 'database'}] + elif token_v == 'schema': + # DROP SCHEMA schema_name + return [{'type': 'schema'}] elif token_v.endswith(',') or token_v == '=': prev_keyword, text_before_cursor = find_prev_keyword(text_before_cursor) if prev_keyword: diff --git a/tests/test_sqlcompletion.py b/tests/test_sqlcompletion.py index 206c77b1..c495e5ac 100644 --- a/tests/test_sqlcompletion.py +++ b/tests/test_sqlcompletion.py @@ -384,3 +384,8 @@ def test_handle_pre_completion_comma_gracefully(text): suggestions = suggest_type(text, text) assert iter(suggestions) + + +def test_drop_schema_suggests_schemas(): + sql = 'DROP SCHEMA ' + assert suggest_type(sql, sql) == [{'type': 'schema'}]
\ No newline at end of file |