diff options
Diffstat (limited to 'test/orm/test_dynamic.py')
| -rw-r--r-- | test/orm/test_dynamic.py | 66 |
1 files changed, 52 insertions, 14 deletions
diff --git a/test/orm/test_dynamic.py b/test/orm/test_dynamic.py index df335f0f6..714878f4e 100644 --- a/test/orm/test_dynamic.py +++ b/test/orm/test_dynamic.py @@ -1791,14 +1791,33 @@ class WriteOnlyBulkTest( "INSERT INTO users (name) VALUES (:name)", [{"name": "x"}], ), - CompiledSQL( - "INSERT INTO addresses (user_id, email_address) " - "VALUES (:user_id, :email_address) " - "RETURNING addresses.id", + Conditional( + testing.requires.insert_executemany_returning.enabled, + [ + CompiledSQL( + "INSERT INTO addresses " + "(user_id, email_address) " + "VALUES (:user_id, :email_address) " + "RETURNING addresses.id", + [ + {"user_id": uid, "email_address": "e1"}, + {"user_id": uid, "email_address": "e2"}, + {"user_id": uid, "email_address": "e3"}, + ], + ) + ], [ - {"user_id": uid, "email_address": "e1"}, - {"user_id": uid, "email_address": "e2"}, - {"user_id": uid, "email_address": "e3"}, + CompiledSQL( + "INSERT INTO addresses " + "(user_id, email_address) " + "VALUES (:user_id, :email_address)", + param, + ) + for param in [ + {"user_id": uid, "email_address": "e1"}, + {"user_id": uid, "email_address": "e2"}, + {"user_id": uid, "email_address": "e3"}, + ] ], ), ], @@ -1863,14 +1882,33 @@ class WriteOnlyBulkTest( "INSERT INTO users (name) VALUES (:name)", [{"name": "x"}], ), - CompiledSQL( - "INSERT INTO addresses (user_id, email_address) " - "VALUES (:user_id, :email_address) " - "RETURNING addresses.id", + Conditional( + testing.requires.insert_executemany_returning.enabled, + [ + CompiledSQL( + "INSERT INTO addresses " + "(user_id, email_address) " + "VALUES (:user_id, :email_address) " + "RETURNING addresses.id", + [ + {"user_id": uid, "email_address": "e1"}, + {"user_id": uid, "email_address": "e2"}, + {"user_id": uid, "email_address": "e3"}, + ], + ) + ], [ - {"user_id": uid, "email_address": "e1"}, - {"user_id": uid, "email_address": "e2"}, - {"user_id": uid, "email_address": "e3"}, + CompiledSQL( + "INSERT INTO addresses " + "(user_id, email_address) " + "VALUES (:user_id, :email_address)", + param, + ) + for param in [ + {"user_id": uid, "email_address": "e1"}, + {"user_id": uid, "email_address": "e2"}, + {"user_id": uid, "email_address": "e3"}, + ] ], ), ], |