def test_atomic_on_connection_plus_that_is_in_progress(in_transaction):
sqlite_conn = sqlite3.connect(':memory:')
conn_plus = ConnectionPlus(sqlite_conn)
# explicitly set to True for testing purposes
conn_plus.atomic_in_progress = True
# implement parametrizing over connection's `in_transaction` attribute
if in_transaction:
conn_plus.cursor().execute('BEGIN')
assert in_transaction is conn_plus.in_transaction
isolation_level = conn_plus.isolation_level
in_transaction = conn_plus.in_transaction
with atomic(conn_plus) as atomic_conn:
assert True is conn_plus.atomic_in_progress
assert isolation_level == conn_plus.isolation_level
assert in_transaction is conn_plus.in_transaction
assert True is atomic_conn.atomic_in_progress
assert isolation_level == atomic_conn.isolation_level
assert in_transaction is atomic_conn.in_transaction
assert True is conn_plus.atomic_in_progress
assert isolation_level == conn_plus.isolation_level
assert in_transaction is conn_plus.in_transaction
assert True is atomic_conn.atomic_in_progress
assert isolation_level == atomic_conn.isolation_level
assert in_transaction is atomic_conn.in_transaction
def _populate_results_table(source_conn: ConnectionPlus,
target_conn: ConnectionPlus,
source_table_name: str,
target_table_name: str) -> None:
"""
Copy over all the entries of the results table
"""
get_data_query = f"""
SELECT *
FROM "{source_table_name}"
"""
source_cursor = source_conn.cursor()
target_cursor = target_conn.cursor()
for row in source_cursor.execute(get_data_query):
column_names = ','.join(row.keys()[1:]) # the first key is "id"
values = tuple(val for val in row[1:])
value_placeholders = sql_placeholder_string(len(values))
insert_data_query = f"""
INSERT INTO "{target_table_name}"
({column_names})
values {value_placeholders}
"""
target_cursor.execute(insert_data_query, values)
def path_to_dbfile(conn: ConnectionPlus) -> str:
"""
Return the path of the database file that the conn object is connected to
"""
cursor = conn.cursor()
cursor.execute("PRAGMA database_list")
row = cursor.fetchall()[0]
return row[2]
def _rewrite_timestamps(target_conn: ConnectionPlus, target_run_id: int,
correct_run_timestamp: float,
correct_completed_timestamp: float) -> None:
"""
Update the timestamp to match the original one
"""
query = """
UPDATE runs
SET run_timestamp = ?
WHERE run_id = ?
"""
cursor = target_conn.cursor()
cursor.execute(query, (correct_run_timestamp, target_run_id))
query = """
UPDATE runs
SET completed_timestamp = ?
WHERE run_id = ?
"""
cursor = target_conn.cursor()
cursor.execute(query, (correct_completed_timestamp, target_run_id))