Adds key value integration tests

This PR primarily adds KV integration tests, but also fixes some parts of the existing KV integration

For tests this PR adds a reset method for the container
I think we can remove this now or add a variable with arbitrary alters?