Python is_single_statement Examples

Example #1

File: test_helpers.py Project: maxious/ckan

    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

Example #2

File: test_helpers.py Project: vivihuang/ckan

    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