Fix incorrect splitting with double quotes and a new line

fredyw · fredyw · commit 002f37dde390 · 2018-03-09T14:36:37.000-06:00
diff --git a/sqlparse/keywords.py b/sqlparse/keywords.py
@@ -64,6 +64,7 @@ def is_keyword(value):
         (r'-?\d+(?![_A-Z&Agrave;-&Uuml;])', tokens.Number.Integer),
         (r"'(''|\\\\|\\'|[^'])*'", tokens.String.Single),
         # not a real string literal in ANSI SQL:
+        (r'"(""|\\\\|\\"|[^"])*"', tokens.String.Symbol),
         (r'(""|".*?[^\\]")', tokens.String.Symbol),
         # sqlite names can be escaped with [square brackets]. left bracket
         # cannot be preceded by word character or a right bracket --
diff --git a/tests/test_split.py b/tests/test_split.py
@@ -139,3 +139,13 @@ def test_split_simple():
     assert len(stmts) == 2
     assert stmts[0] == 'select * from foo;'
     assert stmts[1] == 'select * from bar;'
+
+
+def test_split_quotes_with_new_line():
+    stmts = sqlparse.split('select "foo\nbar"')
+    assert len(stmts) == 1
+    assert stmts[0] == 'select "foo\nbar"'
+
+    stmts = sqlparse.split("select 'foo\n\bar'")
+    assert len(stmts) == 1
+    assert stmts[0] == "select 'foo\n\bar'"
