Fix formatting issue with extended CASE statements.

1 files changed, 8 insertions, 0 deletions

diff --git a/tests/test_format.py b/tests/test_format.py
index b56ceaf..f9740d5 100644
--- a/tests/test_format.py
+++ b/tests/test_format.py
@@ -186,6 +186,14 @@ class TestFormatReindent(TestCaseBase):
                                                '    else 5',
                                                'end']))
 
+    def test_case2(self):
+        f = lambda sql: sqlparse.format(sql, reindent=True)
+        s = 'case(foo) when bar = 1 then 2 else 3 end'
+        self.ndiffAssertEqual(f(s), '\n'.join(['case(foo)',
+                                               '    when bar = 1 then 2',
+                                               '    else 3',
+                                               'end']))
+
     def test_nested_identifier_list(self):  # issue4
         f = lambda sql: sqlparse.format(sql, reindent=True)
         s = '(foo as bar, bar1, bar2 as bar3, b4 as b5)'