Add automatic session rollback for assert_* functions