test_storage: Use the db_transaction context manager instead of raw connections

This minimizes the risk of dangling db connections, which in turn reduces the
chance of cascading test failures.