mirror of
https://github.com/aljazceru/turso.git
synced 2026-02-22 00:15:20 +01:00
antithesis: Add integritycheck parallel driver
This commit is contained in:
33
antithesis-tests/stress-composer/parallel_driver_integritycheck.py
Executable file
33
antithesis-tests/stress-composer/parallel_driver_integritycheck.py
Executable file
@@ -0,0 +1,33 @@
|
||||
#!/usr/bin/env -S python3 -u
|
||||
|
||||
import json
|
||||
import limbo
|
||||
from antithesis.random import get_random
|
||||
from antithesis.assertions import always
|
||||
|
||||
# Get initial state
|
||||
try:
|
||||
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]
|
||||
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'])])
|
||||
|
||||
try:
|
||||
con = limbo.connect('stress_composer.db')
|
||||
except limbo.OperationalError as e:
|
||||
print(f'Failed to open stress_composer.db. Exiting... {e}')
|
||||
exit(0)
|
||||
cur = con.cursor();
|
||||
|
||||
print('Running integrity check...')
|
||||
|
||||
result = cur.execute("PRAGMA integrity_check")
|
||||
row = result.fetchone()
|
||||
always(row == ("ok",), f"Integrity check failed: {row}", {})
|
||||
Reference in New Issue
Block a user