summaryrefslogtreecommitdiff
path: root/lib/sqlalchemy/testing/assertsql.py
diff options
context:
space:
mode:
authorMike Bayer <mike_mp@zzzcomputing.com>2012-11-20 11:03:01 -0500
committerMike Bayer <mike_mp@zzzcomputing.com>2012-11-20 11:03:01 -0500
commitfe8f8349c9f46abe37cd9de7876df0eeb5e12c94 (patch)
tree29b6b1f953d24f4b49a84069f13f3ef72e594212 /lib/sqlalchemy/testing/assertsql.py
parent2799a674038bef5a81102935ae43c70163f0c556 (diff)
parentead37e4ec0dd5750769226befea043d9527f869e (diff)
downloadsqlalchemy-fe8f8349c9f46abe37cd9de7876df0eeb5e12c94.tar.gz
- an enormous merge just because I committed a one line log entry. the joy of DVCS
Diffstat (limited to 'lib/sqlalchemy/testing/assertsql.py')
-rw-r--r--lib/sqlalchemy/testing/assertsql.py9
1 files changed, 8 insertions, 1 deletions
diff --git a/lib/sqlalchemy/testing/assertsql.py b/lib/sqlalchemy/testing/assertsql.py
index 08ee55d57..d955d1554 100644
--- a/lib/sqlalchemy/testing/assertsql.py
+++ b/lib/sqlalchemy/testing/assertsql.py
@@ -3,6 +3,7 @@ from ..engine.default import DefaultDialect
from .. import util
import re
+
class AssertRule(object):
def process_execute(self, clauseelement, *multiparams, **params):
@@ -40,6 +41,7 @@ class AssertRule(object):
assert False, 'Rule has not been consumed'
return self.is_consumed()
+
class SQLMatchRule(AssertRule):
def __init__(self):
self._result = None
@@ -56,6 +58,7 @@ class SQLMatchRule(AssertRule):
return True
+
class ExactSQL(SQLMatchRule):
def __init__(self, sql, params=None):
@@ -138,6 +141,7 @@ class RegexSQL(SQLMatchRule):
_received_statement,
_received_parameters)
+
class CompiledSQL(SQLMatchRule):
def __init__(self, statement, params):
@@ -217,6 +221,7 @@ class CountStatements(AssertRule):
% (self.count, self._statement_count)
return True
+
class AllOf(AssertRule):
def __init__(self, *rules):
@@ -244,6 +249,7 @@ class AllOf(AssertRule):
def consume_final(self):
return len(self.rules) == 0
+
def _process_engine_statement(query, context):
if util.jython:
@@ -256,6 +262,7 @@ def _process_engine_statement(query, context):
query = re.sub(r'\n', '', query)
return query
+
def _process_assertion_statement(query, context):
paramstyle = context.dialect.paramstyle
if paramstyle == 'named':
@@ -275,6 +282,7 @@ def _process_assertion_statement(query, context):
return query
+
class SQLAssert(object):
rules = None
@@ -311,4 +319,3 @@ class SQLAssert(object):
executemany)
asserter = SQLAssert()
-