summaryrefslogtreecommitdiff
path: root/tests
diff options
context:
space:
mode:
authorAndi Albrecht <albrecht.andi@gmail.com>2019-03-11 12:45:11 +0100
committerAndi Albrecht <albrecht.andi@gmail.com>2019-03-11 12:45:11 +0100
commitcff61bb93a080c1becbf2c6cd33911bc206cf587 (patch)
tree3683be35ea59d970be8ff0f46c77151615405636 /tests
parentfcbccb849b56f29fb7f3ddb958d3e10279f8d939 (diff)
downloadsqlparse-cff61bb93a080c1becbf2c6cd33911bc206cf587.tar.gz
Avoid formatting of psql commands (fixes #469).
Diffstat (limited to 'tests')
-rw-r--r--tests/test_regressions.py7
-rw-r--r--tests/test_tokenize.py6
2 files changed, 13 insertions, 0 deletions
diff --git a/tests/test_regressions.py b/tests/test_regressions.py
index 73bc504..1d52ea7 100644
--- a/tests/test_regressions.py
+++ b/tests/test_regressions.py
@@ -366,3 +366,10 @@ def test_issue322_concurrently_is_keyword():
def test_issue359_index_error_assignments(s):
sqlparse.parse(s)
sqlparse.format(s, strip_comments=True)
+
+
+def test_issue469_copy_as_psql_command():
+ formatted = sqlparse.format(
+ '\\copy select * from foo',
+ keyword_case='upper', identifier_case='capitalize')
+ assert formatted == '\\copy SELECT * FROM Foo'
diff --git a/tests/test_tokenize.py b/tests/test_tokenize.py
index 23a5157..fcd1102 100644
--- a/tests/test_tokenize.py
+++ b/tests/test_tokenize.py
@@ -195,3 +195,9 @@ def test_parse_order_by():
p = sqlparse.parse('ORDER BY')[0]
assert len(p.tokens) == 1
assert p.tokens[0].ttype is T.Keyword
+
+
+def test_cli_commands():
+ p = sqlparse.parse('\\copy')[0]
+ assert len(p.tokens) == 1
+ assert p.tokens[0].ttype == T.Command