Add check for single-line quoted strings which are SQL statements

foresmac · foresmac · commit 98609acfb997 · 2015-11-16T20:00:01.000-06:00
diff --git a/grammars/python.cson b/grammars/python.cson
@@ -1536,7 +1536,7 @@
         ]
       }
       {
-        'begin': '(\')(?=\\s*(SELECT|INSERT|UPDATE|DELETE|CREATE|REPLACE|ALTER))'
+        'begin': '(\')(?=\\s*(SELECT|INSERT|UPDATE|DELETE|CREATE|REPLACE|ALTER|WITH))'
         'beginCaptures':
           '1':
             'name': 'punctuation.definition.string.begin.python'
diff --git a/spec/python-spec.coffee b/spec/python-spec.coffee
@@ -327,3 +327,11 @@ describe "Python grammar", ->
       expect(tokens[7][0]).toEqual value: 'SELECT COUNT(*)', scopes: ['source.python', scope]
       expect(tokens[8][0]).toEqual value: 'FROM example_cte', scopes: ['source.python', scope]
       expect(tokens[9][0]).toEqual value: delim, scopes: ['source.python', scope, 'punctuation.definition.string.end.python']
+
+it "tokenizes SQL inline highlighting on single line with a CTE", ->
+
+  tokens = grammar.tokenizeLines('WITH example_cte AS (SELECT bar FROM foo) SELECT COUNT(*) FROM example_cte')
+
+  expect(tokens[0][0]).toEqual value: '\'', scopes: ['source.python', scope, 'punctuation.definition.string.begin.python']
+  expect(tokens[1][0]).toEqual value: 'WITH example_cte AS (SELECT bar FROM foo) SELECT COUNT(*) FROM example_cte', scopes: ['source.python', scope]
+  expect(tokens[2][0]).toEqual value: '\'', scopes: ['source.python', scope, 'punctuation.definition.string.end.python']

Original file line number	Diff line number	Diff line change
`@@ -1536,7 +1536,7 @@`
`1536`	`1536`	`]`
`1537`	`1537`	`}`
`1538`	`1538`	`{`
`1539`		`- 'begin': '(\')(?=\\s*(SELECT\|INSERT\|UPDATE\|DELETE\|CREATE\|REPLACE\|ALTER))'`
	`1539`	`+ 'begin': '(\')(?=\\s*(SELECT\|INSERT\|UPDATE\|DELETE\|CREATE\|REPLACE\|ALTER\|WITH))'`
`1540`	`1540`	`'beginCaptures':`
`1541`	`1541`	`'1':`
`1542`	`1542`	`'name': 'punctuation.definition.string.begin.python'`