Preserve whitespace after subgroups.

1 files changed, 6 insertions, 0 deletions

diff --git a/tests/test_format.py b/tests/test_format.py
index 7a2c655..c9c6f46 100644
--- a/tests/test_format.py
+++ b/tests/test_format.py
@@ -68,6 +68,12 @@ class TestFormat(TestCaseBase):
         self.assertRaises(sqlparse.SQLParseError, sqlparse.format, s,
                           strip_whitespace=None)
 
+    def test_preserve_ws(self):
+        # preserve at least one whitespace after subgroups
+        f = lambda sql: sqlparse.format(sql, strip_whitespace=True)
+        s = 'select\n* /* foo */  from bar '
+        self.ndiffAssertEqual(f(s), 'select * /* foo */ from bar')
+
     def test_outputformat(self):
         sql = 'select * from foo;'
         self.assertRaises(sqlparse.SQLParseError, sqlparse.format, sql,