summaryrefslogtreecommitdiff
path: root/lib/sqlalchemy/testing/assertsql.py
diff options
context:
space:
mode:
authormike bayer <mike_mp@zzzcomputing.com>2020-06-28 02:34:18 +0000
committerGerrit Code Review <gerrit@bbpush.zzzcomputing.com>2020-06-28 02:34:18 +0000
commit021994685b55c0bca0d50c163ff5f9e9441fc037 (patch)
tree45726733668a15f21c5ebe6b3754046f5d527860 /lib/sqlalchemy/testing/assertsql.py
parentba047cc8cab22541e88ce91936162d6e8164991a (diff)
parent08c46eea924d23a234bf3feea1a928eb8ae8a00a (diff)
downloadsqlalchemy-021994685b55c0bca0d50c163ff5f9e9441fc037.tar.gz
Merge "ORM executemany returning"
Diffstat (limited to 'lib/sqlalchemy/testing/assertsql.py')
-rw-r--r--lib/sqlalchemy/testing/assertsql.py8
1 files changed, 8 insertions, 0 deletions
diff --git a/lib/sqlalchemy/testing/assertsql.py b/lib/sqlalchemy/testing/assertsql.py
index ef324635e..caf61a806 100644
--- a/lib/sqlalchemy/testing/assertsql.py
+++ b/lib/sqlalchemy/testing/assertsql.py
@@ -325,6 +325,14 @@ class EachOf(AssertRule):
super(EachOf, self).no_more_statements()
+class Conditional(EachOf):
+ def __init__(self, condition, rules, else_rules):
+ if condition:
+ super(Conditional, self).__init__(*rules)
+ else:
+ super(Conditional, self).__init__(*else_rules)
+
+
class Or(AllOf):
def process_statement(self, execute_observed):
for rule in self.rules: