Add failing test for issue #227.

1 files changed, 7 insertions, 0 deletions

diff --git a/tests/test_regressions.py b/tests/test_regressions.py
index ca7dd5b..acadeff 100644
--- a/tests/test_regressions.py
+++ b/tests/test_regressions.py
@@ -291,3 +291,10 @@ def test_issue212_py2unicode():
 def test_issue213_leadingws():
     sql = " select * from foo"
     assert sqlparse.format(sql, strip_whitespace=True) == "select * from foo"
+
+
+def test_issue227_gettype_cte():
+    select_stmt = sqlparse.parse('SELECT 1, 2, 3 FROM foo;')[0]
+    assert select_stmt.get_type() == 'SELECT'
+    with_stmt = sqlparse.parse('WITH foo AS (SELECT 1, 2, 3) SELECT * FROM foo;')[0]
+    assert with_stmt.get_type() == 'SELECT'