From 84725ee64570622135bf68d87823643b826d05b6 Mon Sep 17 00:00:00 2001 From: Andi Albrecht Date: Sat, 29 Jun 2013 19:12:53 +0200 Subject: Improve parsing of PEP249-style placeholder (fixes #103). --- tests/test_parse.py | 11 +++++++++++ 1 file changed, 11 insertions(+) (limited to 'tests') diff --git a/tests/test_parse.py b/tests/test_parse.py index 7016af9..10da8e1 100644 --- a/tests/test_parse.py +++ b/tests/test_parse.py @@ -2,11 +2,15 @@ """Tests sqlparse function.""" +import pytest + from tests.utils import TestCaseBase import sqlparse import sqlparse.sql +from sqlparse import tokens as T + class SQLParseTest(TestCaseBase): """Tests sqlparse.parse().""" @@ -139,3 +143,10 @@ def test_psql_quotation_marks(): # issue83 .... $PROC_2$ LANGUAGE plpgsql;""") assert len(t) == 2 + + +@pytest.mark.parametrize('ph', ['?', ':1', ':foo', '%s', '%(foo)s']) +def test_placeholder(ph): + p = sqlparse.parse(ph)[0].tokens + assert len(p) == 1 + assert p[0].ttype is T.Name.Placeholder -- cgit v1.2.1