Fix BEGIN READ WRITE regression for Redshift syntax

sqlparse/engine/statement_splitter.py

@@ -132,7 +132,8 @@ def _change_splitlevel(self, ttype, value):
                 (ttype is T.Keyword or ttype is T.Name) and \
                 unified in ('TRANSACTION', 'WORK', 'TRAN',
                             'DISTRIBUTED', 'DEFERRED',
-                            'IMMEDIATE', 'EXCLUSIVE'):
+                            'IMMEDIATE', 'EXCLUSIVE',
+                            'ISOLATION', 'READ'):
             self._seen_begin = False
             if self._block_stack and self._block_stack[-1] == 'BEGIN':
                 self._block_stack.pop()

tests/test_split.py

@@ -285,6 +285,33 @@ def test_split_begin_transaction_formatted():  # issue826
     assert stmts[2].startswith('INSERT')
     assert stmts[3] == 'END\nTRANSACTION;'
 
+# https://github.com/andialbrecht/sqlparse/issues/843
+def test_split_begin_read_write():
+    # Redshift BEGIN READ WRITE should not be treated as a block start
+    sql = """BEGIN READ WRITE;
+DELETE FROM schema.table_a USING table_a_temp
+WHERE schema.table_a.id = table_a_temp.id;
+INSERT INTO schema.table_a SELECT * FROM table_a_temp;
+END TRANSACTION;"""
+    stmts = sqlparse.split(sql)
+    assert len(stmts) == 4
+    assert stmts[0] == 'BEGIN READ WRITE;'
+    assert stmts[1].startswith('DELETE')
+    assert stmts[2].startswith('INSERT')
+    assert stmts[3] == 'END TRANSACTION;'
+
+
+# https://github.com/andialbrecht/sqlparse/issues/843
+def test_split_begin_isolation_level_read_only():
+    sql = """BEGIN ISOLATION LEVEL SERIALIZABLE READ ONLY;
+SELECT 1;
+END TRANSACTION;"""
+    stmts = sqlparse.split(sql)
+    assert len(stmts) == 3
+    assert stmts[0] == 'BEGIN ISOLATION LEVEL SERIALIZABLE READ ONLY;'
+    assert stmts[1] == 'SELECT 1;'
+    assert stmts[2] == 'END TRANSACTION;'
+
 
 def test_split_anonymous_begin_end_for():  # issue845 Case 1
     sql = """
@@ -372,5 +399,3 @@ def test_split_standalone_for_update():
     assert len(stmts) == 2
     assert stmts[0] == "SELECT * FROM foo FOR UPDATE;"
     assert stmts[1] == "SELECT 3;"
-
-