@@ -860,6 +860,34 @@ def test_null_terminated(self):
860860 one = self .exec_naked ("REDISQL.EXEC" , "NULL" , "SELECT 1" + b'\x00 ' )
861861 self .assertEquals (one , [[1 ]])
862862
863+ class TestBlankAfterSemicolon (TestRediSQLWithExec ):
864+ def test_whitespace_after_semicolon (self ):
865+ with DB (self , "NULL" ):
866+ one = self .exec_naked ("REDISQL.EXEC" , "NULL" , "SELECT 1; " )
867+ self .assertEquals (one , [[1 ]])
868+
869+ def test_newline_after_semicolon (self ):
870+ with DB (self , "NULL" ):
871+ one = self .exec_naked ("REDISQL.EXEC" , "NULL" , "SELECT 1;\n " )
872+ self .assertEquals (one , [[1 ]])
873+
874+ def test_mix_after_semicolon (self ):
875+ with DB (self , "NULL" ):
876+ one = self .exec_naked ("REDISQL.EXEC" , "NULL" , "SELECT 1; \n " )
877+ self .assertEquals (one , [[1 ]])
878+
879+ def test_whitespace_after_semicolon_then_query (self ):
880+ with DB (self , "NULL" ):
881+ one = self .exec_naked ("REDISQL.EXEC" , "NULL" , "SELECT 1; SELECT 2;" )
882+ self .assertEquals (one , [[2 ]])
883+
884+ def test_newline_after_semicolon_then_query (self ):
885+ with DB (self , "NULL" ):
886+ one = self .exec_naked ("REDISQL.EXEC" , "NULL" , "SELECT 1;\n SELECT 2;" )
887+ self .assertEquals (one , [[2 ]])
888+
889+
890+
863891if __name__ == '__main__' :
864892 unittest .main ()
865893
0 commit comments