Add persistence integration tests