mirror of
https://gitlab.archlinux.org/archlinux/aurweb.git
synced 2025-02-03 10:43:03 +01:00
aurweb.db: add commit(), add() and autocommit arg
With the addition of these two, some code has been swapped to use these in some of the other db wrappers with an additional autocommit kwarg in create and delete, to control batch transactions. Signed-off-by: Kevin Morris <kevr@0cost.org>
This commit is contained in:
parent
bd8f528011
commit
40448ccd34
2 changed files with 41 additions and 5 deletions
|
@ -273,6 +273,31 @@ def test_create_delete():
|
|||
record = db.query(AccountType, AccountType.AccountType == "test").first()
|
||||
assert record is None
|
||||
|
||||
# Create and delete a record with autocommit=False.
|
||||
db.create(AccountType, AccountType="test", autocommit=False)
|
||||
db.commit()
|
||||
db.delete(AccountType, AccountType.AccountType == "test", autocommit=False)
|
||||
db.commit()
|
||||
record = db.query(AccountType, AccountType.AccountType == "test").first()
|
||||
assert record is None
|
||||
|
||||
|
||||
def test_add_commit():
|
||||
# Use db.add and db.commit to add a temporary record.
|
||||
account_type = AccountType(AccountType="test")
|
||||
db.add(account_type)
|
||||
db.commit()
|
||||
|
||||
# Assert it got created in the DB.
|
||||
assert bool(account_type.ID)
|
||||
|
||||
# Query the DB for it and compare the record with our object.
|
||||
record = db.query(AccountType, AccountType.AccountType == "test").first()
|
||||
assert record == account_type
|
||||
|
||||
# Remove the record.
|
||||
db.delete(AccountType, AccountType.ID == account_type.ID)
|
||||
|
||||
|
||||
def test_connection_executor_mysql_paramstyle():
|
||||
executor = db.ConnectionExecutor(None, backend="mysql")
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue