def test_is_single_statement(self):
singles = ['SELECT * FROM footable',
'SELECT * FROM "bartable"',
'SELECT * FROM "bartable";',
'SELECT * FROM "bart;able";',
"select 'foo'||chr(59)||'bar'"]
multiples = ['SELECT * FROM abc; SET LOCAL statement_timeout to'
'SET LOCAL statement_timeout to; SELECT * FROM abc',
'SELECT * FROM "foo"; SELECT * FROM "abc"']
for single in singles:
assert postgres_backend.is_single_statement(single) is True
for multiple in multiples:
assert postgres_backend.is_single_statement(multiple) is False
def test_is_single_statement(self):
singles = [
'SELECT * FROM footable', 'SELECT * FROM "bartable"',
'SELECT * FROM "bartable";', 'SELECT * FROM "bart;able";',
"select 'foo'||chr(59)||'bar'"
]
multiples = [
'SELECT * FROM abc; SET LOCAL statement_timeout to'
'SET LOCAL statement_timeout to; SELECT * FROM abc',
'SELECT * FROM "foo"; SELECT * FROM "abc"'
]
for single in singles:
assert postgres_backend.is_single_statement(single) is True
for multiple in multiples:
assert postgres_backend.is_single_statement(multiple) is False