Add tests for backup and rollback