aljaz/turso
1
0
Fork 0
mirror of https://github.com/aljazceru/turso.git synced 2025-12-18 00:54:19 +01:00
Code Issues Packages Projects Releases Wiki Activity

ruff lint fix

Browse Source
This commit is contained in:
pedrocarlo
2025-06-19 16:59:49 -03:00
parent 50c8b2ca2e
commit 80ccca8827
29 changed files with 375 additions and 434 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
antithesis-tests/stress-composer/parallel_driver_insert.py
View File

@@ -1,44 +1,44 @@
#!/usr/bin/env -S python3 -u
import json
import limbo
from utils import generate_random_value
from antithesis.random import get_random
import limbo
from antithesis.random import get_random
from utils import generate_random_value
# Get initial state
try:
con_init = limbo.connect('init_state.db')
con_init = limbo.connect("init_state.db")
except Exception as e:
print(f"Error connecting to database: {e}")
exit(0)
cur_init = con_init.cursor()
tbl_len = cur_init.execute('SELECT count FROM tables').fetchone()[0]
tbl_len = cur_init.execute("SELECT count FROM tables").fetchone()[0]
selected_tbl = get_random() % tbl_len
tbl_schema = json.loads(cur_init.execute(f'SELECT schema FROM schemas WHERE tbl = {selected_tbl}').fetchone()[0])
cols = ', '.join([f'col_{col}' for col in range(tbl_schema['colCount'])])
tbl_schema = json.loads(cur_init.execute(f"SELECT schema FROM schemas WHERE tbl = {selected_tbl}").fetchone()[0])
cols = ", ".join([f"col_{col}" for col in range(tbl_schema["colCount"])])
try:
con = limbo.connect('stress_composer.db')
con = limbo.connect("stress_composer.db")
except limbo.OperationalError as e:
print(f'Failed to open stress_composer.db. Exiting... {e}')
print(f"Failed to open stress_composer.db. Exiting... {e}")
exit(0)
cur = con.cursor()
# insert up to 100 rows in the selected table
insertions = get_random() % 100
print(f'Inserting {insertions} rows...')
print(f"Inserting {insertions} rows...")
for i in range(insertions):
values = [generate_random_value(tbl_schema[f'col_{col}']['data_type']) for col in range(tbl_schema['colCount'])]
values = [generate_random_value(tbl_schema[f"col_{col}"]["data_type"]) for col in range(tbl_schema["colCount"])]
try:
cur.execute(f'''
cur.execute(f"""
INSERT INTO tbl_{selected_tbl} ({cols})
VALUES ({', '.join(values)})
''')
VALUES ({", ".join(values)})
""")
except limbo.OperationalError as e:
if "UNIQUE constraint failed" in str(e):
# Ignore UNIQUE constraint violations
@@ -46,4 +46,3 @@ for i in range(insertions):
else:
# Re-raise other operational errors
raise