mirror of
https://github.com/aljazceru/turso.git
synced 2025-12-17 00:24:21 +01:00
4805c6b06b
The ProgrammingError exception is thrown when tables, indexes, or columns are dropped in parallel. Let's not fail the Antithesis test drivers when that happens.
50 lines
1.2 KiB
Python
Executable File
50 lines
1.2 KiB
Python
Executable File
#!/usr/bin/env -S python3 -u
|
|
|
|
|
|
import turso
|
|
from antithesis.random import get_random
|
|
|
|
# Get initial state
|
|
try:
|
|
con_init = turso.connect("init_state.db")
|
|
except Exception as e:
|
|
print(f"Error connecting to database: {e}")
|
|
exit(0)
|
|
|
|
cur_init = con_init.cursor()
|
|
|
|
# Get all existing tables from schemas
|
|
existing_schemas = cur_init.execute("SELECT tbl FROM schemas").fetchall()
|
|
if not existing_schemas:
|
|
print("No tables found in schemas")
|
|
exit(0)
|
|
|
|
# Select a random table
|
|
selected_idx = get_random() % len(existing_schemas)
|
|
selected_tbl = existing_schemas[selected_idx][0]
|
|
|
|
try:
|
|
con = turso.connect("stress_composer.db")
|
|
except Exception as e:
|
|
print(f"Failed to open stress_composer.db. Exiting... {e}")
|
|
exit(0)
|
|
|
|
cur = con.cursor()
|
|
|
|
try:
|
|
cur.execute(f"DROP TABLE tbl_{selected_tbl}")
|
|
con.commit()
|
|
print(f"Successfully dropped table tbl_{selected_tbl}")
|
|
except turso.ProgrammingError as e:
|
|
# Table might have been dropped in parallel - this is expected
|
|
print(f"Table tbl_{selected_tbl} already dropped in parallel: {e}")
|
|
con.rollback()
|
|
|
|
con.close()
|
|
|
|
cur_init.execute("DELETE FROM schemas WHERE tbl = ?", (selected_tbl,))
|
|
|
|
con_init.commit()
|
|
|
|
con_init.close()
|