fix CASE statement when else_ is zero

1 files changed, 33 insertions, 5 deletions

diff --git a/test/sql/case_statement.py b/test/sql/case_statement.py
index 1a75d5289..946279b9d 100644
--- a/test/sql/case_statement.py
+++ b/test/sql/case_statement.py
@@ -17,15 +17,20 @@ class CaseTest(testbase.PersistTest):
         	{'pk':1, 'info':'pk_1_data'},
         	{'pk':2, 'info':'pk_2_data'},
         	{'pk':3, 'info':'pk_3_data'},
-        	{'pk':4, 'info':'pk_4_data'},
-    	    {'pk':5, 'info':'pk_5_data'})
+ 		{'pk':4, 'info':'pk_4_data'},
+		{'pk':5, 'info':'pk_5_data'},
+		{'pk':6, 'info':'pk_6_data'})
     def tearDownAll(self):
         info_table.drop()
     
     def testcase(self):
-        inner = select([case([[info_table.c.pk < 3, literal('lessthan3', type=String)],
-        	[info_table.c.pk >= 3, literal('gt3', type=String)]]).label('x'),
-        	info_table.c.pk, info_table.c.info], from_obj=[info_table]).alias('q_inner')
+        inner = select([case([
+		[info_table.c.pk < 3, 
+                        literal('lessthan3', type=String)],
+        	[and_(info_table.c.pk >= 3, info_table.c.pk < 7), 
+                        literal('gt3', type=String)]]).label('x'),
+        	info_table.c.pk, info_table.c.info], 
+                from_obj=[info_table]).alias('q_inner')
 
         inner_result = inner.execute().fetchall()
 
@@ -35,12 +40,14 @@ class CaseTest(testbase.PersistTest):
         # gt3 3 pk_3_data
         # gt3 4 pk_4_data
         # gt3 5 pk_5_data
+        # gt3 6 pk_6_data
         assert inner_result == [
             ('lessthan3', 1, 'pk_1_data'),
             ('lessthan3', 2, 'pk_2_data'),
             ('gt3', 3, 'pk_3_data'),
             ('gt3', 4, 'pk_4_data'),
             ('gt3', 5, 'pk_5_data'),
+            ('gt3', 6, 'pk_6_data')
         ]
 
         outer = select([inner])
@@ -53,6 +60,27 @@ class CaseTest(testbase.PersistTest):
             ('gt3', 3, 'pk_3_data'),
             ('gt3', 4, 'pk_4_data'),
             ('gt3', 5, 'pk_5_data'),
+            ('gt3', 6, 'pk_6_data')
+        ]
+
+        w_else = select([case([
+		[info_table.c.pk < 3, 
+                        literal(3, type=Integer)],
+        	[and_(info_table.c.pk >= 3, info_table.c.pk < 6), 
+                        literal(6, type=Integer)]],
+                else_ = 0).label('x'),
+        	info_table.c.pk, info_table.c.info], 
+                from_obj=[info_table]).alias('q_inner')
+
+        else_result = w_else.execute().fetchall()
+
+        assert else_result == [
+            (3, 1, 'pk_1_data'),
+            (3, 2, 'pk_2_data'),
+            (6, 3, 'pk_3_data'),
+            (6, 4, 'pk_4_data'),
+            (6, 5, 'pk_5_data'),
+            (0, 6, 'pk_6_data')
         ]
 
 if __name__ == "__main__":