mirror of
https://github.com/aljazceru/turso.git
synced 2025-12-18 09:04:19 +01:00
34 lines
974 B
Python
Executable File
34 lines
974 B
Python
Executable File
#!/usr/bin/env -S python3 -u
|
|
|
|
import json
|
|
import limbo
|
|
from utils import generate_random_value
|
|
from antithesis.random import get_random
|
|
|
|
# Get initial state
|
|
con_init = limbo.connect('init_state.db')
|
|
cur_init = con_init.cursor()
|
|
|
|
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])
|
|
|
|
# get primary key column
|
|
pk = tbl_schema['pk']
|
|
# get non-pk columns
|
|
cols = [f'col_{col}' for col in range(tbl_schema['colCount']) if col != pk]
|
|
|
|
con = limbo.connect('stress_composer.db')
|
|
cur = con.cursor()
|
|
|
|
deletions = get_random() % 100
|
|
print(f'Attempt to delete {deletions} rows in tbl_{selected_tbl}...')
|
|
|
|
for i in range(deletions):
|
|
where_clause = f"col_{pk} = {generate_random_value(tbl_schema[f'col_{pk}']['data_type'])}"
|
|
|
|
cur.execute(f'''
|
|
DELETE FROM tbl_{selected_tbl} WHERE {where_clause}
|
|
''')
|
|
|